@jpool/bond-sdk 0.9.0-next.13 → 0.9.0-next.14

package/dist/index.d.mts

@@ -1911,7 +1911,7 @@ declare class JBondClient {
     /**
      * Get all bond states with total collected collateral
      */
-    getAllBondStates(bondType: BondType): Promise<Array<[BondState, BondStateStats]>>;
+    getAllBondStates(bondType: BondType, session_status?: SessionStatus): Promise<Array<[BondState, BondStateStats]>>;
     /**
      * Get session status for bond state
      */

package/dist/index.d.ts

@@ -1911,7 +1911,7 @@ declare class JBondClient {
     /**
      * Get all bond states with total collected collateral
      */
-    getAllBondStates(bondType: BondType): Promise<Array<[BondState, BondStateStats]>>;
+    getAllBondStates(bondType: BondType, session_status?: SessionStatus): Promise<Array<[BondState, BondStateStats]>>;
     /**
      * Get session status for bond state
      */

package/dist/index.js

@@ -6908,14 +6908,17 @@ var JBondClient = class _JBondClient {
   /**
    * Get all bond states with total collected collateral
    */
-  async getAllBondStates(bondType) {
+  async getAllBondStates(bondType, session_status) {
     const bondStates = [];
     const bondStateAccounts = await this.program.account.bondState.all();
     for (const { account: state } of bondStateAccounts) {
       if (!sameVariant(state.bondType, bondType)) {
         continue;
       }
-      const bondStateStats = await this.getBondStateStats(state.bondType, state.name);
+      if (session_status !== void 0 && this.getBondStateSessionStatus(state) !== session_status) {
+        continue;
+      }
+      const bondStateStats = await this.getBondStateStats(state);
       bondStates.push([state, bondStateStats]);
     }
     return bondStates;
@@ -6923,7 +6926,7 @@ var JBondClient = class _JBondClient {
   /**
    * Get session status for bond state
    */
-  async getBondStateSessionStatus(bondState) {
+  getBondStateSessionStatus(bondState) {
     const now = Math.floor(Date.now() / 1e3);
     if (bondState.sessionFinishTs.toNumber() > now) {
       return 0 /* Live */;
@@ -7005,6 +7008,13 @@ var JBondClient = class _JBondClient {
   async getBondStateTotalCollected(bondType, bondName, votes) {
     const bondStateData = await this.getBondState(bondType, bondName);
     const bondStateCollateralType = bondStateData.collateralType;
+    const chunk = (arr, n = 100) => {
+      const res = [];
+      for (let i = 0; i < arr.length; i += n) {
+        res.push(arr.slice(i, i + n));
+      }
+      return res;
+    };
     return await matchVariant(bondStateCollateralType, {
       native: async () => {
         const addresses = votes.map((vote) => {
@@ -7023,40 +7033,49 @@ var JBondClient = class _JBondClient {
         );
         return balances.reduce((sum, v) => sum + v, 0);
       },
+      // TODO
       token: async (mint) => {
         const tokenAccounts = votes.map((vote) => {
           const [validatorBondAddress] = this.pda.validatorBond(bondType, bondName, new web3_js.PublicKey(vote));
           return splToken.getAssociatedTokenAddressSync(mint, validatorBondAddress, true);
         });
-        const accountInfos = await this.connection.getMultipleAccountsInfo(tokenAccounts);
-        const balances = await Promise.all(
-          accountInfos.map(async (accountInfo) => {
-            if (!accountInfo) {
-              return 0;
+        if (tokenAccounts.length === 0) {
+          return 0;
+        }
+        let total = 0n;
+        for (const part of chunk(tokenAccounts, 100)) {
+          const infos = await this.connection.getMultipleAccountsInfo(part).catch(() => []);
+          for (let i = 0; i < infos.length; i++) {
+            const info = infos[i];
+            if (!info) {
+              continue;
+            }
+            if (info.data.length !== splToken.ACCOUNT_SIZE) {
+              continue;
             }
             try {
-              const balance = await this.connection.getTokenAccountBalance(tokenAccounts[accountInfos.indexOf(accountInfo)]);
-              return Number(balance.value.uiAmount ?? 0) * web3_js.LAMPORTS_PER_SOL;
+              const acc = splToken.AccountLayout.decode(info.data);
+              const raw = BigInt(acc.amount.toString());
+              total += raw;
             } catch {
-              return 0;
             }
-          })
-        );
-        return balances.reduce((sum, v) => sum + v, 0);
+          }
+        }
+        const asNumber = Number(total);
+        return Number.isFinite(asNumber) ? asNumber : Number.MAX_SAFE_INTEGER;
       }
     });
   }
-  async getBondStateStats(bondType, bondName) {
-    const bondState = await this.getBondState(bondType, bondName);
-    const validatorBonds = await this.getValidatorBondsByState(bondState.bondType, bondState.name);
+  async getBondStateStats(state) {
+    const validatorBonds = await this.getValidatorBondsByState(state.bondType, state.name);
     const totalCollected = await this.getBondStateTotalCollected(
-      bondType,
-      bondName,
+      state.bondType,
+      state.name,
       validatorBonds.map((vb) => vb.voteAccount)
     );
     const bondStateStats = {
       totalCollected,
-      status: await this.getBondStateSessionStatus(bondState)
+      status: await this.getBondStateSessionStatus(state)
     };
     return bondStateStats;
   }