Generate for directive

example/pubspec.yaml

@@ -6,7 +6,7 @@ environment:
   sdk: ^3.6.0
 
 dependencies:
-  analyzer: '>=5.2.0 <7.0.0'
+  analyzer: '>=6.9.0 <8.0.0'
   build: ^2.0.0
   source_gen: any
 

source_gen/CHANGELOG.md

@@ -1,4 +1,4 @@
-## 2.0.0-wip
+## 2.0.0
 
 - **Breaking Change**: Change `formatOutput` function to accept a language
   version parameter.
@@ -8,13 +8,14 @@
 - Document deduplication behavior for the output of
   `GeneratorForAnnotation.generateForAnnotatedElement`.
 - Support all the glob quotes.
-- Require `analyzer: ^6.9.0`
-- Require Dart 3.6.0
+- Require `analyzer: '>=6.9.0 <8.0.0'`
+- Support the latest `package:dart_style`
 - `LibraryBuilder`, `PartBuilder`, and `SharedPartBuilder` now take an optional
   `writeDescriptions` boolean. When set to `false`, headers and generator
   descriptions for the files will not be included in the builder output.
 - Include `//dart format width=80` comments in files generated by a
   `LibraryBuilder` or `PartBuilder` and formatted with the default callback.
+- Require Dart 3.6.0
 
 ## 1.5.0
 

source_gen/lib/source_gen.dart

@@ -9,7 +9,8 @@ export 'src/constants/revive.dart' show Revivable;
 export 'src/generator.dart'
     show Generator, InvalidGenerationSource, InvalidGenerationSourceError;
 export 'src/generator_for_annotation.dart' show GeneratorForAnnotation;
-export 'src/library.dart' show AnnotatedElement, LibraryReader;
+export 'src/library.dart'
+    show AnnotatedDirective, AnnotatedElement, LibraryReader;
 export 'src/span_for_element.dart' show spanForElement, spanForElement2;
 export 'src/type_checker.dart' show TypeChecker, UnresolvedAnnotationException;
 export 'src/utils.dart' show typeNameOf;

source_gen/lib/src/generator_for_annotation.dart

@@ -55,6 +55,21 @@ abstract class GeneratorForAnnotation<T> extends Generator {
   FutureOr<String> generate(LibraryReader library, BuildStep buildStep) async {
     final values = <String>{};
 
+    for (var annotatedDirective in library.libraryDirectivesAnnotatedWith(
+      typeChecker,
+      throwOnUnresolved: throwOnUnresolved,
+    )) {
+      final generatedValue = generateForAnnotatedDirective(
+        annotatedDirective.directive,
+        annotatedDirective.annotation,
+        buildStep,
+      );
+      await for (var value in normalizeGeneratorOutput(generatedValue)) {
+        assert(value.length == value.trim().length);
+        values.add(value);
+      }
+    }
+
     for (var annotatedElement in library.annotatedWith(
       typeChecker,
       throwOnUnresolved: throwOnUnresolved,
@@ -123,4 +138,27 @@ abstract class GeneratorForAnnotation<T> extends Generator {
     ConstantReader annotation,
     BuildStep buildStep,
   ) {}
+
+  /// Implement to return source code to generate for [directive].
+  ///
+  /// This method is invoked based on finding directives annotated with an
+  /// instance of [T]. The [annotation] is provided as a [ConstantReader].
+  ///
+  /// Supported return values include a single [String] or multiple [String]
+  /// instances within an [Iterable] or [Stream]. It is also valid to return a
+  /// [Future] of [String], [Iterable], or [Stream]. When multiple values are
+  /// returned through an iterable or stream they will be deduplicated.
+  /// Typically each value will be an independent unit of code and the
+  /// deduplication prevents re-defining the same member multiple times. For
+  /// example if multiple annotated elements may need a specific utility method
+  /// available it can be output for each one, and the single deduplicated
+  /// definition can be shared.
+  ///
+  /// Implementations should return `null` when no content is generated. Empty
+  /// or whitespace-only [String] instances are also ignored.
+  dynamic generateForAnnotatedDirective(
+    ElementDirective directive,
+    ConstantReader annotation,
+    BuildStep buildStep,
+  ) {}
 }

source_gen/lib/src/library.dart

@@ -13,6 +13,16 @@ import 'constants/reader.dart';
 import 'type_checker.dart';
 import 'utils.dart';
 
+/// Result of finding an [annotation] on [directive] through [LibraryReader].
+class AnnotatedDirective {
+  final ConstantReader annotation;
+  final ElementDirective directive;
+
+  const AnnotatedDirective(this.annotation, this.directive);
+
+  Metadata? get metadata2 => directive.metadata2;
+}
+
 /// Result of finding an [annotation] on [element] through [LibraryReader].
 class AnnotatedElement {
   final ConstantReader annotation;
@@ -87,6 +97,30 @@ class LibraryReader {
     }
   }
 
+  /// All of the directives in this library annotated with [checker].
+  Iterable<AnnotatedDirective> libraryDirectivesAnnotatedWith(
+    TypeChecker checker, {
+    bool throwOnUnresolved = true,
+  }) sync* {
+    final firstFragment = element2.firstFragment;
+    final directives = [
+      ...firstFragment.libraryImports2,
+      ...firstFragment.libraryExports2,
+      ...firstFragment.partIncludes,
+    ];
+
+    for (final directive in directives) {
+      final annotation = checker.firstAnnotationOf2(
+        directive,
+        throwOnUnresolved: throwOnUnresolved,
+      );
+
+      if (annotation != null) {
+        yield AnnotatedDirective(ConstantReader(annotation), directive);
+      }
+    }
+  }
+
   /// All of the declarations in this library annotated with exactly [checker].
   Iterable<AnnotatedElement> annotatedWithExact(
     TypeChecker checker, {

source_gen/lib/src/type_checker.dart

@@ -84,7 +84,7 @@ abstract class TypeChecker {
   ///
   /// Throws on unresolved annotations unless [throwOnUnresolved] is `false`.
   DartObject? firstAnnotationOf2(
-    Element2 element, {
+    Object element, {
     bool throwOnUnresolved = true,
   }) {
     if (element case final Annotatable annotatable) {
@@ -188,13 +188,13 @@ abstract class TypeChecker {
   }
 
   DartObject? _computeConstantValue2(
-    Element2 element,
+    Object element,
     ElementAnnotation annotation,
     int annotationIndex, {
     bool throwOnUnresolved = true,
   }) {
     final result = annotation.computeConstantValue();
-    if (result == null && throwOnUnresolved) {
+    if (result == null && throwOnUnresolved && element is Element2) {
       throw UnresolvedAnnotationException._from(element, annotationIndex);
     }
     return result;
@@ -219,7 +219,7 @@ abstract class TypeChecker {
   /// Throws [UnresolvedAnnotationException] on unresolved annotations unless
   /// [throwOnUnresolved] is explicitly set to `false` (default is `true`).
   Iterable<DartObject> annotationsOf2(
-    Element2 element, {
+    Object element, {
     bool throwOnUnresolved = true,
   }) =>
       _annotationsWhere2(
@@ -246,7 +246,7 @@ abstract class TypeChecker {
   }
 
   Iterable<DartObject> _annotationsWhere2(
-    Element2 element,
+    Object element,
     bool Function(DartType) predicate, {
     bool throwOnUnresolved = true,
   }) sync* {

source_gen/pubspec.yaml

@@ -1,5 +1,5 @@
 name: source_gen
-version: 2.0.0-wip
+version: 2.0.0
 description: >-
   Source code generation builders and utilities for the Dart build system
 repository: https://github.com/dart-lang/source_gen/tree/master/source_gen
@@ -10,10 +10,10 @@ environment:
   sdk: ^3.6.0
 
 dependencies:
-  analyzer: ^6.9.0
+  analyzer: '>=6.9.0 <8.0.0'
   async: ^2.5.0
   build: ^2.1.0
-  dart_style: ^2.3.7
+  dart_style: '>=2.3.7 <4.0.0'
   glob: ^2.0.0
   path: ^1.8.0
   pub_semver: ^2.1.4
@@ -29,7 +29,8 @@ dev_dependencies:
   test: ^1.16.0
 
 dependency_overrides:
-  analyzer: ^7.1.0
+  analyzer:
+    path: /Users/scheglov/Source/Dart/sdk.git/sdk/pkg/analyzer
   build:
     git:
       url: https://github.com/dart-lang/build.git