mirror of
https://github.com/rust-lang/rust-analyzer.git
synced 2025-09-27 12:29:21 +00:00
fix: Fix multiple derives in one attribute not expanding all in expand_macro
This commit is contained in:
parent
3f9f63c1bd
commit
0f3617f76f
6 changed files with 59 additions and 19 deletions
|
@ -148,7 +148,7 @@ impl<'db, DB: HirDatabase> Semantics<'db, DB> {
|
||||||
self.imp.expand_attr_macro(item)
|
self.imp.expand_attr_macro(item)
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn expand_derive_macro(&self, derive: &ast::Attr) -> Option<SyntaxNode> {
|
pub fn expand_derive_macro(&self, derive: &ast::Attr) -> Option<Vec<SyntaxNode>> {
|
||||||
self.imp.expand_derive_macro(derive)
|
self.imp.expand_derive_macro(derive)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -389,16 +389,29 @@ impl<'db> SemanticsImpl<'db> {
|
||||||
Some(node)
|
Some(node)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn expand_derive_macro(&self, attr: &ast::Attr) -> Option<SyntaxNode> {
|
fn expand_derive_macro(&self, attr: &ast::Attr) -> Option<Vec<SyntaxNode>> {
|
||||||
let item = attr.syntax().parent().and_then(ast::Item::cast)?;
|
let item = attr.syntax().parent().and_then(ast::Item::cast)?;
|
||||||
let sa = self.analyze(item.syntax());
|
let sa = self.analyze(item.syntax());
|
||||||
let item = InFile::new(sa.file_id, &item);
|
let item = InFile::new(sa.file_id, &item);
|
||||||
let src = InFile::new(sa.file_id, attr.clone());
|
let src = InFile::new(sa.file_id, attr.clone());
|
||||||
let macro_call_id = self.with_ctx(|ctx| ctx.attr_to_derive_macro_call(item, src))?;
|
self.with_ctx(|ctx| {
|
||||||
let file_id = macro_call_id.as_file();
|
let macro_call_ids = ctx.attr_to_derive_macro_call(item, src)?;
|
||||||
let node = self.db.parse_or_expand(file_id)?;
|
|
||||||
self.cache(node.clone(), file_id);
|
let expansions: Vec<_> = macro_call_ids
|
||||||
Some(node)
|
.iter()
|
||||||
|
.map(|call| call.as_file())
|
||||||
|
.flat_map(|file_id| {
|
||||||
|
let node = self.db.parse_or_expand(file_id)?;
|
||||||
|
self.cache(node.clone(), file_id);
|
||||||
|
Some(node)
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
if expansions.is_empty() {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(expansions)
|
||||||
|
}
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
fn is_attr_macro_call(&self, item: &ast::Item) -> bool {
|
fn is_attr_macro_call(&self, item: &ast::Item) -> bool {
|
||||||
|
|
|
@ -246,9 +246,9 @@ impl SourceToDefCtx<'_, '_> {
|
||||||
&mut self,
|
&mut self,
|
||||||
item: InFile<&ast::Item>,
|
item: InFile<&ast::Item>,
|
||||||
src: InFile<ast::Attr>,
|
src: InFile<ast::Attr>,
|
||||||
) -> Option<MacroCallId> {
|
) -> Option<&[MacroCallId]> {
|
||||||
let map = self.dyn_map(item)?;
|
let map = self.dyn_map(item)?;
|
||||||
map[keys::DERIVE_MACRO].get(&src).copied()
|
map[keys::DERIVE_MACRO].get(&src).map(AsRef::as_ref)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn to_def<Ast: AstNode + 'static, ID: Copy + 'static>(
|
fn to_def<Ast: AstNode + 'static, ID: Copy + 'static>(
|
||||||
|
|
|
@ -6,6 +6,7 @@
|
||||||
|
|
||||||
use either::Either;
|
use either::Either;
|
||||||
use hir_expand::HirFileId;
|
use hir_expand::HirFileId;
|
||||||
|
use itertools::Itertools;
|
||||||
use syntax::ast::AttrsOwner;
|
use syntax::ast::AttrsOwner;
|
||||||
|
|
||||||
use crate::{
|
use crate::{
|
||||||
|
@ -109,10 +110,13 @@ impl ChildBySource for ItemScope {
|
||||||
let item = ast_id.with_value(ast_id.to_node(db.upcast()));
|
let item = ast_id.with_value(ast_id.to_node(db.upcast()));
|
||||||
res[keys::ATTR_MACRO].insert(item, call_id);
|
res[keys::ATTR_MACRO].insert(item, call_id);
|
||||||
});
|
});
|
||||||
self.derive_macro_invocs().for_each(|(ast_id, (attr_id, call_id))| {
|
self.derive_macro_invocs().for_each(|(ast_id, calls)| {
|
||||||
let item = ast_id.to_node(db.upcast());
|
let item = ast_id.to_node(db.upcast());
|
||||||
if let Some(attr) = item.attrs().nth(attr_id.ast_index as usize) {
|
let grouped = calls.iter().copied().into_group_map();
|
||||||
res[keys::DERIVE_MACRO].insert(ast_id.with_value(attr), call_id);
|
for (attr_id, calls) in grouped {
|
||||||
|
if let Some(attr) = item.attrs().nth(attr_id.ast_index as usize) {
|
||||||
|
res[keys::DERIVE_MACRO].insert(ast_id.with_value(attr), calls.into());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|
|
@ -195,8 +195,8 @@ impl ItemScope {
|
||||||
|
|
||||||
pub(crate) fn derive_macro_invocs(
|
pub(crate) fn derive_macro_invocs(
|
||||||
&self,
|
&self,
|
||||||
) -> impl Iterator<Item = (AstId<ast::Item>, (AttrId, MacroCallId))> + '_ {
|
) -> impl Iterator<Item = (AstId<ast::Item>, &[(AttrId, MacroCallId)])> + '_ {
|
||||||
self.derive_macros.iter().flat_map(|(k, v)| v.iter().map(move |v| (*k, *v)))
|
self.derive_macros.iter().map(|(k, v)| (*k, v.as_ref()))
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) fn unnamed_trait_vis(&self, tr: TraitId) -> Option<Visibility> {
|
pub(crate) fn unnamed_trait_vis(&self, tr: TraitId) -> Option<Visibility> {
|
||||||
|
|
|
@ -33,7 +33,7 @@ pub const CONST_PARAM: Key<ast::ConstParam, ConstParamId> = Key::new();
|
||||||
|
|
||||||
pub const MACRO: Key<ast::MacroCall, MacroDefId> = Key::new();
|
pub const MACRO: Key<ast::MacroCall, MacroDefId> = Key::new();
|
||||||
pub const ATTR_MACRO: Key<ast::Item, MacroCallId> = Key::new();
|
pub const ATTR_MACRO: Key<ast::Item, MacroCallId> = Key::new();
|
||||||
pub const DERIVE_MACRO: Key<ast::Attr, MacroCallId> = Key::new();
|
pub const DERIVE_MACRO: Key<ast::Attr, Box<[MacroCallId]>> = Key::new();
|
||||||
|
|
||||||
/// XXX: AST Nodes and SyntaxNodes have identity equality semantics: nodes are
|
/// XXX: AST Nodes and SyntaxNodes have identity equality semantics: nodes are
|
||||||
/// equal if they point to exactly the same object.
|
/// equal if they point to exactly the same object.
|
||||||
|
|
|
@ -37,10 +37,11 @@ pub(crate) fn expand_macro(db: &RootDatabase, position: FilePosition) -> Option<
|
||||||
if path == "derive" {
|
if path == "derive" {
|
||||||
let mut tt = tt.syntax().children_with_tokens().skip(1).join("");
|
let mut tt = tt.syntax().children_with_tokens().skip(1).join("");
|
||||||
tt.pop();
|
tt.pop();
|
||||||
return sema
|
let expansions = sema.expand_derive_macro(&attr)?;
|
||||||
.expand_derive_macro(&attr)
|
return Some(ExpandedMacro {
|
||||||
.map(insert_whitespaces)
|
name: tt,
|
||||||
.map(|expansion| ExpandedMacro { name: tt, expansion });
|
expansion: expansions.into_iter().map(insert_whitespaces).join(""),
|
||||||
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -382,4 +383,26 @@ struct Foo {}
|
||||||
"#]],
|
"#]],
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn macro_expand_derive_multi() {
|
||||||
|
check(
|
||||||
|
r#"
|
||||||
|
#[rustc_builtin_macro]
|
||||||
|
pub macro Clone {}
|
||||||
|
#[rustc_builtin_macro]
|
||||||
|
pub macro Copy {}
|
||||||
|
|
||||||
|
#[derive(Cop$0y, Clone)]
|
||||||
|
struct Foo {}
|
||||||
|
"#,
|
||||||
|
expect![[r#"
|
||||||
|
Copy, Clone
|
||||||
|
impl< >crate::marker::Copy for Foo< >{}
|
||||||
|
|
||||||
|
impl< >crate::clone::Clone for Foo< >{}
|
||||||
|
|
||||||
|
"#]],
|
||||||
|
);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue