fix(nodes): find all edges when node has multiple handles

xyflow · Oct 4, 2019 · 35fa292 · 35fa292
1 parent 0bd2102
commit 35fa292
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/graph-utils.js b/src/graph-utils.js
@@ -129,7 +129,15 @@ export const getNodesInside = (nodes, bbox, transform = [0, 0, 1], partially = f
 
 export const getConnectedEdges = (nodes, edges) => {
   const nodeIds = nodes.map(n => n.id);
-  return edges.filter(e => nodeIds.includes(e.source) || nodeIds.includes(e.target));
+  return edges.filter(e => {
+    const hasSourceHandleId = e.source.includes('__');
+    const hasTargetHandleId = e.target.includes('__');
+
+    const sourceId = hasSourceHandleId ? e.source.split('__')[0] : e.source;
+    const targetId = hasTargetHandleId ? e.target.split('__')[0] : e.target;
+
+    return nodeIds.includes(sourceId) || nodeIds.includes(targetId);
+  });
 };
 
 export const fitView = ({ padding = 0 } = {}) => {