refactor(types): use custom node type as generic

2023-12-07 12:24:03 +01:00
parent 0f657a5193
commit 10f31e79e6
12 changed files with 160 additions and 116 deletions
@@ -5,8 +5,8 @@ import type { Node, ReactFlowState } from '../types';

 const nodesSelector = (state: ReactFlowState) => state.nodes;

-function useNodes<NodeData>(): Node<NodeData>[] {
-  const nodes = useStore(nodesSelector, shallow);
+function useNodes<NodeType extends Node = Node>(): NodeType[] {
+  const nodes = useStore(nodesSelector, shallow) as NodeType[];

  return nodes;
 }