Make "expand macro" command work with attribute macros

This commit is contained in:
Jonas Schievink 2021-06-07 16:05:36 +02:00
parent 8d87f9b298
commit 33e747d786
2 changed files with 43 additions and 7 deletions

View file

@ -117,6 +117,12 @@ impl<'db, DB: HirDatabase> Semantics<'db, DB> {
pub fn expand(&self, macro_call: &ast::MacroCall) -> Option<SyntaxNode> {
self.imp.expand(macro_call)
}
/// If `item` has an attribute macro attached to it, expands it.
pub fn expand_attr_macro(&self, item: &ast::Item) -> Option<SyntaxNode> {
self.imp.expand_attr_macro(item)
}
pub fn speculative_expand(
&self,
actual_macro_call: &ast::MacroCall,
@ -332,6 +338,16 @@ impl<'db> SemanticsImpl<'db> {
Some(node)
}
fn expand_attr_macro(&self, item: &ast::Item) -> Option<SyntaxNode> {
let sa = self.analyze(item.syntax());
let src = InFile::new(sa.file_id, item.clone());
let macro_call_id = self.with_ctx(|ctx| ctx.item_to_macro_call(src))?;
let file_id = macro_call_id.as_file();
let node = self.db.parse_or_expand(file_id)?;
self.cache(node.clone(), file_id);
Some(node)
}
fn speculative_expand(
&self,
actual_macro_call: &ast::MacroCall,