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

package/dist/index.mjs

@@ -1,6 +1,6 @@
 import { Program, AnchorProvider, BN } from '@coral-xyz/anchor';
 import { getStakePoolAccount, STAKE_POOL_PROGRAM_ID, StakePoolInstruction } from '@solana/spl-stake-pool';
-import { TOKEN_PROGRAM_ID, ASSOCIATED_TOKEN_PROGRAM_ID, getAssociatedTokenAddressSync } from '@solana/spl-token';
+import { TOKEN_PROGRAM_ID, ASSOCIATED_TOKEN_PROGRAM_ID, getAssociatedTokenAddressSync, ACCOUNT_SIZE, AccountLayout } from '@solana/spl-token';
 import { PublicKey, Transaction, SystemProgram, LAMPORTS_PER_SOL } from '@solana/web3.js';
 import bs58 from 'bs58';
 
@@ -6872,14 +6872,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;
@@ -6887,7 +6890,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 */;
@@ -6969,6 +6972,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) => {
@@ -6987,40 +6997,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 PublicKey(vote));
           return 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 !== ACCOUNT_SIZE) {
+              continue;
             }
             try {
-              const balance = await this.connection.getTokenAccountBalance(tokenAccounts[accountInfos.indexOf(accountInfo)]);
-              return Number(balance.value.uiAmount ?? 0) * LAMPORTS_PER_SOL;
+              const acc = 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;
   }