Split out post process builder state.

Author: davidmorgan

build_runner_core/lib/src/asset_graph/graph.dart

@@ -46,7 +46,7 @@ class AssetGraph implements GeneratedAssetHider {
 
   final BuiltMap<String, LanguageVersion?> packageLanguageVersions;
 
-  Map<AssetId, Set<AssetId>>? _anchorOutputs;
+  final Map<PostProcessAction, Set<AssetId>> _postProcessActions = {};
 
   AssetGraph._(
     this.buildPhasesDigest,
@@ -161,11 +161,6 @@ class AssetGraph implements GeneratedAssetHider {
     }
     _nodesByPackage.putIfAbsent(node.id.package, () => {})[node.id.path] = node;
 
-    if (node.type == NodeType.postProcessAnchor) {
-      // Invalidate `_anchorOutputs`, it will be recomputed when needed.
-      _anchorOutputs = null;
-    }
-
     return node;
   }
 
@@ -255,14 +250,14 @@ class AssetGraph implements GeneratedAssetHider {
     var node = get(id);
     if (node == null) return removedIds;
     removedIds.add(id);
-    anchorOutputs ??= _computeAnchorOutputs();
+    /*anchorOutputs ??= _computeAnchorOutputs();
     for (var anchor in (anchorOutputs[node.id] ?? const <AssetId>{})) {
       _removeRecursive(
         anchor,
         removedIds: removedIds,
         anchorOutputs: anchorOutputs,
       );
-    }
+    }*/
     for (var output in node.primaryOutputs.toList()) {
       _removeRecursive(
         output,
@@ -320,7 +315,7 @@ class AssetGraph implements GeneratedAssetHider {
   /// Returns a cached value if possible. Recomputes if any
   /// [AssetNode.postProcessAnchor] has been added since it was last computed;
   /// not recomputed for removes as values for missing nodes are harmless.
-  Map<AssetId, Set<AssetId>> _computeAnchorOutputs() {
+  /*Map<AssetId, Set<AssetId>> _computeAnchorOutputs() {
     if (_anchorOutputs != null) return _anchorOutputs!;
     final result = <AssetId, Set<AssetId>>{};
     for (final node in allNodes) {
@@ -330,7 +325,7 @@ class AssetGraph implements GeneratedAssetHider {
       }
     }
     return _anchorOutputs = result;
-  }
+  }*/
 
   /// All nodes in the graph, whether source files or generated outputs.
   Iterable<AssetNode> get allNodes =>
@@ -340,6 +335,21 @@ class AssetGraph implements GeneratedAssetHider {
   Iterable<AssetNode> packageNodes(String package) =>
       _nodesByPackage[package]?.values ?? [];
 
+  Iterable<PostProcessAction> packagePostProcessActions(String package) =>
+      _postProcessActions.keys.where((a) => a.primaryInput.package == package);
+
+  void clearPostProcessActionOutputs(PostProcessAction action) {
+    _postProcessActions[action]!.clear();
+  }
+
+  void addPostProcessActionOutput(PostProcessAction action, AssetId output) {
+    _postProcessActions[action]!.add(output);
+  }
+
+  Iterable<AssetId> postProcessActionOutputs(PostProcessAction action) {
+    return _postProcessActions[action]!;
+  }
+
   /// All the generated outputs in the graph.
   Iterable<AssetId> get outputs =>
       allNodes.where((n) => n.type == NodeType.generated).map((n) => n.id);
@@ -649,12 +659,12 @@ class AssetGraph implements GeneratedAssetHider {
       var inputs = allInputs.where((input) => _actionMatches(action, input));
       for (var input in inputs) {
         var buildOptionsNodeId = builderOptionsIdForAction(action, actionNum);
-        var anchor = AssetNode.postProcessAnchorForInputAndAction(
-          input,
-          actionNum,
-          buildOptionsNodeId,
-        );
-        add(anchor);
+        var postProcessAction = PostProcessAction((b) {
+          b.primaryInput = input;
+          b.actionNumber = actionNum;
+          b.builderOptionsId = buildOptionsNodeId;
+        });
+        _postProcessActions[postProcessAction] = {};
       }
       actionNum++;
     }

build_runner_core/lib/src/asset_graph/node.dart

@@ -10,8 +10,6 @@ import 'package:built_value/built_value.dart';
 import 'package:built_value/serializer.dart';
 import 'package:crypto/crypto.dart';
 
-import '../generate/phase.dart';
-
 part 'node.g.dart';
 
 /// Types of [AssetNode].
@@ -348,7 +346,7 @@ abstract class GlobNodeState
 abstract class PostProcessAction
     implements Built<PostProcessAction, PostProcessActionBuilder> {
   static Serializer<PostProcessAction> get serializer =>
-      _$postProcessAnchorNodeSerializer;
+      _$postProcessActionSerializer;
 
   int get actionNumber;
   AssetId get builderOptionsId;
@@ -357,7 +355,7 @@ abstract class PostProcessAction
   PostProcessAction._();
 
   factory PostProcessAction(void Function(PostProcessActionBuilder) updates) =
-      _$PostProcessAnchorNodeConfiguration;
+      _$PostProcessAction;
 }
 
 /// Work that needs doing for a node that tracks its inputs.