@bian-womp/spark-workbench 0.2.72 → 0.2.74

package/lib/cjs/index.cjs

@@ -280,7 +280,7 @@ class InMemoryWorkbench extends AbstractWorkbench {
     }
     // Position and selection APIs for React Flow bridge
     setPositions(map, options) {
-        this.positions = { ...map };
+        this.positions = { ...this.positions, ...map };
         this.emit("graphUiChanged", {
             def: this.def,
             change: { type: "moveNodes" },
@@ -2715,8 +2715,8 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
     }, [wb, registry, overrides?.getDefaultNodeSize]);
     const updateEdgeType = React.useCallback((edgeId, typeId) => wb.updateEdgeType(edgeId, typeId), [wb]);
     const triggerExternal = React.useCallback((nodeId, event) => runner.triggerExternal(nodeId, event), [runner]);
-    // Helper to save runtime metadata to extData.runtime and workbench state
-    const saveRuntimeMetadata = React.useCallback(async () => {
+    // Helper to save runtime metadata and UI state to extData
+    const saveUiRuntimeMetadata = React.useCallback(async () => {
         try {
             const current = wb.getRuntimeState() ?? { nodes: {} };
             const metadata = { nodes: { ...current.nodes } };
@@ -2729,8 +2729,13 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
             }
             // Save cleaned metadata to workbench state
             wb.setRuntimeState(metadata);
-            // Save to extData.runtime via runner (no snapshotFull)
-            await runner.setExtData?.({ runtime: metadata });
+            // Get UI state
+            const uiState = wb.getUIState();
+            // Save both runtime and UI state to extData (merge to preserve both)
+            await runner.setExtData?.({
+                ...(uiState ? { ui: uiState } : {}),
+                runtime: metadata,
+            });
         }
         catch (err) {
             console.warn("[WorkbenchContext] Failed to save runtime metadata:", err);
@@ -3094,7 +3099,7 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
             }
             if (!runner.isRunning()) {
                 if (event.commit) {
-                    await saveRuntimeMetadata();
+                    await saveUiRuntimeMetadata();
                     const history = await runner.commit(reason).catch((err) => {
                         console.error("[WorkbenchContext] Error committing:", err);
                         return undefined;
@@ -3128,7 +3133,7 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
                     await runner.update(event.def, { dry: event.dry });
                 }
                 if (event.commit) {
-                    await saveRuntimeMetadata();
+                    await saveUiRuntimeMetadata();
                     const history = await runner
                         .commit(event.reason ?? reason)
                         .catch((err) => {
@@ -3149,7 +3154,7 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
             setSelectedNodeId(sel.nodes?.[0]);
             setSelectedEdgeId(sel.edges?.[0]);
             if (sel.commit) {
-                await saveRuntimeMetadata();
+                await saveUiRuntimeMetadata();
                 const history = await runner
                     .commit(sel.reason ?? "selection")
                     .catch((err) => {
@@ -3181,7 +3186,7 @@ function WorkbenchProvider({ wb, runner, registry, setRegistry, overrides, uiVer
                         reason = "viewport";
                     }
                 }
-                await saveRuntimeMetadata();
+                await saveUiRuntimeMetadata();
                 const history = await runner
                     .commit(event.reason ?? reason)
                     .catch((err) => {