From e05580a5de6862f3b5a50945f67e0179f78aa470 Mon Sep 17 00:00:00 2001
From: Baptiste Arnaud <baptiste.arnaud95@gmail.com>
Date: Mon, 12 Feb 2024 09:10:04 +0100
Subject: [PATCH] :children_crossing: (editor) Enable groups clipboard accross
 tabs

---
 .../features/graph/hooks/useGroupsStore.ts    | 137 ++++++++++--------
 1 file changed, 73 insertions(+), 64 deletions(-)

diff --git a/apps/builder/src/features/graph/hooks/useGroupsStore.ts b/apps/builder/src/features/graph/hooks/useGroupsStore.ts
index d940ffbb8..85217ed35 100644
--- a/apps/builder/src/features/graph/hooks/useGroupsStore.ts
+++ b/apps/builder/src/features/graph/hooks/useGroupsStore.ts
@@ -1,6 +1,7 @@
 import { createWithEqualityFn } from 'zustand/traditional'
 import { Coordinates, CoordinatesMap } from '../types'
 import { Edge, Group, GroupV6 } from '@typebot.io/schemas'
+import { persist } from 'zustand/middleware'
 
 type Store = {
   focusedGroups: string[]
@@ -19,69 +20,77 @@ type Store = {
   setIsDraggingGraph: (isDragging: boolean) => void
 }
 
-export const useGroupsStore = createWithEqualityFn<Store>((set, get) => ({
-  focusedGroups: [],
-  groupsCoordinates: undefined,
-  groupsInClipboard: undefined,
-  isDraggingGraph: false,
-  getGroupsCoordinates: () => get().groupsCoordinates,
-  focusGroup: (groupId, isShiftKeyPressed) =>
-    set((state) => ({
-      focusedGroups: isShiftKeyPressed
-        ? state.focusedGroups.includes(groupId)
-          ? state.focusedGroups.filter((id) => id !== groupId)
-          : [...state.focusedGroups, groupId]
-        : [groupId],
-    })),
-  blurGroups: () => set({ focusedGroups: [] }),
-  moveFocusedGroups: (delta) =>
-    set(({ focusedGroups, groupsCoordinates }) => ({
-      groupsCoordinates: groupsCoordinates
-        ? {
-            ...groupsCoordinates,
-            ...focusedGroups.reduce(
-              (coords, groupId) => ({
-                ...coords,
-                [groupId]: {
-                  x: groupsCoordinates[groupId].x + delta.x,
-                  y: groupsCoordinates[groupId].y + delta.y,
-                },
-              }),
-              groupsCoordinates
-            ),
-          }
-        : undefined,
-    })),
-  setFocusedGroups: (groupIds) => set({ focusedGroups: groupIds }),
-  setGroupsCoordinates: (groups) =>
-    set({
-      groupsCoordinates: groups
-        ? groups.reduce(
-            (coords, group) => ({
-              ...coords,
-              [group.id]: {
-                x: group.graphCoordinates.x,
-                y: group.graphCoordinates.y,
-              },
-            }),
-            {}
-          )
-        : undefined,
-    }),
-  updateGroupCoordinates: (groupId, newCoord) => {
-    set((state) => ({
-      groupsCoordinates: {
-        ...state.groupsCoordinates,
-        [groupId]: newCoord,
-      },
-    }))
-  },
-  copyGroups: (groups, edges) =>
-    set({
-      groupsInClipboard: {
-        groups,
-        edges,
+export const useGroupsStore = createWithEqualityFn<Store>()(
+  persist(
+    (set, get) => ({
+      focusedGroups: [],
+      groupsCoordinates: undefined,
+      groupsInClipboard: undefined,
+      isDraggingGraph: false,
+      getGroupsCoordinates: () => get().groupsCoordinates,
+      focusGroup: (groupId, isShiftKeyPressed) =>
+        set((state) => ({
+          focusedGroups: isShiftKeyPressed
+            ? state.focusedGroups.includes(groupId)
+              ? state.focusedGroups.filter((id) => id !== groupId)
+              : [...state.focusedGroups, groupId]
+            : [groupId],
+        })),
+      blurGroups: () => set({ focusedGroups: [] }),
+      moveFocusedGroups: (delta) =>
+        set(({ focusedGroups, groupsCoordinates }) => ({
+          groupsCoordinates: groupsCoordinates
+            ? {
+                ...groupsCoordinates,
+                ...focusedGroups.reduce(
+                  (coords, groupId) => ({
+                    ...coords,
+                    [groupId]: {
+                      x: groupsCoordinates[groupId].x + delta.x,
+                      y: groupsCoordinates[groupId].y + delta.y,
+                    },
+                  }),
+                  groupsCoordinates
+                ),
+              }
+            : undefined,
+        })),
+      setFocusedGroups: (groupIds) => set({ focusedGroups: groupIds }),
+      setGroupsCoordinates: (groups) =>
+        set({
+          groupsCoordinates: groups
+            ? groups.reduce(
+                (coords, group) => ({
+                  ...coords,
+                  [group.id]: {
+                    x: group.graphCoordinates.x,
+                    y: group.graphCoordinates.y,
+                  },
+                }),
+                {}
+              )
+            : undefined,
+        }),
+      updateGroupCoordinates: (groupId, newCoord) => {
+        set((state) => ({
+          groupsCoordinates: {
+            ...state.groupsCoordinates,
+            [groupId]: newCoord,
+          },
+        }))
       },
+      copyGroups: (groups, edges) =>
+        set({
+          groupsInClipboard: {
+            groups,
+            edges,
+          },
+        }),
+      setIsDraggingGraph: (isDragging) => set({ isDraggingGraph: isDragging }),
     }),
-  setIsDraggingGraph: (isDragging) => set({ isDraggingGraph: isDragging }),
-}))
+    {
+      name: 'store',
+      partialize: (state) => ({ groupsInClipboard: state.groupsInClipboard }),
+    }
+  )
+)