fix(coinmarket): reflect account updates in coinmarket

(cherry picked from commit 9894500)
trezor · Jan 21, 2025 · ee5fa76 · ee5fa76
1 parent 2eaf105
commit ee5fa76
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/packages/suite/src/hooks/wallet/coinmarket/form/useCoinmarketExchangeForm.ts b/packages/suite/src/hooks/wallet/coinmarket/form/useCoinmarketExchangeForm.ts
@@ -628,6 +628,12 @@ export const useCoinmarketExchangeForm = ({
         }
     }, [defaultValues, values, removeDraft]);
 
+    useEffect(() => {
+        if (account.key === coinmarketAccount?.key) {
+            setAccount(coinmarketAccount);
+        }
+    }, [account, setAccount, coinmarketAccount]);
+
     // react-hook-form auto register custom form fields (without HTMLElement)
     useEffect(() => {
         register('options');

diff --git a/packages/suite/src/middlewares/wallet/coinmarketMiddleware.ts b/packages/suite/src/middlewares/wallet/coinmarketMiddleware.ts
@@ -2,6 +2,7 @@ import { MiddlewareAPI } from 'redux';
 
 import { UI } from '@trezor/connect';
 import { Route } from '@suite-common/suite-types';
+import { accountsActions } from '@suite-common/wallet-core';
 
 import { AppState, Action, Dispatch } from 'src/types/suite';
 import {
@@ -214,5 +215,17 @@ export const coinmarketMiddleware =
             }
         }
 
+        if (isCoinmarketRoute && action.type === accountsActions.updateAccount.type) {
+            const account = action.payload;
+
+            if (state.wallet.coinmarket.sell.coinmarketAccount?.key === account.key) {
+                api.dispatch(coinmarketSellActions.setCoinmarketSellAccount(account));
+            }
+
+            if (state.wallet.coinmarket.exchange.coinmarketAccount?.key === account.key) {
+                api.dispatch(coinmarketExchangeActions.setCoinmarketExchangeAccount(account));
+            }
+        }
+
         return action;
     };