language-servers/rust-analyzer
1
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer.git synced 2025-09-27 04:19:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Add enum variant references CodeLens.

Browse source
This commit is contained in:
Eric Sampson 2021-10-04 02:18:31 -05:00
parent 4b7675fcc3
commit a0bb31587e
4 changed files with 60 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

53
crates/ide/src/annotations.rs
View file

@ -5,7 +5,7 @@ use ide_db::{
helpers::visit_file_defs,
RootDatabase,
};
use syntax::{ast::HasName, AstNode, TextRange};
use syntax::{ast::HasName, AstNode, TextRange, TextSize};
use crate::{
fn_references::find_all_methods,
@ -40,6 +40,7 @@ pub struct AnnotationConfig {
pub annotate_impls: bool,
pub annotate_references: bool,
pub annotate_method_references: bool,
pub annotate_enum_variant_references: bool,
}
pub(crate) fn annotations(
@ -63,18 +64,33 @@ pub(crate) fn annotations(
visit_file_defs(&Semantics::new(db), file_id, &mut |def| match def {
Either::Left(def) => {
let range = match def {
let (range, ranges_variants) = match def {
hir::ModuleDef::Const(konst) => {
konst.source(db).and_then(|node| name_range(&node, file_id))
(konst.source(db).and_then(|node| name_range(&node, file_id)), vec![None])
}
hir::ModuleDef::Trait(trait_) => {
trait_.source(db).and_then(|node| name_range(&node, file_id))
(trait_.source(db).and_then(|node| name_range(&node, file_id)), vec![None])
}
hir::ModuleDef::Adt(adt) => {
adt.source(db).and_then(|node| name_range(&node, file_id))
}
_ => None,
hir::ModuleDef::Adt(adt) => match adt {
hir::Adt::Enum(enum_) => (
enum_.source(db).and_then(|node| name_range(&node, file_id)),
if config.annotate_enum_variant_references {
enum_
.variants(db)
.into_iter()
.map(|variant| {
variant.source(db).and_then(|node| name_range(&node, file_id))
})
.collect()
} else {
vec![None]
},
),
_ => (adt.source(db).and_then(|node| name_range(&node, file_id)), vec![None]),
},
_ => (None, vec![None]),
};
let (range, offset) = match range {
Some(range) => (range, range.start()),
None => return,
@ -99,6 +115,26 @@ pub(crate) fn annotations(
});
}
if config.annotate_enum_variant_references {
let mut variants_metadata: Vec<(TextRange, TextSize)> = Vec::new();
for range_variant in ranges_variants.into_iter() {
let (range, offset) = match range_variant {
Some(range) => (range, range.start()),
None => return,
};
variants_metadata.push((range, offset))
}
for variant_metadata in variants_metadata.into_iter() {
annotations.push(Annotation {
range: variant_metadata.0,
kind: AnnotationKind::HasReferences {
position: FilePosition { file_id, offset: variant_metadata.1 },
data: None,
},
});
}
}
fn name_range<T: HasName>(node: &InFile<T>, file_id: FileId) -> Option<TextRange> {
if node.file_id == file_id.into() {
node.value.name().map(|it| it.syntax().text_range())
@ -173,6 +209,7 @@ mod tests {
annotate_impls: true,
annotate_references: true,
annotate_method_references: true,
annotate_enum_variant_references: true,
},
file_id,
)