@bian-womp/spark-workbench 0.2.80 → 0.2.81

package/lib/cjs/index.cjs

@@ -2772,6 +2772,50 @@ async function upload(parsed, wb, runner) {
     }
 }
 
+/**
+ * Merge UI state from source into target, remapping node IDs using nodeIdMap.
+ * Preserves target state and adds/updates source state with remapped IDs.
+ */
+function mergeUIState(targetUI, sourceUI, nodeIdMap) {
+    const result = {
+        ...targetUI,
+    };
+    if (!sourceUI)
+        return result;
+    // Merge positions (already handled by mergeSnapshotData, but included for completeness)
+    if (sourceUI.positions) {
+        result.positions = {
+            ...(targetUI?.positions || {}),
+            ...Object.fromEntries(Object.entries(sourceUI.positions).map(([oldId, pos]) => [
+                nodeIdMap[oldId] || oldId,
+                pos,
+            ])),
+        };
+    }
+    // Merge selection: remap node IDs and edge IDs
+    if (sourceUI.selection) {
+        const remappedNodes = (sourceUI.selection.nodes || [])
+            .map((id) => nodeIdMap[id] || id)
+            .filter((id) => id); // Filter out invalid mappings
+        const remappedEdges = sourceUI.selection.edges || []; // Edge IDs don't need remapping typically
+        result.selection = {
+            nodes: [...(targetUI?.selection?.nodes || []), ...remappedNodes],
+            edges: [...(targetUI?.selection?.edges || []), ...remappedEdges],
+        };
+    }
+    // Merge nodeNames: remap node IDs
+    if (sourceUI.nodeNames) {
+        result.nodeNames = {
+            ...(targetUI?.nodeNames || {}),
+            ...Object.fromEntries(Object.entries(sourceUI.nodeNames).map(([oldId, name]) => [
+                nodeIdMap[oldId] || oldId,
+                name,
+            ])),
+        };
+    }
+    return result;
+}
+
 const WorkbenchContext = React.createContext(null);
 function useWorkbenchContext() {
     const ctx = React.useContext(WorkbenchContext);
@@ -4293,7 +4337,7 @@ function DefaultNodeHeader({ id, typeId, title, validation, right, showId, onInv
     const [editValue, setEditValue] = React.useState("");
     const inputRef = React.useRef(null);
     // Use getNodeDisplayName if typeId is provided, otherwise use title prop
-    const displayName = typeId ? ctx.getNodeDisplayName(id) : (title ?? id);
+    const displayName = typeId ? ctx.getNodeDisplayName(id) : title ?? id;
     const effectiveTypeId = typeId ?? title ?? id;
     // Get the default display name (without custom name) for comparison
     const getDefaultDisplayName = React.useCallback(() => {
@@ -5871,6 +5915,7 @@ exports.getHandleLayoutY = getHandleLayoutY;
 exports.getNodeBorderClassNames = getNodeBorderClassNames;
 exports.isValidViewport = isValidViewport;
 exports.layoutNode = layoutNode;
+exports.mergeUIState = mergeUIState;
 exports.preformatValueForDisplay = preformatValueForDisplay;
 exports.prettyHandle = prettyHandle;
 exports.resolveOutputDisplay = resolveOutputDisplay;