42 lines
1001 B
TypeScript
42 lines
1001 B
TypeScript
import { internalsSymbol } from '@xyflow/system';
|
|
|
|
import { useStore } from './useStore';
|
|
import type { ReactFlowState } from '../types';
|
|
|
|
export type UseNodesInitializedOptions = {
|
|
includeHiddenNodes?: boolean;
|
|
};
|
|
|
|
const selector = (options: UseNodesInitializedOptions) => (s: ReactFlowState) => {
|
|
if (s.nodes.length === 0) {
|
|
return false;
|
|
}
|
|
|
|
for (const node of s.nodes) {
|
|
if (options.includeHiddenNodes || !node.hidden) {
|
|
if (node[internalsSymbol]?.handleBounds === undefined) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
return true;
|
|
};
|
|
|
|
const defaultOptions = {
|
|
includeHiddenNodes: false,
|
|
};
|
|
|
|
/**
|
|
* Hook which returns true when all nodes are initialized.
|
|
*
|
|
* @public
|
|
* @param options.includeHiddenNodes - defaults to false
|
|
* @returns boolean indicating whether all nodes are initialized
|
|
*/
|
|
export function useNodesInitialized(options: UseNodesInitializedOptions = defaultOptions): boolean {
|
|
const initialized = useStore(selector(options));
|
|
|
|
return initialized;
|
|
}
|