2019-01-19 20:23:26 +00:00
|
|
|
//! Name resolution.
|
|
|
|
|
use std::sync::Arc;
|
|
|
|
|
|
|
|
|
|
use rustc_hash::FxHashMap;
|
|
|
|
|
|
|
|
|
|
use crate::{
|
2019-01-26 21:52:04 +00:00
|
|
|
ModuleDef, Module,
|
|
|
|
|
db::HirDatabase,
|
|
|
|
|
name::{Name, KnownName},
|
|
|
|
|
nameres::{PerNs, ItemMap},
|
2019-01-19 20:23:26 +00:00
|
|
|
generics::GenericParams,
|
2019-01-29 19:49:31 +00:00
|
|
|
expr::{scope::{ExprScopes, ScopeId}, PatId, Body},
|
2019-01-19 20:23:26 +00:00
|
|
|
impl_block::ImplBlock,
|
|
|
|
|
path::Path,
|
|
|
|
|
};
|
|
|
|
|
|
2019-01-23 22:08:41 +00:00
|
|
|
#[derive(Debug, Clone, Default)]
|
2019-01-19 20:23:26 +00:00
|
|
|
pub struct Resolver {
|
2019-01-26 21:52:04 +00:00
|
|
|
scopes: Vec<Scope>,
|
2019-01-19 20:23:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// TODO how to store these best
|
|
|
|
|
#[derive(Debug, Clone)]
|
|
|
|
|
pub(crate) struct ModuleItemMap {
|
|
|
|
|
item_map: Arc<ItemMap>,
|
2019-01-26 21:52:04 +00:00
|
|
|
module: Module,
|
2019-01-19 20:23:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[derive(Debug, Clone)]
|
|
|
|
|
pub(crate) struct ExprScope {
|
|
|
|
|
expr_scopes: Arc<ExprScopes>,
|
|
|
|
|
scope_id: ScopeId,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[derive(Debug, Clone)]
|
|
|
|
|
pub(crate) enum Scope {
|
|
|
|
|
/// All the items and imported names of a module
|
|
|
|
|
ModuleScope(ModuleItemMap),
|
|
|
|
|
/// Brings the generic parameters of an item into scope
|
|
|
|
|
GenericParams(Arc<GenericParams>),
|
|
|
|
|
/// Brings `Self` into scope
|
|
|
|
|
ImplBlockScope(ImplBlock),
|
|
|
|
|
/// Local bindings
|
|
|
|
|
ExprScope(ExprScope),
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq)]
|
|
|
|
|
pub enum Resolution {
|
|
|
|
|
/// An item
|
2019-01-30 21:41:44 +00:00
|
|
|
Def(ModuleDef),
|
2019-01-19 20:23:26 +00:00
|
|
|
/// A local binding (only value namespace)
|
2019-01-30 21:41:44 +00:00
|
|
|
LocalBinding(PatId),
|
2019-01-19 20:23:26 +00:00
|
|
|
/// A generic parameter
|
2019-01-30 21:41:44 +00:00
|
|
|
GenericParam(u32),
|
2019-01-26 21:52:04 +00:00
|
|
|
SelfType(ImplBlock),
|
2019-01-19 20:23:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl Resolver {
|
2019-02-10 19:44:34 +00:00
|
|
|
pub fn resolve_name(&self, db: &impl HirDatabase, name: &Name) -> PerNs<Resolution> {
|
2019-01-26 21:52:04 +00:00
|
|
|
let mut resolution = PerNs::none();
|
2019-01-19 20:23:26 +00:00
|
|
|
for scope in self.scopes.iter().rev() {
|
2019-02-10 19:44:34 +00:00
|
|
|
resolution = resolution.or(scope.resolve_name(db, name));
|
2019-01-26 21:52:04 +00:00
|
|
|
if resolution.is_both() {
|
2019-01-19 20:23:26 +00:00
|
|
|
return resolution;
|
|
|
|
|
}
|
|
|
|
|
}
|
2019-01-26 21:52:04 +00:00
|
|
|
resolution
|
2019-01-19 20:23:26 +00:00
|
|
|
}
|
|
|
|
|
|
2019-01-26 21:52:04 +00:00
|
|
|
pub fn resolve_path(&self, db: &impl HirDatabase, path: &Path) -> PerNs<Resolution> {
|
|
|
|
|
if let Some(name) = path.as_ident() {
|
2019-02-10 19:44:34 +00:00
|
|
|
self.resolve_name(db, name)
|
2019-01-26 21:52:04 +00:00
|
|
|
} else if path.is_self() {
|
2019-02-10 19:44:34 +00:00
|
|
|
self.resolve_name(db, &Name::self_param())
|
2019-01-26 21:52:04 +00:00
|
|
|
} else {
|
|
|
|
|
let (item_map, module) = match self.module() {
|
|
|
|
|
Some(m) => m,
|
|
|
|
|
_ => return PerNs::none(),
|
|
|
|
|
};
|
|
|
|
|
let module_res = item_map.resolve_path(db, module, path);
|
2019-02-06 20:50:26 +00:00
|
|
|
module_res.map(Resolution::Def)
|
2019-01-26 21:52:04 +00:00
|
|
|
}
|
2019-01-19 20:23:26 +00:00
|
|
|
}
|
|
|
|
|
|
2019-02-13 19:53:42 +00:00
|
|
|
pub fn all_names(&self, db: &impl HirDatabase) -> FxHashMap<Name, PerNs<Resolution>> {
|
2019-01-27 19:50:57 +00:00
|
|
|
let mut names = FxHashMap::default();
|
2019-02-01 22:06:57 +00:00
|
|
|
for scope in self.scopes.iter().rev() {
|
2019-02-13 19:53:42 +00:00
|
|
|
scope.collect_names(db, &mut |name, res| {
|
2019-01-27 19:50:57 +00:00
|
|
|
let current: &mut PerNs<Resolution> = names.entry(name).or_default();
|
|
|
|
|
if current.types.is_none() {
|
|
|
|
|
current.types = res.types;
|
|
|
|
|
}
|
|
|
|
|
if current.values.is_none() {
|
|
|
|
|
current.values = res.values;
|
|
|
|
|
}
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
names
|
2019-01-19 20:23:26 +00:00
|
|
|
}
|
2019-01-26 21:52:04 +00:00
|
|
|
|
|
|
|
|
fn module(&self) -> Option<(&ItemMap, Module)> {
|
2019-01-29 19:49:31 +00:00
|
|
|
self.scopes.iter().rev().find_map(|scope| match scope {
|
|
|
|
|
Scope::ModuleScope(m) => Some((&*m.item_map, m.module.clone())),
|
|
|
|
|
|
|
|
|
|
_ => None,
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// The body from which any `LocalBinding` resolutions in this resolver come.
|
|
|
|
|
pub fn body(&self) -> Option<Arc<Body>> {
|
|
|
|
|
self.scopes.iter().rev().find_map(|scope| match scope {
|
|
|
|
|
Scope::ExprScope(expr_scope) => Some(expr_scope.expr_scopes.body()),
|
|
|
|
|
_ => None,
|
|
|
|
|
})
|
2019-01-26 21:52:04 +00:00
|
|
|
}
|
2019-01-19 20:23:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl Resolver {
|
|
|
|
|
pub(crate) fn push_scope(mut self, scope: Scope) -> Resolver {
|
|
|
|
|
self.scopes.push(scope);
|
|
|
|
|
self
|
|
|
|
|
}
|
|
|
|
|
|
2019-01-23 22:08:41 +00:00
|
|
|
pub(crate) fn push_generic_params_scope(self, params: Arc<GenericParams>) -> Resolver {
|
|
|
|
|
self.push_scope(Scope::GenericParams(params))
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pub(crate) fn push_impl_block_scope(self, impl_block: ImplBlock) -> Resolver {
|
|
|
|
|
self.push_scope(Scope::ImplBlockScope(impl_block))
|
|
|
|
|
}
|
|
|
|
|
|
2019-01-26 21:52:04 +00:00
|
|
|
pub(crate) fn push_module_scope(self, item_map: Arc<ItemMap>, module: Module) -> Resolver {
|
|
|
|
|
self.push_scope(Scope::ModuleScope(ModuleItemMap { item_map, module }))
|
2019-01-23 22:08:41 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
pub(crate) fn push_expr_scope(
|
|
|
|
|
self,
|
|
|
|
|
expr_scopes: Arc<ExprScopes>,
|
|
|
|
|
scope_id: ScopeId,
|
|
|
|
|
) -> Resolver {
|
2019-02-08 11:49:43 +00:00
|
|
|
self.push_scope(Scope::ExprScope(ExprScope { expr_scopes, scope_id }))
|
2019-01-23 22:08:41 +00:00
|
|
|
}
|
2019-01-19 20:23:26 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
impl Scope {
|
2019-02-10 19:44:34 +00:00
|
|
|
fn resolve_name(&self, db: &impl HirDatabase, name: &Name) -> PerNs<Resolution> {
|
2019-01-26 21:52:04 +00:00
|
|
|
match self {
|
|
|
|
|
Scope::ModuleScope(m) => {
|
|
|
|
|
if let Some(KnownName::SelfParam) = name.as_known_name() {
|
2019-01-30 21:41:44 +00:00
|
|
|
PerNs::types(Resolution::Def(m.module.into()))
|
2019-01-26 21:52:04 +00:00
|
|
|
} else {
|
2019-02-10 19:44:34 +00:00
|
|
|
m.item_map.resolve_name_in_module(db, m.module, name).map(Resolution::Def)
|
2019-01-26 21:52:04 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
Scope::GenericParams(gp) => match gp.find_by_name(name) {
|
2019-01-30 21:41:44 +00:00
|
|
|
Some(gp) => PerNs::types(Resolution::GenericParam(gp.idx)),
|
2019-01-26 21:52:04 +00:00
|
|
|
None => PerNs::none(),
|
|
|
|
|
},
|
|
|
|
|
Scope::ImplBlockScope(i) => {
|
|
|
|
|
if name.as_known_name() == Some(KnownName::SelfType) {
|
|
|
|
|
PerNs::types(Resolution::SelfType(i.clone()))
|
|
|
|
|
} else {
|
|
|
|
|
PerNs::none()
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
Scope::ExprScope(e) => {
|
2019-02-08 11:49:43 +00:00
|
|
|
let entry =
|
|
|
|
|
e.expr_scopes.entries(e.scope_id).iter().find(|entry| entry.name() == name);
|
2019-01-26 21:52:04 +00:00
|
|
|
match entry {
|
2019-01-30 21:41:44 +00:00
|
|
|
Some(e) => PerNs::values(Resolution::LocalBinding(e.pat())),
|
2019-01-26 21:52:04 +00:00
|
|
|
None => PerNs::none(),
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
2019-01-19 20:23:26 +00:00
|
|
|
}
|
2019-01-27 19:50:57 +00:00
|
|
|
|
2019-02-13 19:53:42 +00:00
|
|
|
fn collect_names(&self, db: &impl HirDatabase, f: &mut dyn FnMut(Name, PerNs<Resolution>)) {
|
2019-01-27 19:50:57 +00:00
|
|
|
match self {
|
|
|
|
|
Scope::ModuleScope(m) => {
|
2019-01-27 16:23:49 +00:00
|
|
|
// TODO: should we provide `self` here?
|
|
|
|
|
// f(
|
|
|
|
|
// Name::self_param(),
|
|
|
|
|
// PerNs::types(Resolution::Def {
|
|
|
|
|
// def: m.module.into(),
|
|
|
|
|
// }),
|
|
|
|
|
// );
|
2019-02-08 11:49:43 +00:00
|
|
|
m.item_map[m.module.module_id].entries().for_each(|(name, res)| {
|
|
|
|
|
f(name.clone(), res.def.map(Resolution::Def));
|
|
|
|
|
});
|
2019-02-04 21:09:56 +00:00
|
|
|
m.item_map.extern_prelude.iter().for_each(|(name, def)| {
|
|
|
|
|
f(name.clone(), PerNs::types(Resolution::Def(*def)));
|
|
|
|
|
});
|
2019-02-13 19:53:42 +00:00
|
|
|
if let Some(prelude) = m.item_map.prelude {
|
|
|
|
|
let prelude_item_map = db.item_map(prelude.krate);
|
|
|
|
|
prelude_item_map[prelude.module_id].entries().for_each(|(name, res)| {
|
|
|
|
|
f(name.clone(), res.def.map(Resolution::Def));
|
|
|
|
|
});
|
|
|
|
|
}
|
2019-01-27 19:50:57 +00:00
|
|
|
}
|
|
|
|
|
Scope::GenericParams(gp) => {
|
|
|
|
|
for param in &gp.params {
|
2019-02-08 11:49:43 +00:00
|
|
|
f(param.name.clone(), PerNs::types(Resolution::GenericParam(param.idx)))
|
2019-01-27 19:50:57 +00:00
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
Scope::ImplBlockScope(i) => {
|
2019-02-08 11:49:43 +00:00
|
|
|
f(Name::self_type(), PerNs::types(Resolution::SelfType(i.clone())));
|
2019-01-27 19:50:57 +00:00
|
|
|
}
|
|
|
|
|
Scope::ExprScope(e) => {
|
|
|
|
|
e.expr_scopes.entries(e.scope_id).iter().for_each(|e| {
|
2019-02-08 11:49:43 +00:00
|
|
|
f(e.name().clone(), PerNs::values(Resolution::LocalBinding(e.pat())));
|
2019-01-27 19:50:57 +00:00
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
2019-01-19 20:23:26 +00:00
|
|
|
}
|
