@ignitionfi/spl-stake-pool 1.1.21 → 1.1.23

package/dist/index.browser.esm.js

@@ -432,6 +432,8 @@ const MAX_VALIDATORS_TO_UPDATE = 4;
 const EPHEMERAL_STAKE_SEED_PREFIX = Buffer$1.from('ephemeral');
 // Seed used to derive transient stake accounts.
 const TRANSIENT_STAKE_SEED_PREFIX = Buffer$1.from('transient');
+// Seed for user stake account created during session withdrawal
+const USER_STAKE_SEED_PREFIX = Buffer$1.from('user_stake');
 // Minimum amount of staked SOL required in a validator stake account to allow
 // for merges without a mismatch on credits observed
 const MINIMUM_ACTIVE_STAKE = LAMPORTS_PER_SOL;
@@ -540,6 +542,19 @@ function findMetadataAddress(stakePoolMintAddress) {
     const [publicKey] = PublicKey.findProgramAddressSync([Buffer$1.from('metadata'), METADATA_PROGRAM_ID.toBuffer(), stakePoolMintAddress.toBuffer()], METADATA_PROGRAM_ID);
     return publicKey;
 }
+/**
+ * Generates the user stake account PDA for session-based withdrawals.
+ * The PDA is derived from the user's wallet and a unique seed.
+ */
+function findUserStakeProgramAddress(programId, userWallet, seed) {
+    const seedBN = typeof seed === 'number' ? new BN(seed) : seed;
+    const [publicKey] = PublicKey.findProgramAddressSync([
+        USER_STAKE_SEED_PREFIX,
+        userWallet.toBuffer(),
+        seedBN.toArrayLike(Buffer$1, 'le', 8),
+    ], programId);
+    return publicKey;
+}
 
 class BNLayout extends Layout {
     constructor(span, signed, property) {
@@ -1102,6 +1117,7 @@ const STAKE_POOL_INSTRUCTION_LAYOUTS = Object.freeze({
             BufferLayout.u8('instruction'),
             BufferLayout.ns64('poolTokensIn'),
             BufferLayout.ns64('minimumLamportsOut'),
+            BufferLayout.ns64('userStakeSeed'),
         ]),
     },
 });
@@ -1587,6 +1603,7 @@ class StakePoolInstruction {
         const data = encodeData(type, {
             poolTokensIn: params.poolTokensIn,
             minimumLamportsOut: params.minimumLamportsOut,
+            userStakeSeed: params.userStakeSeed,
         });
         const keys = [
             { pubkey: params.stakePool, isSigner: false, isWritable: true },
@@ -1603,6 +1620,8 @@ class StakePoolInstruction {
             { pubkey: params.tokenProgramId, isSigner: false, isWritable: false },
             { pubkey: StakeProgram.programId, isSigner: false, isWritable: false },
             { pubkey: params.programSigner, isSigner: false, isWritable: false },
+            { pubkey: params.payer, isSigner: true, isWritable: true },
+            { pubkey: SystemProgram.programId, isSigner: false, isWritable: false },
         ];
         return new TransactionInstruction({
             programId: params.programId,
@@ -2220,10 +2239,46 @@ async function withdrawWsolWithSession(connection, stakePoolAddress, signerOrSes
     };
 }
 /**
- * Creates instructions required to withdraw stake from a stake pool using a Fogo session.
- * The withdrawn stake account will be authorized to the user's wallet.
+ * Finds the next available seed for creating a user stake PDA.
+ * Scans from startSeed until an unused PDA is found.
+ *
+ * @param connection - Solana connection
+ * @param programId - The stake pool program ID
+ * @param userPubkey - User's wallet (used for PDA derivation)
+ * @param startSeed - Starting seed to search from (default: 0)
+ * @param maxSeed - Maximum seed to check before giving up (default: 1000)
+ * @returns The next available seed
+ * @throws Error if no available seed found within maxSeed
+ */
+async function findNextUserStakeSeed(connection, programId, userPubkey, startSeed = 0, maxSeed = 1000) {
+    for (let seed = startSeed; seed < startSeed + maxSeed; seed++) {
+        const pda = findUserStakeProgramAddress(programId, userPubkey, seed);
+        const account = await connection.getAccountInfo(pda);
+        if (!account) {
+            return seed;
+        }
+    }
+    throw new Error(`No available user stake seed found between ${startSeed} and ${startSeed + maxSeed - 1}`);
+}
+/**
+ * Withdraws stake from a stake pool using a Fogo session.
+ *
+ * The on-chain program creates stake account PDAs, so no pre-creation step is needed.
+ * The paymaster payer provides rent for the stake account PDAs.
+ *
+ * @param connection - Solana connection
+ * @param stakePoolAddress - The stake pool to withdraw from
+ * @param signerOrSession - The session signer public key
+ * @param userPubkey - User's wallet (used for PDA derivation and token ownership)
+ * @param payer - Payer for stake account rent (paymaster payer in session context)
+ * @param amount - Amount of pool tokens to withdraw
+ * @param userStakeSeedStart - Starting seed for user stake PDA derivation (default: 0)
+ * @param useReserve - Whether to withdraw from reserve (default: false)
+ * @param voteAccountAddress - Optional specific validator to withdraw from
+ * @param minimumLamportsOut - Minimum lamports to receive (slippage protection)
+ * @param validatorComparator - Optional comparator for validator selection
  */
-async function withdrawStakeWithSession(connection, stakePoolAddress, signerOrSession, userPubkey, amount, useReserve = false, voteAccountAddress, minimumLamportsOut = 0, payer, validatorComparator) {
+async function withdrawStakeWithSession(connection, stakePoolAddress, signerOrSession, userPubkey, payer, amount, userStakeSeedStart = 0, useReserve = false, voteAccountAddress, minimumLamportsOut = 0, validatorComparator) {
     const stakePoolAccount = await getStakePoolAccount(connection, stakePoolAddress);
     const stakePoolProgramId = getStakePoolProgramId(connection.rpcEndpoint);
     const stakePool = stakePoolAccount.account.data;
@@ -2273,8 +2328,8 @@ async function withdrawStakeWithSession(connection, stakePoolAddress, signerOrSe
         withdrawAccounts.push(...(await prepareWithdrawAccounts(connection, stakePool, stakePoolAddress, poolAmount, validatorComparator, poolTokenAccount.equals(stakePool.managerFeeAccount))));
     }
     const instructions = [];
-    const signers = [];
     const stakeAccountPubkeys = [];
+    const userStakeSeeds = [];
     // Max 5 accounts to prevent an error: "Transaction too large"
     const maxWithdrawAccounts = 5;
     let i = 0;
@@ -2282,41 +2337,36 @@ async function withdrawStakeWithSession(connection, stakePoolAddress, signerOrSe
         if (i >= maxWithdrawAccounts) {
             break;
         }
-        // Create a new stake account to receive the withdrawal
-        const stakeReceiver = Keypair.generate();
-        signers.push(stakeReceiver);
-        stakeAccountPubkeys.push(stakeReceiver.publicKey);
-        // Create the stake account that will receive the split stake
-        instructions.push(SystemProgram.createAccount({
-            fromPubkey: payer !== null && payer !== void 0 ? payer : userPubkey,
-            newAccountPubkey: stakeReceiver.publicKey,
-            lamports: stakeAccountRentExemption,
-            space: StakeProgram.space,
-            programId: StakeProgram.programId,
-        }));
-        // Add the withdraw stake with session instruction
+        // Derive the stake account PDA for this withdrawal
+        const userStakeSeed = userStakeSeedStart + i;
+        const stakeReceiverPubkey = findUserStakeProgramAddress(stakePoolProgramId, userPubkey, userStakeSeed);
+        stakeAccountPubkeys.push(stakeReceiverPubkey);
+        userStakeSeeds.push(userStakeSeed);
+        // The on-chain program will create the stake account PDA
         instructions.push(StakePoolInstruction.withdrawStakeWithSession({
             programId: stakePoolProgramId,
             stakePool: stakePoolAddress,
             validatorList: stakePool.validatorList,
             withdrawAuthority,
             stakeToSplit: withdrawAccount.stakeAddress,
-            stakeToReceive: stakeReceiver.publicKey,
+            stakeToReceive: stakeReceiverPubkey,
             sessionSigner: signerOrSession,
             burnFromPool: poolTokenAccount,
             managerFeeAccount: stakePool.managerFeeAccount,
             poolMint: stakePool.poolMint,
             tokenProgramId: stakePool.tokenProgramId,
             programSigner,
+            payer,
             poolTokensIn: withdrawAccount.poolAmount.toNumber(),
-            minimumLamportsOut: solToLamports(minimumLamportsOut),
+            minimumLamportsOut,
+            userStakeSeed,
         }));
         i++;
     }
     return {
         instructions,
-        signers,
         stakeAccountPubkeys,
+        userStakeSeeds,
     };
 }
 async function addValidatorToPool(connection, stakePoolAddress, validatorVote, seed) {
@@ -2681,5 +2731,5 @@ async function updatePoolTokenMetadata(connection, stakePoolAddress, name, symbo
     };
 }
 
-export { DEVNET_STAKE_POOL_PROGRAM_ID, STAKE_POOL_INSTRUCTION_LAYOUTS, STAKE_POOL_PROGRAM_ID, StakePoolInstruction, StakePoolLayout, ValidatorListLayout, ValidatorStakeInfoLayout, addValidatorToPool, createPoolTokenMetadata, decreaseValidatorStake, depositSol, depositStake, depositWsolWithSession, getStakeAccount, getStakePoolAccount, getStakePoolAccounts, getStakePoolProgramId, increaseValidatorStake, removeValidatorFromPool, stakePoolInfo, tokenMetadataLayout, updatePoolTokenMetadata, updateStakePool, withdrawSol, withdrawStake, withdrawStakeWithSession, withdrawWsolWithSession };
+export { DEVNET_STAKE_POOL_PROGRAM_ID, STAKE_POOL_INSTRUCTION_LAYOUTS, STAKE_POOL_PROGRAM_ID, StakePoolInstruction, StakePoolLayout, ValidatorListLayout, ValidatorStakeInfoLayout, addValidatorToPool, createPoolTokenMetadata, decreaseValidatorStake, depositSol, depositStake, depositWsolWithSession, findEphemeralStakeProgramAddress, findNextUserStakeSeed, findStakeProgramAddress, findTransientStakeProgramAddress, findUserStakeProgramAddress, findWithdrawAuthorityProgramAddress, findWsolTransientProgramAddress, getStakeAccount, getStakePoolAccount, getStakePoolAccounts, getStakePoolProgramId, increaseValidatorStake, removeValidatorFromPool, stakePoolInfo, tokenMetadataLayout, updatePoolTokenMetadata, updateStakePool, withdrawSol, withdrawStake, withdrawStakeWithSession, withdrawWsolWithSession };
 //# sourceMappingURL=index.browser.esm.js.map