feat: enrich links between edges and shapes in the internal model

src/component/parser/json/converter/ProcessConverter.ts

@@ -69,8 +69,11 @@ export default class ProcessConverter {
 
   deserialize(processes: string | TProcess | (string | TProcess)[]): void {
     ensureIsArray(processes).forEach(process => this.parseProcess(process));
+
     // Need to call this after all processes have been parsed, because to link a call activity to the elements of the called process, we have to parse all processes before.
     ensureIsArray(processes).forEach(process => this.assignParentOfProcessElementsCalledByCallActivity(process.id));
+
+    this.assignIncomingAndOutgoingIdsFromFlows();
   }
 
   private assignParentOfProcessElementsCalledByCallActivity(processId: string): void {
@@ -87,6 +90,26 @@ export default class ProcessConverter {
     }
   }
 
+  private assignIncomingAndOutgoingIdsFromFlows(): void {
+    const fillShapeBpmnElementAttribute = (
+      shapeBpmnElementId: string,
+      shapeBpmnElementAttributeName: keyof Pick<ShapeBpmnElement, 'outgoingIds' | 'incomingIds'>,
+      valueToAdd: string,
+    ): void => {
+      const shapeBpmnElement =
+        this.convertedElements.findFlowNode(shapeBpmnElementId) ?? this.convertedElements.findLane(shapeBpmnElementId) ?? this.convertedElements.findPoolById(shapeBpmnElementId);
+      if (shapeBpmnElement && !shapeBpmnElement[shapeBpmnElementAttributeName].includes(valueToAdd)) {
+        shapeBpmnElement[shapeBpmnElementAttributeName].push(valueToAdd);
+      }
+    };
+
+    const flows = [...this.convertedElements.getMessageFlows(), ...this.convertedElements.getSequenceFlows(), ...this.convertedElements.getAssociationFlows()];
+    flows.forEach(flow => {
+      fillShapeBpmnElementAttribute(flow.sourceRefId, 'outgoingIds', flow.id);
+      fillShapeBpmnElementAttribute(flow.targetRefId, 'incomingIds', flow.id);
+    });
+  }
+
   private parseProcess(process: TProcess): void {
     const processRef = process.id;
     const pool = this.convertedElements.findPoolByProcessRef(processRef);
@@ -106,7 +129,7 @@ export default class ProcessConverter {
     ShapeUtil.flowNodeKinds()
       .filter(kind => kind != ShapeBpmnElementKind.EVENT_BOUNDARY)
       .forEach(kind => this.buildFlowNodeBpmnElements(process[kind], kind, parentId, process.id));
-    // process boundary events afterwards as we need its parent activity to be available when building it
+    // process boundary events afterward as we need its parent activity to be available when building it
     this.buildFlowNodeBpmnElements(process.boundaryEvent, ShapeBpmnElementKind.EVENT_BOUNDARY, parentId, process.id);
 
     // containers

src/component/parser/json/converter/utils.ts

@@ -45,6 +45,9 @@ export class ConvertedElements {
   findMessageFlow(id: string): MessageFlow {
     return this.messageFlows.get(id);
   }
+  getMessageFlows(): MessageFlow[] {
+    return Array.from(this.messageFlows.values());
+  }
   registerMessageFlow(messageFlow: MessageFlow): void {
     this.messageFlows.set(messageFlow.id, messageFlow);
   }
@@ -69,6 +72,9 @@ export class ConvertedElements {
   findSequenceFlow(id: string): SequenceFlow {
     return this.sequenceFlows.get(id);
   }
+  getSequenceFlows(): SequenceFlow[] {
+    return Array.from(this.sequenceFlows.values());
+  }
   registerSequenceFlow(sequenceFlow: SequenceFlow): void {
     this.sequenceFlows.set(sequenceFlow.id, sequenceFlow);
   }
@@ -77,6 +83,9 @@ export class ConvertedElements {
   findAssociationFlow(id: string): AssociationFlow {
     return this.associationFlows.get(id);
   }
+  getAssociationFlows(): AssociationFlow[] {
+    return Array.from(this.associationFlows.values());
+  }
   registerAssociationFlow(associationFlow: AssociationFlow): void {
     this.associationFlows.set(associationFlow.id, associationFlow);
   }