@ignitionfi/spl-stake-pool 1.1.19 → 1.1.21

package/dist/index.iife.js

@@ -22264,7 +22264,8 @@ var solanaStakePool = (function (exports, node_buffer) {
 	        index: 24,
 	        layout: LayoutExports$1.struct([
 	            LayoutExports$1.u8('instruction'),
-	            LayoutExports$1.ns64('lamports'),
+	            LayoutExports$1.ns64('poolTokensIn'),
+	            LayoutExports$1.ns64('minimumLamportsOut'),
 	        ]),
 	    },
 	    DepositSolWithSlippage: {
@@ -22297,6 +22298,14 @@ var solanaStakePool = (function (exports, node_buffer) {
 	            LayoutExports$1.ns64('minimumLamportsOut'),
 	        ]),
 	    },
+	    WithdrawStakeWithSession: {
+	        index: 29,
+	        layout: LayoutExports$1.struct([
+	            LayoutExports$1.u8('instruction'),
+	            LayoutExports$1.ns64('poolTokensIn'),
+	            LayoutExports$1.ns64('minimumLamportsOut'),
+	        ]),
+	    },
 	});
 	/**
 	 * Stake Pool Instruction class
@@ -22732,6 +22741,7 @@ var solanaStakePool = (function (exports, node_buffer) {
 	     * Creates a transaction instruction to withdraw WSOL from a stake pool using a session.
 	     */
 	    static withdrawWsolWithSession(params) {
+	        var _a;
 	        const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawWsolWithSession;
 	        const data = encodeData(type, {
 	            poolTokensIn: params.poolTokensIn,
@@ -22752,6 +22762,9 @@ var solanaStakePool = (function (exports, node_buffer) {
 	            { 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 },
 	        ];
 	        if (params.solWithdrawAuthority) {
 	            keys.push({
@@ -22760,6 +22773,39 @@ var solanaStakePool = (function (exports, node_buffer) {
 	                isWritable: false,
 	            });
 	        }
+	        // Associated Token Program must be last - only needed in transaction for CPI routing
+	        keys.push({ pubkey: ASSOCIATED_TOKEN_PROGRAM_ID, isSigner: false, isWritable: false });
+	        return new TransactionInstruction({
+	            programId: params.programId,
+	            keys,
+	            data,
+	        });
+	    }
+	    /**
+	     * Creates a transaction instruction to withdraw stake from a stake pool using a Fogo session.
+	     */
+	    static withdrawStakeWithSession(params) {
+	        const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawStakeWithSession;
+	        const data = encodeData(type, {
+	            poolTokensIn: params.poolTokensIn,
+	            minimumLamportsOut: params.minimumLamportsOut,
+	        });
+	        const keys = [
+	            { pubkey: params.stakePool, isSigner: false, isWritable: true },
+	            { pubkey: params.validatorList, isSigner: false, isWritable: true },
+	            { pubkey: params.withdrawAuthority, isSigner: false, isWritable: false },
+	            { pubkey: params.stakeToSplit, isSigner: false, isWritable: true },
+	            { pubkey: params.stakeToReceive, isSigner: false, isWritable: true },
+	            { pubkey: params.sessionSigner, isSigner: true, isWritable: false }, // user_stake_authority_info (signer_or_session)
+	            { pubkey: params.sessionSigner, isSigner: false, isWritable: false }, // user_transfer_authority_info (not used in session path)
+	            { pubkey: params.burnFromPool, isSigner: false, isWritable: true },
+	            { pubkey: params.managerFeeAccount, isSigner: false, isWritable: true },
+	            { pubkey: params.poolMint, isSigner: false, isWritable: true },
+	            { pubkey: SYSVAR_CLOCK_PUBKEY, isSigner: false, isWritable: false },
+	            { pubkey: params.tokenProgramId, isSigner: false, isWritable: false },
+	            { pubkey: StakeProgram.programId, isSigner: false, isWritable: false },
+	            { pubkey: params.programSigner, isSigner: false, isWritable: false },
+	        ];
 	        return new TransactionInstruction({
 	            programId: params.programId,
 	            keys,
@@ -23333,7 +23379,7 @@ var solanaStakePool = (function (exports, node_buffer) {
 	/**
 	 * Creates instructions required to withdraw wSOL from a stake pool.
 	 */
-	async function withdrawWsolWithSession(connection, stakePoolAddress, signerOrSession, userPubkey, amount, minimumLamportsOut = 0, solWithdrawAuthority) {
+	async function withdrawWsolWithSession(connection, stakePoolAddress, signerOrSession, userPubkey, amount, minimumLamportsOut = 0, solWithdrawAuthority, payer) {
 	    const stakePoolAccount = await getStakePoolAccount(connection, stakePoolAddress);
 	    const stakePoolProgramId = getStakePoolProgramId(connection.rpcEndpoint);
 	    const stakePool = stakePoolAccount.account.data;
@@ -23347,7 +23393,8 @@ var solanaStakePool = (function (exports, node_buffer) {
 	    const userWsolAccount = getAssociatedTokenAddressSync(NATIVE_MINT, userPubkey);
 	    const instructions = [];
 	    const signers = [];
-	    instructions.push(createAssociatedTokenAccountIdempotentInstruction(signerOrSession, userWsolAccount, userPubkey, NATIVE_MINT));
+	    // The program handles wSOL ATA creation internally
+	    // This prevents rent drain attacks where paymaster pays for ATA and user reclaims rent
 	    const [programSigner] = PublicKey.findProgramAddressSync([Buffer.from('fogo_session_program_signer')], stakePoolProgramId);
 	    const withdrawAuthority = await findWithdrawAuthorityProgramAddress(stakePoolProgramId, stakePoolAddress);
 	    instructions.push(StakePoolInstruction.withdrawWsolWithSession({
@@ -23364,6 +23411,8 @@ var solanaStakePool = (function (exports, node_buffer) {
 	        solWithdrawAuthority,
 	        wsolMint: NATIVE_MINT,
 	        programSigner,
+	        userWallet: userPubkey,
+	        payer,
 	        poolTokensIn: poolTokens,
 	        minimumLamportsOut,
 	    }));
@@ -23372,6 +23421,106 @@ var solanaStakePool = (function (exports, node_buffer) {
 	        signers,
 	    };
 	}
+	/**
+	 * 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.
+	 */
+	async function withdrawStakeWithSession(connection, stakePoolAddress, signerOrSession, userPubkey, amount, useReserve = false, voteAccountAddress, minimumLamportsOut = 0, payer, validatorComparator) {
+	    const stakePoolAccount = await getStakePoolAccount(connection, stakePoolAddress);
+	    const stakePoolProgramId = getStakePoolProgramId(connection.rpcEndpoint);
+	    const stakePool = stakePoolAccount.account.data;
+	    const poolTokens = solToLamports(amount);
+	    const poolAmount = new BN(poolTokens);
+	    const poolTokenAccount = getAssociatedTokenAddressSync(stakePool.poolMint, userPubkey);
+	    const tokenAccount = await getAccount(connection, poolTokenAccount);
+	    if (tokenAccount.amount < poolTokens) {
+	        throw new Error(`Not enough token balance to withdraw ${amount} pool tokens.
+          Maximum withdraw amount is ${lamportsToSol(tokenAccount.amount)} pool tokens.`);
+	    }
+	    const [programSigner] = PublicKey.findProgramAddressSync([Buffer.from('fogo_session_program_signer')], stakePoolProgramId);
+	    const withdrawAuthority = await findWithdrawAuthorityProgramAddress(stakePoolProgramId, stakePoolAddress);
+	    const stakeAccountRentExemption = await connection.getMinimumBalanceForRentExemption(StakeProgram.space);
+	    // Determine which stake accounts to withdraw from
+	    const withdrawAccounts = [];
+	    if (useReserve) {
+	        withdrawAccounts.push({
+	            stakeAddress: stakePool.reserveStake,
+	            voteAddress: undefined,
+	            poolAmount,
+	        });
+	    }
+	    else if (voteAccountAddress) {
+	        const stakeAccountAddress = await findStakeProgramAddress(stakePoolProgramId, voteAccountAddress, stakePoolAddress);
+	        const stakeAccount = await connection.getAccountInfo(stakeAccountAddress);
+	        if (!stakeAccount) {
+	            throw new Error(`Validator stake account not found for vote address ${voteAccountAddress.toBase58()}`);
+	        }
+	        const availableLamports = new BN(stakeAccount.lamports - MINIMUM_ACTIVE_STAKE - stakeAccountRentExemption);
+	        if (availableLamports.lt(new BN(0))) {
+	            throw new Error('Invalid Stake Account');
+	        }
+	        const availableForWithdrawal = calcLamportsWithdrawAmount(stakePool, availableLamports);
+	        if (availableForWithdrawal.lt(poolAmount)) {
+	            throw new Error(`Not enough lamports available for withdrawal from ${stakeAccountAddress},
+          ${poolAmount} asked, ${availableForWithdrawal} available.`);
+	        }
+	        withdrawAccounts.push({
+	            stakeAddress: stakeAccountAddress,
+	            voteAddress: voteAccountAddress,
+	            poolAmount,
+	        });
+	    }
+	    else {
+	        // Get the list of accounts to withdraw from automatically
+	        withdrawAccounts.push(...(await prepareWithdrawAccounts(connection, stakePool, stakePoolAddress, poolAmount, validatorComparator, poolTokenAccount.equals(stakePool.managerFeeAccount))));
+	    }
+	    const instructions = [];
+	    const signers = [];
+	    const stakeAccountPubkeys = [];
+	    // Max 5 accounts to prevent an error: "Transaction too large"
+	    const maxWithdrawAccounts = 5;
+	    let i = 0;
+	    for (const withdrawAccount of withdrawAccounts) {
+	        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
+	        instructions.push(StakePoolInstruction.withdrawStakeWithSession({
+	            programId: stakePoolProgramId,
+	            stakePool: stakePoolAddress,
+	            validatorList: stakePool.validatorList,
+	            withdrawAuthority,
+	            stakeToSplit: withdrawAccount.stakeAddress,
+	            stakeToReceive: stakeReceiver.publicKey,
+	            sessionSigner: signerOrSession,
+	            burnFromPool: poolTokenAccount,
+	            managerFeeAccount: stakePool.managerFeeAccount,
+	            poolMint: stakePool.poolMint,
+	            tokenProgramId: stakePool.tokenProgramId,
+	            programSigner,
+	            poolTokensIn: withdrawAccount.poolAmount.toNumber(),
+	            minimumLamportsOut: solToLamports(minimumLamportsOut),
+	        }));
+	        i++;
+	    }
+	    return {
+	        instructions,
+	        signers,
+	        stakeAccountPubkeys,
+	    };
+	}
 	async function addValidatorToPool(connection, stakePoolAddress, validatorVote, seed) {
 	    const stakePoolAccount = await getStakePoolAccount(connection, stakePoolAddress);
 	    const stakePoolProgramId = getStakePoolProgramId(connection.rpcEndpoint);
@@ -23759,6 +23908,7 @@ var solanaStakePool = (function (exports, node_buffer) {
 	exports.updateStakePool = updateStakePool;
 	exports.withdrawSol = withdrawSol;
 	exports.withdrawStake = withdrawStake;
+	exports.withdrawStakeWithSession = withdrawStakeWithSession;
 	exports.withdrawWsolWithSession = withdrawWsolWithSession;
 
 	return exports;