Elements. Migrate lib/src/lsp/handlers/handler_implementation.dart

Change-Id: Iaa32be3faad25e838dee04fe15833ebb51acb7db
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/399580
Commit-Queue: Konstantin Shcheglov <[email protected]>
Reviewed-by: Samuel Rawlins <[email protected]>

Author: scheglov

pkg/analysis_server/analyzer_use_new_elements.txt

@@ -14,7 +14,6 @@ lib/src/lsp/handlers/commands/abstract_refactor.dart
 lib/src/lsp/handlers/handler_completion.dart
 lib/src/lsp/handlers/handler_completion_resolve.dart
 lib/src/lsp/handlers/handler_definition.dart
-lib/src/lsp/handlers/handler_implementation.dart
 lib/src/lsp/handlers/handler_references.dart
 lib/src/lsp/handlers/handler_rename.dart
 lib/src/protocol_server.dart

pkg/analysis_server/lib/src/analysis_server.dart

@@ -55,6 +55,7 @@ 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/element/element.dart';
+import 'package:analyzer/dart/element/element2.dart';
 import 'package:analyzer/error/error.dart';
 import 'package:analyzer/exception/exception.dart';
 import 'package:analyzer/file_system/file_system.dart';
@@ -291,8 +292,8 @@ abstract class AnalysisServer {
        ),
        producerGeneratorsForLintRules = AssistProcessor.computeLintRuleMap(),
        messageScheduler = MessageScheduler(
-            testView:
-                retainDataForTesting ? MessageSchedulerTestView() : null) {
+         testView: retainDataForTesting ? MessageSchedulerTestView() : null,
+       ) {
     messageScheduler.setServer(this);
     // Set the default URI converter. This uses the resource providers path
     // context (unlike the initialized value) which allows tests to override it.
@@ -696,6 +697,24 @@ abstract class AnalysisServer {
     return element;
   }
 
+  /// Return the [Element] of the given [node], or `null` if [node] is `null` or
+  /// does not have an element.
+  Element2? getElementOfNode2(AstNode? node) {
+    if (node == null) {
+      return null;
+    }
+    if (node is SimpleIdentifier && node.parent is LibraryIdentifier) {
+      node = node.parent;
+    }
+    if (node is LibraryIdentifier) {
+      node = node.parent;
+    }
+    if (node is StringLiteral && node.parent is UriBasedDirective) {
+      return null;
+    }
+    return ElementLocator.locate2(node);
+  }
+
   /// Return a [LineInfo] for the file with the given [path].
   ///
   /// If the file does not exist or cannot be read, returns `null`.

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

@@ -10,10 +10,9 @@ import 'package:analysis_server/src/lsp/mapping.dart';
 import 'package:analysis_server/src/lsp/registration/feature_registration.dart';
 import 'package:analysis_server/src/search/type_hierarchy.dart';
 import 'package:analyzer/dart/analysis/results.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';
 
 typedef StaticOptions =
     Either3<bool, ImplementationOptions, ImplementationRegistrationOptions>;
@@ -64,23 +63,23 @@ class ImplementationHandler
     OperationPerformanceImpl performance,
   ) async {
     var node = NodeLocator(offset).searchWithin(result.unit);
-    var element = server.getElementOfNode(node);
+    var element = server.getElementOfNode2(node);
     if (element == null) {
       return success([]);
     }
 
-    var helper = TypeHierarchyComputerHelper.fromElement(element.asElement2!);
+    var helper = TypeHierarchyComputerHelper.fromElement(element);
     var interfaceElement = helper.pivotClass;
     if (interfaceElement == null) {
       return success([]);
     }
     var needsMember = helper.findMemberElement(interfaceElement) != null;
 
-    var allSubtypes = <InterfaceElement>{};
+    var allSubtypes = <InterfaceElement2>{};
     await performance.runAsync(
       'appendAllSubtypes',
-      (performance) => server.searchEngine.appendAllSubtypes(
-        interfaceElement.asElement,
+      (performance) => server.searchEngine.appendAllSubtypes2(
+        interfaceElement,
         allSubtypes,
         performance,
       ),
@@ -95,8 +94,8 @@ class ImplementationHandler
                     // Filter based on type, so when searching for members we don't
                     // include any intermediate classes that don't have
                     // implementations for the method.
-                    ? helper.findMemberElement(element.asElement2)?.nonSynthetic2
-                    : element.asElement2;
+                    ? helper.findMemberElement(element)?.nonSynthetic2
+                    : element;
               })
               .nonNulls
               .toSet()