[web] Include tokenAmounts in KLAYswap single staking (#313)

0xinevitable · Oct 12, 2022 · a542ef1 · a542ef1
1 parent a8df8e4
commit a542ef1
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/packages/bento-web/src/defi/klaytn/klayswap/single.ts b/packages/bento-web/src/defi/klaytn/klayswap/single.ts
@@ -65,6 +65,12 @@ export const getSinglePoolBalance = async (
 
   const tokenInfo = getTokenInfo(pool.token);
   const balance = Number(rawBalance) / 10 ** (tokenInfo?.decimals || 18);
+  let tokenAmounts: Record<string, number | undefined> | undefined = undefined;
+  if (tokenInfo) {
+    tokenAmounts = {
+      [tokenInfo.symbol]: balance,
+    };
+  }
 
   return {
     protocol: KlaytnDeFiProtocolType.KLAYSWAP,
@@ -75,6 +81,7 @@ export const getSinglePoolBalance = async (
     tokens: [tokenInfo || null],
     staked: {
       lpAmount: balance,
+      tokenAmounts,
     },
     rewards: {
       tokenAmounts: {