fix: only use initial state if there's no stored state (#92)

Author: Gomorongi

.changeset/wide-words-smash.md

@@ -0,0 +1,5 @@
+---
+"@kohaku-eth/privacy-pools": patch
+---
+
+Use snapshot state only if there's no previous stored state

packages/privacy-pools/src/state/state-manager.ts

@@ -55,7 +55,6 @@ import { ragequitThunk } from "./thunks/ragequitThunk";
 import { SyncAspThunkParams } from "./thunks/syncAspThunk";
 import { syncThunk } from "./thunks/syncThunk";
 import { withdrawThunk } from "./thunks/withdrawThunk";
-import { deserialize } from "./utils/serialize.utils";
 
 export interface StoreFactoryParams
   extends BaseSelectorParams, SyncAspThunkParams {
@@ -179,27 +178,14 @@ const getStoreStorageKey = (
 
 const storeByChainAndEntrypoint = ({
   storageToSyncTo,
-  initialState,
+  initialState: initialStateByChainAndEntrypoint = {},
   ...params
 }: Omit<StoreFactoryParams, "dataService">) => {
-  const initialMapState = Object.entries(initialState || {}).map(
-    ([key, state]) =>
-      [
-        key,
-        initializeSelectors({
-          ...params,
-          store: storeFactory({
-            entrypointInfo: deserialize(state!.entrypointInfo),
-            initialState: state,
-          }),
-        }),
-      ] as const,
-  );
 
   const chainStoreMap = new Map<
     string,
     ReturnType<typeof initializeSelectors<ReturnType<typeof storeFactory>>>
-  >(initialMapState);
+  >();
 
   return {
     getChainStore: (getChainStoreParams: GetChainStoreParams) => {
@@ -210,14 +196,12 @@ const storeByChainAndEntrypoint = ({
       const computedChainKey = getStoreKey(getChainStoreParams);
       let storeWithSelectors = chainStoreMap.get(computedChainKey);
 
-      const storedState = storageToSyncTo?.get(
-        getStoreStorageKey(getChainStoreParams),
-      );
-
       if (!storeWithSelectors) {
-        const initialState: RootState | undefined = storedState
-          ? JSON.parse(storedState)
-          : undefined;
+        const storageKey = getStoreStorageKey(getChainStoreParams);
+        const rawStoredState = storageToSyncTo?.get(storageKey);
+        const storedState: RootState | undefined = rawStoredState ? JSON.parse(rawStoredState) : undefined;
+        const snapshotInitialState = initialStateByChainAndEntrypoint[storageKey];
+        const initialState: RootState | undefined = storedState || snapshotInitialState;
         const store = storeFactory({
           entrypointInfo: {
             chainId,