onNodesChange weirdness

sroussey · Apr 9, 2024 · cf04a27 · cf04a27
1 parent 7d9d689
commit cf04a27
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/packages/web/src/RunGraphFlow.tsx b/packages/web/src/RunGraphFlow.tsx
@@ -1,4 +1,4 @@
-import React, { Dispatch, SetStateAction, useEffect, useRef } from "react";
+import React, { Dispatch, SetStateAction, useCallback, useEffect, useRef } from "react";
 import {
   ReactFlow,
   Controls,
@@ -218,10 +218,18 @@ const defaultEdgeOptions = {
 export const RunGraphFlow: React.FC<{
   graph: TaskGraph;
 }> = ({ graph }) => {
-  const [nodes, setNodes, onNodesChange] = useNodesState<Node<TurboNodeData>>([]);
+  const [nodes, setNodes, onNodesChangeTheirs] = useNodesState<Node<TurboNodeData>>([]);
   const [edges, setEdges, onEdgesChange] = useEdgesState<Edge>([]);
   const graphRef = useRef<TaskGraph | null>(null);
 
+  const onNodesChange = useCallback(
+    (changes: any) => {
+      console.log("Nodes changed", changes);
+      onNodesChangeTheirs(changes);
+    },
+    [onNodesChangeTheirs, nodes, edges]
+  );
+
   const initialized = useNodesInitialized() && !nodes.some((n) => !n.measured);
   const { fitView } = useReactFlow();