3 files changed, 30 insertions, 110 deletions

diff --git a/crates/ra_hir/src/ty/method_resolution.rs b/crates/ra_hir/src/ty/method_resolution.rs
index eb5ca6769..9aad2d3fe 100644
--- a/crates/ra_hir/src/ty/method_resolution.rs
+++ b/crates/ra_hir/src/ty/method_resolution.rs
@@ -5,16 +5,14 @@
 use std::sync::Arc;
 
 use arrayvec::ArrayVec;
-use hir_def::CrateModuleId;
 use rustc_hash::FxHashMap;
 
 use crate::{
     db::HirDatabase,
-    impl_block::{ImplBlock, ImplId},
     resolve::Resolver,
     ty::primitive::{FloatBitness, Uncertain},
     ty::{Ty, TypeCtor},
-    AssocItem, Crate, Function, Module, Mutability, Name, Trait,
+    AssocItem, Crate, Function, ImplBlock, Module, Mutability, Name, Trait,
 };
 
 use super::{autoderef, lower, Canonical, InEnvironment, TraitEnvironment, TraitRef};
@@ -39,65 +37,46 @@ impl TyFingerprint {
 
 #[derive(Debug, PartialEq, Eq)]
 pub struct CrateImplBlocks {
-    /// To make sense of the CrateModuleIds, we need the source root.
-    krate: Crate,
-    impls: FxHashMap<TyFingerprint, Vec<(CrateModuleId, ImplId)>>,
-    impls_by_trait: FxHashMap<Trait, Vec<(CrateModuleId, ImplId)>>,
+    impls: FxHashMap<TyFingerprint, Vec<ImplBlock>>,
+    impls_by_trait: FxHashMap<Trait, Vec<ImplBlock>>,
 }
 
 impl CrateImplBlocks {
-    pub fn lookup_impl_blocks<'a>(&'a self, ty: &Ty) -> impl Iterator<Item = ImplBlock> + 'a {
+    pub(crate) fn impls_in_crate_query(
+        db: &impl HirDatabase,
+        krate: Crate,
+    ) -> Arc<CrateImplBlocks> {
+        let mut crate_impl_blocks =
+            CrateImplBlocks { impls: FxHashMap::default(), impls_by_trait: FxHashMap::default() };
+        if let Some(module) = krate.root_module(db) {
+            crate_impl_blocks.collect_recursive(db, module);
+        }
+        Arc::new(crate_impl_blocks)
+    }
+    pub fn lookup_impl_blocks(&self, ty: &Ty) -> impl Iterator<Item = ImplBlock> + '_ {
         let fingerprint = TyFingerprint::for_impl(ty);
-        fingerprint.and_then(|f| self.impls.get(&f)).into_iter().flat_map(|i| i.iter()).map(
-            move |(module_id, impl_id)| {
-                let module = Module::new(self.krate, *module_id);
-                ImplBlock::from_id(module, *impl_id)
-            },
-        )
+        fingerprint.and_then(|f| self.impls.get(&f)).into_iter().flatten().copied()
     }
 
-    pub fn lookup_impl_blocks_for_trait<'a>(
-        &'a self,
-        tr: Trait,
-    ) -> impl Iterator<Item = ImplBlock> + 'a {
-        self.impls_by_trait.get(&tr).into_iter().flat_map(|i| i.iter()).map(
-            move |(module_id, impl_id)| {
-                let module = Module::new(self.krate, *module_id);
-                ImplBlock::from_id(module, *impl_id)
-            },
-        )
+    pub fn lookup_impl_blocks_for_trait(&self, tr: Trait) -> impl Iterator<Item = ImplBlock> + '_ {
+        self.impls_by_trait.get(&tr).into_iter().flatten().copied()
     }
 
     pub fn all_impls<'a>(&'a self) -> impl Iterator<Item = ImplBlock> + 'a {
-        self.impls.values().chain(self.impls_by_trait.values()).flat_map(|i| i.iter()).map(
-            move |(module_id, impl_id)| {
-                let module = Module::new(self.krate, *module_id);
-                ImplBlock::from_id(module, *impl_id)
-            },
-        )
+        self.impls.values().chain(self.impls_by_trait.values()).flatten().copied()
     }
 
     fn collect_recursive(&mut self, db: &impl HirDatabase, module: Module) {
-        let module_impl_blocks = db.impls_in_module(module);
-
-        for (impl_id, _) in module_impl_blocks.impls.iter() {
-            let impl_block = ImplBlock::from_id(module_impl_blocks.module, impl_id);
-
+        for impl_block in module.impl_blocks(db) {
             let target_ty = impl_block.target_ty(db);
 
             if impl_block.target_trait(db).is_some() {
                 if let Some(tr) = impl_block.target_trait_ref(db) {
-                    self.impls_by_trait
-                        .entry(tr.trait_)
-                        .or_insert_with(Vec::new)
-                        .push((module.id.module_id, impl_id));
+                    self.impls_by_trait.entry(tr.trait_).or_default().push(impl_block);
                 }
             } else {
                 if let Some(target_ty_fp) = TyFingerprint::for_impl(&target_ty) {
-                    self.impls
-                        .entry(target_ty_fp)
-                        .or_insert_with(Vec::new)
-                        .push((module.id.module_id, impl_id));
+                    self.impls.entry(target_ty_fp).or_default().push(impl_block);
                 }
             }
         }
@@ -106,21 +85,6 @@ impl CrateImplBlocks {
             self.collect_recursive(db, child);
         }
     }
-
-    pub(crate) fn impls_in_crate_query(
-        db: &impl HirDatabase,
-        krate: Crate,
-    ) -> Arc<CrateImplBlocks> {
-        let mut crate_impl_blocks = CrateImplBlocks {
-            krate,
-            impls: FxHashMap::default(),
-            impls_by_trait: FxHashMap::default(),
-        };
-        if let Some(module) = krate.root_module(db) {
-            crate_impl_blocks.collect_recursive(db, module);
-        }
-        Arc::new(crate_impl_blocks)
-    }
 }
 
 fn def_crates(db: &impl HirDatabase, cur_crate: Crate, ty: &Ty) -> Option<ArrayVec<[Crate; 2]>> {
diff --git a/crates/ra_hir/src/ty/tests.rs b/crates/ra_hir/src/ty/tests.rs
index 8863c3608..fe9346c78 100644
--- a/crates/ra_hir/src/ty/tests.rs
+++ b/crates/ra_hir/src/ty/tests.rs
@@ -254,7 +254,6 @@ fn test(a: u32, b: isize, c: !, d: &str) {
     1.0f32;
 }"#),
         @r###"
-
     [9; 10) 'a': u32
     [17; 18) 'b': isize
     [27; 28) 'c': !
@@ -317,7 +316,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [15; 20) '{ 1 }': u32
     [17; 18) '1': u32
     [48; 53) '{ 1 }': u32
@@ -354,7 +352,7 @@ fn test() {
     [66; 74) 'S::foo()': i32
     [80; 88) '<S>::foo': fn foo() -> i32
     [80; 90) '<S>::foo()': i32
-"###
+    "###
     );
 }
 
@@ -409,7 +407,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [72; 154) '{     ...a.c; }': ()
     [82; 83) 'c': C
     [86; 87) 'C': C(usize) -> C
@@ -443,7 +440,6 @@ fn test() {
   E::V2;
 }"#),
         @r###"
-
     [48; 82) '{   E:...:V2; }': ()
     [52; 70) 'E::V1 ...d: 1 }': E
     [67; 68) '1': u32
@@ -471,7 +467,6 @@ fn test(a: &u32, b: &mut u32, c: *const u32, d: *mut u32) {
 }
 "#),
         @r###"
-
     [9; 10) 'a': &u32
     [18; 19) 'b': &mut u32
     [31; 32) 'c': *const u32
@@ -524,7 +519,6 @@ fn test() {
 }
 "##),
         @r###"
-
     [11; 221) '{     ...o"#; }': ()
     [17; 21) '5i32': i32
     [27; 31) '5f32': f32
@@ -568,7 +562,6 @@ fn test(x: SomeType) {
 }
 "#),
         @r###"
-
     [27; 28) 'x': SomeType
     [40; 272) '{     ...lo"; }': ()
     [50; 51) 'b': bool
@@ -632,7 +625,6 @@ fn test() -> &mut &f64 {
 }
 "#),
         @r###"
-
     [14; 15) 'x': u32
     [22; 24) '{}': ()
     [78; 231) '{     ...t &c }': &mut &f64
@@ -679,7 +671,6 @@ impl S {
 }
 "#),
         @r###"
-
     [34; 38) 'self': &S
     [40; 61) '{     ...     }': ()
     [50; 54) 'self': &S
@@ -719,7 +710,6 @@ fn test() -> bool {
 }
 "#),
         @r###"
-
     [6; 7) 'x': bool
     [22; 34) '{     0i32 }': i32
     [28; 32) '0i32': i32
@@ -802,7 +792,6 @@ fn test2(a1: *const A, a2: *mut A) {
 }
 "#),
         @r###"
-
     [44; 45) 'a': A
     [50; 213) '{     ...5.b; }': ()
     [60; 62) 'a1': A
@@ -970,7 +959,7 @@ fn test(a: A<i32>) {
     [374; 375) 'B': B<A<i32>>(T) -> B<T>
     [374; 378) 'B(a)': B<A<i32>>
     [376; 377) 'a': A<i32>
-"###
+    "###
     );
 }
 
@@ -983,7 +972,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [11; 37) '{    l... {}; }': ()
     [20; 21) 'x': ()
     [24; 34) 'if true {}': ()
@@ -1105,7 +1093,6 @@ fn test(a: A) {
 }
 "#),
         @r###"
-
     [32; 36) 'self': A
     [38; 39) 'x': u32
     [53; 55) '{}': ()
@@ -1142,7 +1129,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [40; 44) 'self': &str
     [53; 55) '{}': ()
     [69; 89) '{     ...o(); }': ()
@@ -1166,7 +1152,6 @@ fn test(x: &str, y: isize) {
 }
 "#),
         @r###"
-
     [9; 10) 'x': &str
     [18; 19) 'y': isize
     [28; 170) '{     ...d"); }': ()
@@ -1367,7 +1352,6 @@ fn test() {
 }
 "#),
     @r###"
-
     [28; 79) '{     ...(1); }': ()
     [38; 42) 'A(n)': A<i32>
     [40; 41) 'n': &i32
@@ -1396,7 +1380,6 @@ fn test() {
 }
 "#),
     @r###"
-
     [11; 57) '{     ...= v; }': ()
     [21; 22) 'v': &(i32, &i32)
     [25; 33) '&(1, &2)': &(i32, &i32)
@@ -1441,7 +1424,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [68; 289) '{     ...  d; }': ()
     [78; 79) 'e': E
     [82; 95) 'E::A { x: 3 }': E
@@ -1488,7 +1470,6 @@ fn test(a1: A<u32>, i: i32) {
 }
 "#),
         @r###"
-
     [36; 38) 'a1': A<u32>
     [48; 49) 'i': i32
     [56; 147) '{     ...3.x; }': ()
@@ -1569,7 +1550,6 @@ fn test(a1: A<u32>, o: Option<u64>) {
 }
 "#),
         @r###"
-
     [79; 81) 'a1': A<u32>
     [91; 92) 'o': Option<u64>
     [107; 244) '{     ...  }; }': ()
@@ -1604,7 +1584,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [10; 11) 't': T
     [21; 26) '{ t }': T
     [23; 24) 't': T
@@ -1652,7 +1631,6 @@ fn test() -> i128 {
 }
 "#),
         @r###"
-
     [74; 78) 'self': A<X, Y>
     [85; 107) '{     ...     }': X
     [95; 99) 'self': A<X, Y>
@@ -1706,7 +1684,6 @@ fn test(o: Option<u32>) {
 }
 "#),
         @r###"
-
     [78; 82) 'self': &Option<T>
     [98; 100) '{}': ()
     [111; 112) 'o': Option<u32>
@@ -1744,7 +1721,6 @@ fn test() -> i128 {
 }
 "#),
         @r###"
-
     [53; 57) 'self': A<T2>
     [65; 87) '{     ...     }': T2
     [75; 79) 'self': A<T2>
@@ -1921,7 +1897,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [56; 64) '{ A {} }': A
     [58; 62) 'A {}': A
     [126; 132) '{ 99 }': u32
@@ -1961,7 +1936,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [64; 67) 'val': T
     [82; 109) '{     ...     }': Gen<T>
     [92; 103) 'Gen { val }': Gen<T>
@@ -2129,7 +2103,6 @@ fn test(x: X) {
 }
 "#),
         @r###"
-
     [20; 21) 'x': X
     [26; 47) '{     ...eld; }': ()
     [32; 33) 'x': X
@@ -2151,7 +2124,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [11; 89) '{     ...   } }': ()
     [17; 21) 'X {}': {unknown}
     [27; 87) 'match ...     }': ()
@@ -2174,7 +2146,6 @@ fn quux() {
 }
 "#),
         @r###"
-
     [11; 41) '{     ...+ y; }': ()
     [21; 22) 'y': i32
     [25; 27) '92': i32
@@ -2300,7 +2271,6 @@ fn write() {
 }
 "#),
         @r###"
-
     [54; 139) '{     ...   } }': ()
     [60; 137) 'match ...     }': ()
     [66; 83) 'someth...nknown': Maybe<{unknown}>
@@ -2322,7 +2292,6 @@ fn test_line_buffer() {
 }
 "#),
         @r###"
-
     [23; 53) '{     ...n']; }': ()
     [29; 50) '&[0, b...b'\n']': &[u8;_]
     [30; 50) '[0, b'...b'\n']': [u8;_]
@@ -2446,7 +2415,6 @@ fn test<R>(query_response: Canonical<QueryResponse<R>>) {
 }
 "#),
         @r###"
-
     [92; 106) 'query_response': Canonical<QueryResponse<R>>
     [137; 167) '{     ...lue; }': ()
     [143; 164) '&query....value': &QueryResponse<R>
@@ -2472,7 +2440,6 @@ pub fn main_loop() {
 }
 "#),
     @r###"
-
     [144; 146) '{}': ()
     [169; 198) '{     ...t(); }': ()
     [175; 193) 'FxHash...efault': fn default<{unknown}, FxHasher>() -> HashSet<T, H>
@@ -2518,7 +2485,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [49; 50) '0': u32
     [80; 83) '101': u32
     [95; 213) '{     ...NST; }': ()
@@ -2549,7 +2515,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [29; 32) '101': u32
     [70; 73) '101': u32
     [85; 280) '{     ...MUT; }': ()
@@ -2588,7 +2553,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [31; 35) 'self': &Self
     [110; 114) 'self': &Self
     [170; 228) '{     ...i128 }': ()
@@ -2636,7 +2600,6 @@ mod bar_test {
 }
 "#),
         @r###"
-
     [63; 67) 'self': &Self
     [169; 173) 'self': &Self
     [300; 337) '{     ...     }': ()
@@ -2664,7 +2627,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [33; 37) 'self': &Self
     [92; 111) '{     ...d(); }': ()
     [98; 99) 'S': S
@@ -2694,7 +2656,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [43; 47) 'self': &Self
     [82; 86) 'self': &Self
     [210; 361) '{     ..., i8 }': ()
@@ -2725,7 +2686,6 @@ fn test() {
 }
 "#),
         @r###"
-
     [33; 37) 'self': &Self
     [102; 127) '{     ...d(); }': ()
     [108; 109) 'S': S<u32>(T) -> S<T>
@@ -3130,7 +3090,6 @@ fn test<T: Iterable<Item=u32>>() {
 }
 "#),
         @r###"
-
     [67; 100) '{     ...own; }': ()
     [77; 78) 'y': {unknown}
     [90; 97) 'unknown': {unknown}
@@ -3146,7 +3105,6 @@ const A: u32 = 1 + 1;
 static B: u64 = { let x = 1; x };
 "#),
         @r###"
-
     [16; 17) '1': u32
     [16; 21) '1 + 1': u32
     [20; 21) '1': u32
@@ -3170,7 +3128,6 @@ fn test() -> u64 {
 }
 "#),
         @r###"
-
     [38; 87) '{     ... a.1 }': u64
     [48; 49) 'a': S
     [52; 53) 'S': S(i32, u64) -> S
@@ -3225,7 +3182,6 @@ fn indexing_arrays() {
     assert_snapshot!(
         infer("fn main() { &mut [9][2]; }"),
         @r###"
-
     [10; 26) '{ &mut...[2]; }': ()
     [12; 23) '&mut [9][2]': &mut {unknown}
     [17; 20) '[9]': [i32;_]
@@ -4822,9 +4778,9 @@ fn main() {
 }
 "#),
         @r###"
-        ![0; 1) '6': i32
-        [64; 88) '{     ...!(); }': ()
-        [74; 75) 'x': i32
+    ![0; 1) '6': i32
+    [64; 88) '{     ...!(); }': ()
+    [74; 75) 'x': i32
     "###
     );
 }
diff --git a/crates/ra_hir/src/ty/traits/chalk.rs b/crates/ra_hir/src/ty/traits/chalk.rs
index 75351c17d..68304b950 100644
--- a/crates/ra_hir/src/ty/traits/chalk.rs
+++ b/crates/ra_hir/src/ty/traits/chalk.rs
@@ -191,11 +191,11 @@ impl ToChalk for Impl {
     type Chalk = chalk_ir::ImplId;
 
     fn to_chalk(self, db: &impl HirDatabase) -> chalk_ir::ImplId {
-        db.intern_impl(self).into()
+        db.intern_chalk_impl(self).into()
     }
 
     fn from_chalk(db: &impl HirDatabase, impl_id: chalk_ir::ImplId) -> Impl {
-        db.lookup_intern_impl(impl_id.into())
+        db.lookup_intern_chalk_impl(impl_id.into())
     }
 }
 
@@ -630,7 +630,7 @@ fn impl_block_datum(
         .target_trait_ref(db)
         .expect("FIXME handle unresolved impl block trait ref")
         .subst(&bound_vars);
-    let impl_type = if impl_block.module().krate() == krate {
+    let impl_type = if impl_block.krate(db) == krate {
         chalk_rust_ir::ImplType::Local
     } else {
         chalk_rust_ir::ImplType::External