From c0fc95dc5e3eb3875fe800b080a6ff3de7f9be28 Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Tue, 18 Oct 2022 09:24:18 +0200
Subject: [PATCH] refactor(core): add generics to Elements/FlowElements types

---
 packages/core/src/types/flow.ts | 31 +++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/packages/core/src/types/flow.ts b/packages/core/src/types/flow.ts
index 7b91e650..b7938b11 100644
--- a/packages/core/src/types/flow.ts
+++ b/packages/core/src/types/flow.ts
@@ -5,16 +5,39 @@ import type { CoordinateExtent, GraphNode, Node } from './node'
 import type { ConnectionLineOptions, ConnectionLineType, ConnectionMode, Connector } from './connection'
 import type { PanOnScrollMode } from './zoom'
 import type { EdgeTypesObject, NodeTypesObject } from './components'
+import type { CustomEvent } from './hooks'
 
 export type ElementData = any
 
 /** A flow element (after parsing into state)  */
-export type FlowElement<Data = ElementData> = GraphNode<Data> | GraphEdge<Data>
-export type FlowElements<Data = ElementData> = FlowElement<Data>[]
+export type FlowElement<
+  NodeData = ElementData,
+  EdgeData = ElementData,
+  NodeEvents extends Record<string, CustomEvent> = any,
+  EdgeEvents extends Record<string, CustomEvent> = any,
+> = GraphNode<NodeData, NodeEvents> | GraphEdge<EdgeData, EdgeEvents>
+
+export type FlowElements<
+  NodeData = ElementData,
+  EdgeData = ElementData,
+  NodeEvents extends Record<string, CustomEvent> = any,
+  EdgeEvents extends Record<string, CustomEvent> = any,
+> = FlowElement<NodeData, EdgeData, NodeEvents, EdgeEvents>[]
 
 /** Initial elements (before parsing into state) */
-export type Element<Data = ElementData> = Node<Data> | Edge<Data>
-export type Elements<Data = ElementData> = Element<Data>[]
+export type Element<
+  NodeData = ElementData,
+  EdgeData = ElementData,
+  NodeEvents extends Record<string, CustomEvent> = any,
+  EdgeEvents extends Record<string, CustomEvent> = any,
+> = Node<NodeData, NodeEvents> | Edge<EdgeData, EdgeEvents>
+
+export type Elements<
+  NodeData = ElementData,
+  EdgeData = ElementData,
+  NodeEvents extends Record<string, CustomEvent> = any,
+  EdgeEvents extends Record<string, CustomEvent> = any,
+> = Element<NodeData, EdgeData, NodeEvents, EdgeEvents>[]
 
 export type CustomThemeVars = Record<string, string | number>
 export type CSSVars =