This commit is contained in:
Aleksey Kladov 2019-11-26 22:56:07 +03:00
parent cace49e9a7
commit bed6869865
6 changed files with 85 additions and 77 deletions

View file

@ -28,7 +28,8 @@ use crate::{
expr::{BindingAnnotation, Body, BodySourceMap, ExprValidator, Pat, PatId}, expr::{BindingAnnotation, Body, BodySourceMap, ExprValidator, Pat, PatId},
ty::display::HirFormatter, ty::display::HirFormatter,
ty::{ ty::{
self, InEnvironment, InferenceResult, TraitEnvironment, TraitRef, Ty, TypeCtor, TypeWalk, self, InEnvironment, InferenceResult, TraitEnvironment, TraitRef, Ty, TyDefId, TypeCtor,
TypeWalk,
}, },
CallableDef, Either, HirDisplay, Name, Source, CallableDef, Either, HirDisplay, Name, Source,
}; };
@ -498,12 +499,9 @@ impl Adt {
let subst = db.generic_defaults(self.into()); let subst = db.generic_defaults(self.into());
subst.iter().any(|ty| ty == &Ty::Unknown) subst.iter().any(|ty| ty == &Ty::Unknown)
} }
pub fn ty(self, db: &impl HirDatabase) -> Ty { pub fn ty(self, db: &impl HirDatabase) -> Type {
match self { let id = AdtId::from(self);
Adt::Struct(it) => it.ty(db), Type::from_def(db, id.module(db).krate, id)
Adt::Union(it) => it.ty(db),
Adt::Enum(it) => it.ty(db),
}
} }
pub fn module(self, db: &impl DefDatabase) -> Module { pub fn module(self, db: &impl DefDatabase) -> Module {
@ -795,8 +793,8 @@ impl TypeAlias {
db.type_alias_data(self.id).type_ref.clone() db.type_alias_data(self.id).type_ref.clone()
} }
pub fn ty(self, db: &impl HirDatabase) -> Ty { pub fn ty(self, db: &impl HirDatabase) -> Type {
db.ty(self.id.into()) Type::from_def(db, self.id.lookup(db).module(db).krate, self.id)
} }
pub fn name(self, db: &impl DefDatabase) -> Name { pub fn name(self, db: &impl DefDatabase) -> Name {
@ -989,6 +987,17 @@ pub struct Type {
} }
impl Type { impl Type {
fn from_def(
db: &impl HirDatabase,
krate: CrateId,
def: impl HasResolver + Into<TyDefId>,
) -> Type {
let resolver = def.resolver(db);
let environment = TraitEnvironment::lower(db, &resolver);
let ty = db.ty(def.into());
Type { krate, ty: InEnvironment { value: ty, environment } }
}
pub fn is_bool(&self) -> bool { pub fn is_bool(&self) -> bool {
match &self.ty.value { match &self.ty.value {
Ty::Apply(a_ty) => match a_ty.ctor { Ty::Apply(a_ty) => match a_ty.ctor {
@ -1097,6 +1106,28 @@ impl Type {
.map(move |ty| self.derived(ty)) .map(move |ty| self.derived(ty))
} }
// This would be nicer if it just returned an iterator, but that runs into
// lifetime problems, because we need to borrow temp `CrateImplBlocks`.
pub fn iterate_impl_items<T>(
self,
db: &impl HirDatabase,
krate: Crate,
mut callback: impl FnMut(AssocItem) -> Option<T>,
) -> Option<T> {
for krate in self.ty.value.def_crates(db, krate.crate_id)? {
let impls = db.impls_in_crate(krate);
for impl_block in impls.lookup_impl_blocks(&self.ty.value) {
for &item in db.impl_data(impl_block).items.iter() {
if let Some(result) = callback(item.into()) {
return Some(result);
}
}
}
}
None
}
// FIXME: remove // FIXME: remove
pub fn into_ty(self) -> Ty { pub fn into_ty(self) -> Ty {
self.ty.value self.ty.value

View file

@ -389,14 +389,14 @@ impl SourceAnalyzer {
pub fn iterate_path_candidates<T>( pub fn iterate_path_candidates<T>(
&self, &self,
db: &impl HirDatabase, db: &impl HirDatabase,
ty: Ty, ty: &Type,
name: Option<&Name>, name: Option<&Name>,
callback: impl FnMut(&Ty, AssocItem) -> Option<T>, callback: impl FnMut(&Ty, AssocItem) -> Option<T>,
) -> Option<T> { ) -> Option<T> {
// There should be no inference vars in types passed here // There should be no inference vars in types passed here
// FIXME check that? // FIXME check that?
// FIXME replace Unknown by bound vars here // FIXME replace Unknown by bound vars here
let canonical = crate::ty::Canonical { value: ty, num_vars: 0 }; let canonical = crate::ty::Canonical { value: ty.ty.value.clone(), num_vars: 0 };
method_resolution::iterate_method_candidates( method_resolution::iterate_method_candidates(
&canonical, &canonical,
db, db,

View file

@ -189,7 +189,7 @@ impl Ty {
Ty::Param { idx, name } Ty::Param { idx, name }
} }
TypeNs::SelfType(impl_block) => ImplBlock::from(impl_block).target_ty(db), TypeNs::SelfType(impl_block) => ImplBlock::from(impl_block).target_ty(db),
TypeNs::AdtSelfType(adt) => Adt::from(adt).ty(db), TypeNs::AdtSelfType(adt) => db.ty(adt.into()),
TypeNs::AdtId(it) => Ty::from_hir_path_inner(db, resolver, resolved_segment, it.into()), TypeNs::AdtId(it) => Ty::from_hir_path_inner(db, resolver, resolved_segment, it.into()),
TypeNs::BuiltinType(it) => { TypeNs::BuiltinType(it) => {

View file

@ -97,14 +97,15 @@ impl CrateImplBlocks {
} }
} }
fn def_crates( impl Ty {
db: &impl HirDatabase, pub(crate) fn def_crates(
cur_crate: CrateId, &self,
ty: &Ty, db: &impl HirDatabase,
) -> Option<ArrayVec<[CrateId; 2]>> { cur_crate: CrateId,
// Types like slice can have inherent impls in several crates, (core and alloc). ) -> Option<ArrayVec<[CrateId; 2]>> {
// The corresponding impls are marked with lang items, so we can use them to find the required crates. // Types like slice can have inherent impls in several crates, (core and alloc).
macro_rules! lang_item_crate { // The corresponding impls are marked with lang items, so we can use them to find the required crates.
macro_rules! lang_item_crate {
($($name:expr),+ $(,)?) => {{ ($($name:expr),+ $(,)?) => {{
let mut v = ArrayVec::<[LangItemTarget; 2]>::new(); let mut v = ArrayVec::<[LangItemTarget; 2]>::new();
$( $(
@ -114,38 +115,38 @@ fn def_crates(
}}; }};
} }
let lang_item_targets = match ty { let lang_item_targets = match self {
Ty::Apply(a_ty) => match a_ty.ctor { Ty::Apply(a_ty) => match a_ty.ctor {
TypeCtor::Adt(def_id) => { TypeCtor::Adt(def_id) => {
return Some(std::iter::once(def_id.module(db).krate).collect()) return Some(std::iter::once(def_id.module(db).krate).collect())
} }
TypeCtor::Bool => lang_item_crate!("bool"), TypeCtor::Bool => lang_item_crate!("bool"),
TypeCtor::Char => lang_item_crate!("char"), TypeCtor::Char => lang_item_crate!("char"),
TypeCtor::Float(Uncertain::Known(f)) => match f.bitness { TypeCtor::Float(Uncertain::Known(f)) => match f.bitness {
// There are two lang items: one in libcore (fXX) and one in libstd (fXX_runtime) // There are two lang items: one in libcore (fXX) and one in libstd (fXX_runtime)
FloatBitness::X32 => lang_item_crate!("f32", "f32_runtime"), FloatBitness::X32 => lang_item_crate!("f32", "f32_runtime"),
FloatBitness::X64 => lang_item_crate!("f64", "f64_runtime"), FloatBitness::X64 => lang_item_crate!("f64", "f64_runtime"),
},
TypeCtor::Int(Uncertain::Known(i)) => lang_item_crate!(i.ty_to_string()),
TypeCtor::Str => lang_item_crate!("str_alloc", "str"),
TypeCtor::Slice => lang_item_crate!("slice_alloc", "slice"),
TypeCtor::RawPtr(Mutability::Shared) => lang_item_crate!("const_ptr"),
TypeCtor::RawPtr(Mutability::Mut) => lang_item_crate!("mut_ptr"),
_ => return None,
}, },
TypeCtor::Int(Uncertain::Known(i)) => lang_item_crate!(i.ty_to_string()),
TypeCtor::Str => lang_item_crate!("str_alloc", "str"),
TypeCtor::Slice => lang_item_crate!("slice_alloc", "slice"),
TypeCtor::RawPtr(Mutability::Shared) => lang_item_crate!("const_ptr"),
TypeCtor::RawPtr(Mutability::Mut) => lang_item_crate!("mut_ptr"),
_ => return None, _ => return None,
}, };
_ => return None, let res = lang_item_targets
}; .into_iter()
let res = lang_item_targets .filter_map(|it| match it {
.into_iter() LangItemTarget::ImplBlockId(it) => Some(it),
.filter_map(|it| match it { _ => None,
LangItemTarget::ImplBlockId(it) => Some(it), })
_ => None, .map(|it| it.module(db).krate)
}) .collect();
.map(|it| it.module(db).krate) Some(res)
.collect(); }
Some(res)
} }
/// Look up the method with the given name, returning the actual autoderefed /// Look up the method with the given name, returning the actual autoderefed
/// receiver type (but without autoref applied yet). /// receiver type (but without autoref applied yet).
pub(crate) fn lookup_method( pub(crate) fn lookup_method(
@ -286,7 +287,7 @@ fn iterate_inherent_methods<T>(
krate: CrateId, krate: CrateId,
mut callback: impl FnMut(&Ty, AssocItem) -> Option<T>, mut callback: impl FnMut(&Ty, AssocItem) -> Option<T>,
) -> Option<T> { ) -> Option<T> {
for krate in def_crates(db, krate, &ty.value)? { for krate in ty.value.def_crates(db, krate)? {
let impls = db.impls_in_crate(krate); let impls = db.impls_in_crate(krate);
for impl_block in impls.lookup_impl_blocks(&ty.value) { for impl_block in impls.lookup_impl_blocks(&ty.value) {
@ -342,30 +343,6 @@ pub(crate) fn implements_trait(
solution.is_some() solution.is_some()
} }
impl Ty {
// This would be nicer if it just returned an iterator, but that runs into
// lifetime problems, because we need to borrow temp `CrateImplBlocks`.
pub fn iterate_impl_items<T>(
self,
db: &impl HirDatabase,
krate: CrateId,
mut callback: impl FnMut(AssocItem) -> Option<T>,
) -> Option<T> {
for krate in def_crates(db, krate, &self)? {
let impls = db.impls_in_crate(krate);
for impl_block in impls.lookup_impl_blocks(&self) {
for &item in db.impl_data(impl_block).items.iter() {
if let Some(result) = callback(item.into()) {
return Some(result);
}
}
}
}
None
}
}
/// This creates Substs for a trait with the given Self type and type variables /// This creates Substs for a trait with the given Self type and type variables
/// for all other parameters, to query Chalk with it. /// for all other parameters, to query Chalk with it.
fn generic_implements_goal( fn generic_implements_goal(

View file

@ -484,7 +484,7 @@ impl Resolver {
} }
} }
pub trait HasResolver { pub trait HasResolver: Copy {
/// Builds a resolver for type references inside this def. /// Builds a resolver for type references inside this def.
fn resolver(self, db: &impl DefDatabase) -> Resolver; fn resolver(self, db: &impl DefDatabase) -> Resolver;
} }
@ -502,7 +502,7 @@ impl HasResolver for TraitId {
} }
} }
impl<T: Into<AdtId>> HasResolver for T { impl<T: Into<AdtId> + Copy> HasResolver for T {
fn resolver(self, db: &impl DefDatabase) -> Resolver { fn resolver(self, db: &impl DefDatabase) -> Resolver {
let def = self.into(); let def = self.into();
def.module(db) def.module(db)

View file

@ -50,7 +50,7 @@ pub(super) fn complete_path(acc: &mut Completions, ctx: &CompletionContext) {
hir::ModuleDef::TypeAlias(a) => a.ty(ctx.db), hir::ModuleDef::TypeAlias(a) => a.ty(ctx.db),
_ => unreachable!(), _ => unreachable!(),
}; };
ctx.analyzer.iterate_path_candidates(ctx.db, ty.clone(), None, |_ty, item| { ctx.analyzer.iterate_path_candidates(ctx.db, &ty, None, |_ty, item| {
match item { match item {
hir::AssocItem::Function(func) => { hir::AssocItem::Function(func) => {
if !func.has_self_param(ctx.db) { if !func.has_self_param(ctx.db) {