Provide Node Manager

Author: camielvs

src/providers/ComponentSpecProvider.tsx

@@ -1,6 +1,7 @@
 import { type ReactNode, useCallback, useMemo, useRef, useState } from "react";
 
 import { type UndoRedo, useUndoRedo } from "@/hooks/useUndoRedo";
+import { NodeManager } from "@/nodeManager";
 import { loadPipelineByName } from "@/services/pipelineService";
 import { USER_PIPELINES_LIST_NAME } from "@/utils/constants";
 import { prepareComponentRefForEditor } from "@/utils/prepareComponentRefForEditor";
@@ -44,6 +45,7 @@ interface ComponentSpecContextType {
   taskStatusMap: Map<string, string>;
   setTaskStatusMap: (taskStatusMap: Map<string, string>) => void;
   undoRedo: UndoRedo;
+  nodeManager: NodeManager;
 }
 
 const ComponentSpecContext = createRequiredContext<ComponentSpecContextType>(
@@ -59,6 +61,7 @@ export const ComponentSpecProvider = ({
   readOnly?: boolean;
   children: ReactNode;
 }) => {
+  const [nodeManager] = useState(() => new NodeManager());
   const [componentSpec, setComponentSpec] = useState<ComponentSpec>(
     spec ?? EMPTY_GRAPH_COMPONENT_SPEC,
   );
@@ -142,15 +145,25 @@ export const ComponentSpecProvider = ({
     [componentSpec, readOnly],
   );
 
-  const updateGraphSpec = useCallback((newGraphSpec: GraphSpec) => {
-    setComponentSpec((prevSpec) => ({
-      ...prevSpec,
-      implementation: {
-        ...prevSpec.implementation,
-        graph: newGraphSpec,
-      },
-    }));
-  }, []);
+  const updateGraphSpec = useCallback(
+    (newGraphSpec: GraphSpec) => {
+      setComponentSpec((prevSpec) => {
+        const newSpec = {
+          ...prevSpec,
+          implementation: {
+            ...prevSpec.implementation,
+            graph: newGraphSpec,
+          },
+        };
+
+        // Sync node manager with new spec
+        nodeManager.syncWithComponentSpec(newSpec);
+
+        return newSpec;
+      });
+    },
+    [nodeManager],
+  );
 
   const value = useMemo(
     () => ({
@@ -167,6 +180,7 @@ export const ComponentSpecProvider = ({
       updateGraphSpec,
       setTaskStatusMap,
       undoRedo,
+      nodeManager,
     }),
     [
       componentSpec,
@@ -182,6 +196,7 @@ export const ComponentSpecProvider = ({
       updateGraphSpec,
       setTaskStatusMap,
       undoRedo,
+      nodeManager,
     ],
   );