Mirrors/rust-analyzer
2
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer synced 2024-12-29 06:23:25 +00:00
Code Issues Projects Releases Packages Wiki Activity

Retain imports on find-all-references

Browse source
This commit is contained in:
Stanislav 2022-09-04 19:10:04 +03:00
parent 8ddb8b7e8e
commit 29729abc3c
1 changed files with 28 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
crates/ide/src/references.rs
View file

@ -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,