Depositing (staking)

function batchDeposit(
    bytes32[] conditionIds,   // strictly ascending, no duplicates
    bytes32[] questionIds,    // Polymarket questionId per market, same permutation
    uint256[] yesAmounts,     // 6-decimal integers (YES=0)
    uint256[] noAmounts,      // 6-decimal integers (NO=1)
    uint256 nonZeroLength,    // count of non-zero amounts across yes+no
    uint256 referralCode      // 0 if none
) external;

import { encodeFunctionData } from "viem";

// Three calls, batched into one Safe.execTransaction.
const approve = encodeFunctionData({
  abi: conditionalTokensAbi,
  functionName: "setApprovalForAll",
  args: [VAULT, true],
});

const deposit = encodeFunctionData({
  abi: stakingVaultAbi,
  functionName: "batchDeposit",
  args: [conditionIds, questionIds, yesAmounts, noAmounts, nonZeroLength, 0n],
});

const revoke = encodeFunctionData({
  abi: conditionalTokensAbi,
  functionName: "setApprovalForAll",
  args: [VAULT, false],
});

data = abi.encode(
    bytes32[] conditionIds,
    bytes32[] questionIds,
    uint256[] yesAmounts,
    uint256[] noAmounts,
    uint256   nonZeroLength,
    uint256   referralCode
);

import { encodeAbiParameters, encodeFunctionData } from "viem";

// Dense, sorted: per market YES then NO, skipping zero sides.
const ids: bigint[] = [];
const values: bigint[] = [];
for (const r of sortedRows) {
  if (r.yesAmount > 0n) {
    ids.push(r.yesPositionId); // Polymarket `asset` / `oppositeAsset`
    values.push(r.yesAmount);
  }
  if (r.noAmount > 0n) {
    ids.push(r.noPositionId);
    values.push(r.noAmount);
  }
}
if (BigInt(ids.length) !== nonZeroLength)
  throw new Error("ids.length must equal nonZeroLength");

// ABI-encode the deposit payload the hook decodes (same args as batchDeposit).
const data = encodeAbiParameters(
  [
    { type: "bytes32[]" },
    { type: "bytes32[]" },
    { type: "uint256[]" },
    { type: "uint256[]" },
    { type: "uint256" },
    { type: "uint256" },
  ],
  [conditionIds, questionIds, yesAmounts, noAmounts, nonZeroLength, 0n],
);

const transfer = encodeFunctionData({
  abi: conditionalTokensAbi,
  functionName: "safeBatchTransferFrom",
  args: [wallet, VAULT, ids, values, data],
});

const ok = await publicClient.readContract({
  address: LENS,
  abi: lensAbi,
  functionName: "checkBatchDepositCapacity",
  args: [conditionIds, yesAmounts, noAmounts], // pass the WHOLE batch — capacity is vault-wide
});
if (!ok) throw new Error("Deposit would exceed vault capacity");