diff options
Diffstat (limited to 'crates/hir_expand')
-rw-r--r-- | crates/hir_expand/src/builtin_macro.rs | 31 | ||||
-rw-r--r-- | crates/hir_expand/src/db.rs | 4 | ||||
-rw-r--r-- | crates/hir_expand/src/lib.rs | 2 | ||||
-rw-r--r-- | crates/hir_expand/src/name.rs | 2 |
4 files changed, 30 insertions, 9 deletions
diff --git a/crates/hir_expand/src/builtin_macro.rs b/crates/hir_expand/src/builtin_macro.rs index 8db8c7d0c..bd9223825 100644 --- a/crates/hir_expand/src/builtin_macro.rs +++ b/crates/hir_expand/src/builtin_macro.rs | |||
@@ -63,7 +63,7 @@ macro_rules! register_builtin { | |||
63 | pub fn find_builtin_macro( | 63 | pub fn find_builtin_macro( |
64 | ident: &name::Name, | 64 | ident: &name::Name, |
65 | krate: CrateId, | 65 | krate: CrateId, |
66 | ast_id: AstId<ast::MacroCall>, | 66 | ast_id: AstId<ast::MacroRules>, |
67 | ) -> Option<MacroDefId> { | 67 | ) -> Option<MacroDefId> { |
68 | let kind = find_by_name(ident)?; | 68 | let kind = find_by_name(ident)?; |
69 | 69 | ||
@@ -95,6 +95,8 @@ register_builtin! { | |||
95 | // format_args_nl only differs in that it adds a newline in the end, | 95 | // format_args_nl only differs in that it adds a newline in the end, |
96 | // so we use the same stub expansion for now | 96 | // so we use the same stub expansion for now |
97 | (format_args_nl, FormatArgsNl) => format_args_expand, | 97 | (format_args_nl, FormatArgsNl) => format_args_expand, |
98 | (llvm_asm, LlvmAsm) => asm_expand, | ||
99 | (asm, Asm) => asm_expand, | ||
98 | 100 | ||
99 | EAGER: | 101 | EAGER: |
100 | (compile_error, CompileError) => compile_error_expand, | 102 | (compile_error, CompileError) => compile_error_expand, |
@@ -271,6 +273,19 @@ fn format_args_expand( | |||
271 | ExpandResult::ok(expanded) | 273 | ExpandResult::ok(expanded) |
272 | } | 274 | } |
273 | 275 | ||
276 | fn asm_expand( | ||
277 | _db: &dyn AstDatabase, | ||
278 | _id: LazyMacroId, | ||
279 | _tt: &tt::Subtree, | ||
280 | ) -> ExpandResult<tt::Subtree> { | ||
281 | // both asm and llvm_asm don't return anything, so we can expand them to nothing, | ||
282 | // for now | ||
283 | let expanded = quote! { | ||
284 | () | ||
285 | }; | ||
286 | ExpandResult::ok(expanded) | ||
287 | } | ||
288 | |||
274 | fn unquote_str(lit: &tt::Literal) -> Option<String> { | 289 | fn unquote_str(lit: &tt::Literal) -> Option<String> { |
275 | let lit = ast::make::tokens::literal(&lit.to_string()); | 290 | let lit = ast::make::tokens::literal(&lit.to_string()); |
276 | let token = ast::String::cast(lit)?; | 291 | let token = ast::String::cast(lit)?; |
@@ -500,12 +515,16 @@ mod tests { | |||
500 | fn expand_builtin_macro(ra_fixture: &str) -> String { | 515 | fn expand_builtin_macro(ra_fixture: &str) -> String { |
501 | let (db, file_id) = TestDB::with_single_file(&ra_fixture); | 516 | let (db, file_id) = TestDB::with_single_file(&ra_fixture); |
502 | let parsed = db.parse(file_id); | 517 | let parsed = db.parse(file_id); |
518 | let macro_rules: Vec<_> = | ||
519 | parsed.syntax_node().descendants().filter_map(ast::MacroRules::cast).collect(); | ||
503 | let macro_calls: Vec<_> = | 520 | let macro_calls: Vec<_> = |
504 | parsed.syntax_node().descendants().filter_map(ast::MacroCall::cast).collect(); | 521 | parsed.syntax_node().descendants().filter_map(ast::MacroCall::cast).collect(); |
505 | 522 | ||
506 | let ast_id_map = db.ast_id_map(file_id.into()); | 523 | let ast_id_map = db.ast_id_map(file_id.into()); |
507 | 524 | ||
508 | let expander = find_by_name(¯o_calls[0].name().unwrap().as_name()).unwrap(); | 525 | assert_eq!(macro_rules.len(), 1, "test must contain exactly 1 `macro_rules!`"); |
526 | assert_eq!(macro_calls.len(), 1, "test must contain exactly 1 macro call"); | ||
527 | let expander = find_by_name(¯o_rules[0].name().unwrap().as_name()).unwrap(); | ||
509 | 528 | ||
510 | let krate = CrateId(0); | 529 | let krate = CrateId(0); |
511 | let file_id = match expander { | 530 | let file_id = match expander { |
@@ -513,7 +532,7 @@ mod tests { | |||
513 | // the first one should be a macro_rules | 532 | // the first one should be a macro_rules |
514 | let def = MacroDefId { | 533 | let def = MacroDefId { |
515 | krate: Some(CrateId(0)), | 534 | krate: Some(CrateId(0)), |
516 | ast_id: Some(AstId::new(file_id.into(), ast_id_map.ast_id(¯o_calls[0]))), | 535 | ast_id: Some(AstId::new(file_id.into(), ast_id_map.ast_id(¯o_rules[0]))), |
517 | kind: MacroDefKind::BuiltIn(expander), | 536 | kind: MacroDefKind::BuiltIn(expander), |
518 | local_inner: false, | 537 | local_inner: false, |
519 | }; | 538 | }; |
@@ -523,7 +542,7 @@ mod tests { | |||
523 | krate, | 542 | krate, |
524 | kind: MacroCallKind::FnLike(AstId::new( | 543 | kind: MacroCallKind::FnLike(AstId::new( |
525 | file_id.into(), | 544 | file_id.into(), |
526 | ast_id_map.ast_id(¯o_calls[1]), | 545 | ast_id_map.ast_id(¯o_calls[0]), |
527 | )), | 546 | )), |
528 | }; | 547 | }; |
529 | 548 | ||
@@ -534,12 +553,12 @@ mod tests { | |||
534 | // the first one should be a macro_rules | 553 | // the first one should be a macro_rules |
535 | let def = MacroDefId { | 554 | let def = MacroDefId { |
536 | krate: Some(krate), | 555 | krate: Some(krate), |
537 | ast_id: Some(AstId::new(file_id.into(), ast_id_map.ast_id(¯o_calls[0]))), | 556 | ast_id: Some(AstId::new(file_id.into(), ast_id_map.ast_id(¯o_rules[0]))), |
538 | kind: MacroDefKind::BuiltInEager(expander), | 557 | kind: MacroDefKind::BuiltInEager(expander), |
539 | local_inner: false, | 558 | local_inner: false, |
540 | }; | 559 | }; |
541 | 560 | ||
542 | let args = macro_calls[1].token_tree().unwrap(); | 561 | let args = macro_calls[0].token_tree().unwrap(); |
543 | let parsed_args = mbe::ast_to_token_tree(&args).unwrap().0; | 562 | let parsed_args = mbe::ast_to_token_tree(&args).unwrap().0; |
544 | 563 | ||
545 | let arg_id = db.intern_eager_expansion({ | 564 | let arg_id = db.intern_eager_expansion({ |
diff --git a/crates/hir_expand/src/db.rs b/crates/hir_expand/src/db.rs index ffb70f723..11b5b98c8 100644 --- a/crates/hir_expand/src/db.rs +++ b/crates/hir_expand/src/db.rs | |||
@@ -394,8 +394,8 @@ fn to_fragment_kind(db: &dyn AstDatabase, id: MacroCallId) -> FragmentKind { | |||
394 | // FIXME: Handle Pattern | 394 | // FIXME: Handle Pattern |
395 | FragmentKind::Expr | 395 | FragmentKind::Expr |
396 | } | 396 | } |
397 | // FIXME: Expand to statements in appropriate positions; HIR lowering needs to handle that | 397 | EXPR_STMT => FragmentKind::Statements, |
398 | EXPR_STMT | BLOCK_EXPR => FragmentKind::Expr, | 398 | BLOCK_EXPR => FragmentKind::Expr, |
399 | ARG_LIST => FragmentKind::Expr, | 399 | ARG_LIST => FragmentKind::Expr, |
400 | TRY_EXPR => FragmentKind::Expr, | 400 | TRY_EXPR => FragmentKind::Expr, |
401 | TUPLE_EXPR => FragmentKind::Expr, | 401 | TUPLE_EXPR => FragmentKind::Expr, |
diff --git a/crates/hir_expand/src/lib.rs b/crates/hir_expand/src/lib.rs index 1a9428514..ae3086a95 100644 --- a/crates/hir_expand/src/lib.rs +++ b/crates/hir_expand/src/lib.rs | |||
@@ -228,7 +228,7 @@ pub struct MacroDefId { | |||
228 | // (which will probably require touching this code), we can instead use | 228 | // (which will probably require touching this code), we can instead use |
229 | // that (and also remove the hacks for resolving built-in derives). | 229 | // that (and also remove the hacks for resolving built-in derives). |
230 | pub krate: Option<CrateId>, | 230 | pub krate: Option<CrateId>, |
231 | pub ast_id: Option<AstId<ast::MacroCall>>, | 231 | pub ast_id: Option<AstId<ast::MacroRules>>, |
232 | pub kind: MacroDefKind, | 232 | pub kind: MacroDefKind, |
233 | 233 | ||
234 | pub local_inner: bool, | 234 | pub local_inner: bool, |
diff --git a/crates/hir_expand/src/name.rs b/crates/hir_expand/src/name.rs index 6ec5ca0a9..69d8e6803 100644 --- a/crates/hir_expand/src/name.rs +++ b/crates/hir_expand/src/name.rs | |||
@@ -199,6 +199,8 @@ pub mod known { | |||
199 | format_args_nl, | 199 | format_args_nl, |
200 | env, | 200 | env, |
201 | option_env, | 201 | option_env, |
202 | llvm_asm, | ||
203 | asm, | ||
202 | // Builtin derives | 204 | // Builtin derives |
203 | Copy, | 205 | Copy, |
204 | Clone, | 206 | Clone, |