@bian-womp/spark-workbench 0.3.64 → 0.3.66

package/lib/cjs/index.cjs

@@ -1372,10 +1372,15 @@ class LocalGraphRunner extends AbstractGraphRunner {
             Array.isArray(this.extData)) {
             this.extData = {};
         }
+        let hasCustomUpdate = false;
         for (const { path, value } of updates) {
             const pathSegments = sparkGraph.parseJsonPath(path);
-            sparkGraph.setValueAtPathWithCreation(this.extData, pathSegments, value);
+            const updated = sparkGraph.setValueAtPathWithCreation(this.extData, pathSegments, value);
+            if (updated && pathSegments.length > 0 && pathSegments[0] === "custom") {
+                hasCustomUpdate = true;
+            }
         }
+        return hasCustomUpdate;
     }
     async snapshotFull() {
         const def = undefined; // UI will supply def/positions on download for local
@@ -1980,7 +1985,7 @@ class RemoteGraphRunner extends AbstractGraphRunner {
     }
     async updateExtData(updates) {
         const client = await this.ensureClient();
-        await client.api.updateExtData(updates);
+        return await client.api.updateExtData(updates);
     }
     async commit(reason) {
         const client = await this.ensureClient();
@@ -4143,17 +4148,18 @@ function WorkbenchProvider({ wb, runner, overrides, uiVersion, children, }) {
     // Helper to save runtime metadata and UI state to extData
     const saveUiRuntimeMetadata = React.useCallback(async (workbench, graphRunner) => {
         try {
-            const current = workbench.getRuntimeState() ?? { nodes: {} };
-            const metadata = { nodes: { ...current.nodes } };
+            const currentRuntime = workbench.getRuntimeState() ?? { nodes: {} };
+            const runtimeMetaData = { nodes: { ...currentRuntime.nodes } };
+            const customMetaData = workbench.getCustomData();
             // Clean up metadata for nodes that no longer exist
             const nodeIds = new Set(workbench.def.nodes.map((n) => n.nodeId));
-            for (const nodeId of Object.keys(metadata.nodes)) {
+            for (const nodeId of Object.keys(runtimeMetaData.nodes)) {
                 if (!nodeIds.has(nodeId)) {
-                    delete metadata.nodes[nodeId];
+                    delete runtimeMetaData.nodes[nodeId];
                 }
             }
             // Save cleaned metadata to workbench state
-            workbench.setRuntimeState(metadata);
+            workbench.setRuntimeState(runtimeMetaData);
             const fullUiState = workbench.getUIState();
             const uiWithoutViewport = excludeViewportFromUIState(fullUiState);
             // Use updateExtData for efficient batched partial updates
@@ -4161,7 +4167,8 @@ function WorkbenchProvider({ wb, runner, overrides, uiVersion, children, }) {
             if (Object.keys(uiWithoutViewport || {}).length > 0) {
                 updates.push({ path: "ui", value: uiWithoutViewport });
             }
-            updates.push({ path: "runtime", value: metadata });
+            updates.push({ path: "runtime", value: runtimeMetaData });
+            updates.push({ path: "custom", value: customMetaData });
             await graphRunner.updateExtData(updates);
         }
         catch (err) {