@lightprotocol/compressed-token 0.7.0 → 0.9.0

package/dist/es/browser/index.js

@@ -1,6 +1,6 @@
 import { getIndexOrAdd, bn, padOutputStateMerkleTrees, useWallet, confirmConfig, defaultStaticAccountsStruct, toArray, LightSystemProgram, defaultTestStateTreeAccounts, sumUpLamports, validateSufficientBalance, validateSameOwner, dedupeSigner, buildAndSignTx, sendAndConfirmTx } from '@lightprotocol/stateless.js';
 import { PublicKey, SystemProgram, TransactionInstruction, Transaction, sendAndConfirmTransaction, Keypair, Connection, ComputeBudgetProgram } from '@solana/web3.js';
-import { AnchorProvider, setProvider, Program } from '@coral-xyz/anchor';
+import { AnchorProvider, setProvider, Program, BN } from '@coral-xyz/anchor';
 
 const IDL = {
     version: '0.5.0',
@@ -6859,15 +6859,34 @@ class CompressedTokenProgram {
      */
     static async compress(params) {
         const { payer, owner, source, toAddress, mint, outputStateTree } = params;
-        const amount = bn(params.amount);
-        const tokenTransferOutputs = [
-            {
-                owner: toAddress,
-                amount,
-                lamports: bn(0),
-                tlv: null,
-            },
-        ];
+        if (Array.isArray(params.amount) !== Array.isArray(params.toAddress)) {
+            throw new Error('Both amount and toAddress must be arrays or both must be single values');
+        }
+        let tokenTransferOutputs;
+        if (Array.isArray(params.amount) && Array.isArray(params.toAddress)) {
+            if (params.amount.length !== params.toAddress.length) {
+                throw new Error('Amount and toAddress arrays must have the same length');
+            }
+            tokenTransferOutputs = params.amount.map((amt, index) => {
+                const amount = bn(amt);
+                return {
+                    owner: params.toAddress[index],
+                    amount,
+                    lamports: bn(0),
+                    tlv: null,
+                };
+            });
+        }
+        else {
+            tokenTransferOutputs = [
+                {
+                    owner: toAddress,
+                    amount: bn(params.amount),
+                    lamports: bn(0),
+                    tlv: null,
+                },
+            ];
+        }
         const { inputTokenDataWithContext, packedOutputTokenData, remainingAccountMetas, } = packCompressedTokenAccounts({
             inputCompressedTokenAccounts: [],
             outputStateTrees: outputStateTree,
@@ -6880,7 +6899,11 @@ class CompressedTokenProgram {
             delegatedTransfer: null, // TODO: implement
             inputTokenDataWithContext,
             outputCompressedAccounts: packedOutputTokenData,
-            compressOrDecompressAmount: amount,
+            compressOrDecompressAmount: Array.isArray(params.amount)
+                ? params.amount
+                    .map(amt => new BN(amt))
+                    .reduce((sum, amt) => sum.add(amt), new BN(0))
+                : new BN(params.amount),
             isCompress: true,
             cpiContext: null,
             lamportsChangeAccountMerkleTreeIndex: null,
@@ -7012,7 +7035,6 @@ async function approveAndMintTo(rpc, payer, mint, destination, authority, amount
  * @return Signature of the confirmed transaction
  */
 async function compress(rpc, payer, mint, amount, owner, sourceTokenAccount, toAddress, merkleTree, confirmOptions) {
-    amount = bn(amount);
     const compressIx = await CompressedTokenProgram.compress({
         payer: payer.publicKey,
         owner: owner.publicKey,
@@ -7058,7 +7080,7 @@ merkleTree, confirmOptions) {
     const compressedTokenAccounts = await rpc.getCompressedTokenAccountsByOwner(owner.publicKey, {
         mint,
     });
-    const [inputAccounts] = selectMinCompressedTokenAccountsForTransfer(compressedTokenAccounts, amount);
+    const [inputAccounts] = selectMinCompressedTokenAccountsForTransfer(compressedTokenAccounts.items, amount);
     const proof = await rpc.getValidityProof(inputAccounts.map(account => bn(account.compressedAccount.hash)));
     const ix = await CompressedTokenProgram.transfer({
         payer: payer.publicKey,
@@ -7130,7 +7152,7 @@ merkleTree, confirmOptions) {
         mint,
     });
     /// TODO: consider using a different selection algorithm
-    const [inputAccounts] = selectMinCompressedTokenAccountsForTransfer(compressedTokenAccounts, amount);
+    const [inputAccounts] = selectMinCompressedTokenAccountsForTransfer(compressedTokenAccounts.items, amount);
     const proof = await rpc.getValidityProof(inputAccounts.map(account => bn(account.compressedAccount.hash)));
     const ix = await CompressedTokenProgram.decompress({
         payer: payer.publicKey,