fix: mirror deletions between control panel and flow chart (#15)

flojoy-ai · Dec 19, 2023 · a5c3fe0 · a5c3fe0
1 parent e74c60e
commit a5c3fe0
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/renderer/src/stores/flowchart.ts b/src/renderer/src/stores/flowchart.ts
@@ -108,6 +108,13 @@ export const useFlowchartStore = create<FlowchartState>()(
           set({
             nodes: applyNodeChanges(changes, get().nodes)
           });
+          // Delete corresponding nodes in control canvas
+          set({
+            controls: applyNodeChanges(
+              changes.filter((c) => c.type === 'remove'),
+              get().controls
+            )
+          });
         },
         onEdgesChange: (changes: EdgeChange[]) => {
           set({
@@ -118,6 +125,13 @@ export const useFlowchartStore = create<FlowchartState>()(
           set({
             controls: applyNodeChanges(changes, get().controls)
           });
+          // Delete corresponding nodes in flow chart
+          set({
+            nodes: applyNodeChanges(
+              changes.filter((c) => c.type === 'remove'),
+              get().nodes
+            )
+          });
         },
 
         onConnect: (connection: Connection) => {