Iterates over all library fragments to add asset ids.

Author: simphotonics

lib/src/builders/synthetic_builder.dart

@@ -30,10 +30,12 @@ abstract class SyntheticBuilder<S extends SyntheticInput> implements Builder {
     this.header = '',
     this.footer = '',
     Formatter? formatter,
-  })  : formatter = formatter ??
-            DartFormatter(languageVersion: DartFormatter.latestLanguageVersion)
-                .format,
-        syntheticInput = SyntheticInput.instance<S>();
+  }) : formatter =
+           formatter ??
+           DartFormatter(
+             languageVersion: DartFormatter.latestLanguageVersion,
+           ).format,
+       syntheticInput = SyntheticInput.instance<S>();
 
   /// Input files. Specify the complete path relative to the
   /// root directory.
@@ -70,7 +72,8 @@ abstract class SyntheticBuilder<S extends SyntheticInput> implements Builder {
     // Add header to buffer.
     // Expand header:
     final buffer = StringBuffer(
-        '// GENERATED CODE. DO NOT MODIFY. $generatedBy \n\n $header');
+      '// GENERATED CODE. DO NOT MODIFY. $generatedBy \n\n $header',
+    );
     buffer.writeln();
 
     source.trim();
@@ -98,14 +101,16 @@ abstract class SyntheticBuilder<S extends SyntheticInput> implements Builder {
     return result;
   }
 
-  /// Returns a ordered set of library asset ids ordered in reverse topological
+  /// Returns an ordered set of library asset ids ordered in reverse topological
   /// dependency order.
   /// * If a file B includes a file A, then A will be appear
   /// before B.
   /// * Throws [ErrorOf] if a dependency cycle is detected.
   Future<Set<AssetId>> orderedLibraryAssetIds(BuildStep buildStep) async {
-    final assetGraph =
-        DirectedGraph<AssetId>({}, comparator: ((v1, v2) => -v1.compareTo(v2)));
+    final assetGraph = DirectedGraph<AssetId>(
+      {},
+      comparator: ((v1, v2) => -v1.compareTo(v2)),
+    );
 
     // An assetId map of all input libraries with the uri as key.
     final assetMap = <Uri, AssetId>{};
@@ -120,26 +125,30 @@ abstract class SyntheticBuilder<S extends SyntheticInput> implements Builder {
     }
 
     for (final assetId in assetGraph) {
-      final importedAsseIds = <AssetId>{};
+      final importedAssetIds = <AssetId>{};
 
       // Read library.
       final library = await buildStep.resolver.libraryFor(assetId);
-      // Get dependencies
-      for (final import in library.importedLibraries) {
-        final uri = Uri.parse(import.source.uri.toString());
-        // Skip if uri scheme is not "package" or "asset".
-        if (uri.scheme == 'package' ||
-            uri.scheme == 'asset' ||
-            uri.scheme == '') {
-          // Normalise uri to handle relative and package import directives.
-          final importedAssetId = AssetId.resolve(uri, from: assetId);
-          // Add vertex matching import directive.
-          if (assetMap[importedAssetId.uri] != null) {
-            importedAsseIds.add(assetMap[importedAssetId.uri]!);
+
+      for (final fragment in library.fragments) {
+        for (final import in fragment.importedLibraries) {
+          //final uri = Uri.parse(import.source.uri.toString());
+          // TODO:Iterate over all fragments.
+          final uri = import.uri;
+          // Skip if uri scheme is not "package" or "asset".
+          if (uri.scheme == 'package' ||
+              uri.scheme == 'asset' ||
+              uri.scheme == '') {
+            // Normalise uri to handle relative and package import directives.
+            final importedAssetId = AssetId.resolve(uri, from: assetId);
+            // Add vertex matching import directive.
+            if (assetMap[importedAssetId.uri] != null) {
+              importedAssetIds.add(assetMap[importedAssetId.uri]!);
+            }
           }
         }
       }
-      assetGraph.addEdges(assetId, importedAsseIds);
+      assetGraph.addEdges(assetId, importedAssetIds);
     }
 
     final topologicalOrdering = assetGraph.sortedTopologicalOrdering;
@@ -149,10 +158,12 @@ abstract class SyntheticBuilder<S extends SyntheticInput> implements Builder {
       final cycle = assetGraph.cycle;
 
       throw ErrorOf<SyntheticBuilder>(
-          message: 'Circular dependency detected.',
-          expectedState: 'Input files must not include each other. '
-              'Alternatively, set constructor parameter "sortAssets: false".',
-          invalidState: 'File ${cycle.join(' imports ')}.');
+        message: 'Circular dependency detected.',
+        expectedState:
+            'Input files must not include each other. '
+            'Alternatively, set constructor parameter "sortAssets: false".',
+        invalidState: 'File ${cycle.join(' imports ')}.',
+      );
     }
 
     // Return reversed topological ordering of asset ids.