Don't duplicate underlying balance

beefyfinance · May 21, 2024 · 3274185 · 3274185
1 parent f8ea631
commit 3274185
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/vault-interaction.ts b/src/vault-interaction.ts
@@ -158,8 +158,22 @@ function updateVaultBreakDown(block: ethereum.Block, vault: BeefyVault): BeefyVa
 
   // also add the share token and underlying token to the breakdown
   // so we are also computing time weighted balance for those
-  breakdown.push(new TokenBalance(vault.sharesToken, vault.rawSharesTokenTotalSupply))
-  breakdown.push(new TokenBalance(vault.underlyingToken, vault.rawUnderlyingBalance))
+  let foundSharesToken = false
+  let foundUnderlyingToken = false
+  for (let i = 0; i < breakdown.length; i++) {
+    if (breakdown[i].tokenAddress.equals(vault.sharesToken)) {
+      foundSharesToken = true
+    }
+    if (breakdown[i].tokenAddress.equals(vault.underlyingToken)) {
+      foundUnderlyingToken = true
+    }
+  }
+  if (!foundSharesToken) {
+    breakdown.push(new TokenBalance(vault.sharesToken, vault.rawSharesTokenTotalSupply))
+  }
+  if (!foundUnderlyingToken) {
+    breakdown.push(new TokenBalance(vault.underlyingToken, vault.rawUnderlyingBalance))
+  }
 
   for (let i = 0; i < breakdown.length; i++) {
     const tokenBalance = breakdown[i]