From f40788ba4b520ac77efd82282246e1634c4b1ede Mon Sep 17 00:00:00 2001 From: moklick Date: Tue, 30 Aug 2022 22:15:14 +0200 Subject: [PATCH] refactor(useNodesInit): check for handleBounds, false for no nodes --- packages/core/src/hooks/useNodesInitialized.ts | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/packages/core/src/hooks/useNodesInitialized.ts b/packages/core/src/hooks/useNodesInitialized.ts index 3612253d..1f2b1f56 100644 --- a/packages/core/src/hooks/useNodesInitialized.ts +++ b/packages/core/src/hooks/useNodesInitialized.ts @@ -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);