30 lines
874 B
TypeScript
30 lines
874 B
TypeScript
import { getNodesInside } from '@xyflow/system';
|
|
import { shallow } from 'zustand/shallow';
|
|
|
|
import { useStore } from './useStore';
|
|
import type { Node, ReactFlowState } from '../types';
|
|
import { useCallback } from 'react';
|
|
|
|
const selector = (onlyRenderVisible: boolean) => (s: ReactFlowState) => {
|
|
return onlyRenderVisible
|
|
? getNodesInside<Node>(s.nodes, { x: 0, y: 0, width: s.width, height: s.height }, s.transform, true).map(
|
|
(node) => node.id
|
|
)
|
|
: Array.from(s.nodeLookup.keys());
|
|
};
|
|
|
|
/**
|
|
* Hook for getting the visible node ids from the store.
|
|
*
|
|
* @internal
|
|
* @param onlyRenderVisible
|
|
* @returns array with visible node ids
|
|
*/
|
|
function useVisibleNodeIds(onlyRenderVisible: boolean) {
|
|
const nodeIds = useStore(useCallback(selector(onlyRenderVisible), [onlyRenderVisible]), shallow);
|
|
|
|
return nodeIds;
|
|
}
|
|
|
|
export default useVisibleNodeIds;
|