fix: hide zero balance chains (#2563)

novasamatech · Oct 31, 2024 · 55487ae · 55487ae
1 parent 2096b04
commit 55487ae
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/renderer/features/assets/AssetsPortfolioView/lib/tokensService.ts b/src/renderer/features/assets/AssetsPortfolioView/lib/tokensService.ts
@@ -93,9 +93,17 @@ function getChainWithBalance(
 }
 
 function hideZeroBalances(hideZeroBalance: boolean, activeTokensWithBalance: AssetByChains[]): AssetByChains[] {
-  return activeTokensWithBalance.filter((token) => {
-    return !hideZeroBalance || totalAmount(token.totalBalance) !== ZERO_BALANCE;
-  });
+  if (!hideZeroBalance) return activeTokensWithBalance;
+
+  return activeTokensWithBalance.reduce<AssetByChains[]>((acc, token) => {
+    if (totalAmount(token.totalBalance) === ZERO_BALANCE) return acc;
+
+    const chains = token.chains.filter((chain) => {
+      return totalAmount(chain.balance) !== ZERO_BALANCE;
+    });
+
+    return [...acc, { ...token, chains }];
+  }, []);
 }
 
 function sortTokensByBalance(