From 180655077a4d582d49d4aa14bed67b382f336cf0 Mon Sep 17 00:00:00 2001
From: Edwin Cheng <edwin0cheng@gmail.com>
Date: Mon, 15 Apr 2019 02:26:21 +0800
Subject: Fix 1099

---
 crates/ra_hir/src/ty/infer.rs | 51 ++++++++++++++++++++++++++++++++---
 crates/ra_hir/src/ty/tests.rs | 63 +++++++++++++++++++++++++++++++++++++++++--
 2 files changed, 109 insertions(+), 5 deletions(-)

(limited to 'crates/ra_hir')

diff --git a/crates/ra_hir/src/ty/infer.rs b/crates/ra_hir/src/ty/infer.rs
index 651a78fe5..fe6553f79 100644
--- a/crates/ra_hir/src/ty/infer.rs
+++ b/crates/ra_hir/src/ty/infer.rs
@@ -459,6 +459,7 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
             if remaining_index.is_none() { def.take_values()? } else { def.take_types()? };
 
         let remaining_index = remaining_index.unwrap_or(path.segments.len());
+        let mut actual_def_ty: Option<Ty> = None;
 
         // resolve intermediate segments
         for segment in &path.segments[remaining_index..] {
@@ -468,9 +469,20 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
                     let typable: Option<TypableDef> = def.into();
                     let typable = typable?;
 
-                    let substs =
+                    let mut substs =
                         Ty::substs_from_path_segment(self.db, &self.resolver, segment, typable);
-                    self.db.type_for_def(typable, Namespace::Types).subst(&substs)
+
+                    if remaining_index > 0 {
+                        substs = Ty::substs_from_path_segment(
+                            self.db,
+                            &self.resolver,
+                            &path.segments[remaining_index - 1],
+                            typable,
+                        );
+                    }
+
+                    let ty = self.db.type_for_def(typable, Namespace::Types);
+                    ty.subst(&substs)
                 }
                 Resolution::LocalBinding(_) => {
                     // can't have a local binding in an associated item path
@@ -489,6 +501,8 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
             // Attempt to find an impl_item for the type which has a name matching
             // the current segment
             log::debug!("looking for path segment: {:?}", segment);
+            actual_def_ty = Some(ty.clone());
+
             let item: crate::ModuleDef = ty.iterate_impl_items(self.db, |item| {
                 let matching_def: Option<crate::ModuleDef> = match item {
                     crate::ImplItem::Method(func) => {
@@ -528,9 +542,40 @@ impl<'a, D: HirDatabase> InferenceContext<'a, D> {
             Resolution::Def(def) => {
                 let typable: Option<TypableDef> = def.into();
                 let typable = typable?;
+
+                let ty = self.db.type_for_def(typable, Namespace::Values);
                 let substs = Ty::substs_from_path(self.db, &self.resolver, path, typable);
-                let ty = self.db.type_for_def(typable, Namespace::Values).subst(&substs);
+                let ty = ty.subst(&substs);
                 let ty = self.insert_type_vars(ty);
+
+                // plug the old parent_ty in
+                if let Some(actual_def_ty) = actual_def_ty {
+                    if let crate::ModuleDef::Function(func) = def {
+                        let gen = func.generic_params(self.db);
+                        if let Some(target_ty) = func.impl_block(self.db) {
+                            let target_ty = target_ty.target_ty(self.db);
+                            let old_params = target_ty.substs().unwrap().clone();
+
+                            let target_ty = target_ty.subst(&substs);
+                            let target_ty = self.insert_type_vars(target_ty);
+
+                            if gen.count_parent_params() > 0 {
+                                self.unify(&target_ty, &actual_def_ty);
+
+                                if let Ty::Apply(ty) = &ty {
+                                    for (param, pty) in
+                                        old_params.iter().zip(target_ty.substs().unwrap().iter())
+                                    {
+                                        if let Ty::Param { idx, .. } = param {
+                                            self.unify(pty, &ty.parameters.0[*idx as usize]);
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+                }
+
                 Some(ty)
             }
             Resolution::LocalBinding(pat) => {
diff --git a/crates/ra_hir/src/ty/tests.rs b/crates/ra_hir/src/ty/tests.rs
index 82c4aeddb..8d8a0eaaa 100644
--- a/crates/ra_hir/src/ty/tests.rs
+++ b/crates/ra_hir/src/ty/tests.rs
@@ -1426,6 +1426,65 @@ fn test() {
     );
 }
 
+#[test]
+fn infer_associated_method_generics_without_args() {
+    assert_snapshot_matches!(
+        infer(r#"
+struct Gen<T> {
+    val: T
+}
+
+impl<T> Gen<T> {
+    pub fn make() -> Gen<T> {
+        loop { }
+    }
+}
+
+fn test() {
+    let a = Gen::<u32>::make();
+}
+"#),
+        @r###"
+[76; 100) '{     ...     }': !
+[86; 94) 'loop { }': !
+[91; 94) '{ }': ()
+[114; 149) '{     ...e(); }': ()
+[124; 125) 'a': Gen<u32>
+[128; 144) 'Gen::<...::make': fn make<u32>() -> Gen<T>
+[128; 146) 'Gen::<...make()': Gen<u32>"###
+    );
+}
+
+#[test]
+fn infer_associated_method_generics_2_type_params_without_args() {
+    assert_snapshot_matches!(
+        infer(r#"
+struct Gen<T, U> {
+    val: T,
+    val2: U,
+}
+
+impl<T> Gen<u32, T> {
+    pub fn make() -> Gen<u32,T> {
+        loop { }
+    }
+}
+
+fn test() {
+    let a = Gen::<u32, u64>::make();
+}
+"#),
+        @r###"
+[102; 126) '{     ...     }': !
+[112; 120) 'loop { }': !
+[117; 120) '{ }': ()
+[140; 180) '{     ...e(); }': ()
+[150; 151) 'a': Gen<u32, u64>
+[154; 175) 'Gen::<...::make': fn make<u64>() -> Gen<u32, T>
+[154; 177) 'Gen::<...make()': Gen<u32, u64>"###
+    );
+}
+
 #[test]
 fn infer_type_alias() {
     assert_snapshot_matches!(
@@ -1814,8 +1873,8 @@ pub fn main_loop() {
     @r###"
 [144; 146) '{}': ()
 [169; 198) '{     ...t(); }': ()
-[175; 193) 'FxHash...efault': fn default<{unknown}, {unknown}>() -> HashSet<T, H>
-[175; 195) 'FxHash...ault()': HashSet<{unknown}, {unknown}>"###
+[175; 193) 'FxHash...efault': fn default<{unknown}, FxHasher>() -> HashSet<T, H>
+[175; 195) 'FxHash...ault()': HashSet<{unknown}, FxHasher>"###
     );
 }
 
-- 
cgit v1.2.3