chore: update forgot password flow also

Author: 0xnullifier

src/app/pages/ForgotPassword/ForgotPassword.tsx

@@ -6,6 +6,7 @@ import wordsList from 'bip39/src/wordlists/english.json';
 import { formatMnemonic } from 'app/defaults';
 import { useMidenContext } from 'lib/miden/front';
 import { clearClientStorage } from 'lib/miden/reset';
+import type { WalletAccount } from 'lib/shared/types';
 import { navigate } from 'lib/woozie';
 import { ForgotPasswordFlow } from 'screens/onboarding/forgot-password-navigator';
 import { ForgotPasswordAction, ForgotPasswordStep, OnboardingType } from 'screens/onboarding/types';
@@ -17,6 +18,7 @@ const ForgotPassword: FC = () => {
   const [password, setPassword] = useState<string | null>(null);
   const [importedWithFile, setImportedWithFile] = useState(false);
   const [isLoading, setIsLoading] = useState(false);
+  const [importedWalletAccounts, setImportedWalletAccounts] = useState<WalletAccount[]>([]);
 
   const { registerWallet, importWalletFromClient } = useMidenContext();
 
@@ -38,13 +40,21 @@ const ForgotPassword: FC = () => {
       } else {
         try {
           console.log('importing wallet from client');
-          await importWalletFromClient(password, seedPhraseFormatted);
+          await importWalletFromClient(password, seedPhraseFormatted, importedWalletAccounts);
         } catch (e) {
           console.error(e);
         }
       }
     }
-  }, [password, seedPhrase, importedWithFile, registerWallet, onboardingType, importWalletFromClient]);
+  }, [
+    password,
+    seedPhrase,
+    importedWithFile,
+    registerWallet,
+    onboardingType,
+    importWalletFromClient,
+    importedWalletAccounts
+  ]);
 
   const onAction = useCallback(
     async (action: ForgotPasswordAction) => {
@@ -63,8 +73,8 @@ const ForgotPassword: FC = () => {
           break;
         case 'import-wallet-file-submit':
           const seedPhrase = action.payload.split(' ');
-          console.log({ seedPhrase });
           setSeedPhrase(seedPhrase);
+          setImportedWalletAccounts(action.walletAccounts);
           setImportedWithFile(true);
           setStep(ForgotPasswordStep.CreatePassword);
           break;

src/lib/store/index.test.ts

@@ -299,12 +299,13 @@ describe('useWalletStore', () => {
       mockRequest.mockResolvedValueOnce({ type: WalletMessageType.ImportFromClientResponse });
 
       const { importWalletFromClient } = useWalletStore.getState();
-      await importWalletFromClient('password123', 'mnemonic words');
+      await importWalletFromClient('password123', 'mnemonic words', []);
 
       expect(mockRequest).toHaveBeenCalledWith({
         type: WalletMessageType.ImportFromClientRequest,
         password: 'password123',
-        mnemonic: 'mnemonic words'
+        mnemonic: 'mnemonic words',
+        walletAccounts: []
       });
     });
 

src/screens/onboarding/forgot-password-navigator.tsx

@@ -1,5 +1,7 @@
 import React, { FC, useCallback, useMemo, useState } from 'react';
 
+import { WalletAccount } from 'lib/shared/types';
+
 import { ConfirmationScreen } from './common/Confirmation';
 import { CreatePasswordScreen } from './common/CreatePassword';
 import OnboardingHeader from './common/OnboardingHeader';
@@ -144,8 +146,8 @@ export const ForgotPasswordFlow: FC<ForgotPasswordFlowProps> = ({
       onForwardAction?.({ id: 'confirmation' });
     };
 
-    const onImportFileSubmit = (seedPhrase: string) => {
-      onForwardAction?.({ id: 'import-wallet-file-submit', payload: seedPhrase });
+    const onImportFileSubmit = (seedPhrase: string, walletAccounts: WalletAccount[]) => {
+      onForwardAction?.({ id: 'import-wallet-file-submit', payload: seedPhrase, walletAccounts });
     };
 
     const onSelectTransactionTypeSubmit = () =>