fix(core): allow passing regular edges to getConnectedEdges

Signed-off-by: braks <78412429+bcakmakoglu@users.noreply.github.com>
bcakmakoglu · Apr 11, 2023 · a432422 · a432422
1 parent dbcbe78
commit a432422
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/core/src/utils/graph.ts b/packages/core/src/utils/graph.ts
@@ -364,13 +364,13 @@ export function getNodesInside(
   })
 }
 
-export function getConnectedEdges(nodes: (Node | GraphNode | { id: string })[], edges: GraphEdge[]) {
+export function getConnectedEdges<N extends Node | { id: string }, E extends Edge>(nodes: N[], edges: E[]) {
   const nodeIds = nodes.map((node) => node.id)
 
   return edges.filter((edge) => nodeIds.includes(edge.source) || nodeIds.includes(edge.target))
 }
 
-export function getConnectedNodes(nodes: (Node | GraphNode | { id: string })[], edges: GraphEdge[]) {
+export function getConnectedNodes<N extends Node | { id: string }, E extends Edge>(nodes: N[], edges: E[]) {
   const nodeIds = nodes.map((node) => node.id)
   const connectedNodeIds = edges.reduce((acc, edge) => {
     if (nodeIds.includes(edge.source)) {