From 04717b9bd2decfee70a6fbde90540d3dc16bb76f Mon Sep 17 00:00:00 2001 From: Lukas Wirth Date: Sat, 8 May 2021 22:43:26 +0200 Subject: Fix builtintype def_to_ty --- crates/ide_db/src/search.rs | 22 ++++++++++++++-------- 1 file changed, 14 insertions(+), 8 deletions(-) (limited to 'crates/ide_db') diff --git a/crates/ide_db/src/search.rs b/crates/ide_db/src/search.rs index 891b8cf0a..d8abef491 100644 --- a/crates/ide_db/src/search.rs +++ b/crates/ide_db/src/search.rs @@ -318,7 +318,7 @@ pub struct FindUsages<'a> { impl<'a> FindUsages<'a> { /// Enable searching for `Self` when the definition is a type. pub fn include_self_refs(mut self) -> FindUsages<'a> { - self.include_self_kw_refs = def_to_ty(self.sema.db, self.def); + self.include_self_kw_refs = def_to_ty(self.sema, self.def); self } @@ -474,7 +474,7 @@ impl<'a> FindUsages<'a> { sink(file_id, reference) } Some(NameRefClass::Definition(def)) if self.include_self_kw_refs.is_some() => { - if self.include_self_kw_refs == def_to_ty(self.sema.db, &def) { + if self.include_self_kw_refs == def_to_ty(self.sema, &def) { let FileRange { file_id, range } = self.sema.original_range(name_ref.syntax()); let reference = FileReference { range, @@ -536,16 +536,22 @@ impl<'a> FindUsages<'a> { } } -fn def_to_ty(db: &RootDatabase, def: &Definition) -> Option { +fn def_to_ty(sema: &Semantics, def: &Definition) -> Option { match def { Definition::ModuleDef(def) => match def { - ModuleDef::Adt(adt) => Some(adt.ty(db)), - ModuleDef::TypeAlias(it) => Some(it.ty(db)), - ModuleDef::BuiltinType(_it) => None, // FIXME somehow acquire some module to construct the builtin type - ModuleDef::Trait(_it) => None, // FIXME turn trait into its self-type + ModuleDef::Adt(adt) => Some(adt.ty(sema.db)), + ModuleDef::TypeAlias(it) => Some(it.ty(sema.db)), + ModuleDef::BuiltinType(it) => { + let graph = sema.db.crate_graph(); + let krate = graph.iter().next()?; + let root_file = graph[krate].root_file_id; + let module = sema.to_module_def(root_file)?; + Some(it.ty(sema.db, module)) + } + ModuleDef::Trait(_it) => None, // FIXME turn trait into its self-type _ => None, }, - Definition::SelfType(it) => Some(it.self_ty(db)), + Definition::SelfType(it) => Some(it.self_ty(sema.db)), _ => None, } } -- cgit v1.2.3