fix: account list mismatch when reconnecting Snap (#541)

* fix: account list mismatch when reconnecting Snap

* chore: fix comments

Author: khanti42

packages/wallet-ui/src/services/useStarkNetSnap.ts

@@ -30,7 +30,7 @@ import {
   updateCurrentAccount,
   setWalletConnection,
 } from 'slices/walletSlice';
-import { setActiveNetwork, setNetworks } from 'slices/networkSlice';
+import { setNetworksAndActiveNetwork } from 'slices/networkSlice';
 import { disableLoading, enableLoadingWithMessage } from 'slices/UISlice';
 import {
   Transaction,
@@ -156,13 +156,18 @@ export const useStarkNetSnap = () => {
         console.error('No networks found');
         return;
       }
-      dispatch(setNetworks(networks));
 
       const currentNetwork = await getCurrentNetwork();
       const idx = networks.findIndex(
         (network) => network.chainId === currentNetwork.chainId,
       );
-      dispatch(setActiveNetwork(idx));
+
+      dispatch(
+        setNetworksAndActiveNetwork({
+          networks,
+          activeNetwork: idx,
+        }),
+      );
 
       await initWalletData({
         chainId: currentNetwork.chainId,

packages/wallet-ui/src/slices/networkSlice.ts

@@ -15,12 +15,13 @@ export const networkSlice = createSlice({
   name: 'network',
   initialState,
   reducers: {
-    setNetworks: (state, action) => {
-      state.items = action.payload;
-    },
     setActiveNetwork: (state, action) => {
       state.activeNetwork = action.payload;
     },
+    setNetworksAndActiveNetwork: (state, action) => {
+      state.items = action.payload.networks;
+      state.activeNetwork = action.payload.activeNetwork;
+    },
     resetNetwork: () => {
       return {
         ...initialState,
@@ -29,7 +30,7 @@ export const networkSlice = createSlice({
   },
 });
 
-export const { setNetworks, setActiveNetwork, resetNetwork } =
+export const { setActiveNetwork, setNetworksAndActiveNetwork, resetNetwork } =
   networkSlice.actions;
 
 export default networkSlice.reducer;