[analysis_server] Migrate LSP Find References to new element model

Change-Id: I07c78af854c573a759971ee16c21fdc7de376952
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/401060
Commit-Queue: Phil Quitslund <[email protected]>
Reviewed-by: Brian Wilkerson <[email protected]>
Reviewed-by: Phil Quitslund <[email protected]>

Author: DanTup

pkg/analysis_server/lib/src/lsp/handlers/handler_references.dart

@@ -12,7 +12,7 @@ import 'package:analysis_server/src/services/search/search_engine.dart'
     show SearchMatch;
 import 'package:analyzer/dart/analysis/results.dart';
 import 'package:analyzer/dart/ast/ast.dart';
-import 'package:analyzer/dart/element/element.dart';
+import 'package:analyzer/dart/element/element2.dart';
 import 'package:analyzer/src/dart/ast/utilities.dart';
 import 'package:analyzer/src/util/performance/operation_performance.dart';
 import 'package:analyzer/src/utilities/extensions/element.dart';
@@ -51,23 +51,11 @@ class ReferencesHandler
     );
   }
 
-  List<Location> _getDeclarations(Element element) {
-    element = element.nonSynthetic;
-    var unitElement = element.thisOrAncestorOfType<CompilationUnitElement>();
-    if (unitElement == null) {
-      return [];
-    }
-
-    return [
-      Location(
-        uri: uriConverter.toClientUri(unitElement.source.fullName),
-        range: toRange(
-          unitElement.lineInfo,
-          element.nameOffset,
-          element.nameLength,
-        ),
-      ),
-    ];
+  List<Location> _getDeclarations(Element2 element) {
+    return element.nonSynthetic2.fragments
+        .map((fragment) => fragmentToLocation(uriConverter, fragment))
+        .nonNulls
+        .toList();
   }
 
   Future<ErrorOr<List<Location>?>> _getReferences(
@@ -79,9 +67,9 @@ class ReferencesHandler
     var node = NodeLocator(offset).searchWithin(result.unit);
     node = _getReferenceTargetNode(node);
 
-    var element = switch (server.getElementOfNode(node)) {
-      FieldFormalParameterElement(:var field?) => field,
-      PropertyAccessorElement(:var variable2?) => variable2,
+    var element = switch (server.getElementOfNode2(node)) {
+      FieldFormalParameterElement2(:var field2?) => field2,
+      PropertyAccessorElement2(:var variable3?) => variable3,
       (var element) => element,
     };
 
@@ -91,17 +79,11 @@ class ReferencesHandler
 
     var computer = ElementReferencesComputer(server.searchEngine);
     var session = element.session ?? result.session;
-    var element2 =
-        element is LibraryImportElement
-            ? element.prefix?.element.asElement2
-            : element.asElement2;
-    if (element2 == null) {
-      return success(null);
-    }
+
     var results = await performance.runAsync(
       'computer.compute',
       (childPerformance) =>
-          computer.compute(element2, false, performance: childPerformance),
+          computer.compute(element, false, performance: childPerformance),
     );
 
     Location? toLocation(SearchMatch result) {

pkg/analysis_server/test/lsp/references_test.dart

@@ -199,9 +199,9 @@ void f() {
     await _checkRanges(content);
   }
 
-  Future<void> test_import_prefix() async {
+  Future<void> test_import_prefix_declaration() async {
     var content = '''
-imp^ort 'dart:async' as async;
+import 'dart:async' as asy^nc;
 
 /*[0*/async/*0]*/.Future<String>? f() {}
 /*[1*/async/*1]*/.Future<String>? g() {}
@@ -210,6 +210,17 @@ imp^ort 'dart:async' as async;
     await _checkRanges(content);
   }
 
+  Future<void> test_import_prefix_reference() async {
+    var content = '''
+import 'dart:async' as async;
+
+/*[0*/^async/*0]*/.Future<String>? f() {}
+/*[1*/async/*1]*/.Future<String>? g() {}
+''';
+
+    await _checkRanges(content);
+  }
+
   Future<void> test_initializingFormal_argument_withDeclaration() async {
     // Find references on an initializing formal argument should include
     // all references to the field too.