@metamask/transaction-pay-controller 22.4.0 → 22.6.0

package/dist/strategy/relay/polymarket/calldata.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"calldata.cjs","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/calldata.ts"],"names":[],"mappings":";;;AAAA,4CAA+C;AAG/C,MAAM,KAAK,GAAG,IAAI,eAAS,CAAC;IAC1B,mDAAmD;IACnD,mEAAmE;IACnE,iEAAiE;IACjE,sDAAsD;CACvD,CAAC,CAAC;AAEH,SAAgB,aAAa,CAAC,OAAY,EAAE,MAAc;IACxD,OAAO,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAQ,CAAC;AACvE,CAAC;AAFD,sCAEC;AAED,SAAgB,YAAY,CAAC,EAC3B,KAAK,EACL,SAAS,EACT,MAAM,GAKP;IACC,OAAO,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC,CAAQ,CAAC;AAC/E,CAAC;AAVD,oCAUC;AAED,SAAgB,UAAU,CAAC,EACzB,KAAK,EACL,SAAS,EACT,MAAM,GAKP;IACC,OAAO,KAAK,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC,CAAQ,CAAC;AAC7E,CAAC;AAVD,gCAUC;AAED,SAAgB,6BAA6B,CAAC,IAAS;IACrD,MAAM,CAAC,SAAS,CAAC,GAAG,KAAK,CAAC,kBAAkB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;IAC/D,OAAO,SAAgB,CAAC;AAC1B,CAAC;AAHD,sEAGC","sourcesContent":["import { Interface } from '@ethersproject/abi';\nimport type { Hex } from '@metamask/utils';\n\nconst iface = new Interface([\n  'function approve(address spender, uint256 amount)',\n  'function unwrap(address asset, address recipient, uint256 amount)',\n  'function wrap(address asset, address recipient, uint256 amount)',\n  'function transfer(address recipient, uint256 amount)',\n]);\n\nexport function encodeApprove(spender: Hex, amount: bigint): Hex {\n  return iface.encodeFunctionData('approve', [spender, amount]) as Hex;\n}\n\nexport function encodeUnwrap({\n  asset,\n  recipient,\n  amount,\n}: {\n  asset: Hex;\n  recipient: Hex;\n  amount: bigint;\n}): Hex {\n  return iface.encodeFunctionData('unwrap', [asset, recipient, amount]) as Hex;\n}\n\nexport function encodeWrap({\n  asset,\n  recipient,\n  amount,\n}: {\n  asset: Hex;\n  recipient: Hex;\n  amount: bigint;\n}): Hex {\n  return iface.encodeFunctionData('wrap', [asset, recipient, amount]) as Hex;\n}\n\nexport function extractErc20TransferRecipient(data: Hex): Hex {\n  const [recipient] = iface.decodeFunctionData('transfer', data);\n  return recipient as Hex;\n}\n"]}

package/dist/strategy/relay/polymarket/calldata.d.cts

@@ -0,0 +1,14 @@
+import type { Hex } from "@metamask/utils";
+export declare function encodeApprove(spender: Hex, amount: bigint): Hex;
+export declare function encodeUnwrap({ asset, recipient, amount, }: {
+    asset: Hex;
+    recipient: Hex;
+    amount: bigint;
+}): Hex;
+export declare function encodeWrap({ asset, recipient, amount, }: {
+    asset: Hex;
+    recipient: Hex;
+    amount: bigint;
+}): Hex;
+export declare function extractErc20TransferRecipient(data: Hex): Hex;
+//# sourceMappingURL=calldata.d.cts.map

package/dist/strategy/relay/polymarket/calldata.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"calldata.d.cts","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/calldata.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAS3C,wBAAgB,aAAa,CAAC,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,GAAG,GAAG,CAE/D;AAED,wBAAgB,YAAY,CAAC,EAC3B,KAAK,EACL,SAAS,EACT,MAAM,GACP,EAAE;IACD,KAAK,EAAE,GAAG,CAAC;IACX,SAAS,EAAE,GAAG,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,GAAG,CAEN;AAED,wBAAgB,UAAU,CAAC,EACzB,KAAK,EACL,SAAS,EACT,MAAM,GACP,EAAE;IACD,KAAK,EAAE,GAAG,CAAC;IACX,SAAS,EAAE,GAAG,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,GAAG,CAEN;AAED,wBAAgB,6BAA6B,CAAC,IAAI,EAAE,GAAG,GAAG,GAAG,CAG5D"}

package/dist/strategy/relay/polymarket/calldata.d.mts

@@ -0,0 +1,14 @@
+import type { Hex } from "@metamask/utils";
+export declare function encodeApprove(spender: Hex, amount: bigint): Hex;
+export declare function encodeUnwrap({ asset, recipient, amount, }: {
+    asset: Hex;
+    recipient: Hex;
+    amount: bigint;
+}): Hex;
+export declare function encodeWrap({ asset, recipient, amount, }: {
+    asset: Hex;
+    recipient: Hex;
+    amount: bigint;
+}): Hex;
+export declare function extractErc20TransferRecipient(data: Hex): Hex;
+//# sourceMappingURL=calldata.d.mts.map

package/dist/strategy/relay/polymarket/calldata.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"calldata.d.mts","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/calldata.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAS3C,wBAAgB,aAAa,CAAC,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,GAAG,GAAG,CAE/D;AAED,wBAAgB,YAAY,CAAC,EAC3B,KAAK,EACL,SAAS,EACT,MAAM,GACP,EAAE;IACD,KAAK,EAAE,GAAG,CAAC;IACX,SAAS,EAAE,GAAG,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,GAAG,CAEN;AAED,wBAAgB,UAAU,CAAC,EACzB,KAAK,EACL,SAAS,EACT,MAAM,GACP,EAAE;IACD,KAAK,EAAE,GAAG,CAAC;IACX,SAAS,EAAE,GAAG,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;CAChB,GAAG,GAAG,CAEN;AAED,wBAAgB,6BAA6B,CAAC,IAAI,EAAE,GAAG,GAAG,GAAG,CAG5D"}

package/dist/strategy/relay/polymarket/calldata.mjs

@@ -0,0 +1,21 @@
+import { Interface } from "@ethersproject/abi";
+const iface = new Interface([
+    'function approve(address spender, uint256 amount)',
+    'function unwrap(address asset, address recipient, uint256 amount)',
+    'function wrap(address asset, address recipient, uint256 amount)',
+    'function transfer(address recipient, uint256 amount)',
+]);
+export function encodeApprove(spender, amount) {
+    return iface.encodeFunctionData('approve', [spender, amount]);
+}
+export function encodeUnwrap({ asset, recipient, amount, }) {
+    return iface.encodeFunctionData('unwrap', [asset, recipient, amount]);
+}
+export function encodeWrap({ asset, recipient, amount, }) {
+    return iface.encodeFunctionData('wrap', [asset, recipient, amount]);
+}
+export function extractErc20TransferRecipient(data) {
+    const [recipient] = iface.decodeFunctionData('transfer', data);
+    return recipient;
+}
+//# sourceMappingURL=calldata.mjs.map

package/dist/strategy/relay/polymarket/calldata.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"calldata.mjs","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/calldata.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,2BAA2B;AAG/C,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC;IAC1B,mDAAmD;IACnD,mEAAmE;IACnE,iEAAiE;IACjE,sDAAsD;CACvD,CAAC,CAAC;AAEH,MAAM,UAAU,aAAa,CAAC,OAAY,EAAE,MAAc;IACxD,OAAO,KAAK,CAAC,kBAAkB,CAAC,SAAS,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAQ,CAAC;AACvE,CAAC;AAED,MAAM,UAAU,YAAY,CAAC,EAC3B,KAAK,EACL,SAAS,EACT,MAAM,GAKP;IACC,OAAO,KAAK,CAAC,kBAAkB,CAAC,QAAQ,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC,CAAQ,CAAC;AAC/E,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,EACzB,KAAK,EACL,SAAS,EACT,MAAM,GAKP;IACC,OAAO,KAAK,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC,CAAQ,CAAC;AAC7E,CAAC;AAED,MAAM,UAAU,6BAA6B,CAAC,IAAS;IACrD,MAAM,CAAC,SAAS,CAAC,GAAG,KAAK,CAAC,kBAAkB,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;IAC/D,OAAO,SAAgB,CAAC;AAC1B,CAAC","sourcesContent":["import { Interface } from '@ethersproject/abi';\nimport type { Hex } from '@metamask/utils';\n\nconst iface = new Interface([\n  'function approve(address spender, uint256 amount)',\n  'function unwrap(address asset, address recipient, uint256 amount)',\n  'function wrap(address asset, address recipient, uint256 amount)',\n  'function transfer(address recipient, uint256 amount)',\n]);\n\nexport function encodeApprove(spender: Hex, amount: bigint): Hex {\n  return iface.encodeFunctionData('approve', [spender, amount]) as Hex;\n}\n\nexport function encodeUnwrap({\n  asset,\n  recipient,\n  amount,\n}: {\n  asset: Hex;\n  recipient: Hex;\n  amount: bigint;\n}): Hex {\n  return iface.encodeFunctionData('unwrap', [asset, recipient, amount]) as Hex;\n}\n\nexport function encodeWrap({\n  asset,\n  recipient,\n  amount,\n}: {\n  asset: Hex;\n  recipient: Hex;\n  amount: bigint;\n}): Hex {\n  return iface.encodeFunctionData('wrap', [asset, recipient, amount]) as Hex;\n}\n\nexport function extractErc20TransferRecipient(data: Hex): Hex {\n  const [recipient] = iface.decodeFunctionData('transfer', data);\n  return recipient as Hex;\n}\n"]}

package/dist/strategy/relay/polymarket/constants.cjs

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SWEEP_RELAYER_SETTLE_DELAY_MS = exports.SWEEP_BALANCE_RETRY_DELAY_MS = exports.SWEEP_BALANCE_RETRY_ATTEMPTS = exports.POLYMARKET_COLLATERAL_ONRAMP_POLYGON = exports.POLYMARKET_COLLATERAL_OFFRAMP_POLYGON = void 0;
+exports.POLYMARKET_COLLATERAL_OFFRAMP_POLYGON = '0x2957922Eb93258b93368531d39fAcCA3B4dC5854';
+exports.POLYMARKET_COLLATERAL_ONRAMP_POLYGON = '0x93070a847efEf7F70739046A929D47a521F5B8ee';
+exports.SWEEP_BALANCE_RETRY_ATTEMPTS = 5;
+exports.SWEEP_BALANCE_RETRY_DELAY_MS = 1000;
+exports.SWEEP_RELAYER_SETTLE_DELAY_MS = 3000;
+//# sourceMappingURL=constants.cjs.map

package/dist/strategy/relay/polymarket/constants.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.cjs","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/constants.ts"],"names":[],"mappings":";;;AAEa,QAAA,qCAAqC,GAChD,4CAAmD,CAAC;AAEzC,QAAA,oCAAoC,GAC/C,4CAAmD,CAAC;AAEzC,QAAA,4BAA4B,GAAG,CAAC,CAAC;AAEjC,QAAA,4BAA4B,GAAG,IAAI,CAAC;AAEpC,QAAA,6BAA6B,GAAG,IAAI,CAAC","sourcesContent":["import type { Hex } from '@metamask/utils';\n\nexport const POLYMARKET_COLLATERAL_OFFRAMP_POLYGON =\n  '0x2957922Eb93258b93368531d39fAcCA3B4dC5854' as Hex;\n\nexport const POLYMARKET_COLLATERAL_ONRAMP_POLYGON =\n  '0x93070a847efEf7F70739046A929D47a521F5B8ee' as Hex;\n\nexport const SWEEP_BALANCE_RETRY_ATTEMPTS = 5;\n\nexport const SWEEP_BALANCE_RETRY_DELAY_MS = 1000;\n\nexport const SWEEP_RELAYER_SETTLE_DELAY_MS = 3000;\n"]}

package/dist/strategy/relay/polymarket/constants.d.cts

@@ -0,0 +1,6 @@
+export declare const POLYMARKET_COLLATERAL_OFFRAMP_POLYGON: `0x${string}`;
+export declare const POLYMARKET_COLLATERAL_ONRAMP_POLYGON: `0x${string}`;
+export declare const SWEEP_BALANCE_RETRY_ATTEMPTS = 5;
+export declare const SWEEP_BALANCE_RETRY_DELAY_MS = 1000;
+export declare const SWEEP_RELAYER_SETTLE_DELAY_MS = 3000;
+//# sourceMappingURL=constants.d.cts.map

package/dist/strategy/relay/polymarket/constants.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.cts","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/constants.ts"],"names":[],"mappings":"AAEA,eAAO,MAAM,qCAAqC,eACG,CAAC;AAEtD,eAAO,MAAM,oCAAoC,eACI,CAAC;AAEtD,eAAO,MAAM,4BAA4B,IAAI,CAAC;AAE9C,eAAO,MAAM,4BAA4B,OAAO,CAAC;AAEjD,eAAO,MAAM,6BAA6B,OAAO,CAAC"}

package/dist/strategy/relay/polymarket/constants.d.mts

@@ -0,0 +1,6 @@
+export declare const POLYMARKET_COLLATERAL_OFFRAMP_POLYGON: `0x${string}`;
+export declare const POLYMARKET_COLLATERAL_ONRAMP_POLYGON: `0x${string}`;
+export declare const SWEEP_BALANCE_RETRY_ATTEMPTS = 5;
+export declare const SWEEP_BALANCE_RETRY_DELAY_MS = 1000;
+export declare const SWEEP_RELAYER_SETTLE_DELAY_MS = 3000;
+//# sourceMappingURL=constants.d.mts.map

package/dist/strategy/relay/polymarket/constants.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.mts","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/constants.ts"],"names":[],"mappings":"AAEA,eAAO,MAAM,qCAAqC,eACG,CAAC;AAEtD,eAAO,MAAM,oCAAoC,eACI,CAAC;AAEtD,eAAO,MAAM,4BAA4B,IAAI,CAAC;AAE9C,eAAO,MAAM,4BAA4B,OAAO,CAAC;AAEjD,eAAO,MAAM,6BAA6B,OAAO,CAAC"}

package/dist/strategy/relay/polymarket/constants.mjs

@@ -0,0 +1,6 @@
+export const POLYMARKET_COLLATERAL_OFFRAMP_POLYGON = '0x2957922Eb93258b93368531d39fAcCA3B4dC5854';
+export const POLYMARKET_COLLATERAL_ONRAMP_POLYGON = '0x93070a847efEf7F70739046A929D47a521F5B8ee';
+export const SWEEP_BALANCE_RETRY_ATTEMPTS = 5;
+export const SWEEP_BALANCE_RETRY_DELAY_MS = 1000;
+export const SWEEP_RELAYER_SETTLE_DELAY_MS = 3000;
+//# sourceMappingURL=constants.mjs.map

package/dist/strategy/relay/polymarket/constants.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.mjs","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/constants.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,qCAAqC,GAChD,4CAAmD,CAAC;AAEtD,MAAM,CAAC,MAAM,oCAAoC,GAC/C,4CAAmD,CAAC;AAEtD,MAAM,CAAC,MAAM,4BAA4B,GAAG,CAAC,CAAC;AAE9C,MAAM,CAAC,MAAM,4BAA4B,GAAG,IAAI,CAAC;AAEjD,MAAM,CAAC,MAAM,6BAA6B,GAAG,IAAI,CAAC","sourcesContent":["import type { Hex } from '@metamask/utils';\n\nexport const POLYMARKET_COLLATERAL_OFFRAMP_POLYGON =\n  '0x2957922Eb93258b93368531d39fAcCA3B4dC5854' as Hex;\n\nexport const POLYMARKET_COLLATERAL_ONRAMP_POLYGON =\n  '0x93070a847efEf7F70739046A929D47a521F5B8ee' as Hex;\n\nexport const SWEEP_BALANCE_RETRY_ATTEMPTS = 5;\n\nexport const SWEEP_BALANCE_RETRY_DELAY_MS = 1000;\n\nexport const SWEEP_RELAYER_SETTLE_DELAY_MS = 3000;\n"]}

package/dist/strategy/relay/polymarket/withdraw.cjs

@@ -0,0 +1,170 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.sweepPolymarketDepositWallet = exports.submitPolymarketWithdraw = exports.applyPolymarketDepositWalletOverrides = void 0;
+const utils_1 = require("@metamask/utils");
+const constants_1 = require("../../../constants.cjs");
+const logger_1 = require("../../../logger.cjs");
+const token_1 = require("../../../utils/token.cjs");
+const calldata_1 = require("./calldata.cjs");
+const constants_2 = require("./constants.cjs");
+const log = (0, utils_1.createModuleLogger)(logger_1.projectLogger, 'polymarket-withdraw');
+async function applyPolymarketDepositWalletOverrides(body, request, messenger) {
+    const depositWalletAddress = await getDepositWalletAddress(messenger, request.from);
+    body.originCurrency = constants_1.POLYGON_USDCE_ADDRESS;
+    body.user = depositWalletAddress;
+    body.refundTo = depositWalletAddress;
+    body.useDepositAddress = true;
+    body.strict = true;
+}
+exports.applyPolymarketDepositWalletOverrides = applyPolymarketDepositWalletOverrides;
+async function submitPolymarketWithdraw(quote, from, messenger) {
+    const depositWalletAddress = await getDepositWalletAddress(messenger, from);
+    const relayDepositAddress = extractRelayDepositAddress(quote.original);
+    const amount = BigInt(quote.sourceAmount.raw);
+    const preSubmitUsdceBalance = await readUsdceBalanceOrZero(messenger, depositWalletAddress);
+    log('Submitting unwrap batch to Relay deposit address', {
+        depositWalletAddress,
+        relayDepositAddress,
+        amount: amount.toString(),
+        preSubmitUsdceBalance: preSubmitUsdceBalance.toString(),
+    });
+    const result = await submitDepositWalletBatch(messenger, {
+        eoa: from,
+        depositWallet: depositWalletAddress,
+        calls: [
+            {
+                target: constants_1.POLYGON_PUSD_ADDRESS,
+                value: '0',
+                data: (0, calldata_1.encodeApprove)(constants_2.POLYMARKET_COLLATERAL_OFFRAMP_POLYGON, amount),
+            },
+            {
+                target: constants_2.POLYMARKET_COLLATERAL_OFFRAMP_POLYGON,
+                value: '0',
+                data: (0, calldata_1.encodeUnwrap)({
+                    asset: constants_1.POLYGON_USDCE_ADDRESS,
+                    recipient: relayDepositAddress,
+                    amount,
+                }),
+            },
+        ],
+    });
+    return { ...result, preSubmitUsdceBalance };
+}
+exports.submitPolymarketWithdraw = submitPolymarketWithdraw;
+async function sweepPolymarketDepositWallet(from, messenger, options) {
+    const isRefund = options.relayStatus === 'refund' || options.relayStatus === 'refunded';
+    const waitForBalanceAbove = isRefund
+        ? options.preSubmitUsdceBalance
+        : undefined;
+    const depositWalletAddress = await getDepositWalletAddress(messenger, from);
+    const usdceBalance = await readDepositWalletUsdceBalance(messenger, depositWalletAddress, waitForBalanceAbove);
+    if (usdceBalance === undefined) {
+        return;
+    }
+    if (usdceBalance === 0n) {
+        log('USDC.e sweep: nothing to wrap');
+        return;
+    }
+    if (waitForBalanceAbove !== undefined && usdceBalance > waitForBalanceAbove) {
+        log('USDC.e sweep: waiting for relayer RPC to catch up to new balance');
+        await new Promise((resolve) => setTimeout(resolve, constants_2.SWEEP_RELAYER_SETTLE_DELAY_MS));
+    }
+    try {
+        await submitDepositWalletBatch(messenger, {
+            eoa: from,
+            depositWallet: depositWalletAddress,
+            calls: [
+                {
+                    target: constants_1.POLYGON_USDCE_ADDRESS,
+                    value: '0',
+                    data: (0, calldata_1.encodeApprove)(constants_2.POLYMARKET_COLLATERAL_ONRAMP_POLYGON, usdceBalance),
+                },
+                {
+                    target: constants_2.POLYMARKET_COLLATERAL_ONRAMP_POLYGON,
+                    value: '0',
+                    data: (0, calldata_1.encodeWrap)({
+                        asset: constants_1.POLYGON_USDCE_ADDRESS,
+                        recipient: depositWalletAddress,
+                        amount: usdceBalance,
+                    }),
+                },
+            ],
+        });
+    }
+    catch (error) {
+        log('USDC.e sweep: batch submission failed', { error });
+    }
+}
+exports.sweepPolymarketDepositWallet = sweepPolymarketDepositWallet;
+async function readUsdceBalanceOrZero(messenger, depositWalletAddress) {
+    try {
+        const raw = await (0, token_1.getLiveTokenBalance)(messenger, depositWalletAddress, constants_1.CHAIN_ID_POLYGON, constants_1.POLYGON_USDCE_ADDRESS);
+        return BigInt(raw);
+    }
+    catch (error) {
+        log('USDC.e balance read failed, defaulting to zero', { error });
+        return 0n;
+    }
+}
+async function readDepositWalletUsdceBalance(messenger, depositWalletAddress, waitForBalanceAbove) {
+    const shouldRetry = waitForBalanceAbove !== undefined;
+    const maxAttempts = shouldRetry ? constants_2.SWEEP_BALANCE_RETRY_ATTEMPTS : 1;
+    let lastBalance = 0n;
+    for (let attempt = 1; attempt <= maxAttempts; attempt++) {
+        if (attempt > 1) {
+            await new Promise((resolve) => setTimeout(resolve, constants_2.SWEEP_BALANCE_RETRY_DELAY_MS));
+        }
+        try {
+            const raw = await (0, token_1.getLiveTokenBalance)(messenger, depositWalletAddress, constants_1.CHAIN_ID_POLYGON, constants_1.POLYGON_USDCE_ADDRESS);
+            lastBalance = BigInt(raw);
+        }
+        catch (error) {
+            log('USDC.e sweep: failed to read deposit wallet balance', { error });
+            return undefined;
+        }
+        log('USDC.e sweep: deposit wallet balance', {
+            depositWalletAddress,
+            balance: lastBalance.toString(),
+            attempt,
+            waitForBalanceAbove: waitForBalanceAbove?.toString(),
+        });
+        const hasIncreased = waitForBalanceAbove === undefined || lastBalance > waitForBalanceAbove;
+        if (hasIncreased) {
+            return lastBalance;
+        }
+    }
+    return lastBalance;
+}
+async function getDepositWalletAddress(messenger, eoa) {
+    const depositWalletAddress = await messenger.call('TransactionPayController:polymarketGetDepositWalletAddress', { eoa });
+    log('Polymarket callback: getDepositWalletAddress', {
+        eoa,
+        depositWalletAddress,
+    });
+    return depositWalletAddress;
+}
+async function submitDepositWalletBatch(messenger, params) {
+    log('Polymarket callback: submitDepositWalletBatch', {
+        eoa: params.eoa,
+        depositWallet: params.depositWallet,
+        callCount: params.calls.length,
+    });
+    const result = await messenger.call('TransactionPayController:polymarketSubmitDepositWalletBatch', params);
+    log('Polymarket callback: submitDepositWalletBatch returned', {
+        sourceHash: result.sourceHash,
+    });
+    return result;
+}
+function extractRelayDepositAddress(relayQuote) {
+    const depositStep = relayQuote.steps.find((step) => step.id === 'deposit');
+    if (depositStep?.kind !== 'transaction') {
+        throw new Error('Polymarket deposit wallet withdraw: Relay quote has no deposit step');
+    }
+    const transactionStep = depositStep;
+    const depositCallData = transactionStep.items[0]?.data?.data;
+    if (!depositCallData) {
+        throw new Error('Polymarket deposit wallet withdraw: Relay quote deposit step is missing calldata');
+    }
+    return (0, calldata_1.extractErc20TransferRecipient)(depositCallData);
+}
+//# sourceMappingURL=withdraw.cjs.map

package/dist/strategy/relay/polymarket/withdraw.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"withdraw.cjs","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/withdraw.ts"],"names":[],"mappings":";;;AACA,2CAAqD;AAErD,sDAI4B;AAC5B,gDAAgD;AAMhD,oDAA2D;AAO3D,6CAKoB;AACpB,+CAMqB;AAErB,MAAM,GAAG,GAAG,IAAA,0BAAkB,EAAC,sBAAa,EAAE,qBAAqB,CAAC,CAAC;AAE9D,KAAK,UAAU,qCAAqC,CACzD,IAAuB,EACvB,OAAqB,EACrB,SAA4C;IAE5C,MAAM,oBAAoB,GAAG,MAAM,uBAAuB,CACxD,SAAS,EACT,OAAO,CAAC,IAAI,CACb,CAAC;IAEF,IAAI,CAAC,cAAc,GAAG,iCAAqB,CAAC;IAC5C,IAAI,CAAC,IAAI,GAAG,oBAAoB,CAAC;IACjC,IAAI,CAAC,QAAQ,GAAG,oBAAoB,CAAC;IACrC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;AACrB,CAAC;AAfD,sFAeC;AAEM,KAAK,UAAU,wBAAwB,CAC5C,KAAsC,EACtC,IAAS,EACT,SAA4C;IAE5C,MAAM,oBAAoB,GAAG,MAAM,uBAAuB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAC5E,MAAM,mBAAmB,GAAG,0BAA0B,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACvE,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IAE9C,MAAM,qBAAqB,GAAG,MAAM,sBAAsB,CACxD,SAAS,EACT,oBAAoB,CACrB,CAAC;IAEF,GAAG,CAAC,kDAAkD,EAAE;QACtD,oBAAoB;QACpB,mBAAmB;QACnB,MAAM,EAAE,MAAM,CAAC,QAAQ,EAAE;QACzB,qBAAqB,EAAE,qBAAqB,CAAC,QAAQ,EAAE;KACxD,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,SAAS,EAAE;QACvD,GAAG,EAAE,IAAI;QACT,aAAa,EAAE,oBAAoB;QACnC,KAAK,EAAE;YACL;gBACE,MAAM,EAAE,gCAAoB;gBAC5B,KAAK,EAAE,GAAG;gBACV,IAAI,EAAE,IAAA,wBAAa,EAAC,iDAAqC,EAAE,MAAM,CAAC;aACnE;YACD;gBACE,MAAM,EAAE,iDAAqC;gBAC7C,KAAK,EAAE,GAAG;gBACV,IAAI,EAAE,IAAA,uBAAY,EAAC;oBACjB,KAAK,EAAE,iCAAqB;oBAC5B,SAAS,EAAE,mBAAmB;oBAC9B,MAAM;iBACP,CAAC;aACH;SACF;KACF,CAAC,CAAC;IAEH,OAAO,EAAE,GAAG,MAAM,EAAE,qBAAqB,EAAE,CAAC;AAC9C,CAAC;AA3CD,4DA2CC;AAEM,KAAK,UAAU,4BAA4B,CAChD,IAAS,EACT,SAA4C,EAC5C,OAGC;IAED,MAAM,QAAQ,GACZ,OAAO,CAAC,WAAW,KAAK,QAAQ,IAAI,OAAO,CAAC,WAAW,KAAK,UAAU,CAAC;IACzE,MAAM,mBAAmB,GAAG,QAAQ;QAClC,CAAC,CAAC,OAAO,CAAC,qBAAqB;QAC/B,CAAC,CAAC,SAAS,CAAC;IAEd,MAAM,oBAAoB,GAAG,MAAM,uBAAuB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAC5E,MAAM,YAAY,GAAG,MAAM,6BAA6B,CACtD,SAAS,EACT,oBAAoB,EACpB,mBAAmB,CACpB,CAAC;IAEF,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;QAC/B,OAAO;IACT,CAAC;IAED,IAAI,YAAY,KAAK,EAAE,EAAE,CAAC;QACxB,GAAG,CAAC,+BAA+B,CAAC,CAAC;QACrC,OAAO;IACT,CAAC;IAED,IAAI,mBAAmB,KAAK,SAAS,IAAI,YAAY,GAAG,mBAAmB,EAAE,CAAC;QAC5E,GAAG,CAAC,kEAAkE,CAAC,CAAC;QACxE,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5B,UAAU,CAAC,OAAO,EAAE,yCAA6B,CAAC,CACnD,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,wBAAwB,CAAC,SAAS,EAAE;YACxC,GAAG,EAAE,IAAI;YACT,aAAa,EAAE,oBAAoB;YACnC,KAAK,EAAE;gBACL;oBACE,MAAM,EAAE,iCAAqB;oBAC7B,KAAK,EAAE,GAAG;oBACV,IAAI,EAAE,IAAA,wBAAa,EACjB,gDAAoC,EACpC,YAAY,CACb;iBACF;gBACD;oBACE,MAAM,EAAE,gDAAoC;oBAC5C,KAAK,EAAE,GAAG;oBACV,IAAI,EAAE,IAAA,qBAAU,EAAC;wBACf,KAAK,EAAE,iCAAqB;wBAC5B,SAAS,EAAE,oBAAoB;wBAC/B,MAAM,EAAE,YAAY;qBACrB,CAAC;iBACH;aACF;SACF,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,uCAAuC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC;AAhED,oEAgEC;AAED,KAAK,UAAU,sBAAsB,CACnC,SAA4C,EAC5C,oBAAyB;IAEzB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,IAAA,2BAAmB,EACnC,SAAS,EACT,oBAAoB,EACpB,4BAAgB,EAChB,iCAAqB,CACtB,CAAC;QACF,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC;IACrB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,gDAAgD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACjE,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,6BAA6B,CAC1C,SAA4C,EAC5C,oBAAyB,EACzB,mBAAuC;IAEvC,MAAM,WAAW,GAAG,mBAAmB,KAAK,SAAS,CAAC;IACtD,MAAM,WAAW,GAAG,WAAW,CAAC,CAAC,CAAC,wCAA4B,CAAC,CAAC,CAAC,CAAC,CAAC;IACnE,IAAI,WAAW,GAAG,EAAE,CAAC;IAErB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;QACxD,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;YAChB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5B,UAAU,CAAC,OAAO,EAAE,wCAA4B,CAAC,CAClD,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,IAAA,2BAAmB,EACnC,SAAS,EACT,oBAAoB,EACpB,4BAAgB,EAChB,iCAAqB,CACtB,CAAC;YACF,WAAW,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC5B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,qDAAqD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACtE,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,GAAG,CAAC,sCAAsC,EAAE;YAC1C,oBAAoB;YACpB,OAAO,EAAE,WAAW,CAAC,QAAQ,EAAE;YAC/B,OAAO;YACP,mBAAmB,EAAE,mBAAmB,EAAE,QAAQ,EAAE;SACrD,CAAC,CAAC;QAEH,MAAM,YAAY,GAChB,mBAAmB,KAAK,SAAS,IAAI,WAAW,GAAG,mBAAmB,CAAC;QAEzE,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,WAAW,CAAC;QACrB,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,KAAK,UAAU,uBAAuB,CACpC,SAA4C,EAC5C,GAAQ;IAER,MAAM,oBAAoB,GAAG,MAAM,SAAS,CAAC,IAAI,CAC/C,4DAA4D,EAC5D,EAAE,GAAG,EAAE,CACR,CAAC;IACF,GAAG,CAAC,8CAA8C,EAAE;QAClD,GAAG;QACH,oBAAoB;KACrB,CAAC,CAAC;IACH,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,KAAK,UAAU,wBAAwB,CACrC,SAA4C,EAC5C,MAIC;IAED,GAAG,CAAC,+CAA+C,EAAE;QACnD,GAAG,EAAE,MAAM,CAAC,GAAG;QACf,aAAa,EAAE,MAAM,CAAC,aAAa;QACnC,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM;KAC/B,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,CACjC,6DAA6D,EAC7D,MAAM,CACP,CAAC;IACF,GAAG,CAAC,wDAAwD,EAAE;QAC5D,UAAU,EAAE,MAAM,CAAC,UAAU;KAC9B,CAAC,CAAC;IACH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,0BAA0B,CAAC,UAAsB;IACxD,MAAM,WAAW,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,SAAS,CAAC,CAAC;IAE3E,IAAI,WAAW,EAAE,IAAI,KAAK,aAAa,EAAE,CAAC;QACxC,MAAM,IAAI,KAAK,CACb,qEAAqE,CACtE,CAAC;IACJ,CAAC;IAED,MAAM,eAAe,GAAG,WAAmC,CAAC;IAC5D,MAAM,eAAe,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC;IAE7D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,MAAM,IAAI,KAAK,CACb,kFAAkF,CACnF,CAAC;IACJ,CAAC;IAED,OAAO,IAAA,wCAA6B,EAAC,eAAe,CAAC,CAAC;AACxD,CAAC","sourcesContent":["import type { Hex } from '@metamask/utils';\nimport { createModuleLogger } from '@metamask/utils';\n\nimport {\n  CHAIN_ID_POLYGON,\n  POLYGON_PUSD_ADDRESS,\n  POLYGON_USDCE_ADDRESS,\n} from '../../../constants';\nimport { projectLogger } from '../../../logger';\nimport type {\n  QuoteRequest,\n  TransactionPayControllerMessenger,\n  TransactionPayQuote,\n} from '../../../types';\nimport { getLiveTokenBalance } from '../../../utils/token';\nimport type {\n  RelayQuote,\n  RelayQuoteRequest,\n  RelayStatus,\n  RelayTransactionStep,\n} from '../types';\nimport {\n  encodeApprove,\n  encodeUnwrap,\n  encodeWrap,\n  extractErc20TransferRecipient,\n} from './calldata';\nimport {\n  POLYMARKET_COLLATERAL_OFFRAMP_POLYGON,\n  POLYMARKET_COLLATERAL_ONRAMP_POLYGON,\n  SWEEP_BALANCE_RETRY_ATTEMPTS,\n  SWEEP_BALANCE_RETRY_DELAY_MS,\n  SWEEP_RELAYER_SETTLE_DELAY_MS,\n} from './constants';\n\nconst log = createModuleLogger(projectLogger, 'polymarket-withdraw');\n\nexport async function applyPolymarketDepositWalletOverrides(\n  body: RelayQuoteRequest,\n  request: QuoteRequest,\n  messenger: TransactionPayControllerMessenger,\n): Promise<void> {\n  const depositWalletAddress = await getDepositWalletAddress(\n    messenger,\n    request.from,\n  );\n\n  body.originCurrency = POLYGON_USDCE_ADDRESS;\n  body.user = depositWalletAddress;\n  body.refundTo = depositWalletAddress;\n  body.useDepositAddress = true;\n  body.strict = true;\n}\n\nexport async function submitPolymarketWithdraw(\n  quote: TransactionPayQuote<RelayQuote>,\n  from: Hex,\n  messenger: TransactionPayControllerMessenger,\n): Promise<{ sourceHash: Hex; preSubmitUsdceBalance: bigint }> {\n  const depositWalletAddress = await getDepositWalletAddress(messenger, from);\n  const relayDepositAddress = extractRelayDepositAddress(quote.original);\n  const amount = BigInt(quote.sourceAmount.raw);\n\n  const preSubmitUsdceBalance = await readUsdceBalanceOrZero(\n    messenger,\n    depositWalletAddress,\n  );\n\n  log('Submitting unwrap batch to Relay deposit address', {\n    depositWalletAddress,\n    relayDepositAddress,\n    amount: amount.toString(),\n    preSubmitUsdceBalance: preSubmitUsdceBalance.toString(),\n  });\n\n  const result = await submitDepositWalletBatch(messenger, {\n    eoa: from,\n    depositWallet: depositWalletAddress,\n    calls: [\n      {\n        target: POLYGON_PUSD_ADDRESS,\n        value: '0',\n        data: encodeApprove(POLYMARKET_COLLATERAL_OFFRAMP_POLYGON, amount),\n      },\n      {\n        target: POLYMARKET_COLLATERAL_OFFRAMP_POLYGON,\n        value: '0',\n        data: encodeUnwrap({\n          asset: POLYGON_USDCE_ADDRESS,\n          recipient: relayDepositAddress,\n          amount,\n        }),\n      },\n    ],\n  });\n\n  return { ...result, preSubmitUsdceBalance };\n}\n\nexport async function sweepPolymarketDepositWallet(\n  from: Hex,\n  messenger: TransactionPayControllerMessenger,\n  options: {\n    relayStatus: RelayStatus | 'timeout';\n    preSubmitUsdceBalance: bigint;\n  },\n): Promise<void> {\n  const isRefund =\n    options.relayStatus === 'refund' || options.relayStatus === 'refunded';\n  const waitForBalanceAbove = isRefund\n    ? options.preSubmitUsdceBalance\n    : undefined;\n\n  const depositWalletAddress = await getDepositWalletAddress(messenger, from);\n  const usdceBalance = await readDepositWalletUsdceBalance(\n    messenger,\n    depositWalletAddress,\n    waitForBalanceAbove,\n  );\n\n  if (usdceBalance === undefined) {\n    return;\n  }\n\n  if (usdceBalance === 0n) {\n    log('USDC.e sweep: nothing to wrap');\n    return;\n  }\n\n  if (waitForBalanceAbove !== undefined && usdceBalance > waitForBalanceAbove) {\n    log('USDC.e sweep: waiting for relayer RPC to catch up to new balance');\n    await new Promise((resolve) =>\n      setTimeout(resolve, SWEEP_RELAYER_SETTLE_DELAY_MS),\n    );\n  }\n\n  try {\n    await submitDepositWalletBatch(messenger, {\n      eoa: from,\n      depositWallet: depositWalletAddress,\n      calls: [\n        {\n          target: POLYGON_USDCE_ADDRESS,\n          value: '0',\n          data: encodeApprove(\n            POLYMARKET_COLLATERAL_ONRAMP_POLYGON,\n            usdceBalance,\n          ),\n        },\n        {\n          target: POLYMARKET_COLLATERAL_ONRAMP_POLYGON,\n          value: '0',\n          data: encodeWrap({\n            asset: POLYGON_USDCE_ADDRESS,\n            recipient: depositWalletAddress,\n            amount: usdceBalance,\n          }),\n        },\n      ],\n    });\n  } catch (error) {\n    log('USDC.e sweep: batch submission failed', { error });\n  }\n}\n\nasync function readUsdceBalanceOrZero(\n  messenger: TransactionPayControllerMessenger,\n  depositWalletAddress: Hex,\n): Promise<bigint> {\n  try {\n    const raw = await getLiveTokenBalance(\n      messenger,\n      depositWalletAddress,\n      CHAIN_ID_POLYGON,\n      POLYGON_USDCE_ADDRESS,\n    );\n    return BigInt(raw);\n  } catch (error) {\n    log('USDC.e balance read failed, defaulting to zero', { error });\n    return 0n;\n  }\n}\n\nasync function readDepositWalletUsdceBalance(\n  messenger: TransactionPayControllerMessenger,\n  depositWalletAddress: Hex,\n  waitForBalanceAbove: bigint | undefined,\n): Promise<bigint | undefined> {\n  const shouldRetry = waitForBalanceAbove !== undefined;\n  const maxAttempts = shouldRetry ? SWEEP_BALANCE_RETRY_ATTEMPTS : 1;\n  let lastBalance = 0n;\n\n  for (let attempt = 1; attempt <= maxAttempts; attempt++) {\n    if (attempt > 1) {\n      await new Promise((resolve) =>\n        setTimeout(resolve, SWEEP_BALANCE_RETRY_DELAY_MS),\n      );\n    }\n\n    try {\n      const raw = await getLiveTokenBalance(\n        messenger,\n        depositWalletAddress,\n        CHAIN_ID_POLYGON,\n        POLYGON_USDCE_ADDRESS,\n      );\n      lastBalance = BigInt(raw);\n    } catch (error) {\n      log('USDC.e sweep: failed to read deposit wallet balance', { error });\n      return undefined;\n    }\n\n    log('USDC.e sweep: deposit wallet balance', {\n      depositWalletAddress,\n      balance: lastBalance.toString(),\n      attempt,\n      waitForBalanceAbove: waitForBalanceAbove?.toString(),\n    });\n\n    const hasIncreased =\n      waitForBalanceAbove === undefined || lastBalance > waitForBalanceAbove;\n\n    if (hasIncreased) {\n      return lastBalance;\n    }\n  }\n\n  return lastBalance;\n}\n\nasync function getDepositWalletAddress(\n  messenger: TransactionPayControllerMessenger,\n  eoa: Hex,\n): Promise<Hex> {\n  const depositWalletAddress = await messenger.call(\n    'TransactionPayController:polymarketGetDepositWalletAddress',\n    { eoa },\n  );\n  log('Polymarket callback: getDepositWalletAddress', {\n    eoa,\n    depositWalletAddress,\n  });\n  return depositWalletAddress;\n}\n\nasync function submitDepositWalletBatch(\n  messenger: TransactionPayControllerMessenger,\n  params: {\n    eoa: Hex;\n    depositWallet: Hex;\n    calls: { target: Hex; data: Hex; value: string }[];\n  },\n): Promise<{ sourceHash: Hex }> {\n  log('Polymarket callback: submitDepositWalletBatch', {\n    eoa: params.eoa,\n    depositWallet: params.depositWallet,\n    callCount: params.calls.length,\n  });\n  const result = await messenger.call(\n    'TransactionPayController:polymarketSubmitDepositWalletBatch',\n    params,\n  );\n  log('Polymarket callback: submitDepositWalletBatch returned', {\n    sourceHash: result.sourceHash,\n  });\n  return result;\n}\n\nfunction extractRelayDepositAddress(relayQuote: RelayQuote): Hex {\n  const depositStep = relayQuote.steps.find((step) => step.id === 'deposit');\n\n  if (depositStep?.kind !== 'transaction') {\n    throw new Error(\n      'Polymarket deposit wallet withdraw: Relay quote has no deposit step',\n    );\n  }\n\n  const transactionStep = depositStep as RelayTransactionStep;\n  const depositCallData = transactionStep.items[0]?.data?.data;\n\n  if (!depositCallData) {\n    throw new Error(\n      'Polymarket deposit wallet withdraw: Relay quote deposit step is missing calldata',\n    );\n  }\n\n  return extractErc20TransferRecipient(depositCallData);\n}\n"]}

package/dist/strategy/relay/polymarket/withdraw.d.cts

@@ -0,0 +1,13 @@
+import type { Hex } from "@metamask/utils";
+import type { QuoteRequest, TransactionPayControllerMessenger, TransactionPayQuote } from "../../../types.cjs";
+import type { RelayQuote, RelayQuoteRequest, RelayStatus } from "../types.cjs";
+export declare function applyPolymarketDepositWalletOverrides(body: RelayQuoteRequest, request: QuoteRequest, messenger: TransactionPayControllerMessenger): Promise<void>;
+export declare function submitPolymarketWithdraw(quote: TransactionPayQuote<RelayQuote>, from: Hex, messenger: TransactionPayControllerMessenger): Promise<{
+    sourceHash: Hex;
+    preSubmitUsdceBalance: bigint;
+}>;
+export declare function sweepPolymarketDepositWallet(from: Hex, messenger: TransactionPayControllerMessenger, options: {
+    relayStatus: RelayStatus | 'timeout';
+    preSubmitUsdceBalance: bigint;
+}): Promise<void>;
+//# sourceMappingURL=withdraw.d.cts.map

package/dist/strategy/relay/polymarket/withdraw.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"withdraw.d.cts","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/withdraw.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAS3C,OAAO,KAAK,EACV,YAAY,EACZ,iCAAiC,EACjC,mBAAmB,EACpB,2BAAuB;AAExB,OAAO,KAAK,EACV,UAAU,EACV,iBAAiB,EACjB,WAAW,EAEZ,qBAAiB;AAiBlB,wBAAsB,qCAAqC,CACzD,IAAI,EAAE,iBAAiB,EACvB,OAAO,EAAE,YAAY,EACrB,SAAS,EAAE,iCAAiC,GAC3C,OAAO,CAAC,IAAI,CAAC,CAWf;AAED,wBAAsB,wBAAwB,CAC5C,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,EAAE,GAAG,EACT,SAAS,EAAE,iCAAiC,GAC3C,OAAO,CAAC;IAAE,UAAU,EAAE,GAAG,CAAC;IAAC,qBAAqB,EAAE,MAAM,CAAA;CAAE,CAAC,CAuC7D;AAED,wBAAsB,4BAA4B,CAChD,IAAI,EAAE,GAAG,EACT,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE;IACP,WAAW,EAAE,WAAW,GAAG,SAAS,CAAC;IACrC,qBAAqB,EAAE,MAAM,CAAC;CAC/B,GACA,OAAO,CAAC,IAAI,CAAC,CAyDf"}

package/dist/strategy/relay/polymarket/withdraw.d.mts

@@ -0,0 +1,13 @@
+import type { Hex } from "@metamask/utils";
+import type { QuoteRequest, TransactionPayControllerMessenger, TransactionPayQuote } from "../../../types.mjs";
+import type { RelayQuote, RelayQuoteRequest, RelayStatus } from "../types.mjs";
+export declare function applyPolymarketDepositWalletOverrides(body: RelayQuoteRequest, request: QuoteRequest, messenger: TransactionPayControllerMessenger): Promise<void>;
+export declare function submitPolymarketWithdraw(quote: TransactionPayQuote<RelayQuote>, from: Hex, messenger: TransactionPayControllerMessenger): Promise<{
+    sourceHash: Hex;
+    preSubmitUsdceBalance: bigint;
+}>;
+export declare function sweepPolymarketDepositWallet(from: Hex, messenger: TransactionPayControllerMessenger, options: {
+    relayStatus: RelayStatus | 'timeout';
+    preSubmitUsdceBalance: bigint;
+}): Promise<void>;
+//# sourceMappingURL=withdraw.d.mts.map

package/dist/strategy/relay/polymarket/withdraw.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"withdraw.d.mts","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/withdraw.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,GAAG,EAAE,wBAAwB;AAS3C,OAAO,KAAK,EACV,YAAY,EACZ,iCAAiC,EACjC,mBAAmB,EACpB,2BAAuB;AAExB,OAAO,KAAK,EACV,UAAU,EACV,iBAAiB,EACjB,WAAW,EAEZ,qBAAiB;AAiBlB,wBAAsB,qCAAqC,CACzD,IAAI,EAAE,iBAAiB,EACvB,OAAO,EAAE,YAAY,EACrB,SAAS,EAAE,iCAAiC,GAC3C,OAAO,CAAC,IAAI,CAAC,CAWf;AAED,wBAAsB,wBAAwB,CAC5C,KAAK,EAAE,mBAAmB,CAAC,UAAU,CAAC,EACtC,IAAI,EAAE,GAAG,EACT,SAAS,EAAE,iCAAiC,GAC3C,OAAO,CAAC;IAAE,UAAU,EAAE,GAAG,CAAC;IAAC,qBAAqB,EAAE,MAAM,CAAA;CAAE,CAAC,CAuC7D;AAED,wBAAsB,4BAA4B,CAChD,IAAI,EAAE,GAAG,EACT,SAAS,EAAE,iCAAiC,EAC5C,OAAO,EAAE;IACP,WAAW,EAAE,WAAW,GAAG,SAAS,CAAC;IACrC,qBAAqB,EAAE,MAAM,CAAC;CAC/B,GACA,OAAO,CAAC,IAAI,CAAC,CAyDf"}

package/dist/strategy/relay/polymarket/withdraw.mjs

@@ -0,0 +1,164 @@
+import { createModuleLogger } from "@metamask/utils";
+import { CHAIN_ID_POLYGON, POLYGON_PUSD_ADDRESS, POLYGON_USDCE_ADDRESS } from "../../../constants.mjs";
+import { projectLogger } from "../../../logger.mjs";
+import { getLiveTokenBalance } from "../../../utils/token.mjs";
+import { encodeApprove, encodeUnwrap, encodeWrap, extractErc20TransferRecipient } from "./calldata.mjs";
+import { POLYMARKET_COLLATERAL_OFFRAMP_POLYGON, POLYMARKET_COLLATERAL_ONRAMP_POLYGON, SWEEP_BALANCE_RETRY_ATTEMPTS, SWEEP_BALANCE_RETRY_DELAY_MS, SWEEP_RELAYER_SETTLE_DELAY_MS } from "./constants.mjs";
+const log = createModuleLogger(projectLogger, 'polymarket-withdraw');
+export async function applyPolymarketDepositWalletOverrides(body, request, messenger) {
+    const depositWalletAddress = await getDepositWalletAddress(messenger, request.from);
+    body.originCurrency = POLYGON_USDCE_ADDRESS;
+    body.user = depositWalletAddress;
+    body.refundTo = depositWalletAddress;
+    body.useDepositAddress = true;
+    body.strict = true;
+}
+export async function submitPolymarketWithdraw(quote, from, messenger) {
+    const depositWalletAddress = await getDepositWalletAddress(messenger, from);
+    const relayDepositAddress = extractRelayDepositAddress(quote.original);
+    const amount = BigInt(quote.sourceAmount.raw);
+    const preSubmitUsdceBalance = await readUsdceBalanceOrZero(messenger, depositWalletAddress);
+    log('Submitting unwrap batch to Relay deposit address', {
+        depositWalletAddress,
+        relayDepositAddress,
+        amount: amount.toString(),
+        preSubmitUsdceBalance: preSubmitUsdceBalance.toString(),
+    });
+    const result = await submitDepositWalletBatch(messenger, {
+        eoa: from,
+        depositWallet: depositWalletAddress,
+        calls: [
+            {
+                target: POLYGON_PUSD_ADDRESS,
+                value: '0',
+                data: encodeApprove(POLYMARKET_COLLATERAL_OFFRAMP_POLYGON, amount),
+            },
+            {
+                target: POLYMARKET_COLLATERAL_OFFRAMP_POLYGON,
+                value: '0',
+                data: encodeUnwrap({
+                    asset: POLYGON_USDCE_ADDRESS,
+                    recipient: relayDepositAddress,
+                    amount,
+                }),
+            },
+        ],
+    });
+    return { ...result, preSubmitUsdceBalance };
+}
+export async function sweepPolymarketDepositWallet(from, messenger, options) {
+    const isRefund = options.relayStatus === 'refund' || options.relayStatus === 'refunded';
+    const waitForBalanceAbove = isRefund
+        ? options.preSubmitUsdceBalance
+        : undefined;
+    const depositWalletAddress = await getDepositWalletAddress(messenger, from);
+    const usdceBalance = await readDepositWalletUsdceBalance(messenger, depositWalletAddress, waitForBalanceAbove);
+    if (usdceBalance === undefined) {
+        return;
+    }
+    if (usdceBalance === 0n) {
+        log('USDC.e sweep: nothing to wrap');
+        return;
+    }
+    if (waitForBalanceAbove !== undefined && usdceBalance > waitForBalanceAbove) {
+        log('USDC.e sweep: waiting for relayer RPC to catch up to new balance');
+        await new Promise((resolve) => setTimeout(resolve, SWEEP_RELAYER_SETTLE_DELAY_MS));
+    }
+    try {
+        await submitDepositWalletBatch(messenger, {
+            eoa: from,
+            depositWallet: depositWalletAddress,
+            calls: [
+                {
+                    target: POLYGON_USDCE_ADDRESS,
+                    value: '0',
+                    data: encodeApprove(POLYMARKET_COLLATERAL_ONRAMP_POLYGON, usdceBalance),
+                },
+                {
+                    target: POLYMARKET_COLLATERAL_ONRAMP_POLYGON,
+                    value: '0',
+                    data: encodeWrap({
+                        asset: POLYGON_USDCE_ADDRESS,
+                        recipient: depositWalletAddress,
+                        amount: usdceBalance,
+                    }),
+                },
+            ],
+        });
+    }
+    catch (error) {
+        log('USDC.e sweep: batch submission failed', { error });
+    }
+}
+async function readUsdceBalanceOrZero(messenger, depositWalletAddress) {
+    try {
+        const raw = await getLiveTokenBalance(messenger, depositWalletAddress, CHAIN_ID_POLYGON, POLYGON_USDCE_ADDRESS);
+        return BigInt(raw);
+    }
+    catch (error) {
+        log('USDC.e balance read failed, defaulting to zero', { error });
+        return 0n;
+    }
+}
+async function readDepositWalletUsdceBalance(messenger, depositWalletAddress, waitForBalanceAbove) {
+    const shouldRetry = waitForBalanceAbove !== undefined;
+    const maxAttempts = shouldRetry ? SWEEP_BALANCE_RETRY_ATTEMPTS : 1;
+    let lastBalance = 0n;
+    for (let attempt = 1; attempt <= maxAttempts; attempt++) {
+        if (attempt > 1) {
+            await new Promise((resolve) => setTimeout(resolve, SWEEP_BALANCE_RETRY_DELAY_MS));
+        }
+        try {
+            const raw = await getLiveTokenBalance(messenger, depositWalletAddress, CHAIN_ID_POLYGON, POLYGON_USDCE_ADDRESS);
+            lastBalance = BigInt(raw);
+        }
+        catch (error) {
+            log('USDC.e sweep: failed to read deposit wallet balance', { error });
+            return undefined;
+        }
+        log('USDC.e sweep: deposit wallet balance', {
+            depositWalletAddress,
+            balance: lastBalance.toString(),
+            attempt,
+            waitForBalanceAbove: waitForBalanceAbove?.toString(),
+        });
+        const hasIncreased = waitForBalanceAbove === undefined || lastBalance > waitForBalanceAbove;
+        if (hasIncreased) {
+            return lastBalance;
+        }
+    }
+    return lastBalance;
+}
+async function getDepositWalletAddress(messenger, eoa) {
+    const depositWalletAddress = await messenger.call('TransactionPayController:polymarketGetDepositWalletAddress', { eoa });
+    log('Polymarket callback: getDepositWalletAddress', {
+        eoa,
+        depositWalletAddress,
+    });
+    return depositWalletAddress;
+}
+async function submitDepositWalletBatch(messenger, params) {
+    log('Polymarket callback: submitDepositWalletBatch', {
+        eoa: params.eoa,
+        depositWallet: params.depositWallet,
+        callCount: params.calls.length,
+    });
+    const result = await messenger.call('TransactionPayController:polymarketSubmitDepositWalletBatch', params);
+    log('Polymarket callback: submitDepositWalletBatch returned', {
+        sourceHash: result.sourceHash,
+    });
+    return result;
+}
+function extractRelayDepositAddress(relayQuote) {
+    const depositStep = relayQuote.steps.find((step) => step.id === 'deposit');
+    if (depositStep?.kind !== 'transaction') {
+        throw new Error('Polymarket deposit wallet withdraw: Relay quote has no deposit step');
+    }
+    const transactionStep = depositStep;
+    const depositCallData = transactionStep.items[0]?.data?.data;
+    if (!depositCallData) {
+        throw new Error('Polymarket deposit wallet withdraw: Relay quote deposit step is missing calldata');
+    }
+    return extractErc20TransferRecipient(depositCallData);
+}
+//# sourceMappingURL=withdraw.mjs.map

package/dist/strategy/relay/polymarket/withdraw.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"withdraw.mjs","sourceRoot":"","sources":["../../../../src/strategy/relay/polymarket/withdraw.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,wBAAwB;AAErD,OAAO,EACL,gBAAgB,EAChB,oBAAoB,EACpB,qBAAqB,EACtB,+BAA2B;AAC5B,OAAO,EAAE,aAAa,EAAE,4BAAwB;AAMhD,OAAO,EAAE,mBAAmB,EAAE,iCAA6B;AAO3D,OAAO,EACL,aAAa,EACb,YAAY,EACZ,UAAU,EACV,6BAA6B,EAC9B,uBAAmB;AACpB,OAAO,EACL,qCAAqC,EACrC,oCAAoC,EACpC,4BAA4B,EAC5B,4BAA4B,EAC5B,6BAA6B,EAC9B,wBAAoB;AAErB,MAAM,GAAG,GAAG,kBAAkB,CAAC,aAAa,EAAE,qBAAqB,CAAC,CAAC;AAErE,MAAM,CAAC,KAAK,UAAU,qCAAqC,CACzD,IAAuB,EACvB,OAAqB,EACrB,SAA4C;IAE5C,MAAM,oBAAoB,GAAG,MAAM,uBAAuB,CACxD,SAAS,EACT,OAAO,CAAC,IAAI,CACb,CAAC;IAEF,IAAI,CAAC,cAAc,GAAG,qBAAqB,CAAC;IAC5C,IAAI,CAAC,IAAI,GAAG,oBAAoB,CAAC;IACjC,IAAI,CAAC,QAAQ,GAAG,oBAAoB,CAAC;IACrC,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,KAAsC,EACtC,IAAS,EACT,SAA4C;IAE5C,MAAM,oBAAoB,GAAG,MAAM,uBAAuB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAC5E,MAAM,mBAAmB,GAAG,0BAA0B,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACvE,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC;IAE9C,MAAM,qBAAqB,GAAG,MAAM,sBAAsB,CACxD,SAAS,EACT,oBAAoB,CACrB,CAAC;IAEF,GAAG,CAAC,kDAAkD,EAAE;QACtD,oBAAoB;QACpB,mBAAmB;QACnB,MAAM,EAAE,MAAM,CAAC,QAAQ,EAAE;QACzB,qBAAqB,EAAE,qBAAqB,CAAC,QAAQ,EAAE;KACxD,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,MAAM,wBAAwB,CAAC,SAAS,EAAE;QACvD,GAAG,EAAE,IAAI;QACT,aAAa,EAAE,oBAAoB;QACnC,KAAK,EAAE;YACL;gBACE,MAAM,EAAE,oBAAoB;gBAC5B,KAAK,EAAE,GAAG;gBACV,IAAI,EAAE,aAAa,CAAC,qCAAqC,EAAE,MAAM,CAAC;aACnE;YACD;gBACE,MAAM,EAAE,qCAAqC;gBAC7C,KAAK,EAAE,GAAG;gBACV,IAAI,EAAE,YAAY,CAAC;oBACjB,KAAK,EAAE,qBAAqB;oBAC5B,SAAS,EAAE,mBAAmB;oBAC9B,MAAM;iBACP,CAAC;aACH;SACF;KACF,CAAC,CAAC;IAEH,OAAO,EAAE,GAAG,MAAM,EAAE,qBAAqB,EAAE,CAAC;AAC9C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAChD,IAAS,EACT,SAA4C,EAC5C,OAGC;IAED,MAAM,QAAQ,GACZ,OAAO,CAAC,WAAW,KAAK,QAAQ,IAAI,OAAO,CAAC,WAAW,KAAK,UAAU,CAAC;IACzE,MAAM,mBAAmB,GAAG,QAAQ;QAClC,CAAC,CAAC,OAAO,CAAC,qBAAqB;QAC/B,CAAC,CAAC,SAAS,CAAC;IAEd,MAAM,oBAAoB,GAAG,MAAM,uBAAuB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAC5E,MAAM,YAAY,GAAG,MAAM,6BAA6B,CACtD,SAAS,EACT,oBAAoB,EACpB,mBAAmB,CACpB,CAAC;IAEF,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;QAC/B,OAAO;IACT,CAAC;IAED,IAAI,YAAY,KAAK,EAAE,EAAE,CAAC;QACxB,GAAG,CAAC,+BAA+B,CAAC,CAAC;QACrC,OAAO;IACT,CAAC;IAED,IAAI,mBAAmB,KAAK,SAAS,IAAI,YAAY,GAAG,mBAAmB,EAAE,CAAC;QAC5E,GAAG,CAAC,kEAAkE,CAAC,CAAC;QACxE,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5B,UAAU,CAAC,OAAO,EAAE,6BAA6B,CAAC,CACnD,CAAC;IACJ,CAAC;IAED,IAAI,CAAC;QACH,MAAM,wBAAwB,CAAC,SAAS,EAAE;YACxC,GAAG,EAAE,IAAI;YACT,aAAa,EAAE,oBAAoB;YACnC,KAAK,EAAE;gBACL;oBACE,MAAM,EAAE,qBAAqB;oBAC7B,KAAK,EAAE,GAAG;oBACV,IAAI,EAAE,aAAa,CACjB,oCAAoC,EACpC,YAAY,CACb;iBACF;gBACD;oBACE,MAAM,EAAE,oCAAoC;oBAC5C,KAAK,EAAE,GAAG;oBACV,IAAI,EAAE,UAAU,CAAC;wBACf,KAAK,EAAE,qBAAqB;wBAC5B,SAAS,EAAE,oBAAoB;wBAC/B,MAAM,EAAE,YAAY;qBACrB,CAAC;iBACH;aACF;SACF,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,uCAAuC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1D,CAAC;AACH,CAAC;AAED,KAAK,UAAU,sBAAsB,CACnC,SAA4C,EAC5C,oBAAyB;IAEzB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,mBAAmB,CACnC,SAAS,EACT,oBAAoB,EACpB,gBAAgB,EAChB,qBAAqB,CACtB,CAAC;QACF,OAAO,MAAM,CAAC,GAAG,CAAC,CAAC;IACrB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,GAAG,CAAC,gDAAgD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QACjE,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED,KAAK,UAAU,6BAA6B,CAC1C,SAA4C,EAC5C,oBAAyB,EACzB,mBAAuC;IAEvC,MAAM,WAAW,GAAG,mBAAmB,KAAK,SAAS,CAAC;IACtD,MAAM,WAAW,GAAG,WAAW,CAAC,CAAC,CAAC,4BAA4B,CAAC,CAAC,CAAC,CAAC,CAAC;IACnE,IAAI,WAAW,GAAG,EAAE,CAAC;IAErB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;QACxD,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;YAChB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5B,UAAU,CAAC,OAAO,EAAE,4BAA4B,CAAC,CAClD,CAAC;QACJ,CAAC;QAED,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,MAAM,mBAAmB,CACnC,SAAS,EACT,oBAAoB,EACpB,gBAAgB,EAChB,qBAAqB,CACtB,CAAC;YACF,WAAW,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC5B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,GAAG,CAAC,qDAAqD,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YACtE,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,GAAG,CAAC,sCAAsC,EAAE;YAC1C,oBAAoB;YACpB,OAAO,EAAE,WAAW,CAAC,QAAQ,EAAE;YAC/B,OAAO;YACP,mBAAmB,EAAE,mBAAmB,EAAE,QAAQ,EAAE;SACrD,CAAC,CAAC;QAEH,MAAM,YAAY,GAChB,mBAAmB,KAAK,SAAS,IAAI,WAAW,GAAG,mBAAmB,CAAC;QAEzE,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,WAAW,CAAC;QACrB,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,KAAK,UAAU,uBAAuB,CACpC,SAA4C,EAC5C,GAAQ;IAER,MAAM,oBAAoB,GAAG,MAAM,SAAS,CAAC,IAAI,CAC/C,4DAA4D,EAC5D,EAAE,GAAG,EAAE,CACR,CAAC;IACF,GAAG,CAAC,8CAA8C,EAAE;QAClD,GAAG;QACH,oBAAoB;KACrB,CAAC,CAAC;IACH,OAAO,oBAAoB,CAAC;AAC9B,CAAC;AAED,KAAK,UAAU,wBAAwB,CACrC,SAA4C,EAC5C,MAIC;IAED,GAAG,CAAC,+CAA+C,EAAE;QACnD,GAAG,EAAE,MAAM,CAAC,GAAG;QACf,aAAa,EAAE,MAAM,CAAC,aAAa;QACnC,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,MAAM;KAC/B,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,IAAI,CACjC,6DAA6D,EAC7D,MAAM,CACP,CAAC;IACF,GAAG,CAAC,wDAAwD,EAAE;QAC5D,UAAU,EAAE,MAAM,CAAC,UAAU;KAC9B,CAAC,CAAC;IACH,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,0BAA0B,CAAC,UAAsB;IACxD,MAAM,WAAW,GAAG,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,KAAK,SAAS,CAAC,CAAC;IAE3E,IAAI,WAAW,EAAE,IAAI,KAAK,aAAa,EAAE,CAAC;QACxC,MAAM,IAAI,KAAK,CACb,qEAAqE,CACtE,CAAC;IACJ,CAAC;IAED,MAAM,eAAe,GAAG,WAAmC,CAAC;IAC5D,MAAM,eAAe,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,CAAC;IAE7D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,MAAM,IAAI,KAAK,CACb,kFAAkF,CACnF,CAAC;IACJ,CAAC;IAED,OAAO,6BAA6B,CAAC,eAAe,CAAC,CAAC;AACxD,CAAC","sourcesContent":["import type { Hex } from '@metamask/utils';\nimport { createModuleLogger } from '@metamask/utils';\n\nimport {\n  CHAIN_ID_POLYGON,\n  POLYGON_PUSD_ADDRESS,\n  POLYGON_USDCE_ADDRESS,\n} from '../../../constants';\nimport { projectLogger } from '../../../logger';\nimport type {\n  QuoteRequest,\n  TransactionPayControllerMessenger,\n  TransactionPayQuote,\n} from '../../../types';\nimport { getLiveTokenBalance } from '../../../utils/token';\nimport type {\n  RelayQuote,\n  RelayQuoteRequest,\n  RelayStatus,\n  RelayTransactionStep,\n} from '../types';\nimport {\n  encodeApprove,\n  encodeUnwrap,\n  encodeWrap,\n  extractErc20TransferRecipient,\n} from './calldata';\nimport {\n  POLYMARKET_COLLATERAL_OFFRAMP_POLYGON,\n  POLYMARKET_COLLATERAL_ONRAMP_POLYGON,\n  SWEEP_BALANCE_RETRY_ATTEMPTS,\n  SWEEP_BALANCE_RETRY_DELAY_MS,\n  SWEEP_RELAYER_SETTLE_DELAY_MS,\n} from './constants';\n\nconst log = createModuleLogger(projectLogger, 'polymarket-withdraw');\n\nexport async function applyPolymarketDepositWalletOverrides(\n  body: RelayQuoteRequest,\n  request: QuoteRequest,\n  messenger: TransactionPayControllerMessenger,\n): Promise<void> {\n  const depositWalletAddress = await getDepositWalletAddress(\n    messenger,\n    request.from,\n  );\n\n  body.originCurrency = POLYGON_USDCE_ADDRESS;\n  body.user = depositWalletAddress;\n  body.refundTo = depositWalletAddress;\n  body.useDepositAddress = true;\n  body.strict = true;\n}\n\nexport async function submitPolymarketWithdraw(\n  quote: TransactionPayQuote<RelayQuote>,\n  from: Hex,\n  messenger: TransactionPayControllerMessenger,\n): Promise<{ sourceHash: Hex; preSubmitUsdceBalance: bigint }> {\n  const depositWalletAddress = await getDepositWalletAddress(messenger, from);\n  const relayDepositAddress = extractRelayDepositAddress(quote.original);\n  const amount = BigInt(quote.sourceAmount.raw);\n\n  const preSubmitUsdceBalance = await readUsdceBalanceOrZero(\n    messenger,\n    depositWalletAddress,\n  );\n\n  log('Submitting unwrap batch to Relay deposit address', {\n    depositWalletAddress,\n    relayDepositAddress,\n    amount: amount.toString(),\n    preSubmitUsdceBalance: preSubmitUsdceBalance.toString(),\n  });\n\n  const result = await submitDepositWalletBatch(messenger, {\n    eoa: from,\n    depositWallet: depositWalletAddress,\n    calls: [\n      {\n        target: POLYGON_PUSD_ADDRESS,\n        value: '0',\n        data: encodeApprove(POLYMARKET_COLLATERAL_OFFRAMP_POLYGON, amount),\n      },\n      {\n        target: POLYMARKET_COLLATERAL_OFFRAMP_POLYGON,\n        value: '0',\n        data: encodeUnwrap({\n          asset: POLYGON_USDCE_ADDRESS,\n          recipient: relayDepositAddress,\n          amount,\n        }),\n      },\n    ],\n  });\n\n  return { ...result, preSubmitUsdceBalance };\n}\n\nexport async function sweepPolymarketDepositWallet(\n  from: Hex,\n  messenger: TransactionPayControllerMessenger,\n  options: {\n    relayStatus: RelayStatus | 'timeout';\n    preSubmitUsdceBalance: bigint;\n  },\n): Promise<void> {\n  const isRefund =\n    options.relayStatus === 'refund' || options.relayStatus === 'refunded';\n  const waitForBalanceAbove = isRefund\n    ? options.preSubmitUsdceBalance\n    : undefined;\n\n  const depositWalletAddress = await getDepositWalletAddress(messenger, from);\n  const usdceBalance = await readDepositWalletUsdceBalance(\n    messenger,\n    depositWalletAddress,\n    waitForBalanceAbove,\n  );\n\n  if (usdceBalance === undefined) {\n    return;\n  }\n\n  if (usdceBalance === 0n) {\n    log('USDC.e sweep: nothing to wrap');\n    return;\n  }\n\n  if (waitForBalanceAbove !== undefined && usdceBalance > waitForBalanceAbove) {\n    log('USDC.e sweep: waiting for relayer RPC to catch up to new balance');\n    await new Promise((resolve) =>\n      setTimeout(resolve, SWEEP_RELAYER_SETTLE_DELAY_MS),\n    );\n  }\n\n  try {\n    await submitDepositWalletBatch(messenger, {\n      eoa: from,\n      depositWallet: depositWalletAddress,\n      calls: [\n        {\n          target: POLYGON_USDCE_ADDRESS,\n          value: '0',\n          data: encodeApprove(\n            POLYMARKET_COLLATERAL_ONRAMP_POLYGON,\n            usdceBalance,\n          ),\n        },\n        {\n          target: POLYMARKET_COLLATERAL_ONRAMP_POLYGON,\n          value: '0',\n          data: encodeWrap({\n            asset: POLYGON_USDCE_ADDRESS,\n            recipient: depositWalletAddress,\n            amount: usdceBalance,\n          }),\n        },\n      ],\n    });\n  } catch (error) {\n    log('USDC.e sweep: batch submission failed', { error });\n  }\n}\n\nasync function readUsdceBalanceOrZero(\n  messenger: TransactionPayControllerMessenger,\n  depositWalletAddress: Hex,\n): Promise<bigint> {\n  try {\n    const raw = await getLiveTokenBalance(\n      messenger,\n      depositWalletAddress,\n      CHAIN_ID_POLYGON,\n      POLYGON_USDCE_ADDRESS,\n    );\n    return BigInt(raw);\n  } catch (error) {\n    log('USDC.e balance read failed, defaulting to zero', { error });\n    return 0n;\n  }\n}\n\nasync function readDepositWalletUsdceBalance(\n  messenger: TransactionPayControllerMessenger,\n  depositWalletAddress: Hex,\n  waitForBalanceAbove: bigint | undefined,\n): Promise<bigint | undefined> {\n  const shouldRetry = waitForBalanceAbove !== undefined;\n  const maxAttempts = shouldRetry ? SWEEP_BALANCE_RETRY_ATTEMPTS : 1;\n  let lastBalance = 0n;\n\n  for (let attempt = 1; attempt <= maxAttempts; attempt++) {\n    if (attempt > 1) {\n      await new Promise((resolve) =>\n        setTimeout(resolve, SWEEP_BALANCE_RETRY_DELAY_MS),\n      );\n    }\n\n    try {\n      const raw = await getLiveTokenBalance(\n        messenger,\n        depositWalletAddress,\n        CHAIN_ID_POLYGON,\n        POLYGON_USDCE_ADDRESS,\n      );\n      lastBalance = BigInt(raw);\n    } catch (error) {\n      log('USDC.e sweep: failed to read deposit wallet balance', { error });\n      return undefined;\n    }\n\n    log('USDC.e sweep: deposit wallet balance', {\n      depositWalletAddress,\n      balance: lastBalance.toString(),\n      attempt,\n      waitForBalanceAbove: waitForBalanceAbove?.toString(),\n    });\n\n    const hasIncreased =\n      waitForBalanceAbove === undefined || lastBalance > waitForBalanceAbove;\n\n    if (hasIncreased) {\n      return lastBalance;\n    }\n  }\n\n  return lastBalance;\n}\n\nasync function getDepositWalletAddress(\n  messenger: TransactionPayControllerMessenger,\n  eoa: Hex,\n): Promise<Hex> {\n  const depositWalletAddress = await messenger.call(\n    'TransactionPayController:polymarketGetDepositWalletAddress',\n    { eoa },\n  );\n  log('Polymarket callback: getDepositWalletAddress', {\n    eoa,\n    depositWalletAddress,\n  });\n  return depositWalletAddress;\n}\n\nasync function submitDepositWalletBatch(\n  messenger: TransactionPayControllerMessenger,\n  params: {\n    eoa: Hex;\n    depositWallet: Hex;\n    calls: { target: Hex; data: Hex; value: string }[];\n  },\n): Promise<{ sourceHash: Hex }> {\n  log('Polymarket callback: submitDepositWalletBatch', {\n    eoa: params.eoa,\n    depositWallet: params.depositWallet,\n    callCount: params.calls.length,\n  });\n  const result = await messenger.call(\n    'TransactionPayController:polymarketSubmitDepositWalletBatch',\n    params,\n  );\n  log('Polymarket callback: submitDepositWalletBatch returned', {\n    sourceHash: result.sourceHash,\n  });\n  return result;\n}\n\nfunction extractRelayDepositAddress(relayQuote: RelayQuote): Hex {\n  const depositStep = relayQuote.steps.find((step) => step.id === 'deposit');\n\n  if (depositStep?.kind !== 'transaction') {\n    throw new Error(\n      'Polymarket deposit wallet withdraw: Relay quote has no deposit step',\n    );\n  }\n\n  const transactionStep = depositStep as RelayTransactionStep;\n  const depositCallData = transactionStep.items[0]?.data?.data;\n\n  if (!depositCallData) {\n    throw new Error(\n      'Polymarket deposit wallet withdraw: Relay quote deposit step is missing calldata',\n    );\n  }\n\n  return extractErc20TransferRecipient(depositCallData);\n}\n"]}

package/dist/strategy/relay/relay-quotes.cjs

@@ -18,6 +18,7 @@ const quote_gas_1 = require("../../utils/quote-gas.cjs");
 const token_1 = require("../../utils/token.cjs");
 const transaction_1 = require("../../utils/transaction.cjs");
 const constants_2 = require("./constants.cjs");
+const withdraw_1 = require("./polymarket/withdraw.cjs");
 const relay_api_1 = require("./relay-api.cjs");
 const relay_max_gas_station_1 = require("./relay-max-gas-station.cjs");
 const log = (0, utils_1.createModuleLogger)(logger_1.projectLogger, 'relay-strategy');
@@ -137,9 +138,14 @@ async function getSingleQuote(request, fullRequest) {
             tradeType: useExactInput ? 'EXACT_INPUT' : 'EXPECTED_OUTPUT',
             user: from,
         };
+        if (request.isPolymarketDepositWallet) {
+            await (0, withdraw_1.applyPolymarketDepositWalletOverrides)(body, request, messenger);
+        }
         // Skip transaction processing for post-quote flows - the original transaction
-        // will be included in the batch separately, not as part of the quote
-        if (!request.isPostQuote) {
+        // will be included in the batch separately, not as part of the quote.
+        // Skip for Polymarket deposit wallet flows - the source is already a
+        // bridged token transfer, not a contract call to embed.
+        if (!request.isPostQuote && !request.isPolymarketDepositWallet) {
             await processTransactions(transaction, request, body, messenger);
         }
         else if (request.refundTo) {