@bian-womp/spark-workbench 0.2.31 → 0.2.32

package/lib/cjs/index.cjs

@@ -477,7 +477,7 @@ class LocalGraphRunner extends AbstractGraphRunner {
         this.engine.on("error", (e) => this.emit("error", e));
         this.engine.on("invalidate", (e) => this.emit("invalidate", e));
         this.engine.on("stats", (e) => this.emit("stats", e));
-        this.engine.launch();
+        this.engine.launch(opts.invalidate);
         this.runningKind = opts.engine;
         this.emit("status", { running: true, engine: this.runningKind });
         for (const [nodeId, map] of Object.entries(this.stagedInputs)) {
@@ -710,7 +710,7 @@ class RemoteGraphRunner extends AbstractGraphRunner {
                 this.listenersBound = true;
             }
             this.engine = eng;
-            this.engine.launch();
+            this.engine.launch(opts.invalidate);
             this.runningKind = "push";
             this.emit("status", { running: true, engine: this.runningKind });
             for (const [nodeId, map] of Object.entries(this.stagedInputs)) {
@@ -754,9 +754,39 @@ class RemoteGraphRunner extends AbstractGraphRunner {
         }
     }
     async applySnapshotFull(payload) {
+        // Hydrate local cache first so UI can display values immediately
+        this.hydrateCacheFromSnapshot(payload);
+        // Then sync with backend
         const runner = await this.ensureRemoteRunner();
         await runner.applySnapshotFull(payload);
     }
+    /**
+     * 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.
+     */
+    hydrateCacheFromSnapshot(snapshot) {
+        // Hydrate inputs
+        for (const [nodeId, map] of Object.entries(snapshot.inputs || {})) {
+            for (const [handle, value] of Object.entries(map || {})) {
+                this.valueCache.set(`${nodeId}.${handle}`, {
+                    io: "input",
+                    value,
+                });
+                this.emit("value", { nodeId, handle, value, io: "input" });
+            }
+        }
+        // Hydrate outputs
+        for (const [nodeId, map] of Object.entries(snapshot.outputs || {})) {
+            for (const [handle, value] of Object.entries(map || {})) {
+                this.valueCache.set(`${nodeId}.${handle}`, {
+                    io: "output",
+                    value,
+                });
+                this.emit("value", { nodeId, handle, value, io: "output" });
+            }
+        }
+    }
     setEnvironment(env, opts) {
         const t = this.transport;
         if (!t)
@@ -3659,12 +3689,17 @@ function WorkbenchStudioCanvas({ setRegistry, autoScroll, onAutoScrollChange, ex
                             : undefined, children: [jsxRuntime.jsx("option", { value: "local", children: "Local" }), jsxRuntime.jsx("option", { value: "remote-http", children: "Remote (HTTP)" }), jsxRuntime.jsx("option", { value: "remote-ws", children: "Remote (WebSocket)" })] }), backendKind === "remote-http" && !!onHttpBaseUrlChange && (jsxRuntime.jsx("input", { className: "ml-2 border border-gray-300 rounded px-2 py-1 w-72", placeholder: "http://127.0.0.1:18080", value: httpBaseUrl, onChange: (e) => onHttpBaseUrlChange(e.target.value) })), backendKind === "remote-ws" && !!onWsUrlChange && (jsxRuntime.jsx("input", { className: "ml-2 border border-gray-300 rounded px-2 py-1 w-72", placeholder: "ws://127.0.0.1:18081", value: wsUrl, onChange: (e) => onWsUrlChange(e.target.value) })), jsxRuntime.jsxs("select", { className: "border border-gray-300 rounded px-2 py-1", value: runner.getRunningEngine() ?? engine ?? "", onChange: (e) => {
                             const kind = e.target.value || undefined;
                             onEngineChange?.(kind);
-                        }, children: [jsxRuntime.jsx("option", { value: "", children: "Select Engine\u2026" }), jsxRuntime.jsx("option", { value: "push", children: "Push" }), jsxRuntime.jsx("option", { value: "batched", children: "Batched" }), jsxRuntime.jsx("option", { value: "pull", children: "Pull" }), jsxRuntime.jsx("option", { value: "hybrid", children: "Hybrid" }), jsxRuntime.jsx("option", { value: "step", children: "Step" })] }), runner.getRunningEngine() === "step" && (jsxRuntime.jsx("button", { className: "ml-2 border border-gray-300 rounded p-1", onClick: () => runner.step(), disabled: !runner.isRunning(), title: "Step", children: jsxRuntime.jsx(react$1.PlayPauseIcon, { size: 24 }) })), runner.getRunningEngine() === "batched" && (jsxRuntime.jsx("button", { className: "ml-2 border border-gray-300 rounded p-1", onClick: () => runner.flush(), disabled: !runner.isRunning(), title: "Flush", children: jsxRuntime.jsx(react$1.LightningIcon, { size: 24 }) })), runner.isRunning() ? (jsxRuntime.jsxs("button", { className: "border rounded px-2 py-1.5 text-red-700 border-red-600 flex items-center gap-1", onClick: () => runner.dispose(), title: "Stop engine", children: [jsxRuntime.jsx(react$1.StopIcon, { size: 16, weight: "fill" }), jsxRuntime.jsx("span", { className: "font-medium ml-1", children: "Stop" })] })) : (jsxRuntime.jsxs("button", { className: "border rounded px-2 py-1.5 text-green-700 border-green-600 flex items-center gap-1 disabled:text-gray-400 disabled:border-gray-300", onClick: () => {
+                        }, children: [jsxRuntime.jsx("option", { value: "", children: "Select Engine\u2026" }), jsxRuntime.jsx("option", { value: "push", children: "Push" }), jsxRuntime.jsx("option", { value: "batched", children: "Batched" }), jsxRuntime.jsx("option", { value: "pull", children: "Pull" }), jsxRuntime.jsx("option", { value: "hybrid", children: "Hybrid" }), jsxRuntime.jsx("option", { value: "step", children: "Step" })] }), runner.getRunningEngine() === "step" && (jsxRuntime.jsx("button", { className: "ml-2 border border-gray-300 rounded p-1", onClick: () => runner.step(), disabled: !runner.isRunning(), title: "Step", children: jsxRuntime.jsx(react$1.PlayPauseIcon, { size: 24 }) })), runner.getRunningEngine() === "batched" && (jsxRuntime.jsx("button", { className: "ml-2 border border-gray-300 rounded p-1", onClick: () => runner.flush(), disabled: !runner.isRunning(), title: "Flush", children: jsxRuntime.jsx(react$1.LightningIcon, { size: 24 }) })), runner.isRunning() ? (jsxRuntime.jsxs("button", { className: "border rounded px-2 py-1.5 text-red-700 border-red-600 flex items-center gap-1", onClick: () => runner.dispose(), title: "Stop engine", children: [jsxRuntime.jsx(react$1.StopIcon, { size: 16, weight: "fill" }), jsxRuntime.jsx("span", { className: "font-medium ml-1", children: "Stop" })] })) : (jsxRuntime.jsxs("button", { className: "border rounded px-2 py-1.5 text-green-700 border-green-600 flex items-center gap-1 disabled:text-gray-400 disabled:border-gray-300", onClick: (evt) => {
                             const kind = engine;
                             if (!kind)
                                 return alert("Select an engine first.");
+                            if (evt.shiftKey && !confirm("Invalidate and re-run graph?"))
+                                return;
                             try {
-                                runner.launch(wb.export(), { engine: kind });
+                                runner.launch(wb.export(), {
+                                    engine: kind,
+                                    invalidate: evt.shiftKey,
+                                });
                             }
                             catch (err) {
                                 const message = err instanceof Error ? err.message : String(err);