Simplify
This commit is contained in:
@@ -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))
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -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)
|
||||||
|
|||||||
@@ -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(
|
||||||
|
|||||||
Reference in New Issue
Block a user