write path resolution code only once

This commit is contained in:
Aleksey Kladov 2019-01-25 10:15:10 +03:00
parent 857c35ddb0
commit e884ab05c2
3 changed files with 15 additions and 65 deletions

View File

@ -3,7 +3,7 @@
use crate::{ use crate::{
Module, ModuleSource, Problem, ModuleDef, Module, ModuleSource, Problem, ModuleDef,
Crate, Name, Path, PathKind, PerNs, Crate, Name, Path, PerNs,
module_tree::ModuleId, module_tree::ModuleId,
nameres::{ModuleScope, lower::ImportId}, nameres::{ModuleScope, lower::ImportId},
db::HirDatabase, db::HirDatabase,
@ -97,65 +97,8 @@ pub(crate) fn scope_impl(&self, db: &impl HirDatabase) -> ModuleScope {
} }
pub(crate) fn resolve_path_impl(&self, db: &impl HirDatabase, path: &Path) -> PerNs<ModuleDef> { pub(crate) fn resolve_path_impl(&self, db: &impl HirDatabase, path: &Path) -> PerNs<ModuleDef> {
let mut curr_per_ns: PerNs<ModuleDef> = PerNs::types(match path.kind { let item_map = db.item_map(self.krate);
PathKind::Crate => self.crate_root(db).into(), item_map.resolve_path(db, *self, path)
PathKind::Self_ | PathKind::Plain => self.clone().into(),
PathKind::Super => {
if let Some(p) = self.parent(db) {
p.into()
} else {
return PerNs::none();
}
}
PathKind::Abs => {
// TODO: absolute use is not supported
return PerNs::none();
}
});
for segment in path.segments.iter() {
let curr = match curr_per_ns.as_ref().take_types() {
Some(r) => r,
None => {
// we still have path segments left, but the path so far
// didn't resolve in the types namespace => no resolution
// (don't break here because curr_per_ns might contain
// something in the value namespace, and it would be wrong
// to return that)
return PerNs::none();
}
};
// resolve segment in curr
curr_per_ns = match curr {
ModuleDef::Module(m) => {
let scope = m.scope(db);
match scope.get(&segment.name) {
Some(r) => r.def_id.clone(),
None => PerNs::none(),
}
}
ModuleDef::Enum(e) => {
// enum variant
let matching_variant = e
.variants(db)
.into_iter()
.find(|(n, _variant)| n == &segment.name);
match matching_variant {
Some((_n, variant)) => PerNs::both(variant.into(), (*e).into()),
None => PerNs::none(),
}
}
_ => {
// could be an inherent method call in UFCS form
// (`Struct::method`), or some other kind of associated
// item... Which we currently don't handle (TODO)
PerNs::none()
}
};
}
curr_per_ns
} }
pub(crate) fn problems_impl( pub(crate) fn problems_impl(

View File

@ -297,7 +297,7 @@ fn resolve_import(
}; };
let (def_id, reached_fixedpoint) = let (def_id, reached_fixedpoint) =
self.result self.result
.resolve_path(self.db, original_module, &import.path); .resolve_path_fp(self.db, original_module, &import.path);
if reached_fixedpoint == ReachedFixedPoint::Yes { if reached_fixedpoint == ReachedFixedPoint::Yes {
let last_segment = import.path.segments.last().unwrap(); let last_segment = import.path.segments.last().unwrap();
@ -331,10 +331,19 @@ enum ReachedFixedPoint {
} }
impl ItemMap { impl ItemMap {
pub(crate) fn resolve_path(
&self,
db: &impl HirDatabase,
original_module: Module,
path: &Path,
) -> PerNs<ModuleDef> {
self.resolve_path_fp(db, original_module, path).0
}
// returns true if we are sure that additions to `ItemMap` wouldn't change // returns true if we are sure that additions to `ItemMap` wouldn't change
// the result. That is, if we've reached fixed point at this particular // the result. That is, if we've reached fixed point at this particular
// import. // import.
fn resolve_path( fn resolve_path_fp(
&self, &self,
db: &impl HirDatabase, db: &impl HirDatabase,
original_module: Module, original_module: Module,

View File

@ -1,3 +1 @@
use test_utils::mark; test_utils::marks!(inserts_parens_for_function_calls);
mark!(inserts_parens_for_function_calls);