From 7f1b18bebbb9d36fb886461d39f28c0df8f3dc13 Mon Sep 17 00:00:00 2001
From: moklick <info@moritzklack.com>
Date: Tue, 21 Jul 2020 17:08:09 +0200
Subject: [PATCH] fix(addEdge): handle multiple source/target id

---
 src/utils/graph.ts | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/utils/graph.ts b/src/utils/graph.ts
index 79ebfc04..80ff56b3 100644
--- a/src/utils/graph.ts
+++ b/src/utils/graph.ts
@@ -38,8 +38,9 @@ export const addEdge = (edgeParams: Edge | Connection, elements: Elements): Elem
 
   // make sure that there is node with the target and one with the source id
   [edgeParams.source, edgeParams.target].forEach((id) => {
-    if (!elements.find((e) => isNode(e) && e.id === id)) {
-      throw new Error(`Can't create edge. Node with id=${id} does not exist.`);
+    const nodeId = id.includes('__') ? id.split('__')[0] : id;
+    if (!elements.find((e) => isNode(e) && e.id === nodeId)) {
+      throw new Error(`Can't create edge. Node with id=${nodeId} does not exist.`);
     }
   });