Fix Self type retrieval from ProjectionTy

Author: lowr

chalk-ir/src/lib.rs

@@ -1643,17 +1643,6 @@ pub struct ProjectionTy<I: Interner> {
 
 impl<I: Interner> Copy for ProjectionTy<I> where I::InternedSubstitution: Copy {}
 
-impl<I: Interner> ProjectionTy<I> {
-    /// Gets the type parameters of the `Self` type in this alias type.
-    pub fn self_type_parameter(&self, interner: I) -> Ty<I> {
-        self.substitution
-            .iter(interner)
-            .find_map(move |p| p.ty(interner))
-            .unwrap()
-            .clone()
-    }
-}
-
 /// An opaque type `opaque type T<..>: Trait = HiddenTy`.
 #[derive(Clone, PartialEq, Eq, Hash, TypeFoldable, TypeVisitable, HasInterner)]
 pub struct OpaqueTy<I: Interner> {

chalk-solve/src/clauses.rs

@@ -603,11 +603,12 @@ pub fn program_clauses_that_could_match<I: Interner>(
                 // ```
                 let associated_ty_datum = db.associated_ty_data(proj.associated_ty_id);
                 let trait_id = associated_ty_datum.trait_id;
-                let trait_parameters = db.trait_parameters_from_projection(proj);
+                let trait_ref = db.trait_ref_from_projection(proj);
+                let trait_parameters = trait_ref.substitution.as_parameters(interner);
 
                 let trait_datum = db.trait_datum(trait_id);
 
-                let self_ty = proj.self_type_parameter(interner);
+                let self_ty = trait_ref.self_type_parameter(interner);
                 if let TyKind::InferenceVar(_, _) = self_ty.kind(interner) {
                     panic!("Inference vars not allowed when getting program clauses");
                 }
@@ -804,10 +805,11 @@ fn push_alias_alias_eq_clause<I: Interner>(
     alias: AliasTy<I>,
 ) {
     let interner = builder.interner();
-    assert_eq!(
-        *projection_ty.self_type_parameter(interner).kind(interner),
-        TyKind::Alias(alias.clone())
-    );
+    let self_ty = builder
+        .db
+        .trait_ref_from_projection(&projection_ty)
+        .self_type_parameter(interner);
+    assert_eq!(*self_ty.kind(interner), TyKind::Alias(alias.clone()));
 
     // TODO: instead generate clauses without reference to the specific type parameters of the goal?
     let generalized = generalize::Generalize::apply(interner, (projection_ty, ty, alias));

tests/test/projection.rs

@@ -687,6 +687,33 @@ fn forall_projection_gat() {
     }
 }
 
+#[test]
+fn gat_in_non_enumerable_trait() {
+    test! {
+        program {
+            #[non_enumerable]
+            trait Deref { }
+
+            #[non_enumerable]
+            trait PointerFamily {
+                type Pointer<T>: Deref;
+            }
+        }
+
+        goal {
+            forall<T> {
+                forall<U> {
+                    if (T: PointerFamily) {
+                        <T as PointerFamily>::Pointer<U>: Deref
+                    }
+                }
+            }
+        } yields {
+            expect![[r#"Unique"#]]
+        }
+    }
+}
+
 #[test]
 fn normalize_under_binder() {
     test! {