Implement completion for associated items

1 files changed, 20 insertions, 0 deletions

diff --git a/crates/ra_hir/src/ty/method_resolution.rs b/crates/ra_hir/src/ty/method_resolution.rs
index b8e911b88..e81bd3e06 100644
--- a/crates/ra_hir/src/ty/method_resolution.rs
+++ b/crates/ra_hir/src/ty/method_resolution.rs
@@ -174,4 +174,24 @@ impl Ty {
         }
         None
     }
+
+    // This would be nicer if it just returned an iterator, but that runs into
+    // lifetime problems, because we need to borrow temp `CrateImplBlocks`.
+    pub fn iterate_impl_items<T>(
+        self,
+        db: &impl HirDatabase,
+        mut callback: impl FnMut(ImplItem) -> Option<T>,
+    ) -> Option<T> {
+        let krate = def_crate(db, &self)?;
+        let impls = db.impls_in_crate(krate);
+
+        for (_, impl_block) in impls.lookup_impl_blocks(db, &self) {
+            for item in impl_block.items() {
+                if let Some(result) = callback(*item) {
+                    return Some(result);
+                }
+            }
+        }
+        None
+    }
 }