@bian-womp/spark-workbench 0.3.66 → 0.3.68

package/lib/cjs/index.cjs

@@ -2542,7 +2542,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 +2555,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 +6017,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