refactor node functions

src/hooks/store/useGraph.tsx

@@ -1,8 +1,8 @@
 import create from "zustand";
 import { EdgeData, NodeData } from "reaflow/dist/types";
 import { Graph } from "src/components/Graph";
-import { findEdgeChildren } from "src/utils/findEdgeChildren";
-import { findNodeChildren } from "src/utils/findNodeChildren";
+import { getChildrenEdges } from "src/utils/getChildrenEdges";
+import { getOutgoers } from "src/utils/getOutgoers";
 
 export interface Graph {
  nodes: NodeData[];
@@ -34,12 +34,8 @@ const useGraph = create<Graph & GraphActions>((set) => ({
  }),
  expandNodes: (nodeId) =>
  set((state) => {
- const childrenNodes = findNodeChildren(nodeId, state.nodes, state.edges);
- const childrenEdges = findEdgeChildren(
- nodeId,
- childrenNodes,
- state.edges
- );
+ const childrenNodes = getOutgoers(nodeId, state.nodes, state.edges);
+ const childrenEdges = getChildrenEdges(childrenNodes, state.edges);
 
  const nodeIds = childrenNodes.map((node) => node.id);
  const edgeIds = childrenEdges.map((edge) => edge.id);
@@ -56,12 +52,8 @@ const useGraph = create<Graph & GraphActions>((set) => ({
  }),
  collapseNodes: (nodeId) =>
  set((state) => {
- const childrenNodes = findNodeChildren(nodeId, state.nodes, state.edges);
- const childrenEdges = findEdgeChildren(
- nodeId,
- childrenNodes,
- state.edges
- );
+ const childrenNodes = getOutgoers(nodeId, state.nodes, state.edges);
+ const childrenEdges = getChildrenEdges(childrenNodes, state.edges);
 
  const nodeIds = childrenNodes.map((node) => node.id);
  const edgeIds = childrenEdges.map((edge) => edge.id);

src/utils/getChildrenEdges.ts

@@ -0,0 +1,14 @@
+import { NodeData, EdgeData } from "reaflow/dist/types";
+
+export const getChildrenEdges = (
+ nodes: NodeData[],
+ edges: EdgeData[]
+): EdgeData[] => {
+ const nodeIds = nodes.map((node) => node.id);
+
+ return edges.filter(
+ (edge) =>
+ nodeIds.includes(edge.from as string) ||
+ nodeIds.includes(edge.to as string)
+ );
+};

src/utils/getOutgoers.ts

@@ -0,0 +1,20 @@
+import { NodeData, EdgeData } from "reaflow/dist/types";
+
+export const getOutgoers = (
+ nodeId: string,
+ nodes: NodeData[],
+ edges: EdgeData[]
+): NodeData[] => {
+ const allOutgoers: NodeData[] = [];
+
+ const runner = (nodeId: string) => {
+ const outgoerIds = edges.filter((e) => e.from === nodeId).map((e) => e.to);
+ const nodeList = nodes.filter((n) => outgoerIds.includes(n.id));
+ allOutgoers.push(...nodeList);
+ nodeList.forEach((node) => runner(node.id));
+ };
+
+ runner(nodeId);
+
+ return allOutgoers;
+};