refactor(nodes): only use node internals

example/src/Basic/index.tsx

@@ -147,6 +147,7 @@ const BasicFlow = () => {
   };
 
   const onNodesChange = useCallback((changes: NodeChange[]) => {
+    console.log('node change', changes);
     setNodes((ns) => applyNodeChanges(changes, ns));
   }, []);
 

example/src/FloatingEdges/FloatingEdge.tsx

@@ -3,13 +3,13 @@ import { EdgeProps, useStore, getBezierPath, ReactFlowState } from 'react-flow-r
 
 import { getEdgeParams } from './utils';
 
-const nodeSelector = (s: ReactFlowState) => s.nodes;
+const nodeSelector = (s: ReactFlowState) => s.nodeInternals;
 
 const FloatingEdge: FC<EdgeProps> = ({ id, source, target, style }) => {
-  const nodes = useStore(nodeSelector);
+  const nodeInternals = useStore(nodeSelector);
 
-  const sourceNode = useMemo(() => nodes.find((n) => n.id === source), [source, nodes]);
-  const targetNode = useMemo(() => nodes.find((n) => n.id === target), [target, nodes]);
+  const sourceNode = useMemo(() => nodeInternals.get(source), [source, nodeInternals]);
+  const targetNode = useMemo(() => nodeInternals.get(target), [target, nodeInternals]);
 
   if (!sourceNode || !targetNode) {
     return null;

example/src/Hidden/index.tsx

@@ -0,0 +1,62 @@
+import { useState, useCallback } from 'react';
+
+import { useEffect } from 'react';
+import ReactFlow, { addEdge, MiniMap, Controls, Connection, Edge, Node } from 'react-flow-renderer';
+
+const initialNodes: Node[] = [
+  { id: '1', type: 'input', isHidden: true, data: { label: 'Node 1' }, position: { x: 250, y: 5 } },
+  { id: '2', isHidden: true, data: { label: 'Node 2' }, position: { x: 100, y: 100 } },
+  { id: '3', isHidden: true, data: { label: 'Node 3' }, position: { x: 400, y: 100 } },
+  { id: '4', isHidden: true, data: { label: 'Node 4' }, position: { x: 400, y: 200 } },
+];
+
+const initialEdges: Edge[] = [
+  { id: 'e1-2', source: '1', target: '2' },
+  { id: 'e1-3', source: '1', target: '3' },
+  { id: 'e3-4', source: '3', target: '4' },
+];
+
+const setHidden = (isHidden: boolean) => (els: any[]) =>
+  els.map((e: any) => {
+    e.isHidden = isHidden;
+    return e;
+  });
+
+const HiddenFlow = () => {
+  const [nodes, setNodes] = useState<Node[]>(initialNodes);
+  const [edges, setEdges] = useState<Edge[]>(initialEdges);
+  const [isHidden, setIsHidden] = useState<boolean>(true);
+
+  const onConnect = useCallback((params: Edge | Connection) => {
+    setEdges((eds) => addEdge(params, eds));
+  }, []);
+
+  useEffect(() => {
+    setNodes(setHidden(isHidden));
+    setNodes(setHidden(isHidden));
+  }, [isHidden]);
+
+  return (
+    <ReactFlow nodes={nodes} edges={edges} onConnect={onConnect}>
+      <MiniMap />
+      <Controls />
+
+      <div style={{ position: 'absolute', left: 10, top: 10, zIndex: 4 }}>
+        <div>
+          <label htmlFor="ishidden">
+            isHidden
+            <input
+              id="ishidden"
+              type="checkbox"
+              checked={isHidden}
+              onChange={(event) => setIsHidden(event.target.checked)}
+              className="react-flow__ishidden"
+            />
+          </label>
+        </div>
+      </div>
+    </ReactFlow>
+  );
+};
+
+export default HiddenFlow;

example/src/Stress/index.tsx

@@ -51,6 +51,8 @@ const StressFlow = () => {
   };
 
   const onNodesChange = useCallback((changes: NodeChange[]) => {
+    console.log('node change', changes);
+
     setNodes((ns) => applyNodeChanges(changes, ns));
   }, []);
 

example/src/index.tsx

@@ -9,6 +9,7 @@ import CustomNode from './CustomNode';
 import FloatingEdges from './FloatingEdges';
 import Layouting from './Layouting';
 import NestedNodes from './NestedNodes';
+import Hidden from './Hidden';
 
 import './index.css';
 
@@ -41,6 +42,10 @@ const routes = [
     path: '/nested-nodes',
     component: NestedNodes,
   },
+  {
+    path: '/hidden',
+    component: Hidden,
+  },
 ];
 
 const Header = withRouter(({ history, location }) => {