Item search now respects trait impl items

2021-06-11 19:23:59 +02:00
parent c6133fe51c
commit 48f65b3b15
3 changed files with 215 additions and 11 deletions
--- a/crates/hir/src/lib.rs
+++ b/crates/hir/src/lib.rs
@@ -1450,6 +1450,20 @@ impl AssocItem {
            _ => None,
        }
    }
+
+    pub fn containing_trait_impl(self, db: &dyn HirDatabase) -> Option<Trait> {
+        match self.container(db) {
+            AssocItemContainer::Impl(i) => i.trait_(db),
+            _ => None,
+        }
+    }
+
+    pub fn containing_trait_or_trait_impl(self, db: &dyn HirDatabase) -> Option<Trait> {
+        match self.container(db) {
+            AssocItemContainer::Trait(t) => Some(t),
+            AssocItemContainer::Impl(i) => i.trait_(db),
+        }
+    }
 }

 impl HasVisibility for AssocItem {