rust/crates/hir_def/src/resolver.rs

727 lines
25 KiB
Rust
Raw Normal View History

2019-11-21 06:39:09 -06:00
//! Name resolution façade.
2019-01-19 14:23:26 -06:00
use std::sync::Arc;
2020-08-13 09:25:38 -05:00
use base_db::CrateId;
2019-11-21 06:39:09 -06:00
use hir_expand::{
2019-12-13 15:01:06 -06:00
name::{name, Name},
2019-11-21 06:39:09 -06:00
MacroDefId,
};
use rustc_hash::FxHashSet;
use crate::{
2019-11-21 06:24:51 -06:00
body::scope::{ExprScopes, ScopeId},
body::Body,
2019-10-31 02:51:54 -05:00
builtin_type::BuiltinType,
2019-11-23 05:44:43 -06:00
db::DefDatabase,
2019-11-21 06:24:51 -06:00
expr::{ExprId, PatId},
2019-11-21 03:21:46 -06:00
generics::GenericParams,
item_scope::{BuiltinShadowMode, BUILTIN_SCOPE},
2019-12-20 08:43:45 -06:00
nameres::CrateDefMap,
path::{ModPath, PathKind},
2019-11-23 07:53:16 -06:00
per_ns::PerNs,
visibility::{RawVisibility, Visibility},
2021-01-01 03:06:42 -06:00
AdtId, AssocContainerId, ConstId, ConstParamId, ContainerId, DefWithBodyId, EnumId,
EnumVariantId, FunctionId, GenericDefId, HasModule, ImplId, LocalModuleId, Lookup, ModuleDefId,
ModuleId, StaticId, StructId, TraitId, TypeAliasId, TypeParamId, VariantId,
2019-10-30 09:19:30 -05:00
};
2019-01-19 14:23:26 -06:00
2019-01-23 16:08:41 -06:00
#[derive(Debug, Clone, Default)]
2019-11-21 06:39:09 -06:00
pub struct Resolver {
2021-01-08 08:46:48 -06:00
// FIXME: all usages generally call `.rev`, so maybe reverse once in construction?
2019-01-26 15:52:04 -06:00
scopes: Vec<Scope>,
2019-01-19 14:23:26 -06:00
}
2019-03-23 02:53:48 -05:00
// FIXME how to store these best
2019-01-19 14:23:26 -06:00
#[derive(Debug, Clone)]
2019-11-24 12:00:50 -06:00
struct ModuleItemMap {
2019-03-14 04:54:03 -05:00
crate_def_map: Arc<CrateDefMap>,
2019-11-23 07:49:53 -06:00
module_id: LocalModuleId,
2019-01-19 14:23:26 -06:00
}
#[derive(Debug, Clone)]
2019-11-24 12:00:50 -06:00
struct ExprScope {
2019-11-21 03:21:46 -06:00
owner: DefWithBodyId,
2019-01-19 14:23:26 -06:00
expr_scopes: Arc<ExprScopes>,
scope_id: ScopeId,
}
#[derive(Debug, Clone)]
2019-11-24 12:00:50 -06:00
enum Scope {
2019-01-19 14:23:26 -06:00
/// All the items and imported names of a module
ModuleScope(ModuleItemMap),
/// Brings the generic parameters of an item into scope
2019-11-21 03:21:46 -06:00
GenericParams { def: GenericDefId, params: Arc<GenericParams> },
/// Brings `Self` in `impl` block into scope
2020-02-29 14:24:40 -06:00
ImplDefScope(ImplId),
/// Brings `Self` in enum, struct and union definitions into scope
2019-11-21 03:21:46 -06:00
AdtScope(AdtId),
2019-01-19 14:23:26 -06:00
/// Local bindings
ExprScope(ExprScope),
/// Temporary hack to support local items.
LocalItemsScope(Arc<Body>),
2019-01-19 14:23:26 -06:00
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
2019-11-21 06:39:09 -06:00
pub enum TypeNs {
2019-11-21 03:21:46 -06:00
SelfType(ImplId),
GenericParam(TypeParamId),
2019-11-21 03:21:46 -06:00
AdtId(AdtId),
AdtSelfType(AdtId),
2019-11-26 10:30:57 -06:00
// Yup, enum variants are added to the types ns, but any usage of variant as
// type is an error.
2019-11-21 03:21:46 -06:00
EnumVariantId(EnumVariantId),
TypeAliasId(TypeAliasId),
BuiltinType(BuiltinType),
2019-11-21 03:21:46 -06:00
TraitId(TraitId),
// Module belong to type ns, but the resolver is used when all module paths
// are fully resolved.
2019-11-21 03:21:46 -06:00
// ModuleId(ModuleId)
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
2019-11-21 06:39:09 -06:00
pub enum ResolveValueResult {
ValueNs(ValueNs),
Partial(TypeNs, usize),
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
2019-11-21 06:39:09 -06:00
pub enum ValueNs {
ImplSelf(ImplId),
LocalBinding(PatId),
2019-11-21 04:32:03 -06:00
FunctionId(FunctionId),
ConstId(ConstId),
StaticId(StaticId),
StructId(StructId),
EnumVariantId(EnumVariantId),
2021-01-01 03:06:42 -06:00
GenericParam(ConstParamId),
2019-01-19 14:23:26 -06:00
}
impl Resolver {
/// Resolve known trait from std, like `std::futures::Future`
pub fn resolve_known_trait(&self, db: &dyn DefDatabase, path: &ModPath) -> Option<TraitId> {
2019-11-30 09:29:21 -06:00
let res = self.resolve_module_path(db, path, BuiltinShadowMode::Other).take_types()?;
match res {
2019-11-21 03:21:46 -06:00
ModuleDefId::TraitId(it) => Some(it),
_ => None,
}
}
/// Resolve known struct from std, like `std::boxed::Box`
pub fn resolve_known_struct(&self, db: &dyn DefDatabase, path: &ModPath) -> Option<StructId> {
2019-11-30 09:29:21 -06:00
let res = self.resolve_module_path(db, path, BuiltinShadowMode::Other).take_types()?;
match res {
2019-11-21 03:21:46 -06:00
ModuleDefId::AdtId(AdtId::StructId(it)) => Some(it),
_ => None,
}
}
/// Resolve known enum from std, like `std::result::Result`
pub fn resolve_known_enum(&self, db: &dyn DefDatabase, path: &ModPath) -> Option<EnumId> {
2019-11-30 09:29:21 -06:00
let res = self.resolve_module_path(db, path, BuiltinShadowMode::Other).take_types()?;
match res {
2019-11-21 03:21:46 -06:00
ModuleDefId::AdtId(AdtId::EnumId(it)) => Some(it),
_ => None,
}
}
2019-11-30 09:29:21 -06:00
fn resolve_module_path(
&self,
db: &dyn DefDatabase,
path: &ModPath,
2019-11-30 09:29:21 -06:00
shadow: BuiltinShadowMode,
) -> PerNs {
2020-01-10 11:40:45 -06:00
let (item_map, module) = match self.module_scope() {
Some(it) => it,
None => return PerNs::none(),
};
let (module_res, segment_index) = item_map.resolve_path(db, module, &path, shadow);
if segment_index.is_some() {
return PerNs::none();
}
module_res
}
pub fn resolve_module_path_in_items(&self, db: &dyn DefDatabase, path: &ModPath) -> PerNs {
2019-11-30 09:29:21 -06:00
self.resolve_module_path(db, path, BuiltinShadowMode::Module)
}
2019-11-21 06:39:09 -06:00
pub fn resolve_path_in_type_ns(
&self,
db: &dyn DefDatabase,
path: &ModPath,
) -> Option<(TypeNs, Option<usize>)> {
let first_name = path.segments.first()?;
let skip_to_mod = path.kind != PathKind::Plain;
2019-01-19 14:23:26 -06:00
for scope in self.scopes.iter().rev() {
match scope {
Scope::ExprScope(_) => continue,
Scope::GenericParams { .. }
2020-02-29 14:24:40 -06:00
| Scope::ImplDefScope(_)
| Scope::LocalItemsScope(_)
if skip_to_mod =>
{
continue
}
Scope::GenericParams { params, def } => {
2021-01-01 03:06:42 -06:00
if let Some(local_id) = params.find_type_by_name(first_name) {
let idx = if path.segments.len() == 1 { None } else { Some(1) };
return Some((
TypeNs::GenericParam(TypeParamId { local_id, parent: *def }),
idx,
));
}
}
2020-02-29 14:24:40 -06:00
Scope::ImplDefScope(impl_) => {
2019-12-13 15:01:06 -06:00
if first_name == &name![Self] {
let idx = if path.segments.len() == 1 { None } else { Some(1) };
return Some((TypeNs::SelfType(*impl_), idx));
}
}
Scope::AdtScope(adt) => {
2019-12-13 15:01:06 -06:00
if first_name == &name![Self] {
let idx = if path.segments.len() == 1 { None } else { Some(1) };
return Some((TypeNs::AdtSelfType(*adt), idx));
}
}
Scope::ModuleScope(m) => {
2019-11-30 09:29:21 -06:00
let (module_def, idx) = m.crate_def_map.resolve_path(
db,
m.module_id,
&path,
2019-11-30 09:29:21 -06:00
BuiltinShadowMode::Other,
);
let res = to_type_ns(module_def)?;
return Some((res, idx));
}
Scope::LocalItemsScope(body) => {
let def = body.item_scope.get(first_name);
if let Some(res) = to_type_ns(def) {
return Some((res, None));
}
}
2019-01-19 14:23:26 -06:00
}
}
return None;
fn to_type_ns(per_ns: PerNs) -> Option<TypeNs> {
let res = match per_ns.take_types()? {
ModuleDefId::AdtId(it) => TypeNs::AdtId(it),
ModuleDefId::EnumVariantId(it) => TypeNs::EnumVariantId(it),
ModuleDefId::TypeAliasId(it) => TypeNs::TypeAliasId(it),
ModuleDefId::BuiltinType(it) => TypeNs::BuiltinType(it),
ModuleDefId::TraitId(it) => TypeNs::TraitId(it),
ModuleDefId::FunctionId(_)
| ModuleDefId::ConstId(_)
| ModuleDefId::StaticId(_)
| ModuleDefId::ModuleId(_) => return None,
};
Some(res)
}
2019-01-19 14:23:26 -06:00
}
2019-11-21 06:39:09 -06:00
pub fn resolve_path_in_type_ns_fully(
2019-06-01 06:34:19 -05:00
&self,
db: &dyn DefDatabase,
path: &ModPath,
) -> Option<TypeNs> {
let (res, unresolved) = self.resolve_path_in_type_ns(db, path)?;
if unresolved.is_some() {
return None;
}
Some(res)
2019-04-15 05:01:29 -05:00
}
pub fn resolve_visibility(
&self,
db: &dyn DefDatabase,
2019-12-26 08:57:14 -06:00
visibility: &RawVisibility,
) -> Option<Visibility> {
match visibility {
2019-12-26 08:57:14 -06:00
RawVisibility::Module(_) => {
2020-01-10 11:40:45 -06:00
let (item_map, module) = match self.module_scope() {
Some(it) => it,
None => return None,
};
item_map.resolve_visibility(db, module, visibility)
}
RawVisibility::Public => Some(Visibility::Public),
}
}
pub fn resolve_path_in_value_ns(
&self,
db: &dyn DefDatabase,
path: &ModPath,
) -> Option<ResolveValueResult> {
let n_segments = path.segments.len();
2019-12-13 15:01:06 -06:00
let tmp = name![self];
let first_name = if path.is_self() { &tmp } else { path.segments.first()? };
let skip_to_mod = path.kind != PathKind::Plain && !path.is_self();
for scope in self.scopes.iter().rev() {
match scope {
Scope::AdtScope(_)
| Scope::ExprScope(_)
2019-11-20 11:50:34 -06:00
| Scope::GenericParams { .. }
2020-02-29 14:24:40 -06:00
| Scope::ImplDefScope(_)
| Scope::LocalItemsScope(_)
if skip_to_mod =>
{
continue
}
Scope::ExprScope(scope) if n_segments <= 1 => {
let entry = scope
.expr_scopes
.entries(scope.scope_id)
.iter()
.find(|entry| entry.name() == first_name);
if let Some(e) = entry {
2019-09-11 13:01:07 -05:00
return Some(ResolveValueResult::ValueNs(ValueNs::LocalBinding(e.pat())));
}
}
Scope::ExprScope(_) => continue,
Scope::GenericParams { params, def } if n_segments > 1 => {
2021-01-01 03:06:42 -06:00
if let Some(local_id) = params.find_type_by_name(first_name) {
let ty = TypeNs::GenericParam(TypeParamId { local_id, parent: *def });
2019-09-11 13:01:07 -05:00
return Some(ResolveValueResult::Partial(ty, 1));
}
}
2021-01-01 03:06:42 -06:00
Scope::GenericParams { params, def } if n_segments == 1 => {
if let Some(local_id) = params.find_const_by_name(first_name) {
let val = ValueNs::GenericParam(ConstParamId { local_id, parent: *def });
return Some(ResolveValueResult::ValueNs(val));
}
}
2019-11-20 11:50:34 -06:00
Scope::GenericParams { .. } => continue,
Scope::ImplDefScope(impl_) => {
2019-12-13 15:01:06 -06:00
if first_name == &name![Self] {
if n_segments > 1 {
let ty = TypeNs::SelfType(*impl_);
return Some(ResolveValueResult::Partial(ty, 1));
} else {
return Some(ResolveValueResult::ValueNs(ValueNs::ImplSelf(*impl_)));
}
}
}
Scope::AdtScope(adt) => {
if n_segments == 1 {
// bare `Self` doesn't work in the value namespace in a struct/enum definition
continue;
}
2019-12-13 15:01:06 -06:00
if first_name == &name![Self] {
let ty = TypeNs::AdtSelfType(*adt);
return Some(ResolveValueResult::Partial(ty, 1));
}
}
Scope::ModuleScope(m) => {
2019-11-30 09:29:21 -06:00
let (module_def, idx) = m.crate_def_map.resolve_path(
db,
m.module_id,
&path,
2019-11-30 09:29:21 -06:00
BuiltinShadowMode::Other,
);
return match idx {
None => {
let value = to_value_ns(module_def)?;
2019-09-11 13:01:07 -05:00
Some(ResolveValueResult::ValueNs(value))
}
Some(idx) => {
let ty = match module_def.take_types()? {
2019-11-21 03:21:46 -06:00
ModuleDefId::AdtId(it) => TypeNs::AdtId(it),
ModuleDefId::TraitId(it) => TypeNs::TraitId(it),
ModuleDefId::TypeAliasId(it) => TypeNs::TypeAliasId(it),
2019-10-31 10:45:10 -05:00
ModuleDefId::BuiltinType(it) => TypeNs::BuiltinType(it),
ModuleDefId::ModuleId(_)
| ModuleDefId::FunctionId(_)
| ModuleDefId::EnumVariantId(_)
| ModuleDefId::ConstId(_)
| ModuleDefId::StaticId(_) => return None,
};
2019-09-11 13:01:07 -05:00
Some(ResolveValueResult::Partial(ty, idx))
}
};
}
Scope::LocalItemsScope(body) => {
// we don't bother looking in the builtin scope here because there are no builtin values
let def = to_value_ns(body.item_scope.get(first_name));
if let Some(res) = def {
return Some(ResolveValueResult::ValueNs(res));
}
}
}
2019-01-26 15:52:04 -06:00
}
return None;
fn to_value_ns(per_ns: PerNs) -> Option<ValueNs> {
let res = match per_ns.take_values()? {
ModuleDefId::FunctionId(it) => ValueNs::FunctionId(it),
ModuleDefId::AdtId(AdtId::StructId(it)) => ValueNs::StructId(it),
ModuleDefId::EnumVariantId(it) => ValueNs::EnumVariantId(it),
ModuleDefId::ConstId(it) => ValueNs::ConstId(it),
ModuleDefId::StaticId(it) => ValueNs::StaticId(it),
ModuleDefId::AdtId(AdtId::EnumId(_))
| ModuleDefId::AdtId(AdtId::UnionId(_))
| ModuleDefId::TraitId(_)
| ModuleDefId::TypeAliasId(_)
| ModuleDefId::BuiltinType(_)
| ModuleDefId::ModuleId(_) => return None,
};
Some(res)
}
2019-01-19 14:23:26 -06:00
}
2019-11-21 06:39:09 -06:00
pub fn resolve_path_in_value_ns_fully(
2019-06-08 10:38:14 -05:00
&self,
db: &dyn DefDatabase,
path: &ModPath,
) -> Option<ValueNs> {
match self.resolve_path_in_value_ns(db, path)? {
2019-09-11 13:01:07 -05:00
ResolveValueResult::ValueNs(it) => Some(it),
ResolveValueResult::Partial(..) => None,
}
}
pub fn resolve_path_as_macro(
&self,
db: &dyn DefDatabase,
path: &ModPath,
) -> Option<MacroDefId> {
2020-03-14 01:25:30 -05:00
// Search item scope legacy macro first
if let Some(def) = self.resolve_local_macro_def(path) {
return Some(def);
}
2020-01-10 11:40:45 -06:00
let (item_map, module) = self.module_scope()?;
item_map.resolve_path(db, module, &path, BuiltinShadowMode::Other).0.take_macros()
}
pub fn process_all_names(&self, db: &dyn DefDatabase, f: &mut dyn FnMut(Name, ScopeDef)) {
2019-02-01 16:06:57 -06:00
for scope in self.scopes.iter().rev() {
scope.process_names(db, f);
2019-01-27 13:50:57 -06:00
}
2019-01-19 14:23:26 -06:00
}
2019-01-26 15:52:04 -06:00
pub fn traits_in_scope(&self, db: &dyn DefDatabase) -> FxHashSet<TraitId> {
let mut traits = FxHashSet::default();
for scope in &self.scopes {
if let Scope::ModuleScope(m) = scope {
2019-11-24 09:05:12 -06:00
if let Some(prelude) = m.crate_def_map.prelude {
2019-10-31 10:45:10 -05:00
let prelude_def_map = db.crate_def_map(prelude.krate);
2019-11-27 12:31:51 -06:00
traits.extend(prelude_def_map[prelude.local_id].scope.traits());
2019-03-24 11:36:15 -05:00
}
2019-11-21 03:21:46 -06:00
traits.extend(m.crate_def_map[m.module_id].scope.traits());
}
}
traits
2019-03-24 11:36:15 -05:00
}
2020-01-10 11:40:45 -06:00
fn module_scope(&self) -> Option<(&CrateDefMap, LocalModuleId)> {
2019-01-29 13:49:31 -06:00
self.scopes.iter().rev().find_map(|scope| match scope {
2019-03-14 04:54:03 -05:00
Scope::ModuleScope(m) => Some((&*m.crate_def_map, m.module_id)),
2019-01-29 13:49:31 -06:00
_ => None,
})
}
2020-03-14 01:25:30 -05:00
fn resolve_local_macro_def(&self, path: &ModPath) -> Option<MacroDefId> {
let name = path.as_ident()?;
self.scopes.iter().rev().find_map(|scope| {
if let Scope::LocalItemsScope(body) = scope {
return body.item_scope.get_legacy_macro(name);
}
None
})
}
2020-01-10 11:40:45 -06:00
pub fn module(&self) -> Option<ModuleId> {
let (def_map, local_id) = self.module_scope()?;
Some(ModuleId { krate: def_map.krate, local_id })
}
2019-11-21 06:39:09 -06:00
pub fn krate(&self) -> Option<CrateId> {
2020-01-10 11:40:45 -06:00
self.module_scope().map(|t| t.0.krate)
}
2019-11-21 06:39:09 -06:00
pub fn where_predicates_in_scope<'a>(
&'a self,
) -> impl Iterator<Item = &'a crate::generics::WherePredicate> + 'a {
self.scopes
.iter()
.rev()
.filter_map(|scope| match scope {
2019-11-20 11:50:34 -06:00
Scope::GenericParams { params, .. } => Some(params),
_ => None,
})
.flat_map(|params| params.where_predicates.iter())
}
2019-11-21 06:39:09 -06:00
pub fn generic_def(&self) -> Option<GenericDefId> {
self.scopes.iter().rev().find_map(|scope| match scope {
2019-11-20 11:50:34 -06:00
Scope::GenericParams { def, .. } => Some(*def),
_ => None,
})
}
2019-11-21 06:39:09 -06:00
pub fn body_owner(&self) -> Option<DefWithBodyId> {
self.scopes.iter().rev().find_map(|scope| match scope {
Scope::ExprScope(it) => Some(it.owner),
_ => None,
})
}
2019-01-19 14:23:26 -06:00
}
2019-11-21 06:39:09 -06:00
pub enum ScopeDef {
PerNs(PerNs),
ImplSelfType(ImplId),
AdtSelfType(AdtId),
GenericParam(TypeParamId),
Local(PatId),
}
impl Scope {
fn process_names(&self, db: &dyn DefDatabase, f: &mut dyn FnMut(Name, ScopeDef)) {
let mut seen = FxHashSet::default();
2019-01-27 13:50:57 -06:00
match self {
Scope::ModuleScope(m) => {
2019-03-23 02:53:48 -05:00
// FIXME: should we provide `self` here?
2019-01-27 10:23:49 -06:00
// f(
// Name::self_param(),
// PerNs::types(Resolution::Def {
// def: m.module.into(),
// }),
// );
2019-12-22 08:37:07 -06:00
m.crate_def_map[m.module_id].scope.entries().for_each(|(name, def)| {
f(name.clone(), ScopeDef::PerNs(def));
2019-02-08 05:49:43 -06:00
});
m.crate_def_map[m.module_id].scope.legacy_macros().for_each(|(name, macro_)| {
let scope = PerNs::macros(macro_, Visibility::Public);
seen.insert((name.clone(), scope));
f(name.clone(), ScopeDef::PerNs(scope));
});
2019-11-24 09:05:12 -06:00
m.crate_def_map.extern_prelude.iter().for_each(|(name, &def)| {
f(name.clone(), ScopeDef::PerNs(PerNs::types(def, Visibility::Public)));
2019-02-04 15:09:56 -06:00
});
BUILTIN_SCOPE.iter().for_each(|(name, &def)| {
f(name.clone(), ScopeDef::PerNs(def));
});
2019-11-24 09:05:12 -06:00
if let Some(prelude) = m.crate_def_map.prelude {
2019-10-31 10:45:10 -05:00
let prelude_def_map = db.crate_def_map(prelude.krate);
2019-12-22 08:37:07 -06:00
prelude_def_map[prelude.local_id].scope.entries().for_each(|(name, def)| {
let seen_tuple = (name.clone(), def);
if !seen.contains(&seen_tuple) {
f(seen_tuple.0, ScopeDef::PerNs(def));
}
2019-10-31 10:45:10 -05:00
});
2019-02-13 13:53:42 -06:00
}
2019-01-27 13:50:57 -06:00
}
2020-06-30 07:06:23 -05:00
Scope::LocalItemsScope(body) => body.item_scope.entries().for_each(|(name, def)| {
f(name.clone(), ScopeDef::PerNs(def));
}),
Scope::GenericParams { params, def } => {
for (local_id, param) in params.types.iter() {
2020-01-24 12:35:09 -06:00
if let Some(name) = &param.name {
f(
name.clone(),
ScopeDef::GenericParam(TypeParamId { local_id, parent: *def }),
)
}
2019-01-27 13:50:57 -06:00
}
}
2020-02-29 14:24:40 -06:00
Scope::ImplDefScope(i) => {
f(name![Self], ScopeDef::ImplSelfType(*i));
}
Scope::AdtScope(i) => {
f(name![Self], ScopeDef::AdtSelfType(*i));
2019-01-27 13:50:57 -06:00
}
2019-11-15 03:00:36 -06:00
Scope::ExprScope(scope) => {
scope.expr_scopes.entries(scope.scope_id).iter().for_each(|e| {
f(e.name().clone(), ScopeDef::Local(e.pat()));
2019-01-27 13:50:57 -06:00
});
}
}
}
2019-01-19 14:23:26 -06:00
}
2019-11-20 12:55:33 -06:00
2019-11-21 04:28:04 -06:00
// needs arbitrary_self_types to be a method... or maybe move to the def?
pub fn resolver_for_expr(db: &dyn DefDatabase, owner: DefWithBodyId, expr_id: ExprId) -> Resolver {
2019-11-21 04:32:03 -06:00
let scopes = db.expr_scopes(owner);
2019-11-21 04:28:04 -06:00
resolver_for_scope(db, owner, scopes.scope_for(expr_id))
}
2019-11-21 06:39:09 -06:00
pub fn resolver_for_scope(
db: &dyn DefDatabase,
2019-11-21 04:32:03 -06:00
owner: DefWithBodyId,
2019-11-21 04:28:04 -06:00
scope_id: Option<ScopeId>,
) -> Resolver {
2019-11-21 05:13:49 -06:00
let mut r = owner.resolver(db);
r = r.push_local_items_scope(db.body(owner));
2019-11-21 04:32:03 -06:00
let scopes = db.expr_scopes(owner);
2019-11-21 04:28:04 -06:00
let scope_chain = scopes.scope_chain(scope_id).collect::<Vec<_>>();
for scope in scope_chain.into_iter().rev() {
2019-11-21 04:32:03 -06:00
r = r.push_expr_scope(owner, Arc::clone(&scopes), scope);
2019-11-21 04:28:04 -06:00
}
r
}
2019-11-21 07:00:57 -06:00
impl Resolver {
fn push_scope(mut self, scope: Scope) -> Resolver {
self.scopes.push(scope);
self
}
fn push_generic_params_scope(self, db: &dyn DefDatabase, def: GenericDefId) -> Resolver {
2019-11-21 07:00:57 -06:00
let params = db.generic_params(def);
self.push_scope(Scope::GenericParams { def, params })
2019-11-21 07:00:57 -06:00
}
2020-02-29 14:24:40 -06:00
fn push_impl_def_scope(self, impl_def: ImplId) -> Resolver {
self.push_scope(Scope::ImplDefScope(impl_def))
2019-11-21 07:00:57 -06:00
}
fn push_module_scope(
self,
crate_def_map: Arc<CrateDefMap>,
2019-11-23 07:49:53 -06:00
module_id: LocalModuleId,
2019-11-21 07:00:57 -06:00
) -> Resolver {
self.push_scope(Scope::ModuleScope(ModuleItemMap { crate_def_map, module_id }))
}
fn push_local_items_scope(self, body: Arc<Body>) -> Resolver {
self.push_scope(Scope::LocalItemsScope(body))
}
2019-11-21 07:00:57 -06:00
fn push_expr_scope(
self,
owner: DefWithBodyId,
expr_scopes: Arc<ExprScopes>,
scope_id: ScopeId,
) -> Resolver {
self.push_scope(Scope::ExprScope(ExprScope { owner, expr_scopes, scope_id }))
}
}
2019-11-26 13:56:07 -06:00
pub trait HasResolver: Copy {
2019-11-20 12:55:33 -06:00
/// Builds a resolver for type references inside this def.
fn resolver(self, db: &dyn DefDatabase) -> Resolver;
2019-11-20 12:55:33 -06:00
}
2019-11-21 05:13:49 -06:00
impl HasResolver for ModuleId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-11-21 05:13:49 -06:00
let def_map = db.crate_def_map(self.krate);
2019-11-27 12:31:51 -06:00
Resolver::default().push_module_scope(def_map, self.local_id)
2019-11-20 12:55:33 -06:00
}
}
2019-11-21 05:13:49 -06:00
impl HasResolver for TraitId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-12-12 07:34:03 -06:00
self.lookup(db).container.resolver(db).push_generic_params_scope(db, self.into())
2019-11-20 12:55:33 -06:00
}
}
2019-11-26 13:56:07 -06:00
impl<T: Into<AdtId> + Copy> HasResolver for T {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-11-21 06:56:27 -06:00
let def = self.into();
2019-11-24 13:47:58 -06:00
def.module(db)
2019-11-20 12:55:33 -06:00
.resolver(db)
2019-11-21 06:56:27 -06:00
.push_generic_params_scope(db, def.into())
.push_scope(Scope::AdtScope(def))
2019-11-21 05:13:49 -06:00
}
}
impl HasResolver for FunctionId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-11-21 05:13:49 -06:00
self.lookup(db).container.resolver(db).push_generic_params_scope(db, self.into())
}
}
impl HasResolver for ConstId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-11-21 05:13:49 -06:00
self.lookup(db).container.resolver(db)
2019-11-20 12:55:33 -06:00
}
}
2019-11-21 05:13:49 -06:00
impl HasResolver for StaticId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-11-24 06:13:56 -06:00
self.lookup(db).container.resolver(db)
2019-11-20 12:55:33 -06:00
}
}
2019-11-21 05:13:49 -06:00
impl HasResolver for TypeAliasId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-11-21 05:13:49 -06:00
self.lookup(db).container.resolver(db).push_generic_params_scope(db, self.into())
2019-11-20 12:55:33 -06:00
}
}
2019-12-08 05:16:57 -06:00
impl HasResolver for ImplId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-12-12 07:09:13 -06:00
self.lookup(db)
.container
2019-12-08 05:16:57 -06:00
.resolver(db)
.push_generic_params_scope(db, self.into())
2020-02-29 14:24:40 -06:00
.push_impl_def_scope(self)
2019-12-08 05:16:57 -06:00
}
}
impl HasResolver for DefWithBodyId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-12-08 05:16:57 -06:00
match self {
DefWithBodyId::ConstId(c) => c.resolver(db),
DefWithBodyId::FunctionId(f) => f.resolver(db),
DefWithBodyId::StaticId(s) => s.resolver(db),
}
}
}
2019-12-20 05:07:23 -06:00
impl HasResolver for ContainerId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-12-20 05:07:23 -06:00
match self {
ContainerId::ModuleId(it) => it.resolver(db),
ContainerId::DefWithBodyId(it) => it.module(db).resolver(db),
2019-12-20 05:07:23 -06:00
}
}
}
2019-12-20 04:59:50 -06:00
impl HasResolver for AssocContainerId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-11-20 12:55:33 -06:00
match self {
2019-12-20 05:07:23 -06:00
AssocContainerId::ContainerId(it) => it.resolver(db),
2019-12-20 04:59:50 -06:00
AssocContainerId::TraitId(it) => it.resolver(db),
AssocContainerId::ImplId(it) => it.resolver(db),
2019-11-20 12:55:33 -06:00
}
}
}
2019-11-21 05:13:49 -06:00
impl HasResolver for GenericDefId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-11-20 12:55:33 -06:00
match self {
2019-11-21 05:13:49 -06:00
GenericDefId::FunctionId(inner) => inner.resolver(db),
GenericDefId::AdtId(adt) => adt.resolver(db),
GenericDefId::TraitId(inner) => inner.resolver(db),
GenericDefId::TypeAliasId(inner) => inner.resolver(db),
GenericDefId::ImplId(inner) => inner.resolver(db),
GenericDefId::EnumVariantId(inner) => inner.parent.resolver(db),
GenericDefId::ConstId(inner) => inner.resolver(db),
2019-11-20 12:55:33 -06:00
}
}
}
2019-12-08 05:16:57 -06:00
impl HasResolver for VariantId {
fn resolver(self, db: &dyn DefDatabase) -> Resolver {
2019-12-08 05:16:57 -06:00
match self {
VariantId::EnumVariantId(it) => it.parent.resolver(db),
VariantId::StructId(it) => it.resolver(db),
VariantId::UnionId(it) => it.resolver(db),
}
2019-11-21 05:13:49 -06:00
}
}