[element model] migrate omit_local_variable_types

Bug: https://github.com/dart-lang/linter/issues/5099
Change-Id: I199a50d1df93d7002fe8c8179a26634efef8cba3
Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/390624
Commit-Queue: Phil Quitslund <[email protected]>
Auto-Submit: Phil Quitslund <[email protected]>
Reviewed-by: Brian Wilkerson <[email protected]>

Author: pq

pkg/linter/lib/src/rules/omit_local_variable_types.dart

@@ -4,7 +4,7 @@
 
 import 'package:analyzer/dart/ast/ast.dart';
 import 'package:analyzer/dart/ast/visitor.dart';
-import 'package:analyzer/dart/element/element.dart';
+import 'package:analyzer/dart/element/element2.dart';
 import 'package:analyzer/dart/element/type.dart';
 import 'package:analyzer/dart/element/type_provider.dart';
 
@@ -57,8 +57,9 @@ class _Visitor extends SimpleAstVisitor<void> {
       var loopType = loopParts.iterable.staticType;
       if (loopType is! InterfaceType) return;
 
-      var iterableType = loopType.asInstanceOf(typeProvider.iterableElement);
+      var iterableType = loopType.asInstanceOf2(typeProvider.iterableElement2);
       if (iterableType == null) return;
+
       if (iterableType.typeArguments.isNotEmpty &&
           iterableType.typeArguments.first == staticType) {
         rule.reportLint(loopVariableType);
@@ -100,12 +101,12 @@ class _Visitor extends SimpleAstVisitor<void> {
 extension on Expression {
   bool get dependsOnDeclaredTypeForInference {
     if (this case MethodInvocation(:var methodName, typeArguments: null)) {
-      var element = methodName.staticElement;
-      if (element is FunctionElement) {
-        if (element.returnType is TypeParameterType) {
-          return true;
-        }
-      }
+      var element = methodName.element;
+      return switch (element) {
+        LocalFunctionElement() => element.returnType is TypeParameterType,
+        TopLevelFunctionElement() => element.returnType is TypeParameterType,
+        _ => false,
+      };
     }
     return false;
   }