fix(amount-input): balance parsing fix (wormhole-foundation#2567)

wormholelabs-xyz · Sep 11, 2024 · 79dd5a7 · 79dd5a7
1 parent 6b8d602
commit 79dd5a7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/wormhole-connect/src/hooks/useGetTokenBalances.ts b/wormhole-connect/src/hooks/useGetTokenBalances.ts
@@ -14,6 +14,8 @@ import { chainToPlatform } from '@wormhole-foundation/sdk-base';
 import { getTokenBridgeWrappedTokenAddress } from 'utils/sdkv2';
 import { Chain, TokenAddress } from '@wormhole-foundation/sdk';
 
+// TODO: This hook shouldn't format amounts
+// Instead the view should format and render accordingly
 const useGetTokenBalances = (
   walletAddress: string,
   chain: Chain | undefined,

diff --git a/wormhole-connect/src/views/v2/Bridge/AmountInput/index.tsx b/wormhole-connect/src/views/v2/Bridge/AmountInput/index.tsx
@@ -15,7 +15,6 @@ import Typography from '@mui/material/Typography';
 import AlertBannerV2 from 'components/v2/AlertBanner';
 import useGetTokenBalances from 'hooks/useGetTokenBalances';
 import { setAmount } from 'store/transferInput';
-import { toFixedDecimals } from 'utils/balance';
 import { getMaxAmt, validateAmount } from 'utils/transferValidation';
 import type { TokenConfig } from 'config/types';
 import type { RootState } from 'store';
@@ -122,7 +121,7 @@ const AmountInput = (props: Props) => {
           <CircularProgress size={14} />
         ) : (
           <Typography fontSize={14} textAlign="right">
-            {Number.parseFloat(toFixedDecimals(`${tokenBalance}`, 6))}
+            {tokenBalance}
           </Typography>
         )}
       </Stack>
@@ -136,7 +135,8 @@ const AmountInput = (props: Props) => {
         disabled={isInputDisabled || !tokenBalance}
         onClick={() => {
           if (tokenBalance) {
-            const trimmedTokenBalance = toFixedDecimals(`${tokenBalance}`, 6);
+            // TODO: Remove this when useGetTokenBalances returns non formatted amounts
+            const trimmedTokenBalance = tokenBalance.replaceAll(',', '');
             dispatch(setAmount(trimmedTokenBalance));
           }
         }}