This commit is contained in:
Lukas Wirth
2022-02-21 13:34:05 +01:00
parent 1bbef5af85
commit 94e59c9c56
3 changed files with 37 additions and 27 deletions

View File

@@ -1793,10 +1793,7 @@ impl MacroDef {
} }
pub fn is_builtin_derive(&self) -> bool { pub fn is_builtin_derive(&self) -> bool {
match self.id.kind { matches!(self.id.kind, MacroDefKind::BuiltInAttr(exp, _) if exp.is_derive())
MacroDefKind::BuiltInAttr(exp, _) => exp.is_derive(),
_ => false,
}
} }
pub fn is_attr(&self) -> bool { pub fn is_attr(&self) -> bool {
@@ -2433,24 +2430,7 @@ impl Impl {
pub fn is_builtin_derive(self, db: &dyn HirDatabase) -> Option<InFile<ast::Attr>> { pub fn is_builtin_derive(self, db: &dyn HirDatabase) -> Option<InFile<ast::Attr>> {
let src = self.source(db)?; let src = self.source(db)?;
let item = src.file_id.is_builtin_derive(db.upcast())?; src.file_id.is_builtin_derive(db.upcast())
let hygenic = hir_expand::hygiene::Hygiene::new(db.upcast(), item.file_id);
// FIXME: handle `cfg_attr`
let attr = item
.value
.attrs()
.filter_map(|it| {
let path = ModPath::from_src(db.upcast(), it.path()?, &hygenic)?;
if path.as_ident()?.to_smol_str() == "derive" {
Some(it)
} else {
None
}
})
.last()?;
Some(item.with_value(attr))
} }
} }

View File

@@ -262,16 +262,16 @@ impl HirFileId {
} }
/// Indicate it is macro file generated for builtin derive /// Indicate it is macro file generated for builtin derive
pub fn is_builtin_derive(&self, db: &dyn db::AstDatabase) -> Option<InFile<ast::Item>> { pub fn is_builtin_derive(&self, db: &dyn db::AstDatabase) -> Option<InFile<ast::Attr>> {
match self.0 { match self.0 {
HirFileIdRepr::FileId(_) => None, HirFileIdRepr::FileId(_) => None,
HirFileIdRepr::MacroFile(macro_file) => { HirFileIdRepr::MacroFile(macro_file) => {
let loc: MacroCallLoc = db.lookup_intern_macro_call(macro_file.macro_call_id); let loc: MacroCallLoc = db.lookup_intern_macro_call(macro_file.macro_call_id);
let item = match loc.def.kind { let attr = match loc.def.kind {
MacroDefKind::BuiltInDerive(..) => loc.kind.to_node(db), MacroDefKind::BuiltInDerive(..) => loc.kind.to_node(db),
_ => return None, _ => return None,
}; };
Some(item.with_value(ast::Item::cast(item.value.clone())?)) Some(attr.with_value(ast::Attr::cast(attr.value.clone())?))
} }
} }
} }
@@ -383,10 +383,20 @@ impl MacroCallKind {
MacroCallKind::FnLike { ast_id, .. } => { MacroCallKind::FnLike { ast_id, .. } => {
ast_id.with_value(ast_id.to_node(db).syntax().clone()) ast_id.with_value(ast_id.to_node(db).syntax().clone())
} }
MacroCallKind::Derive { ast_id, .. } => { MacroCallKind::Derive { ast_id, derive_attr_index, .. } => {
ast_id.with_value(ast_id.to_node(db).syntax().clone()) // FIXME: handle `cfg_attr`
ast_id.with_value(ast_id.to_node(db)).map(|it| {
it.doc_comments_and_attrs()
.nth(*derive_attr_index as usize)
.and_then(|it| match it {
Either::Left(attr) => Some(attr.syntax().clone()),
Either::Right(_) => None,
})
.unwrap_or_else(|| it.syntax().clone())
})
} }
MacroCallKind::Attr { ast_id, is_derive: true, invoc_attr_index, .. } => { MacroCallKind::Attr { ast_id, is_derive: true, invoc_attr_index, .. } => {
// FIXME: handle `cfg_attr`
ast_id.with_value(ast_id.to_node(db)).map(|it| { ast_id.with_value(ast_id.to_node(db)).map(|it| {
it.doc_comments_and_attrs() it.doc_comments_and_attrs()
.nth(*invoc_attr_index as usize) .nth(*invoc_attr_index as usize)

View File

@@ -735,6 +735,26 @@ mod derive {
) )
} }
#[test]
fn derive_no_attrs() {
check_derive(
r#"
//- proc_macros: identity
//- minicore: derive
#[derive($0)] struct Test;
"#,
expect![[r#""#]],
);
check_derive(
r#"
//- proc_macros: identity
//- minicore: derive
#[derive(i$0)] struct Test;
"#,
expect![[r#""#]],
);
}
#[test] #[test]
fn derive_flyimport() { fn derive_flyimport() {
check_derive( check_derive(