Add zutand persist middleware

hanichris · May 14, 2023 · 9a7f9cd · 9a7f9cd
1 parent 1e4f640
commit 9a7f9cd
Showing 1 changed file with 19 additions and 10 deletions.
diff --git a/frontend/src/store/index.ts b/frontend/src/store/index.ts
@@ -1,4 +1,5 @@
 import { create } from 'zustand';
+import { persist, createJSONStorage } from 'zustand/middleware';
 import { IUser, IVideo, IChannel } from '../utils/types';
 
 type Store = {
@@ -12,15 +13,23 @@ type Store = {
   setRequestLoading: (isLoading: boolean) => void;
 };
 
-const useStore = create<Store>((set) => ({
-  authUser: null,
-  currentVideo: null,
-  currentChannel: null,
-  requestLoading: false,
-  setAuthUser: (user) => set((state) => ({ ...state, authUser: user })),
-  setCurrentVideo: (video) => set((state) => ({ ...state, currentVideo: video })),
-  setCurrentChannel: (channel) => set((state) => ({ ...state, currentChannel: channel })),
-  setRequestLoading: (isLoading) => set((state) => ({ ...state, requestLoading: isLoading })),
-}));
+const useStore = create<Store, any>(
+  persist(
+    (set) => ({
+      authUser: null,
+      currentVideo: null,
+      currentChannel: null,
+      requestLoading: false,
+      setAuthUser: (user) => set((state) => ({ ...state, authUser: user })),
+      setCurrentVideo: (video) => set((state) => ({ ...state, currentVideo: video })),
+      setCurrentChannel: (channel) => set((state) => ({ ...state, currentChannel: channel })),
+      setRequestLoading: (isLoading) => set((state) => ({ ...state, requestLoading: isLoading })),
+    }),
+    {
+      name: 'authUser',
+      storage: createJSONStorage(() => sessionStorage),
+    },
+  ),
+);
 
 export default useStore;