Function getConnectedNodeIds

yaml-flow-editor/src/utils/layout.ts:194–219

(nodeId: string, edges: FlowEdge[])

 * Uses edges as undirected: two nodes are in the same flow if there is a path of edges between them.
 */
export function getConnectedNodeIds(nodeId: string, edges: FlowEdge[]): Set<string> {
  const adj = new Map<string, Set<string>>();
  const add = (a: string, b: string) => {
    if (!adj.has(a)) adj.set(a, new Set());
    adj.get(a)!.add(b);
  };
  edges.forEach((e) => {
    const s = e.source;
    const t = e.target;
    add(s, t);
    add(t, s);
  });
  const out = new Set<string>();
  const stack: string[] = [nodeId];
  out.add(nodeId);
  while (stack.length > 0) {
    const cur = stack.pop()!;
    for (const next of adj.get(cur) ?? []) {
      if (!out.has(next)) {
        out.add(next);
        stack.push(next);
      }
    }
  }
  return out;
}

/**
 * Resolve overlaps for a specific node by pushing down overlapping nodes.