fix: [ADN-409] Fix Cannot Sign Transaction popup

Author: jinoosss

packages/adena-extension/src/common/provider/wallet/wallet-provider.tsx

@@ -52,7 +52,7 @@ export const WalletProvider: React.FC<React.PropsWithChildren<unknown>> = ({ chi
   }, []);
 
   useEffect(() => {
-    if (wallet && networkMetainfos && tokenMetainfos) {
+    if (wallet && networkMetainfos.length > 0 && tokenMetainfos.length > 0) {
       setWalletStatus('FINISH');
     }
   }, [wallet, networkMetainfos, tokenMetainfos]);
@@ -75,6 +75,9 @@ export const WalletProvider: React.FC<React.PropsWithChildren<unknown>> = ({ chi
     setWalletStatus('LOADING');
     try {
       const wallet = await walletService.loadWallet();
+      console.log(wallet);
+      const currentAccountId = await accountService.getCurrentAccountId();
+      wallet.currentAccountId = currentAccountId;
       setWallet(wallet);
       await initCurrentAccount(wallet);
     } catch (e) {

packages/adena-extension/src/hooks/use-current-account.tsx

@@ -32,8 +32,6 @@ export const useCurrentAccount = (): {
       return false;
     }
     await accountService.changeCurrentAccount(changedAccount);
-    const clone = wallet.clone();
-    clone.currentAccountId = changedAccount.id;
     setCurrentAccount(changedAccount);
     dispatchChangedEvent(changedAccount);
     return true;

packages/adena-extension/src/hooks/use-remove-account.tsx

@@ -25,8 +25,8 @@ export const useRemoveAccount = (): {
     clone.removeAccount(account);
     const nextAccount = clone.accounts[clone.accounts.length - 1];
     clone.currentAccountId = nextAccount.id;
-    await updateWallet(clone);
     await changeCurrentAccount(nextAccount);
+    await updateWallet(clone);
     return true;
   };
 

packages/adena-extension/src/hooks/web/google-login/use-google-login-screen.ts

@@ -102,7 +102,11 @@ const useGoogleLoginScreen = (): UseGoogleLoginReturn => {
       const web3AuthKeyring = await Web3AuthKeyring.fromPrivateKeyStr(privateKey);
       const account = await SingleAccount.createBy(web3AuthKeyring, clone.nextAccountName);
       clone.addAccount(account);
-      await changeCurrentAccount(account);
+      clone.addKeyring(web3AuthKeyring);
+      const storedAccount = clone.accounts.find((storedAccount) => storedAccount.id === account.id);
+      if (storedAccount) {
+        await changeCurrentAccount(storedAccount);
+      }
       await updateWallet(clone);
       navigate(RoutePath.WebAccountAddedComplete);
     },

packages/adena-extension/src/hooks/web/setup-airgap/use-setup-airgap-screen.ts

@@ -107,8 +107,10 @@ const useSetupAirgapScreen = (): UseSetupAirgapScreenReturn => {
     const clone = wallet.clone();
     clone.addAccount(account);
     clone.addKeyring(addressKeyring);
-
-    await changeCurrentAccount(account);
+    const storedAccount = clone.accounts.find((storedAccount) => storedAccount.id === account.id);
+    if (storedAccount) {
+      await changeCurrentAccount(storedAccount);
+    }
     await updateWallet(clone);
     navigate(RoutePath.WebAccountAddedComplete);
   }, [address, walletService]);

packages/adena-extension/src/hooks/web/use-account-import-screen.ts

@@ -114,9 +114,11 @@ const useAccountImportScreen = ({ wallet }: { wallet: Wallet }): UseAccountImpor
       const clone = wallet.clone();
       clone.addAccount(account);
       clone.addKeyring(keyring);
-
+      const storedAccount = clone.accounts.find((storedAccount) => storedAccount.id === account.id);
+      if (storedAccount) {
+        await changeCurrentAccount(storedAccount);
+      }
       await updateWallet(clone);
-      await changeCurrentAccount(account);
       navigate(RoutePath.WebAccountAddedComplete);
     }
   }, [step, privateKey, ableToSkipQuestionnaire, makePrivateKeyAccountAndKeyring]);

packages/adena-extension/src/hooks/web/use-wallet-create-screen.ts

@@ -72,8 +72,13 @@ const useWalletCreateScreen = (): UseWalletCreateReturn => {
         const clone = wallet.clone();
         clone.addAccount(account);
         clone.addKeyring(keyring);
+        const storedAccount = clone.accounts.find(
+          (storedAccount) => storedAccount.id === account.id,
+        );
+        if (storedAccount) {
+          await changeCurrentAccount(storedAccount);
+        }
         await updateWallet(clone);
-        await changeCurrentAccount(account);
         navigate(RoutePath.WebAccountAddedComplete);
       } else {
         const createdWallet = await AdenaWallet.createByMnemonic(seeds);

packages/adena-extension/src/pages/popup/certify/remove-account/index.tsx

@@ -28,7 +28,7 @@ export const RemoveAccount = (): JSX.Element => {
     }
     setState('LOADING');
     await removeAccount(currentAccount);
-    navigate(RoutePath.Home);
+    navigate(RoutePath.Wallet);
   };
 
   return (

packages/adena-extension/src/pages/popup/wallet/account-details/index.tsx

@@ -6,7 +6,6 @@ import AccountDetails from '@components/pages/account-details/account-details';
 import { useLoadAccounts } from '@hooks/use-load-accounts';
 import { useNetwork } from '@hooks/use-network';
 import { useAccountName } from '@hooks/use-account-name';
-import { useCurrentAccount } from '@hooks/use-current-account';
 import { CommonFullContentLayout } from '@components/atoms';
 import useLink from '@hooks/use-link';
 import { AdenaStorage } from '@common/storage';
@@ -20,7 +19,6 @@ const AccountDetailsContainer: React.FC = () => {
   const { accounts } = useLoadAccounts();
   const { currentNetwork } = useNetwork();
   const { accountNames, changeAccountName } = useAccountName();
-  const { changeCurrentAccount } = useCurrentAccount();
   const [originName, setOriginName] = useState('');
   const [name, setName] = useState('');
   const [address, setAddress] = useState<string>('');
@@ -55,9 +53,6 @@ const AccountDetailsContainer: React.FC = () => {
   }, [address]);
 
   const moveExportPrivateKey = useCallback(async () => {
-    if (account) {
-      await changeCurrentAccount(account);
-    }
     const sessionStorage = AdenaStorage.session();
     await sessionStorage.set(WALLET_EXPORT_TYPE_STORAGE_KEY, 'PRIVATE_KEY');
     openSecurity();

packages/adena-extension/src/pages/popup/wallet/wallet-create/index.tsx

@@ -14,18 +14,17 @@ export const WalletCreate = (): JSX.Element => {
   useEffect(() => {
     switch (state) {
       case 'NONE':
+      case 'LOADING':
         break;
       case 'CREATE':
       case 'FAIL':
         openRegister();
         break;
-      case 'LOADING':
-        navigate(RoutePath.Wallet);
-        break;
       case 'LOGIN':
         navigate(RoutePath.Login);
         break;
       default:
+        navigate(RoutePath.Wallet);
         break;
     }
   }, [state]);