🩹 map store methods before rehydration (#162)

prazdevs · Dec 28, 2022 · f71a54f · f71a54f
1 parent 682decf
commit f71a54f
Showing 1 changed file with 20 additions and 20 deletions.
diff --git a/packages/plugin/src/plugin.ts b/packages/plugin/src/plugin.ts
@@ -95,6 +95,26 @@ export function createPersistedState(
       }),
     )
 
+    store.$persist = () => {
+      persistences.forEach((persistence) => {
+        persistState(store.$state, persistence)
+      })
+    }
+
+    store.$hydrate = ({ runHooks = true } = {}) => {
+      persistences.forEach((persistence) => {
+        const { beforeRestore, afterRestore } = persistence
+
+        if (runHooks)
+          beforeRestore?.(context)
+
+        hydrateStore(store, persistence)
+
+        if (runHooks)
+          afterRestore?.(context)
+      })
+    }
+
     persistences.forEach((persistence) => {
       const { beforeRestore, afterRestore } = persistence
 
@@ -116,25 +136,5 @@ export function createPersistedState(
         },
       )
     })
-
-    store.$persist = () => {
-      persistences.forEach((persistence) => {
-        persistState(store.$state, persistence)
-      })
-    }
-
-    store.$hydrate = ({ runHooks = true } = {}) => {
-      persistences.forEach((persistence) => {
-        const { beforeRestore, afterRestore } = persistence
-
-        if (runHooks)
-          beforeRestore?.(context)
-
-        hydrateStore(store, persistence)
-
-        if (runHooks)
-          afterRestore?.(context)
-      })
-    }
   }
 }