[element model] fix no_wildcard_variable_uses migration

Bug: dart-lang/linter#5099 Change-Id: Ibde9394aa21fd4d81242e51cfad3f6166c1b2653 Reviewed-on: https://dart-review.googlesource.com/c/sdk/+/388044 Auto-Submit: Phil Quitslund <[email protected]> Reviewed-by: Brian Wilkerson <[email protected]> Commit-Queue: Brian Wilkerson <[email protected]>
dart-lang · Oct 3, 2024 · 7b3718c · 7b3718c
1 parent c514eb5
commit 7b3718c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/pkg/analyzer/lib/src/dart/element/extensions.dart b/pkg/analyzer/lib/src/dart/element/extensions.dart
@@ -4,6 +4,7 @@
 
 import 'package:analyzer/dart/analysis/features.dart';
 import 'package:analyzer/dart/element/element.dart';
+import 'package:analyzer/dart/element/element2.dart';
 import 'package:analyzer/dart/element/nullability_suffix.dart';
 import 'package:analyzer/dart/element/type.dart';
 import 'package:analyzer/src/dart/element/element.dart';
@@ -161,6 +162,11 @@ extension LibraryExtension on LibraryElement? {
   }
 }
 
+extension LibraryExtension2 on LibraryElement2? {
+  bool get hasWildcardVariablesFeatureEnabled2 =>
+      this?.featureSet.isEnabled(Feature.wildcard_variables) ?? false;
+}
+
 extension ParameterElementExtensions on ParameterElement {
   /// Return [ParameterElement] with the specified properties replaced.
   ParameterElement copyWith({

diff --git a/pkg/linter/lib/src/rules/no_wildcard_variable_uses.dart b/pkg/linter/lib/src/rules/no_wildcard_variable_uses.dart
@@ -26,9 +26,7 @@ class NoWildcardVariableUses extends LintRule {
   @override
   void registerNodeProcessors(
       NodeLintRegistry registry, LinterContext context) {
-    if (context.libraryElement.hasWildcardVariablesFeatureEnabled) {
-      return;
-    }
+    if (context.libraryElement2.hasWildcardVariablesFeatureEnabled2) return;
 
     var visitor = _Visitor(this);
     registry.addSimpleIdentifier(this, visitor);