refactor(useNodesInit): check for handleBounds, false for no nodes

2022-08-30 22:15:14 +02:00
parent 596ddd21bc
commit f40788ba4b
1 changed files with 8 additions and 1 deletions
@@ -1,7 +1,14 @@
 import { ReactFlowState } from '../types';
+import { internalsSymbol } from '../utils';
 import { useStore } from './useStore';

-const selector = (s: ReactFlowState) => Array.from(s.nodeInternals.values()).every((n) => n.width && n.height);
+const selector = (s: ReactFlowState) => {
+  if (s.nodeInternals.size === 0) {
+    return false;
+  }
+
+  return Array.from(s.nodeInternals.values()).every((n) => n[internalsSymbol]?.handleBounds !== undefined);
+};

 function useNodesInitialized(): boolean {
  const initialized = useStore(selector);