1 //! `hir_def` crate contains everything between macro expansion and type
4 //! It defines various items (structs, enums, traits) which comprises Rust code,
5 //! as well as an algorithm for resolving paths to such entities.
7 //! Note that `hir_def` is a work in progress, so not all of the above is
11 macro_rules! eprintln {
12 ($($tt:tt)*) => { stdx::eprintln!($($tt)*) };
44 pub mod child_by_source;
53 use std::hash::{Hash, Hasher};
56 use base_db::{impl_intern_key, salsa, CrateId};
58 ast_id_map::FileAstId, eager::expand_eager_macro, hygiene::Hygiene, AstId, HirFileId, InFile,
59 MacroCallId, MacroCallKind, MacroDefId, MacroDefKind,
63 use crate::builtin_type::BuiltinType;
65 Const, Enum, Function, Impl, ItemTreeId, ItemTreeNode, ModItem, Static, Struct, Trait,
70 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
73 pub local_id: LocalModuleId,
76 /// An ID of a module, **local** to a specific crate
77 pub type LocalModuleId = Idx<nameres::ModuleData>;
80 pub struct ItemLoc<N: ItemTreeNode> {
81 pub container: ContainerId,
82 pub id: ItemTreeId<N>,
85 impl<N: ItemTreeNode> Clone for ItemLoc<N> {
86 fn clone(&self) -> Self {
87 Self { container: self.container, id: self.id }
91 impl<N: ItemTreeNode> Copy for ItemLoc<N> {}
93 impl<N: ItemTreeNode> PartialEq for ItemLoc<N> {
94 fn eq(&self, other: &Self) -> bool {
95 self.container == other.container && self.id == other.id
99 impl<N: ItemTreeNode> Eq for ItemLoc<N> {}
101 impl<N: ItemTreeNode> Hash for ItemLoc<N> {
102 fn hash<H: Hasher>(&self, state: &mut H) {
103 self.container.hash(state);
109 pub struct AssocItemLoc<N: ItemTreeNode> {
110 pub container: AssocContainerId,
111 pub id: ItemTreeId<N>,
114 impl<N: ItemTreeNode> Clone for AssocItemLoc<N> {
115 fn clone(&self) -> Self {
116 Self { container: self.container, id: self.id }
120 impl<N: ItemTreeNode> Copy for AssocItemLoc<N> {}
122 impl<N: ItemTreeNode> PartialEq for AssocItemLoc<N> {
123 fn eq(&self, other: &Self) -> bool {
124 self.container == other.container && self.id == other.id
128 impl<N: ItemTreeNode> Eq for AssocItemLoc<N> {}
130 impl<N: ItemTreeNode> Hash for AssocItemLoc<N> {
131 fn hash<H: Hasher>(&self, state: &mut H) {
132 self.container.hash(state);
137 macro_rules! impl_intern {
138 ($id:ident, $loc:ident, $intern:ident, $lookup:ident) => {
139 impl_intern_key!($id);
141 impl Intern for $loc {
143 fn intern(self, db: &dyn db::DefDatabase) -> $id {
148 impl Lookup for $id {
150 fn lookup(&self, db: &dyn db::DefDatabase) -> $loc {
157 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
158 pub struct FunctionId(salsa::InternId);
159 type FunctionLoc = AssocItemLoc<Function>;
160 impl_intern!(FunctionId, FunctionLoc, intern_function, lookup_intern_function);
162 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord)]
163 pub struct StructId(salsa::InternId);
164 type StructLoc = ItemLoc<Struct>;
165 impl_intern!(StructId, StructLoc, intern_struct, lookup_intern_struct);
167 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord)]
168 pub struct UnionId(salsa::InternId);
169 pub type UnionLoc = ItemLoc<Union>;
170 impl_intern!(UnionId, UnionLoc, intern_union, lookup_intern_union);
172 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, PartialOrd, Ord)]
173 pub struct EnumId(salsa::InternId);
174 pub type EnumLoc = ItemLoc<Enum>;
175 impl_intern!(EnumId, EnumLoc, intern_enum, lookup_intern_enum);
177 // FIXME: rename to `VariantId`, only enums can ave variants
178 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
179 pub struct EnumVariantId {
181 pub local_id: LocalEnumVariantId,
184 pub type LocalEnumVariantId = Idx<adt::EnumVariantData>;
186 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
188 pub parent: VariantId,
189 pub local_id: LocalFieldId,
192 pub type LocalFieldId = Idx<adt::FieldData>;
194 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
195 pub struct ConstId(salsa::InternId);
196 type ConstLoc = AssocItemLoc<Const>;
197 impl_intern!(ConstId, ConstLoc, intern_const, lookup_intern_const);
199 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
200 pub struct StaticId(salsa::InternId);
201 pub type StaticLoc = ItemLoc<Static>;
202 impl_intern!(StaticId, StaticLoc, intern_static, lookup_intern_static);
204 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
205 pub struct TraitId(salsa::InternId);
206 pub type TraitLoc = ItemLoc<Trait>;
207 impl_intern!(TraitId, TraitLoc, intern_trait, lookup_intern_trait);
209 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
210 pub struct TypeAliasId(salsa::InternId);
211 type TypeAliasLoc = AssocItemLoc<TypeAlias>;
212 impl_intern!(TypeAliasId, TypeAliasLoc, intern_type_alias, lookup_intern_type_alias);
214 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Ord, PartialOrd)]
215 pub struct ImplId(salsa::InternId);
216 type ImplLoc = ItemLoc<Impl>;
217 impl_intern!(ImplId, ImplLoc, intern_impl, lookup_intern_impl);
219 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
220 pub struct TypeParamId {
221 pub parent: GenericDefId,
222 pub local_id: LocalTypeParamId,
225 pub type LocalTypeParamId = Idx<generics::TypeParamData>;
227 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
228 pub enum ContainerId {
230 DefWithBodyId(DefWithBodyId),
233 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
234 pub enum AssocContainerId {
235 ContainerId(ContainerId),
239 impl_from!(ContainerId for AssocContainerId);
242 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, PartialOrd, Ord)]
248 impl_from!(StructId, UnionId, EnumId for AdtId);
250 /// The defs which can be visible in the module.
251 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
252 pub enum ModuleDefId {
254 FunctionId(FunctionId),
256 // Can't be directly declared, but can be imported.
257 EnumVariantId(EnumVariantId),
261 TypeAliasId(TypeAliasId),
262 BuiltinType(BuiltinType),
267 AdtId(StructId, EnumId, UnionId),
277 /// The defs which have a body.
278 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
279 pub enum DefWithBodyId {
280 FunctionId(FunctionId),
285 impl_from!(FunctionId, ConstId, StaticId for DefWithBodyId);
287 #[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]
288 pub enum AssocItemId {
289 FunctionId(FunctionId),
291 TypeAliasId(TypeAliasId),
293 // FIXME: not every function, ... is actually an assoc item. maybe we should make
294 // sure that you can only turn actual assoc items into AssocItemIds. This would
295 // require not implementing From, and instead having some checked way of
296 // casting them, and somehow making the constructors private, which would be annoying.
297 impl_from!(FunctionId, ConstId, TypeAliasId for AssocItemId);
299 #[derive(Clone, Copy, PartialEq, Eq, Debug, Hash)]
300 pub enum GenericDefId {
301 FunctionId(FunctionId),
304 TypeAliasId(TypeAliasId),
306 // enum variants cannot have generics themselves, but their parent enums
307 // can, and this makes some code easier to write
308 EnumVariantId(EnumVariantId),
309 // consts can have type parameters from their parents (i.e. associated consts of traits)
314 AdtId(StructId, EnumId, UnionId),
323 impl From<AssocItemId> for GenericDefId {
324 fn from(item: AssocItemId) -> Self {
326 AssocItemId::FunctionId(f) => f.into(),
327 AssocItemId::ConstId(c) => c.into(),
328 AssocItemId::TypeAliasId(t) => t.into(),
333 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]
338 FunctionId(FunctionId),
339 EnumVariantId(EnumVariantId),
343 TypeAliasId(TypeAliasId),
344 MacroDefId(MacroDefId),
351 AdtId(StructId, EnumId, UnionId),
363 #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
365 EnumVariantId(EnumVariantId),
369 impl_from!(EnumVariantId, StructId, UnionId for VariantId);
373 fn intern(self, db: &dyn db::DefDatabase) -> Self::ID;
378 fn lookup(&self, db: &dyn db::DefDatabase) -> Self::Data;
381 pub trait HasModule {
382 fn module(&self, db: &dyn db::DefDatabase) -> ModuleId;
385 impl HasModule for ContainerId {
386 fn module(&self, db: &dyn db::DefDatabase) -> ModuleId {
388 ContainerId::ModuleId(it) => it,
389 ContainerId::DefWithBodyId(it) => it.module(db),
394 impl HasModule for AssocContainerId {
395 fn module(&self, db: &dyn db::DefDatabase) -> ModuleId {
397 AssocContainerId::ContainerId(it) => it.module(db),
398 AssocContainerId::ImplId(it) => it.lookup(db).container.module(db),
399 AssocContainerId::TraitId(it) => it.lookup(db).container.module(db),
404 impl<N: ItemTreeNode> HasModule for AssocItemLoc<N> {
405 fn module(&self, db: &dyn db::DefDatabase) -> ModuleId {
406 self.container.module(db)
410 impl HasModule for AdtId {
411 fn module(&self, db: &dyn db::DefDatabase) -> ModuleId {
413 AdtId::StructId(it) => it.lookup(db).container,
414 AdtId::UnionId(it) => it.lookup(db).container,
415 AdtId::EnumId(it) => it.lookup(db).container,
421 impl HasModule for DefWithBodyId {
422 fn module(&self, db: &dyn db::DefDatabase) -> ModuleId {
424 DefWithBodyId::FunctionId(it) => it.lookup(db).module(db),
425 DefWithBodyId::StaticId(it) => it.lookup(db).module(db),
426 DefWithBodyId::ConstId(it) => it.lookup(db).module(db),
432 pub fn as_mod_item(self, db: &dyn db::DefDatabase) -> ModItem {
434 DefWithBodyId::FunctionId(it) => it.lookup(db).id.value.into(),
435 DefWithBodyId::StaticId(it) => it.lookup(db).id.value.into(),
436 DefWithBodyId::ConstId(it) => it.lookup(db).id.value.into(),
441 impl HasModule for GenericDefId {
442 fn module(&self, db: &dyn db::DefDatabase) -> ModuleId {
444 GenericDefId::FunctionId(it) => it.lookup(db).module(db),
445 GenericDefId::AdtId(it) => it.module(db),
446 GenericDefId::TraitId(it) => it.lookup(db).container.module(db),
447 GenericDefId::TypeAliasId(it) => it.lookup(db).module(db),
448 GenericDefId::ImplId(it) => it.lookup(db).container.module(db),
449 GenericDefId::EnumVariantId(it) => it.parent.lookup(db).container.module(db),
450 GenericDefId::ConstId(it) => it.lookup(db).module(db),
455 impl HasModule for StaticLoc {
456 fn module(&self, db: &dyn db::DefDatabase) -> ModuleId {
457 self.container.module(db)
461 /// A helper trait for converting to MacroCallId
462 pub trait AsMacroCall {
465 db: &dyn db::DefDatabase,
467 resolver: impl Fn(path::ModPath) -> Option<MacroDefId>,
468 ) -> Option<MacroCallId> {
469 self.as_call_id_with_errors(db, krate, resolver, &mut |_| ())
472 fn as_call_id_with_errors(
474 db: &dyn db::DefDatabase,
476 resolver: impl Fn(path::ModPath) -> Option<MacroDefId>,
477 error_sink: &mut dyn FnMut(mbe::ExpandError),
478 ) -> Option<MacroCallId>;
481 impl AsMacroCall for InFile<&ast::MacroCall> {
482 fn as_call_id_with_errors(
484 db: &dyn db::DefDatabase,
486 resolver: impl Fn(path::ModPath) -> Option<MacroDefId>,
487 error_sink: &mut dyn FnMut(mbe::ExpandError),
488 ) -> Option<MacroCallId> {
489 let ast_id = AstId::new(self.file_id, db.ast_id_map(self.file_id).ast_id(self.value));
490 let h = Hygiene::new(db.upcast(), self.file_id);
491 let path = self.value.path().and_then(|path| path::ModPath::from_src(path, &h));
494 error_sink(mbe::ExpandError::Other("malformed macro invocation".into()));
497 AstIdWithPath::new(ast_id.file_id, ast_id.value, path?)
498 .as_call_id_with_errors(db, krate, resolver, error_sink)
502 /// Helper wrapper for `AstId` with `ModPath`
503 #[derive(Clone, Debug, Eq, PartialEq)]
504 struct AstIdWithPath<T: ast::AstNode> {
509 impl<T: ast::AstNode> AstIdWithPath<T> {
510 fn new(file_id: HirFileId, ast_id: FileAstId<T>, path: path::ModPath) -> AstIdWithPath<T> {
511 AstIdWithPath { ast_id: AstId::new(file_id, ast_id), path }
515 impl AsMacroCall for AstIdWithPath<ast::MacroCall> {
516 fn as_call_id_with_errors(
518 db: &dyn db::DefDatabase,
520 resolver: impl Fn(path::ModPath) -> Option<MacroDefId>,
521 error_sink: &mut dyn FnMut(mbe::ExpandError),
522 ) -> Option<MacroCallId> {
523 let def: MacroDefId = resolver(self.path.clone()).or_else(|| {
524 error_sink(mbe::ExpandError::Other("could not resolve macro".into()));
528 if let MacroDefKind::BuiltInEager(_) = def.kind {
529 let macro_call = InFile::new(self.ast_id.file_id, self.ast_id.to_node(db.upcast()));
530 let hygiene = Hygiene::new(db.upcast(), self.ast_id.file_id);
538 &|path: ast::Path| resolver(path::ModPath::from_src(path, &hygiene)?),
545 Some(def.as_lazy_macro(db.upcast(), krate, MacroCallKind::FnLike(self.ast_id)).into())
550 impl AsMacroCall for AstIdWithPath<ast::Item> {
551 fn as_call_id_with_errors(
553 db: &dyn db::DefDatabase,
555 resolver: impl Fn(path::ModPath) -> Option<MacroDefId>,
556 error_sink: &mut dyn FnMut(mbe::ExpandError),
557 ) -> Option<MacroCallId> {
558 let def: MacroDefId = resolver(self.path.clone()).or_else(|| {
559 error_sink(mbe::ExpandError::Other("could not resolve macro".into()));
567 MacroCallKind::Attr(self.ast_id, self.path.segments.last()?.to_string()),