feat: add method to get all balances

src/index.ts

@@ -33,7 +33,9 @@ import {
   ValidateBalanceAndApproval,
   WASM_TYPE,
   WasmHookMsg,
-  CosmosChain
+  CosmosChain,
+  CosmosAddress,
+  CosmosBalance
 } from "./types";
 
 import { TxRaw } from "cosmjs-types/cosmos/tx/v1beta1/tx";
@@ -743,27 +745,64 @@ export class Squid {
     userAddress: string;
     chains: number[];
   }): Promise<TokenBalance[]> {
+    // remove invalid and duplicate chains
+    const filteredChains = new Set(
+      chains.filter(c => !Number.isNaN(c) && !isNaN(c))
+    );
+
     return getAllEvmTokensBalance(
-      this.tokens.filter(t => chains.includes(Number(t.chainId))),
+      this.tokens.filter(t => filteredChains.has(Number(t.chainId))),
       userAddress
     );
   }
 
   public async getAllCosmosBalances({
     addresses
   }: {
-    addresses: Array<{
-      coinType: number;
-      chainId: string;
-      address: string;
-    }>;
+    addresses: CosmosAddress[];
   }) {
     const cosmosChains = this.chains.filter(
       c => c.chainType === ChainType.Cosmos
     ) as CosmosChain[];
 
     return getCosmosBalances({ addresses, cosmosChains });
   }
+
+  public async getAllBalances({
+    chainIds,
+    cosmosAddresses,
+    evmAddress
+  }: {
+    chainIds?: string[];
+    cosmosAddresses?: CosmosAddress[];
+    evmAddress?: string;
+  }): Promise<{
+    cosmosBalances?: CosmosBalance[];
+    evmBalances?: TokenBalance[];
+  }> {
+    if (!chainIds) {
+      // fetch balances for all chains compatible with provided addresses
+      const evmBalances = evmAddress
+        ? await this.getAllEvmBalances({
+            chains: this.tokens.map(t => Number(t.chainId)),
+            userAddress: evmAddress
+          })
+        : [];
+
+      const cosmosBalances = cosmosAddresses
+        ? await this.getAllCosmosBalances({
+            addresses: cosmosAddresses
+          })
+        : [];
+
+      return {
+        evmBalances,
+        cosmosBalances
+      };
+    }
+
+    return {};
+  }
 }
 
 export * from "./types";

src/services/getCosmosBalances.ts

@@ -1,5 +1,5 @@
 import { StargateClient } from "@cosmjs/stargate";
-import { ChainType, CosmosChain } from "../types";
+import { ChainType, CosmosAddress, CosmosBalance, CosmosChain } from "../types";
 import { fromBech32, toBech32 } from "@cosmjs/encoding";
 
 const deriveCosmosAddress = (chainPrefix: string, address: string): string => {
@@ -10,18 +10,11 @@ export async function getCosmosBalances({
   addresses,
   cosmosChains
 }: {
-  addresses: Array<{
-    coinType: number;
-    chainId: string;
-    address: string;
-  }>;
+  addresses: CosmosAddress[];
   cosmosChains: CosmosChain[];
-}) {
-  const balances: {
-    amount: string;
-    denom: string;
-    chainId: string;
-  }[] = [];
+}): Promise<CosmosBalance[]> {
+  const balances: CosmosBalance[] = [];
+
   for (const chain of cosmosChains) {
     if (chain.chainType !== ChainType.Cosmos) continue;
 

src/types/index.ts

@@ -515,3 +515,15 @@ export type AllBalancesResult = {
   cosmosBalances: TokenBalance[];
   evmBalances: TokenBalance[];
 };
+
+export type CosmosAddress = {
+  coinType: number;
+  chainId: string;
+  address: string;
+};
+
+export type CosmosBalance = {
+  amount: string;
+  denom: string;
+  chainId: string;
+};