wip

Author: SwenSchaeferjohann

package-lock.json

@@ -35,8 +35,8 @@
   },
   "homepage": "https://github.com/Lightprotocol/example-nodejs-client#readme",
   "dependencies": {
-    "@lightprotocol/compressed-token": "0.20.9",
-    "@lightprotocol/stateless.js": "0.20.9",
+    "@lightprotocol/compressed-token": "0.21.0",
+    "@lightprotocol/stateless.js": "0.21.0",
     "@solana/spl-token": "0.3.9",
     "@solana/spl-token-metadata": "^0.1.6",
     "@solana/web3.js": "^1.98.0",

package.json

@@ -3,6 +3,7 @@ import {
   createRpc,
   LightSystemProgram,
   Rpc,
+  selectStateTreeInfo,
 } from "@lightprotocol/stateless.js";
 import { PAYER_KEYPAIR, RPC_ENDPOINT } from "../constants";
 import { randomBytes } from "crypto";
@@ -26,7 +27,9 @@ const trees = [
 (async () => {
   try {
     while (true) {
-      const pseudoRandomTree = trees[Math.floor(Math.random() * trees.length)];
+      const stateTreeInfos = await connection.getStateTreeInfos();
+      const treeInfo = selectStateTreeInfo(stateTreeInfos);
+
       // Create account with random address
       const randomSeed = new Uint8Array(randomBytes(32));
       const txId = await createAccount(
@@ -35,8 +38,7 @@ const trees = [
         [randomSeed],
         LightSystemProgram.programId,
         undefined,
-        undefined,
-        new PublicKey(pseudoRandomTree)
+        treeInfo
       );
       console.log(
         `Compressed Account Creation Success. Transaction Signature:`,

src/internal/create-account-pulse.ts

@@ -6,6 +6,7 @@ import {
   LightSystemProgram,
   pickRandomTreeAndQueue,
   Rpc,
+  selectStateTreeInfo,
 } from "@lightprotocol/stateless.js";
 import { PAYER_KEYPAIR, RPC_ENDPOINT } from "../constants";
 import { randomBytes } from "crypto";
@@ -16,16 +17,16 @@ const connection: Rpc = createRpc(RPC_ENDPOINT);
 (async () => {
   try {
     while (true) {
-      const activeStateTrees = await connection.getCachedActiveStateTreeInfo();
+      const stateTreeInfos = await connection.getStateTreeInfos();
+      const treeInfo = selectStateTreeInfo(stateTreeInfos);
 
-      const { tree, queue } = pickRandomTreeAndQueue(activeStateTrees);
-      console.log("Picked output state tree:", tree.toBase58());
+      console.log("Picked output state tree:", treeInfo.tree.toBase58());
       const compressedTxId = await compress(
         connection,
         fromKeypair,
         bn(10),
         fromKeypair.publicKey,
-        tree
+        treeInfo
       );
       console.log("Compressed TxId", compressedTxId);
       // Creat account with random address
@@ -36,8 +37,7 @@ const connection: Rpc = createRpc(RPC_ENDPOINT);
         10,
         LightSystemProgram.programId,
         undefined,
-        undefined,
-        tree
+        treeInfo
       );
       console.log(
         `Compressed Account Creation Success. Transaction Signature:`,

src/internal/create-account-with-lamports-pulse.ts

@@ -3,7 +3,7 @@ import {
   buildAndSignTx,
   createRpc,
   dedupeSigner,
-  getLightStateTreeInfo,
+  getAllStateTreeInfos,
   sendAndConfirmTx,
 } from "@lightprotocol/stateless.js";
 import {
@@ -16,7 +16,7 @@ import { AddressLookupTableProgram, Keypair, PublicKey } from "@solana/web3.js";
 
 const payer = PAYER_KEYPAIR;
 const authority = LUT_DEVNET_AUTHORITY_KEYPAIR;
-const connection: Rpc = createRpc(RPC_ENDPOINT, RPC_ENDPOINT, RPC_ENDPOINT);
+const connection: Rpc = createRpc(RPC_ENDPOINT);
 
 (async () => {
   // const address = new PublicKey("9NYFyEqPkyXUhkerbGHXUXkvb4qpzeEdHuGpgbgpH1NJ");
@@ -36,10 +36,14 @@ const connection: Rpc = createRpc(RPC_ENDPOINT, RPC_ENDPOINT, RPC_ENDPOINT);
     "5dhaJLBjnVBQFErr8oiCJmcVsx3Zj6xDekGB2zULPsnP"
   );
 
-  const info = await getLightStateTreeInfo({
+  const info = await getAllStateTreeInfos({
     connection,
-    stateTreeLookupTableAddress: stateTreeLookupTableDevnet,
-    nullifyTableAddress: nullifiedStateTreeLookupTableDevnet,
+    stateTreeLUTPairs: [
+      {
+        stateTreeLookupTable: stateTreeLookupTableDevnet,
+        nullifyLookupTable: nullifiedStateTreeLookupTableDevnet,
+      },
+    ],
   });
 
   await extend(

src/internal/extend-lut.ts

@@ -8,7 +8,7 @@ import { PAYER_KEYPAIR, RPC_ENDPOINT, AUTHORITY_KEYPAIR } from "../constants";
 import { PublicKey } from "@solana/web3.js";
 
 const payer = PAYER_KEYPAIR;
-const connection: Rpc = createRpc(RPC_ENDPOINT, RPC_ENDPOINT, RPC_ENDPOINT);
+const connection: Rpc = createRpc(RPC_ENDPOINT);
 
 const stateTreeAddresses = [
   "smt1NamzXdq4AMqS2fS2F1i5KTYPZRhoHgWx38d8WsT",

src/internal/extend-state-lut.ts

@@ -4,29 +4,33 @@ import {
   createRpc,
   pickRandomTreeAndQueue,
   Rpc,
+  selectStateTreeInfo,
   sleep,
   transfer,
 } from "@lightprotocol/stateless.js";
 import { PAYER_KEYPAIR, RPC_ENDPOINT } from "../constants";
 import { PublicKey } from "@solana/web3.js";
+import {
+  getTokenPoolInfos,
+  selectTokenPoolInfo,
+} from "@lightprotocol/compressed-token";
 
 const fromKeypair = PAYER_KEYPAIR;
 const connection: Rpc = createRpc(RPC_ENDPOINT, RPC_ENDPOINT);
 
 const batchSize = 10;
 (async () => {
   try {
-    const activeStateTrees = await connection.getCachedActiveStateTreeInfo();
-
-    const { tree, queue } = pickRandomTreeAndQueue(activeStateTrees);
-    console.log("Picked output state tree:", tree.toBase58());
+    const stateTreeInfos = await connection.getStateTreeInfos();
+    const treeInfo = selectStateTreeInfo(stateTreeInfos);
+    console.log("Picked output state tree:", treeInfo.tree.toBase58());
 
     const compressedTxId = await compress(
       connection,
       fromKeypair,
       bn(1e5),
       fromKeypair.publicKey,
-      tree
+      treeInfo
     );
     while (true) {
       console.log("Compressed TxId", compressedTxId);
@@ -41,7 +45,7 @@ const batchSize = 10;
             1,
             fromKeypair,
             fromKeypair.publicKey,
-            tree,
+            treeInfo,
             {
               skipPreflight: false,
             }

src/internal/transfer-pulse.ts

@@ -3,8 +3,9 @@ import {
   calculateComputeUnitPrice,
   createRpc,
   Rpc,
+  selectStateTreeInfo,
 } from "@lightprotocol/stateless.js";
-import { createMint } from "@lightprotocol/compressed-token";
+import { createMint, getTokenPoolInfos } from "@lightprotocol/compressed-token";
 import { getOrCreateAssociatedTokenAccount, mintTo } from "@solana/spl-token";
 import { createAirdropInstructions } from "./create-instructions";
 import { BatchResultType, signAndSendAirdropBatches } from "./sign-and-send";
@@ -75,15 +76,16 @@ const recipients = [
   );
   console.log(`mint-to success! txId: ${mintToTxId}`);
 
-  const activeStateTrees = await connection.getCachedActiveStateTreeInfo();
-
+  const stateTreeInfos = await connection.getStateTreeInfos();
+  const tokenPoolInfos = await getTokenPoolInfos(connection, mint);
   const instructionBatches = await createAirdropInstructions({
     amount: 1e6,
     recipients,
     payer: PAYER.publicKey,
     sourceTokenAccount: ata.address,
     mint,
-    stateTrees: activeStateTrees,
+    stateTreeInfos,
+    tokenPoolInfos,
     computeUnitPrice: calculateComputeUnitPrice(10_000, 500_000),
   });
 

src/scripts/airdrop/airdrop.ts

@@ -1,8 +1,14 @@
-import { CompressedTokenProgram } from "@lightprotocol/compressed-token";
+import {
+  CompressedTokenProgram,
+  selectTokenPoolInfo,
+  TokenPoolInfo,
+} from "@lightprotocol/compressed-token";
 import {
   bn,
   ActiveTreeBundle,
   pickRandomTreeAndQueue,
+  selectStateTreeInfo,
+  StateTreeInfo,
 } from "@lightprotocol/stateless.js";
 import {
   ComputeBudgetProgram,
@@ -16,7 +22,8 @@ interface CreateAirdropInstructionsParams {
   payer: PublicKey;
   sourceTokenAccount: PublicKey;
   mint: PublicKey;
-  stateTrees: ActiveTreeBundle[];
+  stateTreeInfos: StateTreeInfo[];
+  tokenPoolInfos: TokenPoolInfo[];
   maxRecipientsPerInstruction?: number;
   maxInstructionsPerTransaction?: number;
   computeUnitLimit?: number;
@@ -31,7 +38,8 @@ export async function createAirdropInstructions({
   payer,
   sourceTokenAccount,
   mint,
-  stateTrees,
+  stateTreeInfos,
+  tokenPoolInfos,
   maxRecipientsPerInstruction = 5,
   maxInstructionsPerTransaction = 3,
   computeUnitLimit = 500_000,
@@ -59,7 +67,8 @@ export async function createAirdropInstructions({
       );
     }
 
-    const { tree } = pickRandomTreeAndQueue(stateTrees);
+    const treeInfo = selectStateTreeInfo(stateTreeInfos);
+    const tokenPoolInfo = selectTokenPoolInfo(tokenPoolInfos);
 
     for (let j = 0; j < maxInstructionsPerTransaction; j++) {
       const startIdx = i + j * maxRecipientsPerInstruction;
@@ -77,7 +86,8 @@ export async function createAirdropInstructions({
         toAddress: recipientBatch,
         amount: recipientBatch.map(() => amountBn),
         mint,
-        outputStateTree: tree,
+        outputStateTreeInfo: treeInfo,
+        tokenPoolInfo,
       });
 
       instructions.push(compressIx);

src/scripts/airdrop/create-instructions.ts

@@ -1,6 +1,10 @@
 import * as web3 from "@solana/web3.js";
 import { RPC_ENDPOINT, PAYER_KEYPAIR, MINT_ADDRESS } from "../constants";
-import { CompressedTokenProgram } from "@lightprotocol/compressed-token";
+import {
+  CompressedTokenProgram,
+  getTokenPoolInfos,
+  selectTokenPoolInfo,
+} from "@lightprotocol/compressed-token";
 
 import {
   bn,
@@ -9,6 +13,7 @@ import {
   dedupeSigner,
   pickRandomTreeAndQueue,
   Rpc,
+  selectStateTreeInfo,
   sendAndConfirmTx,
 } from "@lightprotocol/stateless.js";
 import * as splToken from "@solana/spl-token";
@@ -19,10 +24,13 @@ import * as splToken from "@solana/spl-token";
     const mintAddress = MINT_ADDRESS;
     const payer = PAYER_KEYPAIR;
 
-    const activeStateTrees = await connection.getCachedActiveStateTreeInfo();
+    const activeStateTrees = await connection.getStateTreeInfos();
+    const treeInfo = selectStateTreeInfo(activeStateTrees);
+    console.log("Picked output state tree:", treeInfo.tree.toBase58());
 
-    const { tree } = pickRandomTreeAndQueue(activeStateTrees);
-    console.log("Picked output state tree:", tree.toBase58());
+    const tokenPoolInfo = selectTokenPoolInfo(
+      await getTokenPoolInfos(connection, mintAddress)
+    );
 
     // Get the source token account for the mint address
     const sourceTokenAccount = await splToken.getOrCreateAssociatedTokenAccount(
@@ -78,7 +86,8 @@ import * as splToken from "@solana/spl-token";
         toAddress: recipientBatch,
         amount: recipientBatch.map(() => amount),
         mint: mintAddress,
-        outputStateTree: tree,
+        outputStateTreeInfo: treeInfo,
+        tokenPoolInfo,
       });
       instructions.push(compressIx);
       i += maxRecipientsPerInstruction;