diff --git a/packages/api/src/hooks/useAccountsList.ts b/packages/api/src/hooks/useAccountsList.ts index e3523120336a..0814e33cffe6 100644 --- a/packages/api/src/hooks/useAccountsList.ts +++ b/packages/api/src/hooks/useAccountsList.ts @@ -1,9 +1,11 @@ import { useMemo } from 'react'; import useAuthorize from './useAuthorize'; +import useBalance from './useBalance'; /** A custom hook that returns the list of accounts of the logged in user. */ const useAccountsList = () => { const { data: authorize_data, ...rest } = useAuthorize(); + const { data: balance_data } = useBalance(); // Add additional information to the authorize response. const modified_accounts = useMemo(() => { @@ -30,9 +32,20 @@ const useAccountsList = () => { }); }, [authorize_data.account_list, authorize_data.loginid]); + // Add balance to each account + const modified_accounts_with_balance = useMemo( + () => + modified_accounts?.map(account => ({ + ...account, + /** The balance of the account. */ + balance: balance_data?.accounts?.[account.loginid]?.balance || 0, + })), + [balance_data?.accounts, modified_accounts] + ); + return { /** The list of accounts. */ - data: modified_accounts, + data: modified_accounts_with_balance, ...rest, }; };