@avalabs/avalanche-module 0.0.18 → 0.0.19

package/src/module.ts

@@ -12,26 +12,30 @@ import type {
 import { parseManifest } from '@avalabs/vm-module-types';
 import { rpcErrors } from '@metamask/rpc-errors';
 import ManifestJson from '../manifest.json';
-import { getNetworkFee } from './handlers/get-network-fee';
+import { getNetworkFee } from './handlers/get-network-fee/get-network-fee';
 import { getTransactionHistory } from './handlers/get-transaction-history/get-transaction-history';
 import { getEnv } from './env';
 import { AvalancheGlacierService } from './services/glacier-service/glacier-service';
-import { hashBlockchainId } from '@internal/utils';
+import { hashBlockchainId, TokenService } from '@internal/utils';
+import { getBalances } from './handlers/get-balances/get-balances';
 
 export class AvalancheModule implements Module {
   #glacierService: AvalancheGlacierService;
+  #proxyApiUrl: string;
 
   constructor({ environment }: { environment: Environment }) {
-    const { glacierApiUrl } = getEnv(environment);
+    const { glacierApiUrl, proxyApiUrl } = getEnv(environment);
     this.#glacierService = new AvalancheGlacierService({ glacierApiUrl });
+    this.#proxyApiUrl = proxyApiUrl;
   }
 
   getAddress(): Promise<string> {
     return Promise.resolve('Avalanche address');
   }
 
-  getBalances(_: GetBalancesParams): Promise<GetBalancesResponse> {
-    return Promise.resolve({});
+  getBalances({ addresses, network, storage, currency }: GetBalancesParams): Promise<GetBalancesResponse> {
+    const tokenService = new TokenService({ storage, proxyApiUrl: this.#proxyApiUrl });
+    return getBalances({ addresses, currency, network, glacierService: this.#glacierService, tokenService });
   }
 
   getManifest(): Manifest | undefined {

package/src/services/glacier-service/glacier-service.ts

@@ -1,8 +1,11 @@
 import {
   BlockchainId,
   Glacier,
+  type ListCChainAtomicBalancesResponse,
   type ListCChainAtomicTransactionsResponse,
+  type ListPChainBalancesResponse,
   type ListPChainTransactionsResponse,
+  type ListXChainBalancesResponse,
   type ListXChainTransactionsResponse,
   Network,
   PrimaryNetworkTxType,
@@ -53,4 +56,20 @@ export class AvalancheGlacierService {
       throw error;
     }
   }
+
+  async getChainBalance(params: {
+    blockchainId: BlockchainId;
+    network: Network;
+    blockTimestamp?: number;
+    addresses?: string;
+  }): Promise<ListPChainBalancesResponse | ListXChainBalancesResponse | ListCChainAtomicBalancesResponse> {
+    try {
+      return this.glacierSdk.primaryNetworkBalances.getBalancesByAddresses(params);
+    } catch (error) {
+      if (error instanceof GlacierUnhealthyError) {
+        this.setGlacierToUnhealthy();
+      }
+      throw error;
+    }
+  }
 }