refactor(getElements): use internal node position #603

xyflow · Nov 9, 2020 · 2db5a7b · 2db5a7b
1 parent cf87acc
commit 2db5a7b
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/src/utils/graph.ts b/src/utils/graph.ts
@@ -210,21 +210,18 @@ export const getNodesInside = (
 };
 
 export const getConnectedEdges = (nodes: Node[], edges: Edge[]): Edge[] => {
-  const nodeIds = nodes.map((n) => n.id);
+  const nodeIds = nodes.map((node) => node.id);
 
-  return edges.filter((e) => {
-    const sourceId = e.source.split('__')[0];
-    const targetId = e.target.split('__')[0];
-
-    return nodeIds.includes(sourceId) || nodeIds.includes(targetId);
-  });
+  return edges.filter((edge) => nodeIds.includes(edge.source) || nodeIds.includes(edge.target));
 };
 
 const parseElements = (nodes: Node[], edges: Edge[]): Elements => {
   return [
     ...nodes.map((node) => {
       const n = { ...node };
 
+      n.position = n.__rf.position;
+
       delete n.__rf;
       return n;
     }),