From b3addf63d2ef029d22c29d8f0d5960ba325341b1 Mon Sep 17 00:00:00 2001
From: braks <78412429+bcakmakoglu@users.noreply.github.com>
Date: Tue, 11 Apr 2023 16:42:03 +0200
Subject: [PATCH] feat(core): allow passing node ids as string to
 `getConnectedEdges`

Signed-off-by: braks <78412429+bcakmakoglu@users.noreply.github.com>
---
 packages/core/src/utils/graph.ts | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/packages/core/src/utils/graph.ts b/packages/core/src/utils/graph.ts
index 5b509b13..64bf9215 100644
--- a/packages/core/src/utils/graph.ts
+++ b/packages/core/src/utils/graph.ts
@@ -364,25 +364,28 @@ export function getNodesInside(
   })
 }
 
-export function getConnectedEdges<N extends Node | { id: string }, E extends Edge>(nodes: N[], edges: E[]) {
-  const nodeIds = nodes.map((node) => node.id)
+export function getConnectedEdges<N extends Node | { id: string } | string, E extends Edge>(nodes: N[], edges: E[]) {
+  const nodeIds = nodes.map((node) => (isString(node) ? node : node.id))
 
   return edges.filter((edge) => nodeIds.includes(edge.source) || nodeIds.includes(edge.target))
 }
 
-export function getConnectedNodes<N extends Node | { id: string }, E extends Edge>(nodes: N[], edges: E[]) {
-  const nodeIds = nodes.map((node) => node.id)
+export function getConnectedNodes<N extends Node | { id: string } | string, E extends Edge>(nodes: N[], edges: E[]) {
+  const nodeIds = nodes.map((node) => (isString(node) ? node : node.id))
+
   const connectedNodeIds = edges.reduce((acc, edge) => {
     if (nodeIds.includes(edge.source)) {
       acc.add(edge.target)
     }
+
     if (nodeIds.includes(edge.target)) {
       acc.add(edge.source)
     }
+
     return acc
   }, new Set<string>())
 
-  return nodes.filter((node) => connectedNodeIds.has(node.id))
+  return nodes.filter((node) => connectedNodeIds.has(isString(node) ? node : node.id))
 }
 
 export function getTransformForBounds(