Auto merge of rust-lang#12198 - jonas-schievink:ide-resolve-supertrait-assoc-types, r=jonas-schievink

fix: Resolve assoc. types of supertraits in the IDE layer

Fixes rust-lang/rust-analyzer#12166

Author: bors

crates/hir/src/lib.rs

@@ -54,7 +54,7 @@ use hir_def::{
 };
 use hir_expand::{name::name, MacroCallKind};
 use hir_ty::{
-    autoderef,
+    all_super_traits, autoderef,
     consteval::{unknown_const_as_generic, ComputedExpr, ConstEvalError, ConstExt},
     diagnostics::BodyValidationDiagnostic,
     method_resolution::{self, TyFingerprint},
@@ -1676,6 +1676,11 @@ impl Trait {
         db.trait_data(self.id).items.iter().map(|(_name, it)| (*it).into()).collect()
     }
 
+    pub fn items_with_supertraits(self, db: &dyn HirDatabase) -> Vec<AssocItem> {
+        let traits = all_super_traits(db.upcast(), self.into());
+        traits.iter().flat_map(|tr| Trait::from(*tr).items(db)).collect()
+    }
+
     pub fn is_auto(self, db: &dyn HirDatabase) -> bool {
         db.trait_data(self.id).is_auto
     }

crates/ide-completion/src/completions/type.rs

@@ -168,8 +168,9 @@ pub(crate) fn complete_type_path(acc: &mut Completions, ctx: &CompletionContext)
                     if let Some(hir::PathResolution::Def(hir::ModuleDef::Trait(trait_))) =
                         ctx.sema.resolve_path(&path_seg.parent_path())
                     {
-                        trait_.items(ctx.sema.db).into_iter().for_each(|it| {
+                        trait_.items_with_supertraits(ctx.sema.db).into_iter().for_each(|it| {
                             if let hir::AssocItem::TypeAlias(alias) = it {
+                                cov_mark::hit!(complete_assoc_type_in_generics_list);
                                 acc.add_type_alias_with_eq(ctx, alias)
                             }
                         });

crates/ide-completion/src/tests/type_pos.rs

@@ -336,9 +336,13 @@ fn foo<'lt, T, const C: usize>() {
 
 #[test]
 fn completes_types_and_const_in_arg_list() {
+    cov_mark::check!(complete_assoc_type_in_generics_list);
     check(
         r#"
-trait Trait2 {
+trait Trait1 {
+    type Super;
+}
+trait Trait2: Trait1 {
     type Foo;
 }
 
@@ -348,14 +352,16 @@ fn foo<'lt, T: Trait2<$0>, const CONST_PARAM: usize>(_: T) {}
             ct CONST
             cp CONST_PARAM
             en Enum
-            ma makro!(…)          macro_rules! makro
+            ma makro!(…)            macro_rules! makro
             md module
             st Record
             st Tuple
             st Unit
             tt Trait
+            tt Trait1
             tt Trait2
-            ta Foo =  (as Trait2) type Foo
+            ta Foo =  (as Trait2)   type Foo
+            ta Super =  (as Trait1) type Super
             tp T
             un Union
             bt u32

crates/ide-db/src/defs.rs

@@ -386,9 +386,8 @@ impl NameRefClass {
                     let containing_path = name_ref.syntax().ancestors().find_map(ast::Path::cast)?;
                     let resolved = sema.resolve_path(&containing_path)?;
                     if let PathResolution::Def(ModuleDef::Trait(tr)) = resolved {
-                        // FIXME: resolve in supertraits
                         if let Some(ty) = tr
-                            .items(sema.db)
+                            .items_with_supertraits(sema.db)
                             .iter()
                             .filter_map(|&assoc| match assoc {
                                 hir::AssocItem::TypeAlias(it) => Some(it),

crates/ide/src/goto_definition.rs

@@ -1012,6 +1012,22 @@ fn f() -> impl Iterator<Item$0 = u8> {}
         );
     }
 
+    #[test]
+    fn goto_def_for_super_assoc_ty_in_path() {
+        check(
+            r#"
+trait Super {
+    type Item;
+       //^^^^
+}
+
+trait Sub: Super {}
+
+fn f() -> impl Sub<Item$0 = u8> {}
+"#,
+        );
+    }
+
     #[test]
     fn unknown_assoc_ty() {
         check_unresolved(