@bian-womp/spark-workbench 0.3.17 → 0.3.19

package/lib/cjs/index.cjs

@@ -889,7 +889,7 @@ class InMemoryWorkbench extends AbstractWorkbench {
         return this.customData.nodes?.[nodeId];
     }
     /**
-     * Set custom data for a specific node.
+     * Set custom data for a specific node (supports partial updates).
      */
     setCustomNodeData(nodeId, data, options) {
         if (!this.customData.nodes) {
@@ -902,10 +902,23 @@ class InMemoryWorkbench extends AbstractWorkbench {
             }
         }
         else {
-            this.customData.nodes[nodeId] = data;
+            // If merge is true, merge with existing data; otherwise replace
+            if (options?.merge && this.customData.nodes[nodeId]) {
+                this.customData.nodes[nodeId] = {
+                    ...this.customData.nodes[nodeId],
+                    ...data,
+                };
+            }
+            else {
+                this.customData.nodes[nodeId] = data;
+            }
         }
         this.emit("graphUiChanged", {
-            change: { type: "customNodeData", nodeId, data },
+            change: {
+                type: "customNodeData",
+                nodeId,
+                data: this.customData.nodes?.[nodeId],
+            },
             ...options,
         });
     }
@@ -916,7 +929,7 @@ class InMemoryWorkbench extends AbstractWorkbench {
         return this.customData.edges?.[edgeId];
     }
     /**
-     * Set custom data for a specific edge.
+     * Set custom data for a specific edge (supports partial updates).
      */
     setCustomEdgeData(edgeId, data, options) {
         if (!this.customData.edges) {
@@ -929,10 +942,23 @@ class InMemoryWorkbench extends AbstractWorkbench {
             }
         }
         else {
-            this.customData.edges[edgeId] = data;
+            // If merge is true, merge with existing data; otherwise replace
+            if (options?.merge && this.customData.edges[edgeId]) {
+                this.customData.edges[edgeId] = {
+                    ...this.customData.edges[edgeId],
+                    ...data,
+                };
+            }
+            else {
+                this.customData.edges[edgeId] = data;
+            }
         }
         this.emit("graphUiChanged", {
-            change: { type: "customEdgeData", edgeId, data },
+            change: {
+                type: "customEdgeData",
+                edgeId,
+                data: this.customData.edges?.[edgeId],
+            },
             ...options,
         });
     }
@@ -943,17 +969,26 @@ class InMemoryWorkbench extends AbstractWorkbench {
         return this.customData.meta;
     }
     /**
-     * Set custom metadata.
+     * Set custom metadata (supports partial updates).
      */
     setCustomMetaData(meta, options) {
         if (meta === undefined) {
             delete this.customData.meta;
         }
         else {
-            this.customData.meta = meta;
+            // If merge is true, merge with existing metadata; otherwise replace
+            if (options?.merge && this.customData.meta) {
+                this.customData.meta = {
+                    ...this.customData.meta,
+                    ...meta,
+                };
+            }
+            else {
+                this.customData.meta = meta;
+            }
         }
         this.emit("graphUiChanged", {
-            change: { type: "customMetaData", meta },
+            change: { type: "customMetaData", meta: this.customData.meta },
             ...options,
         });
     }
@@ -4123,8 +4158,9 @@ function WorkbenchProvider({ wb, runner, overrides, uiVersion, children, }) {
                 const err = nodeError.err;
                 let errorSummary;
                 if (err && typeof err === "object") {
-                    const message = err.message || String(err);
-                    const code = err.code || err.statusCode;
+                    const errAny = err;
+                    const message = errAny.userMessage || errAny.message || String(err);
+                    const code = errAny.statusCode || errAny.code;
                     errorSummary = {
                         message: typeof message === "string" ? message : String(message),
                         code: typeof code === "number" ? code : undefined,
@@ -4140,7 +4176,7 @@ function WorkbenchProvider({ wb, runner, overrides, uiVersion, children, }) {
                     ...s,
                     [nodeId]: {
                         ...s[nodeId],
-                        lastError: nodeError.err,
+                        lastError: errorSummary,
                     },
                 }));
                 // Mark this runId as errored
@@ -4228,6 +4264,10 @@ function WorkbenchProvider({ wb, runner, overrides, uiVersion, children, }) {
                 // Start fallback animation window
                 setFallbackStarts((prev) => ({ ...prev, [id]: now }));
             }
+            else if (s.kind === "node-custom-data") {
+                const id = s.nodeId;
+                wb.setCustomNodeData(id, s.data, { commit: false, merge: true });
+            }
             else if (s.kind === "node-progress") {
                 const id = s.nodeId;
                 setNodeStatus((prev) => ({