use std::sync::Arc;
-use hir_expand::{name::Name, InFile};
+use hir_expand::{name::Name, AstId, ExpandResult, InFile, MacroCallId};
use syntax::ast;
use crate::{
attr::Attrs,
- body::Expander,
+ body::{Expander, Mark},
db::DefDatabase,
intern::Interned,
item_tree::{self, AssocItem, FnFlags, ItemTreeId, ModItem, Param},
+ nameres::attr_resolution::ResolvedAttr,
type_ref::{TraitRef, TypeBound, TypeRef},
visibility::RawVisibility,
- AssocContainerId, AssocItemId, ConstId, ConstLoc, FunctionId, FunctionLoc, HasModule, ImplId,
- Intern, Lookup, ModuleId, StaticId, TraitId, TypeAliasId, TypeAliasLoc,
+ AssocItemId, AstIdWithPath, ConstId, ConstLoc, FunctionId, FunctionLoc, HasModule, ImplId,
+ Intern, ItemContainerId, Lookup, ModuleId, StaticId, TraitId, TypeAliasId, TypeAliasLoc,
};
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct FunctionData {
pub name: Name,
- pub params: Vec<Interned<TypeRef>>,
+ pub params: Vec<(Option<Name>, Interned<TypeRef>)>,
pub ret_type: Interned<TypeRef>,
pub async_ret_type: Option<Interned<TypeRef>>,
pub attrs: Attrs,
pub visibility: RawVisibility,
pub abi: Option<Interned<str>>,
+ pub legacy_const_generics_indices: Vec<u32>,
flags: FnFlags,
}
flags.bits |= FnFlags::IS_VARARGS;
}
+ if matches!(loc.container, ItemContainerId::ExternBlockId(_)) {
+ flags.bits |= FnFlags::IS_IN_EXTERN_BLOCK;
+ }
+
+ let legacy_const_generics_indices = item_tree
+ .attrs(db, krate, ModItem::from(loc.id.value).into())
+ .by_key("rustc_legacy_const_generics")
+ .tt_values()
+ .next()
+ .map(|arg| parse_rustc_legacy_const_generics(arg))
+ .unwrap_or_default();
+
Arc::new(FunctionData {
name: func.name.clone(),
params: enabled_params
.clone()
.filter_map(|id| match &item_tree[id] {
- Param::Normal(ty) => Some(ty.clone()),
+ Param::Normal(name, ty) => Some((name.clone(), ty.clone())),
Param::Varargs => None,
})
.collect(),
attrs: item_tree.attrs(db, krate, ModItem::from(loc.id.value).into()),
visibility: item_tree[func.visibility].clone(),
abi: func.abi.clone(),
+ legacy_const_generics_indices,
flags,
})
}
}
}
+fn parse_rustc_legacy_const_generics(tt: &tt::Subtree) -> Vec<u32> {
+ let mut indices = Vec::new();
+ for args in tt.token_trees.chunks(2) {
+ match &args[0] {
+ tt::TokenTree::Leaf(tt::Leaf::Literal(lit)) => match lit.text.parse() {
+ Ok(index) => indices.push(index),
+ Err(_) => break,
+ },
+ _ => break,
+ }
+
+ if let Some(comma) = args.get(1) {
+ match comma {
+ tt::TokenTree::Leaf(tt::Leaf::Punct(punct)) if punct.char == ',' => {}
+ _ => break,
+ }
+ }
+ }
+
+ indices
+}
+
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct TypeAliasData {
pub name: Name,
name: typ.name.clone(),
type_ref: typ.type_ref.clone(),
visibility: item_tree[typ.visibility].clone(),
- is_extern: typ.is_extern,
+ is_extern: matches!(loc.container, ItemContainerId::ExternBlockId(_)),
bounds: typ.bounds.to_vec(),
})
}
/// method calls to this trait's methods when the receiver is an array and the crate edition is
/// 2015 or 2018.
pub skip_array_during_method_dispatch: bool,
+ // box it as the vec is usually empty anyways
+ pub attribute_calls: Option<Box<Vec<(AstId<ast::Item>, MacroCallId)>>>,
}
impl TraitData {
let is_auto = tr_def.is_auto;
let is_unsafe = tr_def.is_unsafe;
let module_id = tr_loc.container;
- let container = AssocContainerId::TraitId(tr);
+ let container = ItemContainerId::TraitId(tr);
let visibility = item_tree[tr_def.visibility].clone();
let mut expander = Expander::new(db, tr_loc.id.file_id(), module_id);
let skip_array_during_method_dispatch = item_tree
.by_key("rustc_skip_array_during_method_dispatch")
.exists();
- let items = collect_items(
- db,
- module_id,
- &mut expander,
- tr_def.items.iter().copied(),
- tr_loc.id.tree_id(),
- container,
- 100,
- );
+ let (items, attribute_calls) =
+ do_collect(db, module_id, &mut expander, &tr_def.items, tr_loc.id.tree_id(), container);
Arc::new(TraitData {
name,
is_unsafe,
visibility,
skip_array_during_method_dispatch,
+ attribute_calls,
})
}
_ => None,
})
}
+
+ pub fn attribute_calls(&self) -> impl Iterator<Item = (AstId<ast::Item>, MacroCallId)> + '_ {
+ self.attribute_calls.iter().flat_map(|it| it.iter()).copied()
+ }
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub self_ty: Interned<TypeRef>,
pub items: Vec<AssocItemId>,
pub is_negative: bool,
+ // box it as the vec is usually empty anyways
+ pub attribute_calls: Option<Box<Vec<(AstId<ast::Item>, MacroCallId)>>>,
}
impl ImplData {
let self_ty = impl_def.self_ty.clone();
let is_negative = impl_def.is_negative;
let module_id = impl_loc.container;
- let container = AssocContainerId::ImplId(id);
+ let container = ItemContainerId::ImplId(id);
let mut expander = Expander::new(db, impl_loc.id.file_id(), module_id);
- let items = collect_items(
+ let (items, attribute_calls) = do_collect(
db,
module_id,
&mut expander,
- impl_def.items.iter().copied(),
+ &impl_def.items,
impl_loc.id.tree_id(),
container,
- 100,
);
let items = items.into_iter().map(|(_, item)| item).collect();
- Arc::new(ImplData { target_trait, self_ty, items, is_negative })
+ Arc::new(ImplData { target_trait, self_ty, items, is_negative, attribute_calls })
+ }
+
+ pub fn attribute_calls(&self) -> impl Iterator<Item = (AstId<ast::Item>, MacroCallId)> + '_ {
+ self.attribute_calls.iter().flat_map(|it| it.iter()).copied()
}
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct ConstData {
- /// const _: () = ();
+ /// `None` for `const _: () = ();`
pub name: Option<Name>,
pub type_ref: Interned<TypeRef>,
pub visibility: RawVisibility,
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct StaticData {
- pub name: Option<Name>,
+ pub name: Name,
pub type_ref: Interned<TypeRef>,
pub visibility: RawVisibility,
pub mutable: bool,
impl StaticData {
pub(crate) fn static_data_query(db: &dyn DefDatabase, konst: StaticId) -> Arc<StaticData> {
- let node = konst.lookup(db);
- let item_tree = node.id.item_tree(db);
- let statik = &item_tree[node.id.value];
+ let loc = konst.lookup(db);
+ let item_tree = loc.id.item_tree(db);
+ let statik = &item_tree[loc.id.value];
Arc::new(StaticData {
- name: Some(statik.name.clone()),
+ name: statik.name.clone(),
type_ref: statik.type_ref.clone(),
visibility: item_tree[statik.visibility].clone(),
mutable: statik.mutable,
- is_extern: statik.is_extern,
+ is_extern: matches!(loc.container, ItemContainerId::ExternBlockId(_)),
})
}
}
+fn do_collect(
+ db: &dyn DefDatabase,
+ module_id: ModuleId,
+ expander: &mut Expander,
+ assoc_items: &[AssocItem],
+ tree_id: item_tree::TreeId,
+ container: ItemContainerId,
+) -> (Vec<(Name, AssocItemId)>, Option<Box<Vec<(AstId<ast::Item>, MacroCallId)>>>) {
+ let mut items = Vec::new();
+ let mut attribute_calls = Vec::new();
+
+ collect_items(
+ db,
+ &mut items,
+ &mut attribute_calls,
+ module_id,
+ expander,
+ assoc_items.iter().copied(),
+ tree_id,
+ container,
+ );
+
+ let attribute_calls =
+ if attribute_calls.is_empty() { None } else { Some(Box::new(attribute_calls)) };
+ (items, attribute_calls)
+}
+
fn collect_items(
db: &dyn DefDatabase,
+ items: &mut Vec<(Name, AssocItemId)>,
+ attr_calls: &mut Vec<(AstId<ast::Item>, MacroCallId)>,
module: ModuleId,
expander: &mut Expander,
assoc_items: impl Iterator<Item = AssocItem>,
tree_id: item_tree::TreeId,
- container: AssocContainerId,
- limit: usize,
-) -> Vec<(Name, AssocItemId)> {
- if limit == 0 {
- return Vec::new();
- }
-
+ container: ItemContainerId,
+) {
let item_tree = tree_id.item_tree(db);
let crate_graph = db.crate_graph();
let cfg_options = &crate_graph[module.krate].cfg_options;
+ let def_map = module.def_map(db);
- let mut items = Vec::new();
- for item in assoc_items {
+ 'items: for item in assoc_items {
let attrs = item_tree.attrs(db, module.krate, ModItem::from(item).into());
if !attrs.is_cfg_enabled(cfg_options) {
continue;
}
+ for attr in &*attrs {
+ let ast_id = AstId::new(expander.current_file_id(), item.ast_id(&item_tree).upcast());
+ let ast_id_with_path = AstIdWithPath { path: (*attr.path).clone(), ast_id };
+
+ if let Ok(ResolvedAttr::Macro(call_id)) =
+ def_map.resolve_attr_macro(db, module.local_id, ast_id_with_path, attr)
+ {
+ attr_calls.push((ast_id, call_id));
+ let res = expander.enter_expand_id(db, call_id);
+ collect_macro_items(db, items, attr_calls, module, expander, container, res);
+ continue 'items;
+ }
+ }
+
match item {
AssocItem::Function(id) => {
let item = &item_tree[id];
let res = expander.enter_expand(db, call);
if let Ok(res) = res {
- if let Some((mark, mac)) = res.value {
- let src: InFile<ast::MacroItems> = expander.to_source(mac);
- let tree_id = item_tree::TreeId::new(src.file_id, None);
- let item_tree = tree_id.item_tree(db);
- let iter =
- item_tree.top_level_items().iter().filter_map(ModItem::as_assoc_item);
- items.extend(collect_items(
- db,
- module,
- expander,
- iter,
- tree_id,
- container,
- limit - 1,
- ));
-
- expander.exit(db, mark);
- }
+ collect_macro_items(db, items, attr_calls, module, expander, container, res);
}
}
}
}
+}
- items
+fn collect_macro_items(
+ db: &dyn DefDatabase,
+ items: &mut Vec<(Name, AssocItemId)>,
+ attr_calls: &mut Vec<(AstId<ast::Item>, MacroCallId)>,
+ module: ModuleId,
+ expander: &mut Expander,
+ container: ItemContainerId,
+ res: ExpandResult<Option<(Mark, ast::MacroItems)>>,
+) {
+ if let Some((mark, mac)) = res.value {
+ let src: InFile<ast::MacroItems> = expander.to_source(mac);
+ let tree_id = item_tree::TreeId::new(src.file_id, None);
+ let item_tree = tree_id.item_tree(db);
+ let iter = item_tree.top_level_items().iter().filter_map(ModItem::as_assoc_item);
+ collect_items(db, items, attr_calls, module, expander, iter, tree_id, container);
+
+ expander.exit(db, mark);
+ }
}