@bian-womp/spark-workbench 0.3.63 → 0.3.65

package/lib/cjs/index.cjs

@@ -416,8 +416,9 @@ class InMemoryWorkbench extends AbstractWorkbench {
     setViewport(viewport) {
         if (lod.isEqual(this.viewport, viewport))
             return;
+        const init = this.viewport === null;
         this.viewport = { ...viewport };
-        this.emit("graphUiChanged", { change: { type: "viewport" } });
+        this.emit("graphUiChanged", { change: { type: "viewport" }, init });
     }
     getViewport() {
         return this.viewport ? { ...this.viewport } : null;
@@ -1371,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
@@ -1687,7 +1693,7 @@ class RemoteGraphRunner extends AbstractGraphRunner {
         this.clientPromise = promise = (async () => {
             // Build client config from backend config
             const clientConfig = this.buildClientConfig(backend);
-            // Wrap custom event handler to intercept flow-viewport events and emit viewport event
+            // Wrap custom event handler to intercept viewport events and emit viewport event
             const wrappedOnCustomEvent = (event) => {
                 const msg = event?.message;
                 if (msg &&
@@ -1979,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();
@@ -4142,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
@@ -4160,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) {