31 lines
1.3 KiB
TypeScript
31 lines
1.3 KiB
TypeScript
import { useMemo, useRef } from 'react';
|
|
import shallow from 'zustand/shallow';
|
|
|
|
import { devWarn } from '../../utils';
|
|
import { CreateEdgeTypes } from '../EdgeRenderer/utils';
|
|
import { CreateNodeTypes } from '../NodeRenderer/utils';
|
|
import type { EdgeTypes, EdgeTypesWrapped, NodeTypes, NodeTypesWrapped } from '../../types';
|
|
|
|
export function useNodeOrEdgeTypes(nodeOrEdgeTypes: NodeTypes, createTypes: CreateNodeTypes): NodeTypesWrapped;
|
|
export function useNodeOrEdgeTypes(nodeOrEdgeTypes: EdgeTypes, createTypes: CreateEdgeTypes): EdgeTypesWrapped;
|
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
export function useNodeOrEdgeTypes(nodeOrEdgeTypes: any, createTypes: any): any {
|
|
const typesKeysRef = useRef<string[] | null>(null);
|
|
|
|
const typesParsed = useMemo(() => {
|
|
if (process.env.NODE_ENV === 'development') {
|
|
const typeKeys = Object.keys(nodeOrEdgeTypes);
|
|
if (shallow(typesKeysRef.current, typeKeys)) {
|
|
devWarn(
|
|
"It looks like you have created a new nodeTypes or edgeTypes object. If this wasn't on purpose please define the nodeTypes/edgeTypes outside of the component or memoize them. Help: https://reactflow.dev/error#200"
|
|
);
|
|
}
|
|
|
|
typesKeysRef.current = typeKeys;
|
|
}
|
|
return createTypes(nodeOrEdgeTypes);
|
|
}, [nodeOrEdgeTypes]);
|
|
|
|
return typesParsed;
|
|
}
|