deletion service

Author: jogibear9988

src/services/BpmnDeletionService.ts

@@ -0,0 +1,48 @@
+import { DeletionService, IDesignItem } from '@node-projects/web-component-designer';
+import { edgeTags } from './bpmnRegistry.js';
+
+function collectItems(roots: IDesignItem[]) {
+  const all: IDesignItem[] = [];
+  const stack = [...roots];
+  while (stack.length) {
+    const item = stack.pop()!;
+    all.push(item);
+    stack.push(...item.children());
+  }
+  return all;
+}
+
+function getBpmnId(item: IDesignItem) {
+  return item.element.getAttribute('bpmn-id');
+}
+
+export class BpmnDeletionService extends DeletionService {
+  override removeItems(items: IDesignItem[]) {
+    if (!items.length) {
+      return;
+    }
+
+    const selectedAndDescendants = collectItems(items);
+    const deletedIds = new Set(
+      selectedAndDescendants
+        .map(item => getBpmnId(item))
+        .filter((id): id is string => !!id)
+    );
+
+    const rootItems = Array.from(items[0].instanceServiceContainer.designerCanvas.rootDesignItem.children());
+    const allCanvasItems = collectItems(rootItems);
+
+    const connectedEdges = allCanvasItems.filter(item => {
+      if (!edgeTags.has(item.element.localName)) {
+        return false;
+      }
+
+      const sourceRef = item.element.getAttribute('source-ref');
+      const targetRef = item.element.getAttribute('target-ref');
+      return !!sourceRef && !!targetRef && (deletedIds.has(sourceRef) || deletedIds.has(targetRef));
+    });
+
+    const finalItems = Array.from(new Set([...selectedAndDescendants, ...connectedEdges]));
+    super.removeItems(finalItems);
+  }
+}

src/setupBpmnDesignerServiceContainer.ts

@@ -7,7 +7,6 @@ import {
   DefaultPropertyEditorTypesService,
   DefaultInstanceService,
   DefaultModelCommandService,
-  DeletionService,
   DesignItemDocumentPositionService,
   DesignItemService,
   DragDropService,
@@ -42,6 +41,7 @@ import {
   ZoomToolButtonProvider
 } from '@node-projects/web-component-designer';
 import { BpmnParserService } from './services/BpmnParserService.js';
+import { BpmnDeletionService } from './services/BpmnDeletionService.js';
 import { BpmnPlacementService } from './services/BpmnPlacementService.js';
 import { BpmnPropertyGroupsService } from './services/BpmnPropertyGroupsService.js';
 import { BpmnContextPadExtensionProvider } from './extensions/BpmnContextPadExtension.js';
@@ -71,7 +71,7 @@ export function createBpmnDesignerServiceContainer() {
   serviceContainer.register('propertyGroupsService', new BpmnPropertyGroupsService());
   serviceContainer.register('propertyService', new BaseCustomWebComponentPropertiesService(true));
   serviceContainer.register('designItemService', new DesignItemService());
-  serviceContainer.register('deletionService', new DeletionService());
+  serviceContainer.register('deletionService', new BpmnDeletionService());
   serviceContainer.register('elementsService', new PreDefinedElementsService('bpmn', bpmnElements));
 
   serviceContainer.register('undoService', (designerCanvas: IDesignerCanvas) => new UndoService(designerCanvas));