Add extensions support for avoid_using_api

CHANGELOG.md

@@ -1,7 +1,7 @@
 ## 0.4.0
 
 - Added `allow_with_comments` parameter for `no_empty_block` lint.
-
+- Added extension support for `avoid_using_api`
 ## 0.3.0
 
 - Added `exclude` parameter for the following lints:

lib/src/lints/avoid_using_api/avoid_using_api_linter.dart

@@ -245,5 +245,48 @@ class AvoidUsingApiLinter {
           );
       }
     });
+
+    context.registry.addMethodInvocation((node) {
+      final methodName = node.methodName.name;
+      if (methodName != identifier) {
+        return;
+      }
+
+      final element = node.methodName.staticElement;
+      if (element == null || element.enclosingElement3 is! ExtensionElement) {
+        return;
+      }
+
+      final extensionElement = element.enclosingElement3! as ExtensionElement;
+
+      if (extensionElement.name != className) {
+        return;
+      }
+
+      final sourcePath = extensionElement.librarySource.uri.toString();
+      if (!_matchesSource(sourcePath, source)) {
+        return;
+      }
+
+      reporter.atNode(node.methodName, entryCode);
+    });
+
+    context.registry.addPrefixedIdentifier((node) {
+      final propertyName = node.identifier.name;
+      if (propertyName != identifier) return;
+
+      final element = node.identifier.staticElement;
+      if (element == null || element.enclosingElement3 is! ExtensionElement) {
+        return;
+      }
+
+      final extensionElement = element.enclosingElement3! as ExtensionElement;
+      if (extensionElement.name != className) return;
+
+      final sourcePath = extensionElement.librarySource.uri.toString();
+      if (!_matchesSource(sourcePath, source)) return;
+
+      reporter.atNode(node.identifier, entryCode);
+    });
   }
 }

lint_test/avoid_using_api/external_source/lib/external_source.dart

@@ -22,3 +22,9 @@ const test2 = 'Hello World';
 void test() {}
 
 int banned = 5;
+
+extension BannedExtension on int {
+  int banned() => this + 10;
+
+  int get bannedGetter => 10;
+}

lint_test/avoid_using_api/identifier_extension_source_ban/analysis_options.yaml

@@ -0,0 +1,18 @@
+analyzer:
+  plugins:
+    - ../../custom_lint
+
+custom_lint:
+  rules:
+    - avoid_using_api:
+      severity: warning
+      entries:
+        - class_name: BannedExtension
+          identifier: banned
+          source: package:external_source
+          reason: "Banned identifier from BannedExtension from package:external_source is not allowed"
+        - class_name: BannedExtension
+          identifier: bannedGetter
+          source: package:external_source
+          reason: "bannedGetter identifier from BannedExtension from package:external_source is not allowed"
+

lint_test/avoid_using_api/identifier_extension_source_ban/lib/identifier_extension_source_ban_test.dart

@@ -0,0 +1,11 @@
+import 'package:external_source/external_source.dart';
+
+void extensionIdentifierBanTesting(){
+  const int a = 10;
+
+  // expect_lint: avoid_using_api
+  a.banned();
+
+  // expect_lint: avoid_using_api
+  a.bannedGetter;
+}

lint_test/avoid_using_api/identifier_extension_source_ban/pubspec.yaml

@@ -0,0 +1,17 @@
+name: identifier_extension_source_ban
+description: A sample command-line application.
+version: 1.0.0
+publish_to: none
+
+environment:
+  sdk: ^3.7.2
+
+dependencies:
+  external_source:
+    path: ../external_source
+
+dev_dependencies:
+  lints: ^3.0.0
+  test: ^1.21.0
+  solid_lints:
+    path: ../../../