@bian-womp/spark-workbench 0.3.69 → 0.3.70

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/lib/cjs/index.cjs CHANGED
@@ -281,11 +281,22 @@ class InMemoryWorkbench extends AbstractWorkbench {
281
281
  return id;
282
282
  }
283
283
  removeNode(nodeId, options) {
284
+ const removedEdges = new Set();
284
285
  this._def.nodes = this._def.nodes.filter((n) => n.nodeId !== nodeId);
285
- this._def.edges = this._def.edges.filter((e) => e.source.nodeId !== nodeId && e.target.nodeId !== nodeId);
286
+ this._def.edges = this._def.edges.filter((e) => {
287
+ if (e.source.nodeId === nodeId || e.target.nodeId === nodeId) {
288
+ removedEdges.add(e.id);
289
+ return false;
290
+ }
291
+ return true;
292
+ });
286
293
  delete this.positions[nodeId];
287
294
  delete this.sizes[nodeId];
288
295
  delete this.nodeNames[nodeId];
296
+ this.setSelectionInternal({
297
+ nodes: this.selection.nodes.filter((id) => id !== nodeId),
298
+ edges: this.selection.edges.filter((id) => !removedEdges.has(id)),
299
+ }, options);
289
300
  this.emit("graphChanged", {
290
301
  def: this._def,
291
302
  change: { type: "removeNode", nodeId },