Multi trees (#15)

* pick pseudorandom

* batch transfer

* for mainnet

* 0.19.0

* add airdrop.ts

Author: SwenSchaeferjohann

.env.example

@@ -2,3 +2,5 @@
 RPC_ENDPOINT="https://devnet.helius-rpc.com?api-key=<YOUR_API_KEY>"
 PAYER_KEYPAIR="bs58encoded string of secret key"
 MINT_ADDRESS="D3tw8PvgsJqqFBUKcJxiLAe78wbvu3jzospbtbTt9u18"
+AUTHORITY_KEYPAIR="bs58encoded string of secret key"
+HELIUS_API_KEY="your helius api key"

package-lock.json

@@ -9,6 +9,7 @@
     "token": "ts-node src/scripts/token.ts",
     "connection": "ts-node src/scripts/connection.ts",
     "lookup-table": "ts-node src/scripts/lookup-table.ts",
+    "state-lut": "ts-node src/internal/state-lut.ts",
     "batch-compress": "ts-node src/scripts/batch-compress.ts",
     "compress-with-delegate": "ts-node src/scripts/compress-with-delegate.ts",
     "mint-spl": "ts-node src/scripts/mint-spl.ts",
@@ -17,6 +18,7 @@
     "pulse": "ts-node src/internal/transfer-pulse.ts",
     "create-account-pulse": "ts-node src/internal/create-account-pulse.ts",
     "create-account-with-lamports-pulse": "ts-node src/internal/create-account-with-lamports-pulse.ts",
+    "airdrop": "ts-node src/scripts/airdrop/airdrop.ts",
     "test": "npm run lamports && npm run token && npm run connection && npm run batch-compress && npm run mint-spl && npm run mint-spl-22 && npm run mint-spl-22-manual"
   },
   "repository": {
@@ -31,15 +33,19 @@
   },
   "homepage": "https://github.com/Lightprotocol/example-nodejs-client#readme",
   "dependencies": {
-    "@lightprotocol/compressed-token": "0.18.0",
-    "@lightprotocol/stateless.js": "0.18.0",
+    "@lightprotocol/compressed-token": "0.20.3",
+    "@lightprotocol/stateless.js": "0.20.3",
     "@solana/spl-token": "0.3.9",
     "@solana/spl-token-metadata": "^0.1.6",
     "@solana/web3.js": "^1.98.0",
     "@types/node": "^20.12.7",
+    "bn.js": "^5.2.1",
     "bs58": "^6.0.0",
     "dotenv": "^16.4.5",
     "ts-node": "^10.9.2",
     "typescript": "^5.4.5"
+  },
+  "devDependencies": {
+    "@types/bn.js": "^5.1.6"
   }
 }

package.json

@@ -7,6 +7,23 @@ export const RPC_ENDPOINT = process.env.RPC_ENDPOINT;
 export const PAYER_KEYPAIR = Keypair.fromSecretKey(
   bs58.decode(process.env.PAYER_KEYPAIR!)
 );
+
+export const AUTHORITY_KEYPAIR = process.env.LUT_AUTHORITY_KEYPAIR
+  ? Keypair.fromSecretKey(
+      Uint8Array.from(JSON.parse(process.env.LUT_AUTHORITY_KEYPAIR))
+    )
+  : undefined;
+
+export const LUT_MAINNET_AUTHORITY_KEYPAIR = process.env.LUT_AUTHORITY_KEYPAIR
+  ? Keypair.fromSecretKey(
+      Uint8Array.from(JSON.parse(process.env.LUT_AUTHORITY_KEYPAIR))
+    )
+  : undefined;
+
+export const LUT_DEVNET_AUTHORITY_KEYPAIR = process.env.LUT_DEVNET_AUTH_KEYPAIR
+  ? Keypair.fromSecretKey(bs58.decode(process.env.LUT_DEVNET_AUTH_KEYPAIR))
+  : undefined;
+
 export const MINT_ADDRESS = new PublicKey(process.env.MINT_ADDRESS!);
 if (!RPC_ENDPOINT) throw new Error("Please set RPC_ENDPOINT in .env");
 if (!PAYER_KEYPAIR)

src/constants.ts

@@ -6,21 +6,37 @@ import {
 } from "@lightprotocol/stateless.js";
 import { PAYER_KEYPAIR, RPC_ENDPOINT } from "../constants";
 import { randomBytes } from "crypto";
+import { PublicKey } from "@solana/web3.js";
 
 const fromKeypair = PAYER_KEYPAIR;
 
 const connection: Rpc = createRpc(RPC_ENDPOINT, RPC_ENDPOINT);
 
+const trees = [
+  "smt2rJAFdyJJupwMKAqTNAJwvjhmiZ4JYGZmbVRw1Ho",
+  "smt3AFtReRGVcrP11D6bSLEaKdUmrGfaTNowMVccJeu",
+  "smt4vjXvdjDFzvRMUxwTWnSy4c7cKkMaHuPrGsdDH7V",
+  "smt5uPaQT9n6b1qAkgyonmzRxtuazA53Rddwntqistc",
+  "smt6ukQDSPPYHSshQovmiRUjG9jGFq2hW9vgrDFk5Yz",
+  "smt7onMFkvi3RbyhQCMajudYQkB1afAFt9CDXBQTLz6",
+  "smt8TYxNy8SuhAdKJ8CeLtDkr2w6dgDmdz5ruiDw9Y9",
+  "smt9ReAYRF5eFjTd5gBJMn5aKwNRcmp3ub2CQr2vW7j",
+  "smtAvYA5UbTRyKAkAj5kHs1CmrA42t6WkVLi4c6mA1f",
+];
 (async () => {
   try {
     while (true) {
+      const pseudoRandomTree = trees[Math.floor(Math.random() * trees.length)];
       // Create account with random address
       const randomSeed = new Uint8Array(randomBytes(32));
       const txId = await createAccount(
         connection,
         fromKeypair,
         [randomSeed],
-        LightSystemProgram.programId
+        LightSystemProgram.programId,
+        undefined,
+        undefined,
+        new PublicKey(pseudoRandomTree)
       );
       console.log(
         `Compressed Account Creation Success. Transaction Signature:`,

src/internal/create-account-pulse.ts

@@ -1,25 +1,43 @@
 import {
+  bn,
+  compress,
   createAccountWithLamports,
   createRpc,
   LightSystemProgram,
+  pickRandomTreeAndQueue,
   Rpc,
 } from "@lightprotocol/stateless.js";
 import { PAYER_KEYPAIR, RPC_ENDPOINT } from "../constants";
 import { randomBytes } from "crypto";
 
 const fromKeypair = PAYER_KEYPAIR;
-const connection: Rpc = createRpc(RPC_ENDPOINT, RPC_ENDPOINT);
+const connection: Rpc = createRpc(RPC_ENDPOINT);
 
 (async () => {
   try {
     while (true) {
+      const activeStateTrees = await connection.getCachedActiveStateTreeInfo();
+
+      const { tree, queue } = pickRandomTreeAndQueue(activeStateTrees);
+      console.log("Picked output state tree:", tree.toBase58());
+      const compressedTxId = await compress(
+        connection,
+        fromKeypair,
+        bn(10),
+        fromKeypair.publicKey,
+        tree
+      );
+      console.log("Compressed TxId", compressedTxId);
       // Creat account with random address
       const txId = await createAccountWithLamports(
         connection,
         fromKeypair,
         [randomBytes(32)],
         10,
-        LightSystemProgram.programId
+        LightSystemProgram.programId,
+        undefined,
+        undefined,
+        tree
       );
       console.log(
         `Compressed Account Creation Success. Transaction Signature:`,

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

@@ -0,0 +1,75 @@
+import {
+  Rpc,
+  buildAndSignTx,
+  createRpc,
+  dedupeSigner,
+  getLightStateTreeInfo,
+  sendAndConfirmTx,
+} from "@lightprotocol/stateless.js";
+import {
+  LUT_MAINNET_AUTHORITY_KEYPAIR,
+  LUT_DEVNET_AUTHORITY_KEYPAIR,
+  PAYER_KEYPAIR,
+  RPC_ENDPOINT,
+} from "../constants";
+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);
+
+(async () => {
+  // const address = new PublicKey("9NYFyEqPkyXUhkerbGHXUXkvb4qpzeEdHuGpgbgpH1NJ");
+  const address = new PublicKey("qAJZMgnQJ8G6vA3WRcjD9Jan1wtKkaCFWLWskxJrR5V");
+
+  // const stateTreeLookupTableMainnet = new PublicKey(
+  //   "7i86eQs3GSqHjN47WdWLTCGMW6gde1q96G2EVnUyK2st"
+  // );
+  // const nullifiedStateTreeLookupTableMainnet = new PublicKey(
+  //   "H9QD4u1fG7KmkAzn2tDXhheushxFe1EcrjGGyEFXeMqT"
+  // );
+
+  const stateTreeLookupTableDevnet = new PublicKey(
+    "8n8rH2bFRVA6cSGNDpgqcKHCndbFCT1bXxAQG89ejVsh"
+  );
+  const nullifiedStateTreeLookupTableDevnet = new PublicKey(
+    "5dhaJLBjnVBQFErr8oiCJmcVsx3Zj6xDekGB2zULPsnP"
+  );
+
+  const info = await getLightStateTreeInfo({
+    connection,
+    stateTreeLookupTableAddress: stateTreeLookupTableDevnet,
+    nullifyTableAddress: nullifiedStateTreeLookupTableDevnet,
+  });
+
+  await extend(
+    payer,
+    authority,
+    address,
+    info.flatMap((i) => [i.tree])
+  );
+})();
+
+async function extend(
+  payer: Keypair,
+  authority: Keypair,
+  lookupTableAddress: PublicKey,
+  addresses: PublicKey[]
+) {
+  const extendInstruction = AddressLookupTableProgram.extendLookupTable({
+    payer: payer.publicKey,
+    authority: authority.publicKey,
+    lookupTable: lookupTableAddress,
+    addresses,
+  });
+  const bhash = await connection.getLatestBlockhash();
+  const tx = await buildAndSignTx(
+    [extendInstruction],
+    payer,
+    bhash.blockhash,
+    dedupeSigner(payer, [authority])
+  );
+  await sendAndConfirmTx(connection, tx);
+
+  console.log("extended lookup table:", lookupTableAddress.toBase58());
+}