refactor(nodelookup): add treelevel

src/container/EdgeRenderer/index.tsx

@@ -251,8 +251,8 @@ const EdgeRenderer = (props: EdgeRendererProps) => {
       <g transform={`translate(${transform[0]},${transform[1]}) scale(${transform[2]})`}>
         {edges.map((edge: Edge) => {
           // @todo: getSourceTargetNodes is called many times during dragging/creating edges
-          const sourceNode = nodeLookup.current.get(edge.source);
-          const targetNode = nodeLookup.current.get(edge.target);
+          const sourceNode = nodeLookup.get(edge.source);
+          const targetNode = nodeLookup.get(edge.target);
 
           return (
             <Edge

src/container/NodeRenderer/index.tsx

@@ -236,11 +236,13 @@ const NodeRenderer = (props: NodeRendererProps) => {
     });
   }, []);
 
+  console.log(nodeLookup);
+
   return (
     <div className="react-flow__nodes" style={transformStyle}>
       <MemoizedNodes
         nodes={nodes}
-        nodeLookup={nodeLookup.current}
+        nodeLookup={nodeLookup}
         snapToGrid={snapToGrid}
         snapGrid={snapGrid}
         nodesDraggable={nodesDraggable}