@ignitionfi/fogo-stake-pool 1.0.3 → 1.1.0

package/dist/index.cjs.js

@@ -459,6 +459,139 @@ const USER_STAKE_SEED_PREFIX = node_buffer.Buffer.from('user_stake');
 // for merges without a mismatch on credits observed
 const MINIMUM_ACTIVE_STAKE = 1000000;
 
+/**
+ * BN-based layout for data decoding using buffer-layout.
+ * Used for decoding on-chain account data (StakePool, ValidatorList, etc.)
+ */
+class BNDataLayout extends bufferLayout.Layout {
+    constructor(span, signed, property) {
+        super(span, property);
+        this.blobLayout = bufferLayout.blob(span);
+        this.signed = signed;
+    }
+    decode(b, offset = 0) {
+        const num = new BN(this.blobLayout.decode(b, offset), 10, 'le');
+        if (this.signed) {
+            return num.fromTwos(this.span * 8).clone();
+        }
+        return num;
+    }
+    encode(src, b, offset = 0) {
+        if (this.signed) {
+            src = src.toTwos(this.span * 8);
+        }
+        return this.blobLayout.encode(src.toArrayLike(Buffer, 'le', this.span), b, offset);
+    }
+}
+/**
+ * Creates a u64 layout for data decoding (account layouts).
+ * Used in StakePoolLayout, ValidatorListLayout, etc.
+ */
+function u64(property) {
+    return new BNDataLayout(8, false, property);
+}
+/**
+ * BN-based layout for 64-bit unsigned integers using @solana/buffer-layout.
+ * Used for encoding instruction data with support for values > MAX_SAFE_INTEGER.
+ */
+class BNInstructionLayout extends BufferLayout__namespace.Layout {
+    constructor(span, signed, property) {
+        super(span, property);
+        this.blobLayout = BufferLayout__namespace.blob(span);
+        this.signed = signed;
+    }
+    decode(b, offset = 0) {
+        const num = new BN(this.blobLayout.decode(b, offset), 10, 'le');
+        if (this.signed) {
+            return num.fromTwos(this.span * 8).clone();
+        }
+        return num;
+    }
+    encode(src, b, offset = 0) {
+        if (this.signed) {
+            src = src.toTwos(this.span * 8);
+        }
+        return this.blobLayout.encode(src.toArrayLike(Buffer, 'le', this.span), b, offset);
+    }
+    getSpan(_b, _offset) {
+        return this.span;
+    }
+}
+/**
+ * Creates a u64 layout for instruction encoding.
+ * Properly handles BN values larger than Number.MAX_SAFE_INTEGER.
+ * Compatible with @solana/buffer-layout.struct().
+ */
+// eslint-disable-next-line ts/no-explicit-any
+function u64Instruction(property) {
+    return new BNInstructionLayout(8, false, property);
+}
+class WrappedLayout extends bufferLayout.Layout {
+    constructor(layout, decoder, encoder, property) {
+        super(layout.span, property);
+        this.layout = layout;
+        this.decoder = decoder;
+        this.encoder = encoder;
+    }
+    decode(b, offset) {
+        return this.decoder(this.layout.decode(b, offset));
+    }
+    encode(src, b, offset) {
+        return this.layout.encode(this.encoder(src), b, offset);
+    }
+    getSpan(b, offset) {
+        return this.layout.getSpan(b, offset);
+    }
+}
+function publicKey(property) {
+    return new WrappedLayout(bufferLayout.blob(32), (b) => new web3_js.PublicKey(b), (key) => key.toBuffer(), property);
+}
+class OptionLayout extends bufferLayout.Layout {
+    constructor(layout, property) {
+        super(-1, property);
+        this.layout = layout;
+        this.discriminator = bufferLayout.u8();
+    }
+    encode(src, b, offset = 0) {
+        if (src === null || src === undefined) {
+            return this.discriminator.encode(0, b, offset);
+        }
+        this.discriminator.encode(1, b, offset);
+        return this.layout.encode(src, b, offset + 1) + 1;
+    }
+    decode(b, offset = 0) {
+        const discriminator = this.discriminator.decode(b, offset);
+        if (discriminator === 0) {
+            return null;
+        }
+        else if (discriminator === 1) {
+            return this.layout.decode(b, offset + 1);
+        }
+        throw new Error(`Invalid option ${this.property}`);
+    }
+    getSpan(b, offset = 0) {
+        const discriminator = this.discriminator.decode(b, offset);
+        if (discriminator === 0) {
+            return 1;
+        }
+        else if (discriminator === 1) {
+            return this.layout.getSpan(b, offset + 1) + 1;
+        }
+        throw new Error(`Invalid option ${this.property}`);
+    }
+}
+function option(layout, property) {
+    return new OptionLayout(layout, property);
+}
+function vec(elementLayout, property) {
+    const length = bufferLayout.u32('length');
+    const layout = bufferLayout.struct([
+        length,
+        bufferLayout.seq(elementLayout, bufferLayout.offset(length, -length.span), 'values'),
+    ]);
+    return new WrappedLayout(layout, ({ values }) => values, values => ({ values }), property);
+}
+
 /**
  * Populate a buffer of instruction data using an InstructionType
  * @internal
@@ -577,95 +710,6 @@ function findUserStakeProgramAddress(programId, userWallet, seed) {
     return publicKey;
 }
 
-class BNLayout extends bufferLayout.Layout {
-    constructor(span, signed, property) {
-        super(span, property);
-        this.blob = bufferLayout.blob(span);
-        this.signed = signed;
-    }
-    decode(b, offset = 0) {
-        const num = new BN(this.blob.decode(b, offset), 10, 'le');
-        if (this.signed) {
-            return num.fromTwos(this.span * 8).clone();
-        }
-        return num;
-    }
-    encode(src, b, offset = 0) {
-        if (this.signed) {
-            src = src.toTwos(this.span * 8);
-        }
-        return this.blob.encode(src.toArrayLike(Buffer, 'le', this.span), b, offset);
-    }
-}
-function u64(property) {
-    return new BNLayout(8, false, property);
-}
-class WrappedLayout extends bufferLayout.Layout {
-    constructor(layout, decoder, encoder, property) {
-        super(layout.span, property);
-        this.layout = layout;
-        this.decoder = decoder;
-        this.encoder = encoder;
-    }
-    decode(b, offset) {
-        return this.decoder(this.layout.decode(b, offset));
-    }
-    encode(src, b, offset) {
-        return this.layout.encode(this.encoder(src), b, offset);
-    }
-    getSpan(b, offset) {
-        return this.layout.getSpan(b, offset);
-    }
-}
-function publicKey(property) {
-    return new WrappedLayout(bufferLayout.blob(32), (b) => new web3_js.PublicKey(b), (key) => key.toBuffer(), property);
-}
-class OptionLayout extends bufferLayout.Layout {
-    constructor(layout, property) {
-        super(-1, property);
-        this.layout = layout;
-        this.discriminator = bufferLayout.u8();
-    }
-    encode(src, b, offset = 0) {
-        if (src === null || src === undefined) {
-            return this.discriminator.encode(0, b, offset);
-        }
-        this.discriminator.encode(1, b, offset);
-        return this.layout.encode(src, b, offset + 1) + 1;
-    }
-    decode(b, offset = 0) {
-        const discriminator = this.discriminator.decode(b, offset);
-        if (discriminator === 0) {
-            return null;
-        }
-        else if (discriminator === 1) {
-            return this.layout.decode(b, offset + 1);
-        }
-        throw new Error(`Invalid option ${this.property}`);
-    }
-    getSpan(b, offset = 0) {
-        const discriminator = this.discriminator.decode(b, offset);
-        if (discriminator === 0) {
-            return 1;
-        }
-        else if (discriminator === 1) {
-            return this.layout.getSpan(b, offset + 1) + 1;
-        }
-        throw new Error(`Invalid option ${this.property}`);
-    }
-}
-function option(layout, property) {
-    return new OptionLayout(layout, property);
-}
-function vec(elementLayout, property) {
-    const length = bufferLayout.u32('length');
-    const layout = bufferLayout.struct([
-        length,
-        bufferLayout.seq(elementLayout, bufferLayout.offset(length, -length.span), 'values'),
-    ]);
-    return new WrappedLayout(layout, ({ values }) => values, values => ({ values }), property);
-}
-
 const feeFields = [u64('denominator'), u64('numerator')];
 var AccountType;
 (function (AccountType) {
@@ -1008,11 +1052,50 @@ function arrayChunk(array, size) {
     return result;
 }
 
+/**
+ * Converts various numeric types to BN for safe large number handling.
+ * @internal
+ */
+function toBN(value) {
+    if (BN.isBN(value)) {
+        return value;
+    }
+    if (typeof value === 'bigint') {
+        return new BN(value.toString());
+    }
+    if (typeof value === 'string') {
+        // Validate string is a valid non-negative integer
+        const trimmed = value.trim();
+        if (!/^\d+$/.test(trimmed)) {
+            throw new Error(`Invalid amount string: "${value}". Must be a non-negative integer.`);
+        }
+        return new BN(trimmed);
+    }
+    if (typeof value === 'number') {
+        if (!Number.isFinite(value)) {
+            throw new Error('Invalid amount: must be a finite number');
+        }
+        if (value < 0) {
+            throw new Error('Invalid amount: must be non-negative');
+        }
+        if (!Number.isInteger(value)) {
+            throw new Error('Invalid amount: must be an integer (lamports)');
+        }
+        // CRITICAL: Numbers > MAX_SAFE_INTEGER have already lost precision
+        // We throw an error instead of silently corrupting data
+        if (value > Number.MAX_SAFE_INTEGER) {
+            throw new Error(`Amount ${value} exceeds Number.MAX_SAFE_INTEGER (9,007,199,254,740,991). ` +
+                `Use BigInt or BN for large values to avoid precision loss.`);
+        }
+        return new BN(value);
+    }
+    throw new Error(`Invalid amount type: ${typeof value}`);
+}
 // 'UpdateTokenMetadata' and 'CreateTokenMetadata' have dynamic layouts
 const MOVE_STAKE_LAYOUT = BufferLayout__namespace.struct([
     BufferLayout__namespace.u8('instruction'),
-    BufferLayout__namespace.ns64('lamports'),
-    BufferLayout__namespace.ns64('transientStakeSeed'),
+    u64Instruction('lamports'),
+    u64Instruction('transientStakeSeed'),
 ]);
 const UPDATE_VALIDATOR_LIST_BALANCE_LAYOUT = BufferLayout__namespace.struct([
     BufferLayout__namespace.u8('instruction'),
@@ -1087,7 +1170,7 @@ const STAKE_POOL_INSTRUCTION_LAYOUTS = Object.freeze({
         index: 10,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('poolTokens'),
+            u64Instruction('poolTokens'),
         ]),
     },
     /// Deposit SOL directly into the pool's reserve account. The output is a "pool" token
@@ -1096,7 +1179,7 @@ const STAKE_POOL_INSTRUCTION_LAYOUTS = Object.freeze({
         index: 14,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('lamports'),
+            u64Instruction('lamports'),
         ]),
     },
     /// Withdraw SOL directly from the pool's reserve account. Fails if the
@@ -1105,25 +1188,25 @@ const STAKE_POOL_INSTRUCTION_LAYOUTS = Object.freeze({
         index: 16,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('poolTokens'),
+            u64Instruction('poolTokens'),
         ]),
     },
     IncreaseAdditionalValidatorStake: {
         index: 19,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('lamports'),
-            BufferLayout__namespace.ns64('transientStakeSeed'),
-            BufferLayout__namespace.ns64('ephemeralStakeSeed'),
+            u64Instruction('lamports'),
+            u64Instruction('transientStakeSeed'),
+            u64Instruction('ephemeralStakeSeed'),
         ]),
     },
     DecreaseAdditionalValidatorStake: {
         index: 20,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('lamports'),
-            BufferLayout__namespace.ns64('transientStakeSeed'),
-            BufferLayout__namespace.ns64('ephemeralStakeSeed'),
+            u64Instruction('lamports'),
+            u64Instruction('transientStakeSeed'),
+            u64Instruction('ephemeralStakeSeed'),
         ]),
     },
     DecreaseValidatorStakeWithReserve: {
@@ -1138,62 +1221,62 @@ const STAKE_POOL_INSTRUCTION_LAYOUTS = Object.freeze({
         index: 23,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('lamports'),
+            u64Instruction('lamports'),
         ]),
     },
     WithdrawStakeWithSlippage: {
         index: 24,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('poolTokensIn'),
-            BufferLayout__namespace.ns64('minimumLamportsOut'),
+            u64Instruction('poolTokensIn'),
+            u64Instruction('minimumLamportsOut'),
         ]),
     },
     DepositSolWithSlippage: {
         index: 25,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('lamports'),
+            u64Instruction('lamports'),
         ]),
     },
     WithdrawSolWithSlippage: {
         index: 26,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('lamports'),
+            u64Instruction('lamports'),
         ]),
     },
     DepositWsolWithSession: {
         index: 27,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('lamportsIn'),
-            BufferLayout__namespace.ns64('minimumPoolTokensOut'),
+            u64Instruction('lamportsIn'),
+            u64Instruction('minimumPoolTokensOut'),
         ]),
     },
     WithdrawWsolWithSession: {
         index: 28,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('poolTokensIn'),
-            BufferLayout__namespace.ns64('minimumLamportsOut'),
+            u64Instruction('poolTokensIn'),
+            u64Instruction('minimumLamportsOut'),
         ]),
     },
     WithdrawStakeWithSession: {
         index: 29,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('poolTokensIn'),
-            BufferLayout__namespace.ns64('minimumLamportsOut'),
-            BufferLayout__namespace.ns64('userStakeSeed'),
+            u64Instruction('poolTokensIn'),
+            u64Instruction('minimumLamportsOut'),
+            u64Instruction('userStakeSeed'),
         ]),
     },
     WithdrawFromStakeAccountWithSession: {
         index: 30,
         layout: BufferLayout__namespace.struct([
             BufferLayout__namespace.u8('instruction'),
-            BufferLayout__namespace.ns64('lamports'),
-            BufferLayout__namespace.ns64('userStakeSeed'),
+            u64Instruction('lamports'),
+            u64Instruction('userStakeSeed'),
         ]),
     },
 });
@@ -1325,7 +1408,10 @@ class StakePoolInstruction {
     static increaseValidatorStake(params) {
         const { programId, stakePool, staker, withdrawAuthority, validatorList, reserveStake, transientStake, validatorStake, validatorVote, lamports, transientStakeSeed, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.IncreaseValidatorStake;
-        const data = encodeData(type, { lamports, transientStakeSeed });
+        const data = encodeData(type, {
+            lamports: toBN(lamports),
+            transientStakeSeed: toBN(transientStakeSeed),
+        });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: false },
             { pubkey: staker, isSigner: true, isWritable: false },
@@ -1355,7 +1441,11 @@ class StakePoolInstruction {
     static increaseAdditionalValidatorStake(params) {
         const { programId, stakePool, staker, withdrawAuthority, validatorList, reserveStake, transientStake, validatorStake, validatorVote, lamports, transientStakeSeed, ephemeralStake, ephemeralStakeSeed, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.IncreaseAdditionalValidatorStake;
-        const data = encodeData(type, { lamports, transientStakeSeed, ephemeralStakeSeed });
+        const data = encodeData(type, {
+            lamports: toBN(lamports),
+            transientStakeSeed: toBN(transientStakeSeed),
+            ephemeralStakeSeed: toBN(ephemeralStakeSeed),
+        });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: false },
             { pubkey: staker, isSigner: true, isWritable: false },
@@ -1385,7 +1475,10 @@ class StakePoolInstruction {
     static decreaseValidatorStake(params) {
         const { programId, stakePool, staker, withdrawAuthority, validatorList, validatorStake, transientStake, lamports, transientStakeSeed, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.DecreaseValidatorStake;
-        const data = encodeData(type, { lamports, transientStakeSeed });
+        const data = encodeData(type, {
+            lamports: toBN(lamports),
+            transientStakeSeed: toBN(transientStakeSeed),
+        });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: false },
             { pubkey: staker, isSigner: true, isWritable: false },
@@ -1411,7 +1504,10 @@ class StakePoolInstruction {
     static decreaseValidatorStakeWithReserve(params) {
         const { programId, stakePool, staker, withdrawAuthority, validatorList, reserveStake, validatorStake, transientStake, lamports, transientStakeSeed, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.DecreaseValidatorStakeWithReserve;
-        const data = encodeData(type, { lamports, transientStakeSeed });
+        const data = encodeData(type, {
+            lamports: toBN(lamports),
+            transientStakeSeed: toBN(transientStakeSeed),
+        });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: false },
             { pubkey: staker, isSigner: true, isWritable: false },
@@ -1438,7 +1534,11 @@ class StakePoolInstruction {
     static decreaseAdditionalValidatorStake(params) {
         const { programId, stakePool, staker, withdrawAuthority, validatorList, reserveStake, validatorStake, transientStake, lamports, transientStakeSeed, ephemeralStakeSeed, ephemeralStake, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.DecreaseAdditionalValidatorStake;
-        const data = encodeData(type, { lamports, transientStakeSeed, ephemeralStakeSeed });
+        const data = encodeData(type, {
+            lamports: toBN(lamports),
+            transientStakeSeed: toBN(transientStakeSeed),
+            ephemeralStakeSeed: toBN(ephemeralStakeSeed),
+        });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: false },
             { pubkey: staker, isSigner: true, isWritable: false },
@@ -1495,7 +1595,7 @@ class StakePoolInstruction {
     static depositSol(params) {
         const { programId, stakePool, withdrawAuthority, depositAuthority, reserveStake, fundingAccount, destinationPoolAccount, managerFeeAccount, referralPoolAccount, poolMint, lamports, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.DepositSol;
-        const data = encodeData(type, { lamports });
+        const data = encodeData(type, { lamports: toBN(lamports) });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: true },
             { pubkey: withdrawAuthority, isSigner: false, isWritable: false },
@@ -1528,8 +1628,8 @@ class StakePoolInstruction {
         var _a;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.DepositWsolWithSession;
         const data = encodeData(type, {
-            lamportsIn: params.lamportsIn,
-            minimumPoolTokensOut: params.minimumPoolTokensOut,
+            lamportsIn: toBN(params.lamportsIn),
+            minimumPoolTokensOut: toBN(params.minimumPoolTokensOut),
         });
         const keys = [
             { pubkey: params.stakePool, isSigner: false, isWritable: true },
@@ -1571,7 +1671,7 @@ class StakePoolInstruction {
     static withdrawStake(params) {
         const { programId, stakePool, validatorList, withdrawAuthority, validatorStake, destinationStake, destinationStakeAuthority, sourceTransferAuthority, sourcePoolAccount, managerFeeAccount, poolMint, poolTokens, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawStake;
-        const data = encodeData(type, { poolTokens });
+        const data = encodeData(type, { poolTokens: toBN(poolTokens) });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: true },
             { pubkey: validatorList, isSigner: false, isWritable: true },
@@ -1599,7 +1699,7 @@ class StakePoolInstruction {
     static withdrawSol(params) {
         const { programId, stakePool, withdrawAuthority, sourceTransferAuthority, sourcePoolAccount, reserveStake, destinationSystemAccount, managerFeeAccount, solWithdrawAuthority, poolMint, poolTokens, } = params;
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawSol;
-        const data = encodeData(type, { poolTokens });
+        const data = encodeData(type, { poolTokens: toBN(poolTokens) });
         const keys = [
             { pubkey: stakePool, isSigner: false, isWritable: true },
             { pubkey: withdrawAuthority, isSigner: false, isWritable: false },
@@ -1634,8 +1734,8 @@ class StakePoolInstruction {
     static withdrawWsolWithSession(params) {
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawWsolWithSession;
         const data = encodeData(type, {
-            poolTokensIn: params.poolTokensIn,
-            minimumLamportsOut: params.minimumLamportsOut,
+            poolTokensIn: toBN(params.poolTokensIn),
+            minimumLamportsOut: toBN(params.minimumLamportsOut),
         });
         const keys = [
             { pubkey: params.stakePool, isSigner: false, isWritable: true },
@@ -1677,9 +1777,9 @@ class StakePoolInstruction {
     static withdrawStakeWithSession(params) {
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawStakeWithSession;
         const data = encodeData(type, {
-            poolTokensIn: params.poolTokensIn,
-            minimumLamportsOut: params.minimumLamportsOut,
-            userStakeSeed: params.userStakeSeed,
+            poolTokensIn: toBN(params.poolTokensIn),
+            minimumLamportsOut: toBN(params.minimumLamportsOut),
+            userStakeSeed: toBN(params.userStakeSeed),
         });
         const keys = [
             { pubkey: params.stakePool, isSigner: false, isWritable: true },
@@ -1714,8 +1814,8 @@ class StakePoolInstruction {
     static withdrawFromStakeAccountWithSession(params) {
         const type = STAKE_POOL_INSTRUCTION_LAYOUTS.WithdrawFromStakeAccountWithSession;
         const data = encodeData(type, {
-            lamports: params.lamports,
-            userStakeSeed: params.userStakeSeed,
+            lamports: toBN(params.lamports),
+            userStakeSeed: toBN(params.userStakeSeed),
         });
         const keys = [
             { pubkey: params.userStakeAccount, isSigner: false, isWritable: true },
@@ -2004,10 +2104,18 @@ skipBalanceCheck = false) {
     if (!skipBalanceCheck) {
         const tokenAccountInfo = await connection.getTokenAccountBalance(wsolTokenAccount, 'confirmed');
         const wsolBalance = tokenAccountInfo
-            ? parseInt(tokenAccountInfo.value.amount)
-            : 0;
-        if (wsolBalance < lamports) {
-            throw new Error(`Not enough WSOL to deposit into pool. Maximum deposit amount is ${lamportsToSol(wsolBalance)} WSOL.`);
+            ? BigInt(tokenAccountInfo.value.amount)
+            : BigInt(0);
+        // Convert lamports to BigInt for comparison
+        const lamportsBigInt = typeof lamports === 'bigint'
+            ? lamports
+            : typeof lamports === 'string'
+                ? BigInt(lamports)
+                : BN.isBN(lamports)
+                    ? BigInt(lamports.toString())
+                    : BigInt(lamports);
+        if (wsolBalance < lamportsBigInt) {
+            throw new Error(`Not enough WSOL to deposit into pool. Maximum deposit amount is ${lamportsToSol(Number(wsolBalance))} WSOL.`);
         }
     }
     const stakePoolAccount = await getStakePoolAccount(connection, stakePoolAddress);
@@ -2053,7 +2161,15 @@ skipBalanceCheck = false) {
  */
 async function depositSol(connection, stakePoolAddress, from, lamports, destinationTokenAccount, referrerTokenAccount, depositAuthority) {
     const fromBalance = await connection.getBalance(from, 'confirmed');
-    if (fromBalance < lamports) {
+    // Convert lamports to BigInt for comparison
+    const lamportsBigInt = typeof lamports === 'bigint'
+        ? lamports
+        : typeof lamports === 'string'
+            ? BigInt(lamports)
+            : BN.isBN(lamports)
+                ? BigInt(lamports.toString())
+                : BigInt(lamports);
+    if (BigInt(fromBalance) < lamportsBigInt) {
         throw new Error(`Not enough SOL to deposit into pool. Maximum deposit amount is ${lamportsToSol(fromBalance)} SOL.`);
     }
     const stakePoolAccount = await getStakePoolAccount(connection, stakePoolAddress);
@@ -2067,7 +2183,7 @@ async function depositSol(connection, stakePoolAddress, from, lamports, destinat
     instructions.push(web3_js.SystemProgram.transfer({
         fromPubkey: from,
         toPubkey: userSolTransfer.publicKey,
-        lamports,
+        lamports: lamportsBigInt,
     }));
     // Create token account if not specified
     if (!destinationTokenAccount) {