fetch balance if wallet exists

Author: nadiia-balaian

src/blockchain/FuelNetwork.ts

@@ -64,6 +64,10 @@ export class FuelNetwork {
     return this.walletManager.getBalance(accountAddress, assetAddress);
   };
 
+  getWallet = (): Nullable<any> => {
+    return this.walletManager.wallet;
+  };
+
   getIsExternalWallet = () => false;
 
   getTokenList = (): Token[] => {

src/components/Header/Header.tsx

@@ -40,8 +40,12 @@ const Header: React.FC = observer(() => {
   useEffect(() => {
     accountStore.setAddress(address);
     if (address && wallet) {
-      const bcNetwork = FuelNetwork.getInstance();
-      bcNetwork.setWallet(address, wallet);
+      const setWallet = async () => {
+        const bcNetwork = FuelNetwork.getInstance();
+        await bcNetwork.setWallet(address, wallet);
+      };
+
+      setWallet();
     }
   }, [address, wallet?.address]);
 

src/stores/BalanceStore.ts

@@ -49,8 +49,9 @@ export class BalanceStore {
   update = async () => {
     const { accountStore } = this.rootStore;
     const bcNetwork = FuelNetwork.getInstance();
+    const wallet = bcNetwork.getWallet();
 
-    if (!accountStore.address) return;
+    if (!accountStore.address || !wallet) return;
 
     try {
       for (const token of bcNetwork!.getTokenList()) {

src/stores/TradeStore.ts

@@ -158,7 +158,7 @@ class TradeStore {
       this.setSpotMarkets(spotMarkets);
       await this.updateMarketPrices();
     } catch (error) {
-      console.error("[PERP] Error init spot market", error);
+      console.error("Error init spot market", error);
     }
   };