fix: [lw-12781] set proper blockchain on add/restore wallet (#1871)

Author: vetalcore

apps/browser-extension-wallet/src/components/MainMenu/DropdownMenuOverlay/components/UserInfo.tsx

@@ -17,6 +17,8 @@ import { AnyBip32Wallet, AnyWallet, Bip32WalletAccount, ScriptWallet, WalletType
 import { Wallet } from '@lace/cardano';
 import { Separator } from './Separator';
 import { getUiWalletType } from '@src/utils/get-ui-wallet-type';
+import { isScriptWallet } from '@lace/core';
+import { useCurrentBlockchain } from '@src/multichain';
 
 const ADRESS_FIRST_PART_LENGTH = 10;
 const ADRESS_LAST_PART_LENGTH = 5;
@@ -65,6 +67,7 @@ export const UserInfo = ({
   const [lastActiveAccount, setLastActiveAccount] = useState<number>(0);
   const [handle] = useGetHandles();
   const { activeWalletAvatar, getAvatar } = useWalletAvatar();
+  const { blockchain } = useCurrentBlockchain();
 
   const handleName = handle?.nftMetadata?.name;
 
@@ -153,6 +156,9 @@ export const UserInfo = ({
               duration: TOAST_DEFAULT_DURATION,
               text: t('multiWallet.activated.wallet', { walletName: wallet.metadata.name })
             });
+            // reload if we switch from, to or between btc wallet(s)
+            if ((!isScriptWallet(wallet) && wallet.blockchainName === 'Bitcoin') || blockchain === 'bitcoin')
+              window.location.reload();
           }}
           type={walletType}
           profile={
@@ -180,7 +186,8 @@ export const UserInfo = ({
       onOpenWalletAccounts,
       setIsDropdownMenuOpen,
       activeWalletId,
-      t
+      t,
+      blockchain
     ]
   );
 

apps/browser-extension-wallet/src/hooks/__tests__/useWalletManager.test.tsx

@@ -403,6 +403,7 @@ describe('Testing useWalletManager hook', () => {
 
   describe('createWallet', () => {
     test('should store wallet in repository', async () => {
+      const setBackgroundStorage = jest.fn();
       const walletId = 'walletId';
       (walletApiUi.walletRepository as any).addWallet = jest.fn().mockResolvedValue(walletId);
       (walletApiUi.walletManager as any).activate = jest.fn().mockResolvedValue(undefined);
@@ -432,7 +433,13 @@ describe('Testing useWalletManager hook', () => {
           current: { createWallet }
         }
       } = renderHook(() => useWalletManager(), {
-        wrapper: getWrapper({})
+        wrapper: getWrapper({
+          backgroundService: {
+            clearBackgroundStorage: jest.fn(),
+            getBackgroundStorage: jest.fn().mockResolvedValue({ activeBlockchain: 'cardano' }),
+            setBackgroundStorage
+          } as unknown as BackgroundServiceAPIProviderProps['value']
+        })
       });
 
       expect(createWallet).toBeDefined();
@@ -453,6 +460,8 @@ describe('Testing useWalletManager hook', () => {
       expect(walletApiUi.walletRepository.addWallet).toBeCalledTimes(1);
       expect(walletApiUi.walletManager.activate).toBeCalledTimes(1);
       expect(mockUseSecrets.clearSecrets).toBeCalledTimes(1);
+      expect(setBackgroundStorage).toBeCalledTimes(1);
+      expect(setBackgroundStorage).toBeCalledWith({ activeBlockchain: 'cardano' });
     });
   });
 

apps/browser-extension-wallet/src/hooks/useWalletManager.ts

@@ -822,6 +822,10 @@ export const useWalletManager = (): UseWalletManager => {
         accountIndex
       });
 
+      await backgroundService.setBackgroundStorage({
+        activeBlockchain: 'cardano'
+      });
+
       // Needed for reset password flow
       saveValueInLocalStorage({ key: 'wallet', value: { name } });
 
@@ -842,7 +846,7 @@ export const useWalletManager = (): UseWalletManager => {
         }
       };
     },
-    [getCurrentChainId, clearSecrets]
+    [getCurrentChainId, backgroundService, clearSecrets]
   );
 
   /**