fix(Staking): limit parsing

interlay · Nov 6, 2023 · 7e849c0 · 7e849c0
1 parent b1b3f42
commit 7e849c0
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 4 deletions.
diff --git a/src/hooks/api/escrow/use-get-account-staking-data.tsx b/src/hooks/api/escrow/use-get-account-staking-data.tsx
@@ -1,4 +1,4 @@
-import { CurrencyExt } from '@interlay/interbtc-api';
+import { CurrencyExt, newMonetaryAmount } from '@interlay/interbtc-api';
 import { MonetaryAmount } from '@interlay/monetary-js';
 import { AccountId } from '@polkadot/types/interfaces';
 import Big from 'big.js';
@@ -7,6 +7,7 @@ import { useErrorHandler } from 'react-error-boundary';
 import { useQuery } from 'react-query';
 
 import { BLOCK_TIME } from '@/config/parachain';
+import { GOVERNANCE_TOKEN } from '@/config/relay-chains';
 import { REFETCH_INTERVAL } from '@/utils/constants/api';
 
 import useAccountId from '../../use-account-id';
@@ -61,14 +62,17 @@ const getAccountStakingData = async (accountId: AccountId): Promise<AccountStaki
     votingBalancePromise
   ]);
 
+  const unparsedLimit = limit?.values().next().value?.toString() as string;
+  const parsedLimit = newMonetaryAmount(unparsedLimit || 0, GOVERNANCE_TOKEN);
+
   const unlock = getUnlockData(stakedBalance.endBlock, currentBlockNumber);
 
   return {
     unlock,
     balance: stakedBalance.amount,
     votingBalance,
     projected,
-    limit
+    limit: parsedLimit
   };
 };
 

diff --git a/src/pages/Staking/components/StakingAccountDetails/StakingAccountDetails.tsx b/src/pages/Staking/components/StakingAccountDetails/StakingAccountDetails.tsx
@@ -29,7 +29,10 @@ const StakingAccountDetails = ({
   const [isOpen, setOpen] = useState(false);
 
   const transaction = useTransaction(Transaction.ESCROW_WITHDRAW_REWARDS, {
-    onSuccess: onClaimRewards
+    onSuccess: () => {
+      onClaimRewards();
+      setOpen(false);
+    }
   });
 
   const handleSubmit = () => transaction.execute();

diff --git a/src/pages/Staking/components/StakingWithdrawCard/StakingWithdrawCard.tsx b/src/pages/Staking/components/StakingWithdrawCard/StakingWithdrawCard.tsx
@@ -23,7 +23,10 @@ const StakingWithdrawCard = ({ data, onWithdraw, ...props }: StakingWithdrawCard
   const [isOpen, setOpen] = useState(false);
 
   const transaction = useTransaction(Transaction.ESCROW_WITHDRAW, {
-    onSuccess: onWithdraw
+    onSuccess: () => {
+      onWithdraw();
+      setOpen(false);
+    }
   });
 
   const handleSubmit = () => transaction.execute();