chore(react): use connection state generics internally

packages/react/src/types/store.ts

@@ -78,7 +78,7 @@ export type ReactFlowStore<NodeType extends Node = Node, EdgeType extends Edge =
   userSelectionActive: boolean;
   userSelectionRect: SelectionRect | null;
 
-  connection: ConnectionState;
+  connection: ConnectionState<InternalNode<NodeType>>;
   connectionMode: ConnectionMode;
   connectionClickStartHandle: (Pick<Handle, 'nodeId' | 'id'> & Required<Pick<Handle, 'type'>>) | null;
 
@@ -163,7 +163,7 @@ export type ReactFlowActions<NodeType extends Node, EdgeType extends Edge> = {
   setTranslateExtent: (translateExtent: CoordinateExtent) => void;
   setNodeExtent: (nodeExtent: CoordinateExtent) => void;
   cancelConnection: () => void;
-  updateConnection: UpdateConnection;
+  updateConnection: UpdateConnection<InternalNode<NodeType>>;
   reset: () => void;
   triggerNodeChanges: (changes: NodeChange<NodeType>[]) => void;
   triggerEdgeChanges: (changes: EdgeChange<EdgeType>[]) => void;

packages/system/src/types/general.ts

@@ -173,7 +173,9 @@ export type ConnectionState<NodeType extends InternalNodeBase = InternalNodeBase
   | ConnectionInProgress<NodeType>
   | NoConnection;
 
-export type UpdateConnection = (params: ConnectionState) => void;
+export type UpdateConnection<NodeType extends InternalNodeBase = InternalNodeBase> = (
+  params: ConnectionState<NodeType>
+) => void;
 
 export type ColorModeClass = 'light' | 'dark';
 export type ColorMode = ColorModeClass | 'system';