feat(use-nodes-initialized): add options, exclude hidden nodes

2023-03-24 16:42:14 +01:00
parent 0cb000a84d
commit 586b7ca629
5 changed files with 108 additions and 8 deletions
@@ -2,16 +2,27 @@ import { internalsSymbol } from '../utils';
 import { useStore } from './useStore';
 import type { ReactFlowState } from '../types';

-const selector = (s: ReactFlowState) => {
+export type UseNodesInitializedOptions = {
+  includeHiddenNodes?: boolean;
+};
+
+const selector = (options: UseNodesInitializedOptions) => (s: ReactFlowState) => {
  if (s.nodeInternals.size === 0) {
    return false;
  }

-  return s.getNodes().every((n) => n[internalsSymbol]?.handleBounds !== undefined);
+  return s
+    .getNodes()
+    .filter((n) => (options.includeHiddenNodes ? true : !n.hidden))
+    .every((n) => n[internalsSymbol]?.handleBounds !== undefined);
 };

-function useNodesInitialized(): boolean {
-  const initialized = useStore(selector);
+const defaultOptions = {
+  includeHiddenNodes: false,
+};
+
+function useNodesInitialized(options: UseNodesInitializedOptions = defaultOptions): boolean {
+  const initialized = useStore(selector(options));

  return initialized;
 }