@ignitionfi/spl-stake-pool 1.1.22 → 1.1.24

package/dist/index.d.ts

@@ -6,6 +6,7 @@ export { DEVNET_STAKE_POOL_PROGRAM_ID, STAKE_POOL_PROGRAM_ID, } from './constant
 export * from './instructions';
 export type { AccountType, StakePool, ValidatorList, ValidatorStakeInfo, } from './layouts';
 export { StakePoolLayout, ValidatorListLayout, ValidatorStakeInfoLayout, } from './layouts';
+export { findEphemeralStakeProgramAddress, findStakeProgramAddress, findTransientStakeProgramAddress, findUserStakeProgramAddress, findWithdrawAuthorityProgramAddress, findWsolTransientProgramAddress, } from './utils';
 export interface ValidatorListAccount {
     pubkey: PublicKey;
     account: AccountInfo<ValidatorList>;
@@ -85,18 +86,46 @@ export declare function withdrawSol(connection: Connection, stakePoolAddress: Pu
 }>;
 /**
  * Creates instructions required to withdraw wSOL from a stake pool.
+ * Rent for ATA creation (if needed) is paid from the withdrawal amount.
  */
-export declare function withdrawWsolWithSession(connection: Connection, stakePoolAddress: PublicKey, signerOrSession: PublicKey, userPubkey: PublicKey, amount: number, minimumLamportsOut?: number, solWithdrawAuthority?: PublicKey, payer?: PublicKey): Promise<{
+export declare function withdrawWsolWithSession(connection: Connection, stakePoolAddress: PublicKey, signerOrSession: PublicKey, userPubkey: PublicKey, amount: number, minimumLamportsOut?: number, solWithdrawAuthority?: PublicKey): Promise<{
     instructions: TransactionInstruction[];
     signers: Signer[];
 }>;
 /**
- * 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.
- */
-export declare function withdrawStakeWithSession(connection: Connection, stakePoolAddress: PublicKey, signerOrSession: PublicKey, userPubkey: PublicKey, amount: number, payer: PublicKey, useReserve?: boolean, voteAccountAddress?: PublicKey, minimumLamportsOut?: number, validatorComparator?: (_a: ValidatorAccount, _b: ValidatorAccount) => number): Promise<{
+ * 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
+ */
+export declare function findNextUserStakeSeed(connection: Connection, programId: PublicKey, userPubkey: PublicKey, startSeed?: number, maxSeed?: number): Promise<number>;
+/**
+ * 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.
+ * Rent for the stake account PDAs is paid from the withdrawal amount.
+ *
+ * @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 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
+ */
+export declare function withdrawStakeWithSession(connection: Connection, stakePoolAddress: PublicKey, signerOrSession: PublicKey, userPubkey: PublicKey, amount: number, userStakeSeedStart?: number, useReserve?: boolean, voteAccountAddress?: PublicKey, minimumLamportsOut?: number, validatorComparator?: (_a: ValidatorAccount, _b: ValidatorAccount) => number): Promise<{
     instructions: TransactionInstruction[];
     stakeAccountPubkeys: PublicKey[];
+    userStakeSeeds: number[];
 }>;
 export declare function addValidatorToPool(connection: Connection, stakePoolAddress: PublicKey, validatorVote: PublicKey, seed?: number): Promise<{
     instructions: TransactionInstruction[];

package/dist/index.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'),
         ]),
     },
 });
@@ -1537,9 +1553,9 @@ class StakePoolInstruction {
     }
     /**
      * Creates a transaction instruction to withdraw WSOL from a stake pool using a session.
+     * Rent for ATA creation (if needed) is paid from the withdrawal amount.
      */
     static withdrawWsolWithSession(params) {
-        var _a;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawWsolWithSession;
         const data = encodeData(type, {
             poolTokensIn: params.poolTokensIn,
@@ -1560,7 +1576,6 @@ class StakePoolInstruction {
             { pubkey: params.tokenProgramId, isSigner: false, isWritable: false },
             { pubkey: params.wsolMint, isSigner: false, isWritable: false },
             { pubkey: params.programSigner, isSigner: false, isWritable: true },
-            { pubkey: (_a = params.payer) !== null && _a !== void 0 ? _a : params.userTransferAuthority, isSigner: true, isWritable: true },
             { pubkey: params.userWallet, isSigner: false, isWritable: false },
             { pubkey: SystemProgram.programId, isSigner: false, isWritable: false },
         ];
@@ -1581,12 +1596,14 @@ class StakePoolInstruction {
     }
     /**
      * Creates a transaction instruction to withdraw stake from a stake pool using a Fogo session.
+     * The stake account is created as a PDA and rent is paid from the withdrawal amount.
      */
     static withdrawStakeWithSession(params) {
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawStakeWithSession;
         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,7 @@ class StakePoolInstruction {
             { pubkey: params.tokenProgramId, isSigner: false, isWritable: false },
             { pubkey: StakeProgram.programId, isSigner: false, isWritable: false },
             { pubkey: params.programSigner, isSigner: false, isWritable: false },
+            { pubkey: SystemProgram.programId, isSigner: false, isWritable: false },
         ];
         return new TransactionInstruction({
             programId: params.programId,
@@ -2176,8 +2194,9 @@ async function withdrawSol(connection, stakePoolAddress, tokenOwner, solReceiver
 }
 /**
  * Creates instructions required to withdraw wSOL from a stake pool.
+ * Rent for ATA creation (if needed) is paid from the withdrawal amount.
  */
-async function withdrawWsolWithSession(connection, stakePoolAddress, signerOrSession, userPubkey, amount, minimumLamportsOut = 0, solWithdrawAuthority, payer) {
+async function withdrawWsolWithSession(connection, stakePoolAddress, signerOrSession, userPubkey, amount, minimumLamportsOut = 0, solWithdrawAuthority) {
     const stakePoolAccount = await getStakePoolAccount(connection, stakePoolAddress);
     const stakePoolProgramId = getStakePoolProgramId(connection.rpcEndpoint);
     const stakePool = stakePoolAccount.account.data;
@@ -2210,7 +2229,6 @@ async function withdrawWsolWithSession(connection, stakePoolAddress, signerOrSes
         wsolMint: NATIVE_MINT,
         programSigner,
         userWallet: userPubkey,
-        payer,
         poolTokensIn: poolTokens,
         minimumLamportsOut,
     }));
@@ -2220,10 +2238,45 @@ 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 withdrawStakeWithSession(connection, stakePoolAddress, signerOrSession, userPubkey, amount, payer, useReserve = false, voteAccountAddress, minimumLamportsOut = 0, validatorComparator) {
+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.
+ * Rent for the stake account PDAs is paid from the withdrawal amount.
+ *
+ * @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 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, userStakeSeedStart = 0, useReserve = false, voteAccountAddress, minimumLamportsOut = 0, validatorComparator) {
     const stakePoolAccount = await getStakePoolAccount(connection, stakePoolAddress);
     const stakePoolProgramId = getStakePoolProgramId(connection.rpcEndpoint);
     const stakePool = stakePoolAccount.account.data;
@@ -2274,6 +2327,7 @@ async function withdrawStakeWithSession(connection, stakePoolAddress, signerOrSe
     }
     const instructions = [];
     const stakeAccountPubkeys = [];
+    const userStakeSeeds = [];
     // Max 5 accounts to prevent an error: "Transaction too large"
     const maxWithdrawAccounts = 5;
     let i = 0;
@@ -2281,26 +2335,12 @@ async function withdrawStakeWithSession(connection, stakePoolAddress, signerOrSe
         if (i >= maxWithdrawAccounts) {
             break;
         }
-        // Create a deterministic stake account address using a seed
-        // This avoids needing the new account to sign (required for sessions)
-        // We use payer as base so only payer needs to sign (payer = basePubkey)
-        // The seed includes timestamp + random component to ensure uniqueness
-        const uniqueId = `${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
-        const seed = `ws:${uniqueId}:${i}`.slice(0, 32); // Seed max 32 chars
-        const stakeReceiverPubkey = await PublicKey.createWithSeed(payer, seed, StakeProgram.programId);
+        // Derive the stake account PDA for this withdrawal
+        const userStakeSeed = userStakeSeedStart + i;
+        const stakeReceiverPubkey = findUserStakeProgramAddress(stakePoolProgramId, userPubkey, userStakeSeed);
         stakeAccountPubkeys.push(stakeReceiverPubkey);
-        // Create the stake account using createAccountWithSeed
-        // Since basePubkey === fromPubkey (both are payer), only payer needs to sign
-        instructions.push(SystemProgram.createAccountWithSeed({
-            fromPubkey: payer,
-            newAccountPubkey: stakeReceiverPubkey,
-            basePubkey: payer,
-            seed,
-            lamports: stakeAccountRentExemption,
-            space: StakeProgram.space,
-            programId: StakeProgram.programId,
-        }));
-        // Add the withdraw stake with session instruction
+        userStakeSeeds.push(userStakeSeed);
+        // The on-chain program will create the stake account PDA
         instructions.push(StakePoolInstruction.withdrawStakeWithSession({
             programId: stakePoolProgramId,
             stakePool: stakePoolAddress,
@@ -2316,12 +2356,14 @@ async function withdrawStakeWithSession(connection, stakePoolAddress, signerOrSe
             programSigner,
             poolTokensIn: withdrawAccount.poolAmount.toNumber(),
             minimumLamportsOut,
+            userStakeSeed,
         }));
         i++;
     }
     return {
         instructions,
         stakeAccountPubkeys,
+        userStakeSeeds,
     };
 }
 async function addValidatorToPool(connection, stakePoolAddress, validatorVote, seed) {
@@ -2686,5 +2728,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.esm.js.map