refactor(core): change generic of useNode, useEdges

Signed-off-by: braks <[email protected]>

Author: bcakmakoglu

packages/core/src/composables/useEdge.ts

@@ -1,4 +1,4 @@
-import type { CustomEvent, ElementData } from '~/types'
+import type { GraphEdge } from '~/types'
 import { VueFlowError } from '~/utils/errors'
 
 /**
@@ -8,13 +8,13 @@ import { VueFlowError } from '~/utils/errors'
  *
  * Meaning if you do not provide an id, this composable has to be called in a child of your custom edge component, or it will throw
  */
-export default function useEdge<Data = ElementData, CustomEvents extends Record<string, CustomEvent> = any>(id?: string) {
+export default function useEdge<T extends GraphEdge = GraphEdge>(id?: string) {
   const edgeId = id ?? inject(EdgeId, '')
   const edgeEl = inject(EdgeRef, null)
 
   const { findEdge, emits } = useVueFlow()
 
-  const edge = findEdge<Data, CustomEvents>(edgeId)
+  const edge = findEdge<T>(edgeId)
 
   if (!edge) {
     emits.error(new VueFlowError(ErrorCode.EDGE_NOT_FOUND, edgeId))

packages/core/src/composables/useNode.ts

@@ -1,4 +1,4 @@
-import type { CustomEvent, ElementData } from '~/types'
+import type { GraphNode } from '~/types'
 
 /**
  * Access a node, it's parent (if one exists) and connected edges
@@ -7,13 +7,13 @@ import type { CustomEvent, ElementData } from '~/types'
  *
  * Meaning if you do not provide an id, this composable has to be called in a child of your custom node component, or it will throw
  */
-export default function useNode<Data = ElementData, CustomEvents extends Record<string, CustomEvent> = any>(id?: string) {
+export default function useNode<T extends GraphNode = GraphNode>(id?: string) {
   const nodeId = id ?? inject(NodeId, '')
   const nodeEl = inject(NodeRef, null)
 
   const { findNode, getEdges, emits } = useVueFlow()
 
-  const node = findNode<Data, CustomEvents>(nodeId)!
+  const node = findNode<T>(nodeId)!
 
   if (!node) {
     emits.error(new VueFlowError(ErrorCode.NODE_NOT_FOUND, nodeId))