Add support for associated type bindings (`where Trait<Type = X>`)

1 files changed, 27 insertions, 0 deletions

diff --git a/crates/ra_hir/src/ty/traits.rs b/crates/ra_hir/src/ty/traits.rs
index b634f0b79..25316bc02 100644
--- a/crates/ra_hir/src/ty/traits.rs
+++ b/crates/ra_hir/src/ty/traits.rs
@@ -124,6 +124,9 @@ impl Obligation {
     pub fn from_predicate(predicate: GenericPredicate) -> Option<Obligation> {
         match predicate {
             GenericPredicate::Implemented(trait_ref) => Some(Obligation::Trait(trait_ref)),
+            GenericPredicate::Projection(projection_pred) => {
+                Some(Obligation::Projection(projection_pred))
+            }
             GenericPredicate::Error => None,
         }
     }
@@ -135,6 +138,30 @@ pub struct ProjectionPredicate {
     pub ty: Ty,
 }
 
+impl ProjectionPredicate {
+    pub fn subst(mut self, substs: &super::Substs) -> ProjectionPredicate {
+        self.walk_mut(&mut |ty| match ty {
+            Ty::Param { idx, .. } => {
+                if let Some(t) = substs.get(*idx as usize).cloned() {
+                    *ty = t;
+                }
+            }
+            _ => {}
+        });
+        self
+    }
+
+    pub fn walk(&self, f: &mut impl FnMut(&Ty)) {
+        self.projection_ty.walk(f);
+        self.ty.walk(f);
+    }
+
+    pub fn walk_mut(&mut self, f: &mut impl FnMut(&mut Ty)) {
+        self.projection_ty.walk_mut(f);
+        self.ty.walk_mut(f);
+    }
+}
+
 /// Solve a trait goal using Chalk.
 pub(crate) fn trait_solve_query(
     db: &impl HirDatabase,