refactor(ide): use target alternatives for declarations

Author: hongjr03

crates/ide/src/goto_declaration.rs

@@ -1,5 +1,6 @@
-use hir::semantics::Semantics;
+use hir::{file::HirFileId, semantics::Semantics};
 use itertools::Itertools;
+use utils::line_index::TextRange;
 
 use crate::{
     FilePosition, RangeInfo,
@@ -8,6 +9,7 @@ use crate::{
     goto_definition,
     navigation_target::{NavTarget, ToNav},
     semantic_target::{SemanticTarget, TargetIntent, resolve_semantic_target},
+    source_targets::SourceTarget,
 };
 
 pub(crate) fn goto_declaration(
@@ -24,14 +26,46 @@ pub(crate) fn goto_declaration(
         parsed_file.root(),
         goto_definition::token_precedence,
     );
-    let SemanticTarget::Source(target) = target.unique_for_intent(TargetIntent::Navigate)? else {
-        return None;
-    };
+    render_declaration_target(
+        db,
+        hir_file_id,
+        &sema,
+        target.targets_for_intent(TargetIntent::Navigate),
+    )
+}
+
+fn render_declaration_target(
+    db: &RootDb,
+    hir_file_id: HirFileId,
+    sema: &Semantics<RootDb>,
+    targets: Vec<SemanticTarget<'_>>,
+) -> Option<RangeInfo<Vec<NavTarget>>> {
+    let mut ranges = Vec::new();
+    let mut navs = Vec::new();
+    for target in targets {
+        let SemanticTarget::Source(target) = target else {
+            return None;
+        };
+        let target = render_source_declaration_target(db, hir_file_id, sema, target)?;
+        ranges.push(target.range);
+        navs.extend(target.info);
+    }
+
+    let range = covering_range(&ranges)?;
+    Some(RangeInfo::new(range, navs.into_iter().unique().collect()))
+}
+
+fn render_source_declaration_target(
+    db: &RootDb,
+    hir_file_id: HirFileId,
+    sema: &Semantics<RootDb>,
+    target: SourceTarget<'_>,
+) -> Option<RangeInfo<Vec<NavTarget>>> {
     let (range, tokens) = target.into_parts();
 
     let origins = tokens
         .into_iter()
-        .filter_map(|token| match DefinitionClass::resolve(&sema, hir_file_id, token)? {
+        .filter_map(|token| match DefinitionClass::resolve(sema, hir_file_id, token)? {
             DefinitionClass::Definition(definition) => {
                 Some(definition.declaration_origins().into_iter().collect_vec())
             }
@@ -52,3 +86,9 @@ pub(crate) fn goto_declaration(
 
     Some(RangeInfo::new(range, navs))
 }
+
+fn covering_range(ranges: &[TextRange]) -> Option<TextRange> {
+    let start = ranges.iter().map(|range| range.start()).min()?;
+    let end = ranges.iter().map(|range| range.end()).max()?;
+    Some(TextRange::new(start, end))
+}