polymarket-terminal/scripts/patch-clob-client.cjs

/**
 * patch-clob-client.cjs
 *
 * Patches @polymarket/clob-client to inject proxy support.
 * Runs automatically via `npm install` (postinstall hook).
 *
 * What it does:
 *   - Adds HttpsProxyAgent import to http-helpers/index.js
 *   - Injects proxy agent into every axios request to polymarket.com
 *   - Reads PROXY_URL from process.env at runtime
 *
 * Uses regex matching so it works regardless of code formatting
 * (minified, prettified, different whitespace, etc.)
 */

const fs = require('fs');
const path = require('path');

const TARGET = path.join(
    __dirname,
    '..',
    'node_modules',
    '@polymarket',
    'clob-client',
    'dist',
    'http-helpers',
    'index.js',
);

if (!fs.existsSync(TARGET)) {
    console.log('[patch] @polymarket/clob-client not found — skipping');
    process.exit(0);
}

let code = fs.readFileSync(TARGET, 'utf8');

// Check if already patched
if (code.includes('getProxyAgent')) {
    console.log('[patch] @polymarket/clob-client already patched — skipping');
    process.exit(0);
}

// ── Step 1: Inject proxy helper code ────────────────────────────────────────

const PROXY_CODE = `
// Proxy support for Polymarket API (auto-patched by scripts/patch-clob-client.cjs)
const https_proxy_agent_1 = require("https-proxy-agent");
const getProxyAgent = () => {
    if (process.env.PROXY_URL) {
        return new https_proxy_agent_1.HttpsProxyAgent(process.env.PROXY_URL);
    }
    return undefined;
};
`;

// Try multiple insertion points — different versions format differently
const insertionPatterns = [
    // Pattern 1: require("browser-or-node") with semicolon
    /require\s*\(\s*["']browser-or-node["']\s*\)\s*;/,
    // Pattern 2: require("axios") line (fallback)
    /require\s*\(\s*["']axios["']\s*\)\s*;/,
    // Pattern 3: any require line near the top (last resort)
    /const\s+\w+\s*=\s*require\s*\(\s*["']axios["']\s*\)/,
];

let inserted = false;
for (const pattern of insertionPatterns) {
    const match = code.match(pattern);
    if (match) {
        code = code.replace(match[0], match[0] + PROXY_CODE);
        console.log(`[patch] Injected proxy helper after: ${match[0].substring(0, 50)}...`);
        inserted = true;
        break;
    }
}

if (!inserted) {
    console.error('[patch] Could not find insertion point for proxy code — skipping');
    process.exit(0);
}

// ── Step 2: Patch the request function to inject proxy agent ────────────────

// The compiled code can have two formats:
//
// Format A (some versions): uses a `config` variable
//   const config = { method, url: endpoint, headers, data, params };
//   ... axios_1.default(config) ...
//
// Format B (VPS/fresh install): inline object passed to axios
//   return yield (0, axios_1.default)({ method, url: endpoint, headers, data, params });

let patched = false;

// ── Try Format A: explicit config variable ──────────────────────────────────
const configRegex = /(?:const|let|var)\s+config\s*=\s*\{\s*method\s*,\s*url\s*:\s*endpoint\s*,\s*headers\s*,\s*data\s*,\s*params\s*\}\s*;/;
const configMatch = code.match(configRegex);

if (configMatch) {
    const PROXY_INJECT = `${configMatch[0]}
    // Add proxy agent for Polymarket API
    if (endpoint && endpoint.includes('polymarket.com')) {
        const agent = getProxyAgent();
        if (agent) {
            config.httpsAgent = agent;
            config.proxy = false;
        }
    }`;
    code = code.replace(configMatch[0], PROXY_INJECT);
    console.log('[patch] Injected proxy agent into request config (Format A)');
    patched = true;
}

// ── Try Format B: inline axios call with object literal ─────────────────────
// Matches: (0, axios_1.default)({ method, url: endpoint, headers, data, params })
// or:      axios_1.default({ method, url: endpoint, headers, data, params })
if (!patched) {
    const inlineRegex = /(\(0,\s*axios_1\.default\)\s*\(\s*)\{\s*method\s*,\s*url\s*:\s*endpoint\s*,\s*headers\s*,\s*data\s*,\s*params\s*\}/;
    const inlineMatch = code.match(inlineRegex);

    if (inlineMatch) {
        // Replace the inline object with a function that builds + injects proxy
        const replacement = `(() => { const _cfg = { method, url: endpoint, headers, data, params }; const _agent = getProxyAgent(); if (_agent && endpoint && endpoint.includes('polymarket.com')) { _cfg.httpsAgent = _agent; _cfg.proxy = false; } return _cfg; })()`;
        // Replace just the object literal part: { method, url: endpoint, ... }
        // Keep the (0, axios_1.default)( prefix
        code = code.replace(inlineMatch[0], inlineMatch[1] + replacement);
        console.log('[patch] Injected proxy agent into inline axios call (Format B)');
        patched = true;
    }
}

// ── Try Format C: any axios call with config object ─────────────────────────
if (!patched) {
    const axiosCallRegex = /axios_1\.default\s*\(\s*\{\s*method/;
    const axiosMatch = code.match(axiosCallRegex);

    if (axiosMatch) {
        // Find the full object literal by counting braces
        const startIdx = code.indexOf(axiosMatch[0]);
        const braceStart = code.indexOf('{', startIdx);
        let depth = 0;
        let braceEnd = braceStart;
        for (let i = braceStart; i < code.length; i++) {
            if (code[i] === '{') depth++;
            if (code[i] === '}') depth--;
            if (depth === 0) { braceEnd = i; break; }
        }
        const objectLiteral = code.substring(braceStart, braceEnd + 1);
        const replacement = `(() => { const _cfg = ${objectLiteral}; const _agent = getProxyAgent(); if (_agent && _cfg.url && _cfg.url.includes('polymarket.com')) { _cfg.httpsAgent = _agent; _cfg.proxy = false; } return _cfg; })()`;
        code = code.substring(0, braceStart) + replacement + code.substring(braceEnd + 1);
        console.log('[patch] Injected proxy agent via brace-matching (Format C)');
        patched = true;
    }
}

if (!patched) {
    console.error('[patch] Could not find request config or axios call — skipping');
    console.error('[patch] File content preview (first 2000 chars):');
    console.error(code.substring(0, 2000));
    process.exit(0);
}

fs.writeFileSync(TARGET, code, 'utf8');
console.log('[patch] @polymarket/clob-client patched with proxy support ✅');