From 29729abc3c9736ca70bc0828e36e6670fb36b5bc Mon Sep 17 00:00:00 2001
From: Stanislav <enomado@users.noreply.github.com>
Date: Sun, 4 Sep 2022 19:10:04 +0300
Subject: [PATCH] Retain imports on find-all-references

---
 crates/ide/src/references.rs | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/crates/ide/src/references.rs b/crates/ide/src/references.rs
index 99614b645e4..fad44930fec 100644
--- a/crates/ide/src/references.rs
+++ b/crates/ide/src/references.rs
@@ -79,6 +79,8 @@ pub(crate) fn find_all_refs(
                 retain_adt_literal_usages(&mut usages, def, sema);
             }
 
+            retain_import_usages(&mut usages, sema);
+
             let references = usages
                 .into_iter()
                 .map(|(file_id, refs)| {
@@ -112,6 +114,32 @@ pub(crate) fn find_all_refs(
     }
 }
 
+fn retain_import_usages(usages: &mut UsageSearchResult, sema: &Semantics<'_, RootDatabase>) {
+    for (file_id, refs) in &mut usages.references {
+        refs.retain(|x| {
+            let file_sema = sema.parse(file_id.clone()).syntax().clone();
+
+            let maybe_node = file_sema.child_or_token_at_range(x.range.clone());
+
+            if let Some(node) = maybe_node {
+                let res = match node {
+                    syntax::NodeOrToken::Node(x) => {
+                        if matches!(x.kind(), USE) {
+                            false
+                        } else {
+                            true
+                        }
+                    }
+                    syntax::NodeOrToken::Token(_) => true,
+                };
+                res
+            } else {
+                true
+            }
+        });
+    }
+}
+
 pub(crate) fn find_defs<'a>(
     sema: &'a Semantics<'_, RootDatabase>,
     syntax: &SyntaxNode,
-- 
2.44.0