feat: return all chain balances, rename amount to balance, and return…

… coin decimals
0xsquid · Oct 25, 2023 · 4ad643a · 4ad643a
1 parent 81c5267
commit 4ad643a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 10 deletions.
diff --git a/src/services/getCosmosBalances.ts b/src/services/getCosmosBalances.ts
@@ -13,7 +13,7 @@ export async function getCosmosBalances({
   addresses: CosmosAddress[];
   cosmosChains: CosmosChain[];
 }): Promise<CosmosBalance[]> {
-  const balances: CosmosBalance[] = [];
+  const cosmosBalances: CosmosBalance[] = [];
 
   for (const chain of cosmosChains) {
     if (chain.chainType !== ChainType.Cosmos) continue;
@@ -31,21 +31,26 @@ export async function getCosmosBalances({
 
     try {
       const client = await StargateClient.connect(chain.rpc);
-      const [account] = (await client.getAllBalances(cosmosAddress)) ?? [];
+      const balances = (await client.getAllBalances(cosmosAddress)) ?? [];
 
-      if (!account) continue;
+      if (balances.length === 0) continue;
 
-      const { amount, denom } = account;
+      balances.forEach(balance => {
+        const { amount, denom } = balance;
 
-      balances.push({
-        amount,
-        denom,
-        chainId: String(chain.chainId)
+        cosmosBalances.push({
+          balance: amount,
+          denom,
+          chainId: String(chain.chainId),
+          decimals:
+            chain.currencies.find(currency => currency.coinDenom === denom)
+              ?.coinDecimals ?? 6
+        });
       });
     } catch (error) {
       //
     }
   }
 
-  return balances;
+  return cosmosBalances;
 }
diff --git a/src/types/index.ts b/src/types/index.ts
@@ -523,7 +523,8 @@ export type CosmosAddress = {
 };
 
 export type CosmosBalance = {
-  amount: string;
+  decimals: number;
+  balance: string;
   denom: string;
   chainId: string;
 };