Refactor account store to remove fetchBalance method and directly use…

… connectedAccount's balance method

src/lib/ConnectMassaWallets/store/accountStore.ts

@@ -21,7 +21,6 @@ export interface AccountStoreState {
   setWallets: (wallets: Wallet[]) => void;
   setConnectedAccount: (account?: Provider) => void;
   setCurrentNetwork: (network: Network) => void;
-  fetchBalance: (provider: Provider, final: boolean) => Promise<void>;
   refreshBalance: (final: boolean) => void;
 }
 
@@ -95,23 +94,20 @@ export const useAccountStore = create<AccountStoreState>((set, get) => ({
   setConnectedAccount: async (connectedAccount?: Provider) => {
     set({ connectedAccount });
     if (!connectedAccount) return;
-    get().fetchBalance(connectedAccount, false);
+    const balance = await connectedAccount.balance(false);
+    set({ balance });
   },
 
   setCurrentNetwork: (network: Network) => {
     if (network === get().network) return;
     set({ network });
   },
 
-  fetchBalance: async (provider: Provider) => {
-    const balance = await provider.balance(false);
-    set({ balance });
-  },
-
   refreshBalance: async (final: boolean) => {
     const { connectedAccount } = get();
     if (!connectedAccount) return;
-    get().fetchBalance(connectedAccount, final);
+    const balance = await connectedAccount.balance(final);
+    set({ balance });
   },
 }));