@bian-womp/spark-workbench 0.2.80 → 0.2.82

package/lib/cjs/index.cjs

@@ -1174,9 +1174,9 @@ class LocalGraphRunner extends AbstractGraphRunner {
             this.build(payload.def);
         }
         this.setEnvironment?.(payload.environment || {}, { merge: false });
-        this.hydrate(payload);
+        this.hydrate(payload, { dry: options?.dry });
     }
-    hydrate(snapshot) {
+    hydrate(snapshot, opts) {
         // Hydrate via runtime for exact restore (this emits events on runtime emitter)
         this.runtime?.hydrate({
             inputs: snapshot.inputs || {},
@@ -1185,12 +1185,24 @@ class LocalGraphRunner extends AbstractGraphRunner {
         // Also emit directly from runner to ensure UI gets events even if engine isn't running
         for (const [nodeId, map] of Object.entries(snapshot.inputs || {})) {
             for (const [handle, value] of Object.entries(map || {})) {
-                this.emit("value", { nodeId, handle, value, io: "input" });
+                this.emit("value", {
+                    nodeId,
+                    handle,
+                    value,
+                    io: "input",
+                    dry: opts?.dry,
+                });
             }
         }
         for (const [nodeId, map] of Object.entries(snapshot.outputs || {})) {
             for (const [handle, value] of Object.entries(map || {})) {
-                this.emit("value", { nodeId, handle, value, io: "output" });
+                this.emit("value", {
+                    nodeId,
+                    handle,
+                    value,
+                    io: "output",
+                    dry: opts?.dry,
+                });
             }
         }
     }
@@ -1744,17 +1756,17 @@ class RemoteGraphRunner extends AbstractGraphRunner {
     }
     async applySnapshotFull(payload, options) {
         // Hydrate local cache first so UI can display values immediately
-        this.hydrateValueCache(payload);
+        this.hydrateValueCache(payload, { dry: options?.dry });
         // Then sync with backend
         const client = await this.ensureClient();
-        await client.applySnapshotFull(payload, options);
+        await client.applySnapshotFull(payload, { skipBuild: options?.skipBuild });
     }
     /**
      * Hydrates the local valueCache from a snapshot and emits value events.
      * This ensures the UI can display inputs/outputs immediately without waiting
      * for value events from the remote backend.
      */
-    hydrateValueCache(snapshot) {
+    hydrateValueCache(snapshot, opts) {
         // Hydrate inputs
         for (const [nodeId, map] of Object.entries(snapshot.inputs || {})) {
             for (const [handle, value] of Object.entries(map || {})) {
@@ -1762,7 +1774,13 @@ class RemoteGraphRunner extends AbstractGraphRunner {
                     io: "input",
                     value,
                 });
-                this.emit("value", { nodeId, handle, value, io: "input" });
+                this.emit("value", {
+                    nodeId,
+                    handle,
+                    value,
+                    io: "input",
+                    dry: opts?.dry,
+                });
             }
         }
         // Hydrate outputs
@@ -1772,7 +1790,13 @@ class RemoteGraphRunner extends AbstractGraphRunner {
                     io: "output",
                     value,
                 });
-                this.emit("value", { nodeId, handle, value, io: "output" });
+                this.emit("value", {
+                    nodeId,
+                    handle,
+                    value,
+                    io: "output",
+                    dry: opts?.dry,
+                });
             }
         }
     }
@@ -2772,6 +2796,50 @@ async function upload(parsed, wb, runner) {
     }
 }
 
+/**
+ * Merge UI state from source into target, remapping node IDs using nodeIdMap.
+ * Preserves target state and adds/updates source state with remapped IDs.
+ */
+function mergeUIState(targetUI, sourceUI, nodeIdMap) {
+    const result = {
+        ...targetUI,
+    };
+    if (!sourceUI)
+        return result;
+    // Merge positions (already handled by mergeSnapshotData, but included for completeness)
+    if (sourceUI.positions) {
+        result.positions = {
+            ...(targetUI?.positions || {}),
+            ...Object.fromEntries(Object.entries(sourceUI.positions).map(([oldId, pos]) => [
+                nodeIdMap[oldId] || oldId,
+                pos,
+            ])),
+        };
+    }
+    // Merge selection: remap node IDs and edge IDs
+    if (sourceUI.selection) {
+        const remappedNodes = (sourceUI.selection.nodes || [])
+            .map((id) => nodeIdMap[id] || id)
+            .filter((id) => id); // Filter out invalid mappings
+        const remappedEdges = sourceUI.selection.edges || []; // Edge IDs don't need remapping typically
+        result.selection = {
+            nodes: [...(targetUI?.selection?.nodes || []), ...remappedNodes],
+            edges: [...(targetUI?.selection?.edges || []), ...remappedEdges],
+        };
+    }
+    // Merge nodeNames: remap node IDs
+    if (sourceUI.nodeNames) {
+        result.nodeNames = {
+            ...(targetUI?.nodeNames || {}),
+            ...Object.fromEntries(Object.entries(sourceUI.nodeNames).map(([oldId, name]) => [
+                nodeIdMap[oldId] || oldId,
+                name,
+            ])),
+        };
+    }
+    return result;
+}
+
 const WorkbenchContext = React.createContext(null);
 function useWorkbenchContext() {
     const ctx = React.useContext(WorkbenchContext);
@@ -3060,6 +3128,8 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
                 wb.refreshValidation();
         };
         const offRunnerValue = runner.on("value", (e) => {
+            if (e?.dry)
+                return;
             const now = Date.now();
             if (e?.io === "input") {
                 const nodeId = e.nodeId;
@@ -4293,7 +4363,7 @@ function DefaultNodeHeader({ id, typeId, title, validation, right, showId, onInv
     const [editValue, setEditValue] = React.useState("");
     const inputRef = React.useRef(null);
     // Use getNodeDisplayName if typeId is provided, otherwise use title prop
-    const displayName = typeId ? ctx.getNodeDisplayName(id) : (title ?? id);
+    const displayName = typeId ? ctx.getNodeDisplayName(id) : title ?? id;
     const effectiveTypeId = typeId ?? title ?? id;
     // Get the default display name (without custom name) for comparison
     const getDefaultDisplayName = React.useCallback(() => {
@@ -5871,6 +5941,7 @@ exports.getHandleLayoutY = getHandleLayoutY;
 exports.getNodeBorderClassNames = getNodeBorderClassNames;
 exports.isValidViewport = isValidViewport;
 exports.layoutNode = layoutNode;
+exports.mergeUIState = mergeUIState;
 exports.preformatValueForDisplay = preformatValueForDisplay;
 exports.prettyHandle = prettyHandle;
 exports.resolveOutputDisplay = resolveOutputDisplay;