@bian-womp/spark-workbench 0.3.68 → 0.3.70

package/lib/cjs/index.cjs

@@ -281,11 +281,22 @@ class InMemoryWorkbench extends AbstractWorkbench {
         return id;
     }
     removeNode(nodeId, options) {
+        const removedEdges = new Set();
         this._def.nodes = this._def.nodes.filter((n) => n.nodeId !== nodeId);
-        this._def.edges = this._def.edges.filter((e) => e.source.nodeId !== nodeId && e.target.nodeId !== nodeId);
+        this._def.edges = this._def.edges.filter((e) => {
+            if (e.source.nodeId === nodeId || e.target.nodeId === nodeId) {
+                removedEdges.add(e.id);
+                return false;
+            }
+            return true;
+        });
         delete this.positions[nodeId];
         delete this.sizes[nodeId];
         delete this.nodeNames[nodeId];
+        this.setSelectionInternal({
+            nodes: this.selection.nodes.filter((id) => id !== nodeId),
+            edges: this.selection.edges.filter((id) => !removedEdges.has(id)),
+        }, options);
         this.emit("graphChanged", {
             def: this._def,
             change: { type: "removeNode", nodeId },
@@ -564,6 +575,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 +600,9 @@ class InMemoryWorkbench extends AbstractWorkbench {
                     : undefined,
                 size,
                 inputs: inputsToCopy,
+                customData: customNodeData !== undefined
+                    ? lod.cloneDeep(customNodeData)
+                    : undefined,
                 originalNodeId: node.nodeId,
             };
         });
@@ -864,6 +879,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) {