Feat: Pull balances /w block-height | fetch tokenPrice for native token

examples/wallet-manager.ts

@@ -18,29 +18,42 @@ const allChainWallets: WalletManagerFullConfig['config'] = {
         tokens: ["WBTC"]
       }
     ],
-    priceFeedConfig: {
+    walletBalanceConfig: {
+      enabled: true,
       scheduled: {
-        enabled: true
-      },
+        enabled: false,
+      }
+    },
+    priceFeedConfig: {
       supportedTokens: [{
         chainId: 2,
+        chainName: "ethereum",
         tokenContract: "0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48",
         coingeckoId: "usd-coin",
         symbol: "USDC"
       },
       {
         chainId: 2,
+        chainName: "ethereum",
         tokenContract: "0x2260fac5e5542a773aa44fbcfedf7c193bc2c599",
         coingeckoId: "wrapped-bitcoin",
         symbol: "WBTC"
       }],
-      enabled: true
     }
   },
   solana: {
     wallets: [
       { address: "6VnfVsLdLwNuuCmooLTziQ99PFXZ5vc3yyqyb9tMDhhw", tokens: ['usdc'] },
     ],
+    walletBalanceConfig: {
+      enabled: true,
+      scheduled: {
+        enabled: false,
+      }
+    },
+    priceFeedConfig: {
+      supportedTokens: []
+    }
   },
   sui: {
     rebalance: {
@@ -62,14 +75,20 @@ const allChainWallets: WalletManagerFullConfig['config'] = {
         tokens: ['USDC', 'USDT'] 
       },
     ],
+    walletBalanceConfig: {
+      enabled: true,
+      scheduled: {
+        enabled: false,
+      }
+    },
     priceFeedConfig: {
       supportedTokens: [{
         chainId: 21,
+        chainName: "sui",
         tokenContract: "0x5d4b302506645c37ff133b98c4b50a5ae14841659738d6d733d59d0d217a93bf",
         coingeckoId: "usd-coin",
         symbol: "USDC"
       }],
-      enabled: false,
     }
   },
   klatyn: {
@@ -88,7 +107,16 @@ const allChainWallets: WalletManagerFullConfig['config'] = {
         address: "0x8d0d970225597085A59ADCcd7032113226C0419d",
         tokens: []
       }
-    ]
+    ],
+    walletBalanceConfig: {
+      enabled: true,
+      scheduled: {
+        enabled: false,
+      }
+    },
+    priceFeedConfig: {
+      supportedTokens: []
+    }
   }
 }
 
@@ -103,15 +131,22 @@ export const manager = buildWalletManager({
       enabled: true,
       serve: true,
       port: 9091,
+    },
+    priceFeedOptions: {
+      enabled: true,
+      scheduled: {
+        enabled: false,
+      }
     }
   }
 });
 
-
-// Note: Below code needs wallet's private key to be set in config abopve for aquiring wallet
-// manager.withWallet('ethereum', async (wallet) => {
-//   console.log('Address', wallet.address);
-//   console.log('Block height', wallet.walletToolbox.getBlockHeight());
-//   console.log('Native balances', await wallet.walletToolbox.pullNativeBalance(wallet.address));
-//   console.log('Token balances', await wallet.walletToolbox.pullTokenBalances(wallet.address, ['0xa0b86991c6218b36c1d19d4a2e9eb0ce3606eb48', '0x2260fac5e5542a773aa44fbcfedf7c193bc2c599']));
-// })
+(async () => {
+  try {
+    console.time('balances')
+    const balances = await manager.pullBalancesAtBlockHeight();
+    console.timeLog('balances', JSON.stringify(balances))
+  } catch (err) {
+    console.error('Failed to pullBalancesAtBlockHeight', err);
+  }
+})();

src/balances/evm/index.ts

@@ -42,9 +42,10 @@ export async function pullEvmTokenBalance(
 export async function pullEvmNativeBalance(
   provider: ethers.providers.JsonRpcProvider,
   address: string,
+  blockHeight?: number,
 ): Promise<Balance>{
-  const weiAmount = await provider.getBalance(address);
-  
+  const weiAmount = await provider.getBalance(address, blockHeight);
+
   return {
     isNative: true,
     rawBalance: weiAmount.toString(),

src/balances/solana.ts

@@ -1,12 +1,14 @@
 import bs58 from 'bs58';
 import { Connection, PublicKey, Keypair } from "@solana/web3.js";
 import { Balance } from "./index";
+import { SOLANA_DEFAULT_COMMITMENT } from '../wallets/solana/solana.config';
 
 export async function pullSolanaNativeBalance(
   connection: Connection,
-  address: string
+  address: string,
 ): Promise<Balance> {
-  const lamports = await connection.getBalance(new PublicKey(address))
+  // solana web3.js doesn't support passing exact slot(block number) only minSlot, while fetching balance 
+  const lamports = await connection.getBalance(new PublicKey(address), SOLANA_DEFAULT_COMMITMENT)
 
   return {
     isNative: true,

src/balances/sui.ts

@@ -18,6 +18,8 @@ export interface SuiTransactionDetails {
 export async function pullSuiNativeBalance(conn: Connection, address: string): Promise<WalletBalance> {
   const provider = new JsonRpcProvider(conn);
 
+  // mysten SDK doesn't support passing checkpoint (block number) to getBalance
+  // https://github.com/MystenLabs/sui/issues/14137
   const rawBalance = await provider.getBalance({ owner: address });
 
   return {
@@ -54,7 +56,15 @@ export async function pullSuiTokenData(
 export async function pullSuiTokenBalances(
     conn: Connection,
     address: string
-): Promise<any> {
+): Promise<{
+  coinType: string;
+  coinObjectCount: number;
+  totalBalance: string;
+  lockedBalance: {
+      number?: number | undefined;
+      epochId?: number | undefined;
+  };
+}[]> {
   const provider = new JsonRpcProvider(conn);
 
   return provider.getAllBalances({ owner: address });

src/chain-wallet-manager.ts

@@ -17,13 +17,7 @@ import { CosmosProvider, CosmosWallet } from "./wallets/cosmos";
 import { EVMProvider, EVMWallet } from "./wallets/evm";
 import { SolanaProvider, SolanaWallet } from "./wallets/solana";
 import { SuiProvider, SuiWallet } from "./wallets/sui";
-import {
-  PriceFeed,
-  WalletPriceFeedConfig,
-  WalletRebalancingConfig,
-} from "./wallet-manager";
-import { ScheduledPriceFeed } from "./price-assistant/scheduled-price-feed";
-import { OnDemandPriceFeed } from "./price-assistant/ondemand-price-feed";
+import { PriceFeed, WalletBalanceConfig, WalletPriceFeedConfig, WalletRebalancingConfig } from "./wallet-manager";
 
 const DEFAULT_POLL_INTERVAL = 60 * 1000;
 const DEFAULT_REBALANCE_INTERVAL = 60 * 1000;
@@ -42,6 +36,7 @@ export type ChainWalletManagerOptions = {
     maxGasPrice?: number;
     gasLimit?: number;
   };
+  walletBalanceConfig: WalletBalanceConfig;
   priceFeedConfig: WalletPriceFeedConfig;
   balancePollInterval?: number;
   walletOptions?: WalletOptions;
@@ -85,7 +80,11 @@ export class ChainWalletManager {
   public walletToolbox: Wallet;
   protected priceFeed?: PriceFeed;
 
-  constructor(options: any, private wallets: WalletConfig[]) {
+  constructor(
+    options: any,
+    private wallets: WalletConfig[],
+    priceFeedInstance?: PriceFeed,
+  ) {
     this.validateOptions(options);
     this.options = this.parseOptions(options);
 
@@ -94,15 +93,7 @@ export class ChainWalletManager {
     }
 
     this.logger = createLogger(this.options.logger);
-    const { priceFeedConfig } = this.options;
-
-    if (priceFeedConfig?.enabled) {
-      if (priceFeedConfig?.scheduled?.enabled) {
-        this.priceFeed = new ScheduledPriceFeed(priceFeedConfig, this.logger);
-      } else {
-        this.priceFeed = new OnDemandPriceFeed(priceFeedConfig, this.logger);
-      }
-    }
+    this.priceFeed = priceFeedInstance
 
     this.walletToolbox = createWalletToolbox(
       options.network,
@@ -274,9 +265,20 @@ export class ChainWalletManager {
       );
       this.priceFeed?.start();
     }
-    this.interval = setInterval(async () => {
-      await this.refreshBalances();
-    }, this.options.balancePollInterval);
+
+    if (this.options.walletBalanceConfig?.enabled) {
+      if (this.options.walletBalanceConfig?.scheduled?.enabled) {
+        this.interval = setInterval(async () => {
+          await this.refreshBalances();
+        }, this.options.walletBalanceConfig?.scheduled?.interval ?? this.options.balancePollInterval);
+      } else {
+        // no op: Don't poll balances, fetch on demand instead
+      }
+    } else {
+      this.interval = setInterval(async () => {
+        await this.refreshBalances();
+      }, this.options.balancePollInterval);
+    }
 
     if (this.options.rebalance.enabled) {
       this.logger.info(
@@ -419,4 +421,19 @@ export class ChainWalletManager {
       this.availableWalletsByChainName[chainName],
     );
   }
+
+  public getBlockHeight () {
+    return this.walletToolbox.getBlockHeight();
+  }
+  /** Pull balances on demand */
+  public async pullBalances () {
+    const balances = await this.walletToolbox.pullBalances();
+    return this.mapBalances(balances);
+  }
+
+  /** Pull balances on demand with block height */
+  public async pullBalancesAtBlockHeight(blockHeight: number) {
+    const balances = await this.walletToolbox.pullBalancesAtBlockHeight(blockHeight);
+    return this.mapBalances(balances);
+  }
 }