[DAS] Fixes import prefix highlighting

[email protected]

Fixes: #59942
Change-Id: Id5c971f040dc2dcdddde6e36055e9045eb4b2aef
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/404925
Auto-Submit: Felipe Morschel <[email protected]>
Reviewed-by: Konstantin Shcheglov <[email protected]>
Commit-Queue: Phil Quitslund <[email protected]>
Reviewed-by: Phil Quitslund <[email protected]>

Author: FMorschel

pkg/analysis_server/lib/src/domains/analysis/occurrences_dart.dart

@@ -125,6 +125,13 @@ class _DartUnitOccurrencesComputerVisitor extends RecursiveAstVisitor<void> {
     super.visitGenericTypeAlias(node);
   }
 
+  @override
+  void visitImportPrefixReference(ImportPrefixReference node) {
+    _addOccurrence(node.element2!, node.offset);
+
+    super.visitImportPrefixReference(node);
+  }
+
   @override
   void visitMethodDeclaration(MethodDeclaration node) {
     _addOccurrence(node.declaredFragment!.element, node.name.offset);

pkg/analysis_server/test/analysis/notification_occurrences_test.dart

@@ -576,6 +576,23 @@ String f(int char) {
       ''');
   }
 
+  Future<void> test_prefix() async {
+    await assertOccurrences(kind: ElementKind.PREFIX, '''
+import '' as /*[0*/p/*0]*/;
+
+class A {
+  void m() {
+    /*[1*/p/*1]*/.foo();
+    print(/*[2*/p/*2]*/.a);
+  }
+}
+
+void foo() {}
+
+/*[3*/p/*3]*/.A? a;
+''');
+  }
+
   Future<void> test_prefix_wildcard() async {
     // Ensure no crash.
     await assertOccurrences(kind: ElementKind.PREFIX, '''

pkg/analysis_server/test/lsp/document_highlights_test.dart

@@ -189,6 +189,21 @@ void f() {
     /*[2*/key/*2]*/;
   }
 }
+''');
+
+  Future<void> test_prefix() => _testMarkedContent('''
+import '' as /*[0*/p/*0]*/;
+
+class A {
+  void m() {
+    /*[1*/p/*1]*/.foo();
+    print(/*[2*/p/*2]*/.a);
+  }
+}
+
+void foo() {}
+
+/*[3*/p^/*3]*/.A? a;
 ''');
 
   Future<void> test_prefixed() => _testMarkedContent('''