@bian-womp/spark-workbench 0.3.67 → 0.3.69

package/lib/cjs/index.cjs

@@ -564,6 +564,7 @@ class InMemoryWorkbench extends AbstractWorkbench {
         const copiedNodes = nodesToCopy.map((node) => {
             const pos = positions[node.nodeId];
             const size = sizes[node.nodeId];
+            const customNodeData = this.getCustomNodeData(node.nodeId);
             // Get all inbound edges for this node
             const inboundEdges = this.def.edges.filter((e) => e.target.nodeId === node.nodeId);
             // Build set of handles that have inbound edges
@@ -588,6 +589,9 @@ class InMemoryWorkbench extends AbstractWorkbench {
                     : undefined,
                 size,
                 inputs: inputsToCopy,
+                customData: customNodeData !== undefined
+                    ? lod.cloneDeep(customNodeData)
+                    : undefined,
                 originalNodeId: node.nodeId,
             };
         });
@@ -864,6 +868,12 @@ class InMemoryWorkbench extends AbstractWorkbench {
                 dry: true,
             });
             nodeIdMap.set(nodeData.originalNodeId, newNodeId);
+            if (nodeData.customData !== undefined) {
+                this.setCustomNodeData(newNodeId, lod.cloneDeep(nodeData.customData), {
+                    commit: false,
+                    reason: options?.reason,
+                });
+            }
         }
         // Add edges
         for (const edgeData of data.edges) {
@@ -2542,7 +2552,7 @@ function layoutNode(args, overrides) {
     };
 }
 
-function useWorkbenchBridge(wb) {
+function useWorkbenchBridge(wb, handlesMap) {
     const onConnect = React.useCallback((params) => {
         if (!params.source || !params.target)
             return;
@@ -2555,11 +2565,27 @@ function useWorkbenchBridge(wb) {
             e.target.handle === params.targetHandle);
         if (exists)
             return;
+        // Check if target input handle is a non-array type (typeId doesn't end with [])
+        // If so, remove any existing connections to that handle before connecting
+        const targetHandles = handlesMap?.[params.target];
+        const targetTypeId = targetHandles
+            ? sparkGraph.getInputTypeId(targetHandles.inputs, params.targetHandle)
+            : undefined;
+        const isArrayInput = targetTypeId?.endsWith("[]") ?? false;
+        if (!isArrayInput) {
+            // Find existing edges to this input handle
+            const existingEdges = wb.def.edges.filter((e) => e.target.nodeId === params.target &&
+                e.target.handle === params.targetHandle);
+            // Remove existing edges (without committing yet)
+            for (const edge of existingEdges) {
+                wb.disconnect(edge.id, { commit: false });
+            }
+        }
         wb.connect({
             source: { nodeId: params.source, handle: params.sourceHandle },
             target: { nodeId: params.target, handle: params.targetHandle },
         }, { commit: true });
-    }, [wb]);
+    }, [wb, handlesMap]);
     const onNodesChange = React.useCallback((changes) => {
         const positions = {};
         let commit = false;
@@ -6001,7 +6027,7 @@ const WorkbenchCanvasComponent = React.forwardRef((props, ref) => {
             }
         },
     }), []);
-    const { onConnect, onNodesChange, onEdgesChange, onEdgesDelete, onNodesDelete, } = useWorkbenchBridge(wb);
+    const { onConnect, onNodesChange, onEdgesChange, onEdgesDelete, onNodesDelete, } = useWorkbenchBridge(wb, handlesMap);
     const ui = React.useMemo(() => wb.getUI(), [wb, uiVersion]);
     const { nodeTypes, resolveNodeType } = React.useMemo(() => {
         // Build nodeTypes map using UI extension registry