From 54b9b03ca2e90083fd1d1fe199c5dde595423b53 Mon Sep 17 00:00:00 2001 From: Lukas Wirth Date: Mon, 4 Jan 2021 15:44:19 +0100 Subject: [PATCH] Show GotoTypeAction for TypeParam --- crates/hir/src/code_model.rs | 12 ++++++ crates/ide/src/hover.rs | 71 +++++++++++++++++++++++++++--------- 2 files changed, 66 insertions(+), 17 deletions(-) diff --git a/crates/hir/src/code_model.rs b/crates/hir/src/code_model.rs index a7a38d43a3..071e553a88 100644 --- a/crates/hir/src/code_model.rs +++ b/crates/hir/src/code_model.rs @@ -1276,6 +1276,18 @@ impl TypeParam { } } + pub fn trait_bounds(self, db: &dyn HirDatabase) -> Vec { + db.generic_predicates_for_param(self.id) + .into_iter() + .filter_map(|pred| match &pred.value { + hir_ty::GenericPredicate::Implemented(trait_ref) => { + Some(Trait::from(trait_ref.trait_)) + } + _ => None, + }) + .collect() + } + pub fn default(self, db: &dyn HirDatabase) -> Option { let params = db.generic_defaults(self.id.parent); let local_idx = hir_ty::param_idx(db, self.id)?; diff --git a/crates/ide/src/hover.rs b/crates/ide/src/hover.rs index 932279a063..f2ad95cb60 100644 --- a/crates/ide/src/hover.rs +++ b/crates/ide/src/hover.rs @@ -228,11 +228,6 @@ fn runnable_action( } fn goto_type_action(db: &RootDatabase, def: Definition) -> Option { - let ty = match def { - Definition::Local(it) => it.ty(db), - Definition::ConstParam(it) => it.ty(db), - _ => return None, - }; let mut targets: Vec = Vec::new(); let mut push_new_def = |item: ModuleDef| { if !targets.contains(&item) { @@ -240,17 +235,27 @@ fn goto_type_action(db: &RootDatabase, def: Definition) -> Option { } }; - ty.walk(db, |t| { - if let Some(adt) = t.as_adt() { - push_new_def(adt.into()); - } else if let Some(trait_) = t.as_dyn_trait() { - push_new_def(trait_.into()); - } else if let Some(traits) = t.as_impl_traits(db) { - traits.into_iter().for_each(|it| push_new_def(it.into())); - } else if let Some(trait_) = t.as_associated_type_parent_trait(db) { - push_new_def(trait_.into()); - } - }); + if let Definition::TypeParam(it) = def { + it.trait_bounds(db).into_iter().for_each(|it| push_new_def(it.into())); + } else { + let ty = match def { + Definition::Local(it) => it.ty(db), + Definition::ConstParam(it) => it.ty(db), + _ => return None, + }; + + ty.walk(db, |t| { + if let Some(adt) = t.as_adt() { + push_new_def(adt.into()); + } else if let Some(trait_) = t.as_dyn_trait() { + push_new_def(trait_.into()); + } else if let Some(traits) = t.as_impl_traits(db) { + traits.into_iter().for_each(|it| push_new_def(it.into())); + } else if let Some(trait_) = t.as_associated_type_parent_trait(db) { + push_new_def(trait_.into()); + } + }); + } let targets = targets .into_iter() @@ -3086,7 +3091,7 @@ fn main() { let s<|>t = test().get(); } struct Bar; struct Foo; -impl Foo> {} +impl Foo> {} "#, expect![[r#" [ @@ -3112,6 +3117,38 @@ impl Foo> {} ); } + #[test] + fn test_hover_type_param_has_goto_type_action() { + check_actions( + r#" +trait Foo {} + +fn foo(t: T<|>){} +"#, + expect![[r#" + [ + GoToType( + [ + HoverGotoTypeData { + mod_path: "test::Foo", + nav: NavigationTarget { + file_id: FileId( + 0, + ), + full_range: 0..12, + focus_range: 6..9, + name: "Foo", + kind: Trait, + description: "trait Foo", + }, + }, + ], + ), + ] + "#]], + ); + } + #[test] fn hover_displays_normalized_crate_names() { check(