refactor(core): change generic of useNode, useEdges

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

Author: bcakmakoglu

packages/core/src/composables/useEdge.ts

@@ -1,6 +1,6 @@
 import { inject } from 'vue'
 import { useVueFlow } from './useVueFlow'
-import type { CustomEvent, ElementData } from '~/types'
+import type { GraphEdge } from '~/types'
 import { ErrorCode, VueFlowError } from '~/utils'
 import { EdgeId, EdgeRef } from '~/context'
 
@@ -11,13 +11,13 @@ import { EdgeId, EdgeRef } from '~/context'
  *
  * 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 function useEdge<Data = ElementData, CustomEvents extends Record<string, CustomEvent> = any>(id?: string) {
+export 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,6 +1,6 @@
 import { computed, inject } from 'vue'
 import { useVueFlow } from './useVueFlow'
-import type { CustomEvent, ElementData } from '~/types'
+import type { GraphNode } from '~/types'
 import { NodeId, NodeRef } from '~/context'
 import { ErrorCode, VueFlowError, getConnectedEdges } from '~/utils'
 
@@ -11,13 +11,13 @@ import { ErrorCode, VueFlowError, getConnectedEdges } from '~/utils'
  *
  * 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 function useNode<Data = ElementData, CustomEvents extends Record<string, CustomEvent> = any>(id?: string) {
+export function useNode<T extends GraphNode = GraphNode>(id?: string) {
   const nodeId = id ?? inject(NodeId, '')
   const nodeEl = inject(NodeRef, null)
 
   const { findNode, edges, emits } = useVueFlow()
 
-  const node = findNode<Data, CustomEvents>(nodeId)!
+  const node = findNode<T>(nodeId)!
 
   if (!node) {
     emits.error(new VueFlowError(ErrorCode.NODE_NOT_FOUND, nodeId))