@bian-womp/spark-graph 0.3.44 → 0.3.45

package/lib/cjs/index.cjs

@@ -495,7 +495,8 @@ Registry.idCounter = 0;
  * This is the ONLY place where nodes, edges, and resolvedByNode are directly updated.
  */
 class Graph {
-    constructor(registry) {
+    constructor(eventEmitter, registry) {
+        this.eventEmitter = eventEmitter;
         this.registry = registry;
         this.nodes = new Map();
         this.edges = [];
@@ -552,7 +553,7 @@ class Graph {
     /**
      * Update node inputs
      */
-    updateNodeInput(nodeId, handle, value) {
+    updateNodeInput(nodeId, handle, value, calculated = true) {
         const node = this.getNodeMutable(nodeId);
         if (!node)
             return;
@@ -562,15 +563,16 @@ class Graph {
         else {
             node.inputs[handle] = value;
         }
-    }
-    /**
-     * Delete a node input handle
-     */
-    deleteNodeInput(nodeId, handle) {
-        const node = this.nodes.get(nodeId);
-        if (!node)
-            return;
-        delete node.inputs[handle];
+        if (calculated) {
+            this.updateNodeLastInputAt(nodeId, handle, Date.now());
+        }
+        this.eventEmitter.emit("value", {
+            nodeId,
+            handle,
+            value,
+            io: "input",
+            runtimeTypeId: unwrapTypeId(value),
+        });
     }
     /**
      * Update node outputs
@@ -579,7 +581,19 @@ class Graph {
         const node = this.nodes.get(nodeId);
         if (!node)
             return;
-        node.outputs[handle] = value;
+        if (value === undefined) {
+            delete node.outputs[handle];
+        }
+        else {
+            node.outputs[handle] = value;
+        }
+        this.eventEmitter.emit("value", {
+            nodeId,
+            handle,
+            value,
+            io: "output",
+            runtimeTypeId: unwrapTypeId(value),
+        });
     }
     /**
      * Update node state
@@ -652,9 +666,8 @@ class Graph {
         const node = this.nodes.get(nodeId);
         if (!node)
             return;
-        if (!node.lastInputAt) {
+        if (!node.lastInputAt)
             node.lastInputAt = {};
-        }
         node.lastInputAt[handle] = timestamp;
     }
     /**
@@ -1977,13 +1990,6 @@ class EdgePropagator {
             return false;
         }
         this.graph.updateNodeOutput(srcNodeId, srcHandle, value);
-        this.eventEmitter.emit("value", {
-            nodeId: srcNodeId,
-            handle: srcHandle,
-            value,
-            io: "output",
-            runtimeTypeId: unwrapTypeId(value),
-        });
         return true;
     }
     /**
@@ -2205,16 +2211,7 @@ class EdgePropagator {
      * Set target input value and emit event
      */
     setTargetInput(edge, dstNode, value) {
-        const now = Date.now();
         this.graph.updateNodeInput(edge.target.nodeId, edge.target.handle, value);
-        this.graph.updateNodeLastInputAt(edge.target.nodeId, edge.target.handle, now);
-        this.eventEmitter.emit("value", {
-            nodeId: edge.target.nodeId,
-            handle: edge.target.handle,
-            value,
-            io: "input",
-            runtimeTypeId: unwrapTypeId(value),
-        });
         this.handleResolver.scheduleRecomputeHandles(edge.target.nodeId);
     }
     /**
@@ -3024,8 +3021,8 @@ class GraphRuntime {
         this.pauseRefCount = 0;
         this.persistentPauseToken = null;
         // Initialize components
-        this.graph = new Graph();
         this.eventEmitter = new EventEmitter();
+        this.graph = new Graph(this.eventEmitter);
         this.runContextManager = new RunContextManager(this.graph, "debug");
         this.handleResolver = new HandleResolver(this.graph, this.eventEmitter, this.runContextManager, this);
         this.edgePropagator = new EdgePropagator(this.graph, this.eventEmitter, this.runContextManager, this.handleResolver, this);
@@ -3439,14 +3436,7 @@ class GraphRuntime {
                     const clonedValue = structuredClone(v);
                     const same = valuesEqual(prev, clonedValue);
                     if (!same) {
-                        this.graph.updateNodeInput(nodeId, h, clonedValue);
-                        this.eventEmitter.emit("value", {
-                            nodeId,
-                            handle: h,
-                            value: clonedValue,
-                            io: "input",
-                            runtimeTypeId: unwrapTypeId(clonedValue),
-                        });
+                        this.graph.updateNodeInput(nodeId, h, clonedValue, false);
                         nodeChanged = true;
                     }
                 }
@@ -3461,13 +3451,6 @@ class GraphRuntime {
                 for (const [h, v] of Object.entries(map || {})) {
                     const clonedValue = structuredClone(v);
                     this.graph.updateNodeOutput(nodeId, h, clonedValue);
-                    this.eventEmitter.emit("value", {
-                        nodeId,
-                        handle: h,
-                        value: clonedValue,
-                        io: "output",
-                        runtimeTypeId: unwrapTypeId(clonedValue),
-                    });
                 }
             }
             // Trigger handle resolution for nodes with changed inputs
@@ -3657,7 +3640,7 @@ class GraphRuntime {
                         if (!currSet.has(handle)) {
                             const node = this.graph.getNode(nodeId);
                             if (node && handle in node.inputs) {
-                                this.graph.deleteNodeInput(nodeId, handle);
+                                this.graph.updateNodeInput(nodeId, handle, undefined);
                                 changed = true;
                             }
                         }