Format balances after math (#151)

airswap · Feb 15, 2024 · 23ccfed · 23ccfed
1 parent de94877
commit 23ccfed
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 11 deletions.
diff --git a/src/features/staking/PieBar.tsx b/src/features/staking/PieBar.tsx
@@ -15,14 +15,15 @@ export const PieBar = () => {
   } = useTokenBalances();
 
   const unstaked = formatNumber(astBalanceRaw, 4) || 0;
-  const staked = formatNumber(sAstBalanceRaw, 4) || 0;
-  const unstakable = formatNumber(unstakableSastBalanceRaw, 4) || 0;
-  const unstakableV4 =
-    formatNumber(unstakableSastBalanceV4_DeprecatedRaw, 4) || 0;
-  const stakedV4 = formatNumber(sAstBalanceV4_DeprecatedRaw, 4) || 0;
 
-  const totalStaked = String(Number(staked) + Number(stakedV4));
-  const totalUnstakable = String(Number(unstakable) + Number(unstakableV4));
+  const totalStaked = formatNumber(
+    sAstBalanceRaw + sAstBalanceV4_DeprecatedRaw,
+    4,
+  );
+  const totalUnstakable = formatNumber(
+    unstakableSastBalanceRaw + unstakableSastBalanceV4_DeprecatedRaw,
+    4,
+  );
 
   // the following 2 values are only used for calculateTokenProportions
   const totalUnstakableRaw =

diff --git a/src/features/staking/StakingButton.tsx b/src/features/staking/StakingButton.tsx
@@ -11,10 +11,10 @@ export const StakingButton = () => {
   const { isConnected } = useAccount();
   const { sAstBalanceRaw, sAstBalanceV4_DeprecatedRaw } = useTokenBalances();
 
-  const sAstBalance = formatNumber(sAstBalanceRaw, 4) || 0;
-  const sAstBalanceV4 = formatNumber(sAstBalanceV4_DeprecatedRaw, 4) || 0;
-
-  const totalSastBalance = Number(sAstBalance) + Number(sAstBalanceV4);
+  const totalSastBalance = formatNumber(
+    sAstBalanceRaw + sAstBalanceV4_DeprecatedRaw,
+    4,
+  );
 
   return (
     <>