Elements. Migrate src/dart/analysis/search.dart

Replace index / search of `CompilationUnitElement` with special
`referencesLibraryFragment(LibraryFragment)`, because they are not
elements.


Change-Id: I4de4ce42270bdad92996bedbd06f108a7c71dffe
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/412681
Reviewed-by: Paul Berry <[email protected]>
Commit-Queue: Konstantin Shcheglov <[email protected]>

Author: scheglov

pkg/analysis_server/lib/src/services/search/search_engine_internal.dart

@@ -35,7 +35,8 @@ class SearchEngineImpl implements SearchEngine {
             _searchDirectSubtypes(type, searchEngineCache, performance),
       );
       for (var directResult in directResults) {
-        var directSubtype = directResult.enclosingElement2 as InterfaceElement2;
+        var directSubtype =
+            directResult.enclosingFragment.element as InterfaceElement2;
         if (allSubtypes.add(directSubtype)) {
           await addSubtypes(directSubtype);
         }
@@ -124,7 +125,7 @@ class SearchEngineImpl implements SearchEngine {
     var drivers = _drivers.toList();
     var searchedFiles = _createSearchedFiles(drivers);
     for (var driver in drivers) {
-      var elements = await driver.search.classMembers2(name, searchedFiles);
+      var elements = await driver.search.classMembers(name, searchedFiles);
       allDeclarations.addAll(elements.map(SearchMatchImpl.forElement));
     }
     return allDeclarations;
@@ -161,7 +162,7 @@ class SearchEngineImpl implements SearchEngine {
     var drivers = _drivers.toList();
     var searchedFiles = _createSearchedFiles(drivers);
     for (var driver in drivers) {
-      var results = await driver.search.references2(element, searchedFiles);
+      var results = await driver.search.references(element, searchedFiles);
       allResults.addAll(results);
     }
     return allResults.map(SearchMatchImpl.forSearchResult).toList();
@@ -194,7 +195,7 @@ class SearchEngineImpl implements SearchEngine {
     var regExp = RegExp(pattern);
     var drivers = _drivers.toList();
     for (var driver in drivers) {
-      var elements = await driver.search.topLevelElements2(regExp);
+      var elements = await driver.search.topLevelElements(regExp);
       allElements.addAll(elements);
     }
     return allElements.map(SearchMatchImpl.forElement).toList();
@@ -336,7 +337,7 @@ class SearchMatchImpl implements SearchMatch {
       libraryFragment.element.firstFragment.source,
       libraryFragment.source,
       libraryFragment.element,
-      result.enclosingElement2,
+      result.enclosingFragment.element,
       result.isResolved,
       result.isQualified,
       toMatchKind(result.kind),

pkg/analyzer/lib/src/dart/analysis/index.dart

@@ -830,9 +830,17 @@ class _IndexContributor extends GeneralizingAstVisitor {
   }
 
   @override
-  void visitExportDirective(ExportDirective node) {
-    var element = node.libraryExport;
-    recordUriReference(element?.exportedLibrary2, node.uri);
+  void visitExportDirective(covariant ExportDirectiveImpl node) {
+    if (node.libraryExport case var libraryExport?) {
+      if (libraryExport.exportedLibrary2 case var exportedLibrary?) {
+        assembler.addLibraryFragmentReference(
+          target: exportedLibrary.firstFragment,
+          uriOffset: node.uri.offset,
+          uriLength: node.uri.length,
+        );
+      }
+    }
+
     super.visitExportDirective(node);
   }
 
@@ -898,9 +906,17 @@ class _IndexContributor extends GeneralizingAstVisitor {
   }
 
   @override
-  void visitImportDirective(ImportDirective node) {
-    var element = node.libraryImport;
-    recordUriReference(element?.importedLibrary2, node.uri);
+  void visitImportDirective(covariant ImportDirectiveImpl node) {
+    if (node.libraryImport case var libraryImport?) {
+      if (libraryImport.importedLibrary2 case var importedLibrary?) {
+        assembler.addLibraryFragmentReference(
+          target: importedLibrary.firstFragment,
+          uriOffset: node.uri.offset,
+          uriLength: node.uri.length,
+        );
+      }
+    }
+
     super.visitImportDirective(node);
   }
 

pkg/analyzer/lib/src/dart/analysis/library_analyzer.dart

@@ -1111,7 +1111,7 @@ class LibraryAnalyzer {
 /// Analysis result for single file.
 class UnitAnalysisResult {
   final FileState file;
-  final CompilationUnit unit;
+  final CompilationUnitImpl unit;
   final List<AnalysisError> errors;
 
   UnitAnalysisResult(this.file, this.unit, this.errors);

pkg/analyzer/lib/src/dart/analysis/results.dart

@@ -6,7 +6,6 @@
 
 import 'package:analyzer/dart/analysis/results.dart';
 import 'package:analyzer/dart/analysis/session.dart';
-import 'package:analyzer/dart/ast/ast.dart';
 import 'package:analyzer/dart/ast/token.dart';
 import 'package:analyzer/dart/ast/visitor.dart';
 import 'package:analyzer/dart/element/element.dart';
@@ -16,6 +15,7 @@ import 'package:analyzer/error/error.dart';
 import 'package:analyzer/file_system/file_system.dart';
 import 'package:analyzer/source/line_info.dart';
 import 'package:analyzer/src/dart/analysis/file_state.dart';
+import 'package:analyzer/src/dart/ast/ast.dart';
 import 'package:analyzer/src/dart/element/element.dart';
 import 'package:analyzer/src/dart/element/type_provider.dart';
 import 'package:analyzer/src/dart/element/type_system.dart';
@@ -490,7 +490,7 @@ class ResolvedLibraryResultImpl extends AnalysisResultImpl
 class ResolvedUnitResultImpl extends FileResultImpl
     implements ResolvedUnitResult {
   @override
-  final CompilationUnit unit;
+  final CompilationUnitImpl unit;
 
   @override
   final List<AnalysisError> errors;
@@ -538,5 +538,5 @@ class UnitElementResultImpl extends FileResultImpl
   });
 
   @override
-  LibraryFragment get fragment => element;
+  CompilationUnitElementImpl get fragment => element;
 }