@pineforge/codegen-mcp 0.8.1 → 0.8.3

package/dist/server.js

@@ -0,0 +1,758 @@
+/**
+ * @pineforge/codegen-mcp — shared MCP server factory.
+ *
+ * Single source of truth for all tool logic. Both entrypoints (docker via
+ * `index.ts`, local/in-container via `index.local.ts`) build their server here;
+ * they differ only in which EngineRunner they construct and which tool surface
+ * they request via `opts.imageTools`. Tool COPY is made mode-aware off
+ * `runner.mode` so descriptions stay accurate per backend.
+ */
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { z } from "zod";
+import { mkdtemp, mkdir, writeFile, rm, stat, readFile } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join, resolve, isAbsolute, dirname } from "node:path";
+import { VERSION } from "./version.js";
+import { DEFAULT_IMAGE, stringifyParams, } from "./engine.js";
+// ─── Config ───────────────────────────────────────────────────────────────
+const ALLOW_ANYWHERE = process.env.PINEFORGE_ALLOW_ANYWHERE === "1";
+const BINANCE_SPOT_BASE = "https://api.binance.com";
+const BINANCE_FAPI_BASE = "https://fapi.binance.com";
+const BINANCE_KLINES_LIMIT = 1000;
+const BINANCE_PAGE_DELAY_MS = 200;
+async function runBacktest(runner, args) {
+    const csvPath = await resolveCsvPath(args.ohlcv_csv_path);
+    const image = args.image ?? DEFAULT_IMAGE;
+    const cpp = await runner.transpile(args.source, args.image);
+    const tmp = await mkdtemp(join(tmpdir(), "pineforge-bt-"));
+    const cppPath = join(tmp, "strategy.cpp");
+    await writeFile(cppPath, cpp, "utf8");
+    try {
+        const report = await runner.backtest({
+            cppPath,
+            csvPath,
+            image: args.image,
+            inputs: args.inputs,
+            overrides: args.overrides,
+            runtime: args.runtime,
+        });
+        return {
+            ...report,
+            _meta: { strategy_cpp_bytes: cpp.length, image: runner.mode === "docker" ? image : "local" },
+        };
+    }
+    finally {
+        rm(tmp, { recursive: true, force: true }).catch(() => undefined);
+    }
+}
+async function runBacktestGrid(runner, args) {
+    const csvPath = await resolveCsvPath(args.ohlcv_csv_path);
+    const image = args.image ?? DEFAULT_IMAGE;
+    const includeTrades = args.include_trades === true;
+    const sortBy = args.sort_by ?? "net_pnl";
+    const maxCombos = args.max_combinations ?? 64;
+    const concurrency = Math.max(1, Math.min(args.concurrency ?? 1, 8));
+    const combos = buildCombinations(args.inputs, args.overrides, args.fixed_inputs, args.fixed_overrides);
+    if (combos.length === 0) {
+        throw new Error("no parameter combinations produced — provide at least one inputs/overrides axis");
+    }
+    if (combos.length > maxCombos) {
+        throw new Error(`${combos.length} combinations exceeds max_combinations=${maxCombos}. ` +
+            `Either reduce the grid or raise max_combinations.`);
+    }
+    const cpp = await runner.transpile(args.source, args.image);
+    const tmp = await mkdtemp(join(tmpdir(), "pineforge-grid-"));
+    const cppPath = join(tmp, "strategy.cpp");
+    await writeFile(cppPath, cpp, "utf8");
+    try {
+        const rows = await pMap(combos, concurrency, async (combo) => {
+            try {
+                const report = await runner.backtest({
+                    cppPath, csvPath,
+                    image: args.image,
+                    inputs: combo.inputs, overrides: combo.overrides,
+                    runtime: args.runtime,
+                });
+                return {
+                    ok: true,
+                    inputs: combo.inputs,
+                    overrides: combo.overrides,
+                    summary: report.summary,
+                    applied_inputs: report.applied_inputs,
+                    applied_overrides: report.applied_overrides,
+                    elapsed_seconds: report.elapsed_seconds,
+                    ...(includeTrades ? { trades: report.trades } : {}),
+                };
+            }
+            catch (e) {
+                return {
+                    ok: false,
+                    inputs: combo.inputs,
+                    overrides: combo.overrides,
+                    error: e instanceof Error ? e.message : String(e),
+                };
+            }
+        });
+        const succeeded = rows.filter(r => r.ok);
+        const failed = rows.filter(r => !r.ok);
+        const sortValue = (r) => {
+            if (!r.ok || !r.summary)
+                return -Infinity;
+            const v = r.summary[sortBy];
+            return typeof v === "number" ? v : -Infinity;
+        };
+        succeeded.sort((a, b) => sortValue(b) - sortValue(a));
+        return {
+            total_combinations: combos.length,
+            succeeded: succeeded.length,
+            failed: failed.length,
+            sort_by: sortBy,
+            image: runner.mode === "docker" ? image : "local",
+            _meta: { strategy_cpp_bytes: cpp.length, concurrency },
+            best: succeeded[0] ?? null,
+            results: [...succeeded, ...failed],
+        };
+    }
+    finally {
+        rm(tmp, { recursive: true, force: true }).catch(() => undefined);
+    }
+}
+// ─── Param-grid helpers ───────────────────────────────────────────────────
+function cartesianProduct(grid) {
+    const keys = Object.keys(grid);
+    if (keys.length === 0)
+        return [{}];
+    let acc = [{}];
+    for (const k of keys) {
+        const values = grid[k] ?? [];
+        const next = [];
+        for (const partial of acc) {
+            for (const v of values) {
+                next.push({ ...partial, [k]: String(v) });
+            }
+        }
+        acc = next;
+    }
+    return acc;
+}
+function buildCombinations(inputsGrid, overridesGrid, fixedInputs, fixedOverrides) {
+    const inputCombos = cartesianProduct(inputsGrid ?? {});
+    const overrideCombos = cartesianProduct(overridesGrid ?? {});
+    const fixIn = stringifyParams(fixedInputs);
+    const fixOv = stringifyParams(fixedOverrides);
+    const out = [];
+    for (const i of inputCombos) {
+        for (const o of overrideCombos) {
+            out.push({ inputs: { ...fixIn, ...i }, overrides: { ...fixOv, ...o } });
+        }
+    }
+    return out;
+}
+async function pMap(items, n, fn) {
+    const results = new Array(items.length);
+    let next = 0;
+    const worker = async () => {
+        while (true) {
+            const i = next++;
+            if (i >= items.length)
+                return;
+            results[i] = await fn(items[i], i);
+        }
+    };
+    await Promise.all(Array.from({ length: Math.min(n, items.length) }, worker));
+    return results;
+}
+// ─── Path / CSV helpers ───────────────────────────────────────────────────
+function resolveScopedPath(p, label) {
+    const abs = isAbsolute(p) ? p : resolve(process.cwd(), p);
+    if (!ALLOW_ANYWHERE && !abs.startsWith(process.cwd() + "/") && abs !== process.cwd()) {
+        throw new Error(`${label} path '${abs}' is outside cwd '${process.cwd()}'. ` +
+            `Set PINEFORGE_ALLOW_ANYWHERE=1 to override.`);
+    }
+    return abs;
+}
+async function resolveCsvPath(p) {
+    const abs = resolveScopedPath(p, "OHLCV");
+    const st = await stat(abs).catch(() => null);
+    if (!st || !st.isFile())
+        throw new Error(`OHLCV file not found: ${abs}`);
+    const head = (await readFile(abs, "utf8")).slice(0, 200).split(/\r?\n/)[0] ?? "";
+    const expected = ["timestamp", "open", "high", "low", "close", "volume"];
+    const cols = head.toLowerCase().split(",").map((s) => s.trim());
+    if (expected.some((c, i) => cols[i] !== c)) {
+        throw new Error(`OHLCV header mismatch. Expected: ${expected.join(",")}\nGot: ${head}`);
+    }
+    return abs;
+}
+// ─── Binance public-API client ────────────────────────────────────────────
+const BINANCE_INTERVAL_MS = {
+    "1s": 1_000,
+    "1m": 60_000, "3m": 180_000, "5m": 300_000, "15m": 900_000, "30m": 1_800_000,
+    "1h": 3_600_000, "2h": 7_200_000, "4h": 14_400_000, "6h": 21_600_000,
+    "8h": 28_800_000, "12h": 43_200_000,
+    "1d": 86_400_000, "3d": 259_200_000,
+    "1w": 604_800_000,
+    "1M": 30 * 86_400_000, // approximate
+};
+const BINANCE_INTERVALS = Object.keys(BINANCE_INTERVAL_MS);
+function binanceKlinesUrl(market) {
+    return market === "spot"
+        ? `${BINANCE_SPOT_BASE}/api/v3/klines`
+        : `${BINANCE_FAPI_BASE}/fapi/v1/klines`;
+}
+function binanceExchangeInfoUrl(market) {
+    return market === "spot"
+        ? `${BINANCE_SPOT_BASE}/api/v3/exchangeInfo`
+        : `${BINANCE_FAPI_BASE}/fapi/v1/exchangeInfo`;
+}
+async function binanceGet(url) {
+    const resp = await fetch(url, {
+        headers: { "user-agent": `pineforge-codegen-mcp/${VERSION}` },
+    });
+    const text = await resp.text();
+    if (!resp.ok) {
+        throw new Error(`Binance ${resp.status} for ${url}: ${text.slice(0, 500)}`);
+    }
+    try {
+        return JSON.parse(text);
+    }
+    catch {
+        throw new Error(`Binance non-JSON response: ${text.slice(0, 200)}`);
+    }
+}
+async function fetchKlinesPage(market, symbol, interval, startTime, endTime, limit) {
+    const params = new URLSearchParams({
+        symbol: symbol.toUpperCase(),
+        interval,
+        limit: String(limit),
+    });
+    if (startTime !== undefined)
+        params.set("startTime", String(startTime));
+    if (endTime !== undefined)
+        params.set("endTime", String(endTime));
+    const url = `${binanceKlinesUrl(market)}?${params.toString()}`;
+    const data = await binanceGet(url);
+    if (!Array.isArray(data))
+        throw new Error(`Binance unexpected klines payload: ${JSON.stringify(data).slice(0, 200)}`);
+    return data;
+}
+async function fetchBinanceOhlcv(args) {
+    const intervalMs = BINANCE_INTERVAL_MS[args.interval];
+    if (intervalMs === undefined) {
+        throw new Error(`Unknown interval '${args.interval}'. Valid: ${Object.keys(BINANCE_INTERVAL_MS).join(", ")}`);
+    }
+    if (args.limit <= 0)
+        throw new Error("limit must be > 0");
+    if (args.limit > 100_000)
+        throw new Error("limit must be ≤ 100000 (sanity cap)");
+    const outAbs = resolveScopedPath(args.output_path, "output");
+    await mkdir(dirname(outAbs), { recursive: true });
+    const now = Date.now();
+    const endTime = args.end_time ?? now;
+    const startTime = args.start_time ?? Math.max(0, endTime - args.limit * intervalMs);
+    const collected = [];
+    let cursor = startTime;
+    let pages = 0;
+    while (collected.length < args.limit && cursor <= endTime) {
+        const remaining = args.limit - collected.length;
+        const pageLimit = Math.min(BINANCE_KLINES_LIMIT, remaining);
+        const page = await fetchKlinesPage(args.market, args.symbol, args.interval, cursor, endTime, pageLimit);
+        pages++;
+        if (page.length === 0)
+            break;
+        // Dedup against previous tail (Binance is inclusive on startTime).
+        const tail = collected[collected.length - 1];
+        const lastSeen = tail ? tail[0] : -1;
+        for (const k of page) {
+            if (k[0] > lastSeen)
+                collected.push(k);
+        }
+        if (page.length < pageLimit)
+            break;
+        const lastPage = page[page.length - 1];
+        cursor = lastPage[0] + intervalMs;
+        if (collected.length < args.limit && cursor <= endTime) {
+            await sleep(BINANCE_PAGE_DELAY_MS);
+        }
+    }
+    if (collected.length === 0) {
+        throw new Error(`Binance returned 0 bars for ${args.symbol} ${args.interval} (${args.market})`);
+    }
+    const lines = ["timestamp,open,high,low,close,volume"];
+    for (const k of collected) {
+        const ts = Number(k[0]);
+        if (!Number.isFinite(ts))
+            continue;
+        const o = sanitizeNumeric(k[1]);
+        const h = sanitizeNumeric(k[2]);
+        const l = sanitizeNumeric(k[3]);
+        const c = sanitizeNumeric(k[4]);
+        const v = sanitizeNumeric(k[5]);
+        lines.push(`${ts},${o},${h},${l},${c},${v}`);
+    }
+    const csv = lines.join("\n") + "\n";
+    await writeFile(outAbs, csv, "utf8");
+    const first = collected[0];
+    const last = collected[collected.length - 1];
+    return {
+        output_path: outAbs,
+        market: args.market,
+        symbol: args.symbol.toUpperCase(),
+        interval: args.interval,
+        bars: collected.length,
+        pages,
+        first_open_time: first[0],
+        last_open_time: last[0],
+        first_open_iso: new Date(first[0]).toISOString(),
+        last_open_iso: new Date(last[0]).toISOString(),
+        bytes: Buffer.byteLength(csv, "utf8"),
+    };
+}
+function sanitizeNumeric(raw) {
+    // Binance returns numeric strings; we keep them as strings to avoid
+    // float round-trip loss but reject anything non-numeric so the CSV
+    // stays parseable downstream.
+    const s = String(raw);
+    if (!/^-?\d+(\.\d+)?([eE][-+]?\d+)?$/.test(s)) {
+        throw new Error(`Non-numeric kline field: ${s}`);
+    }
+    return s;
+}
+function sleep(ms) {
+    return new Promise((r) => setTimeout(r, ms));
+}
+// Cache exchangeInfo for 5 min — symbol list rarely changes.
+const SYMBOL_CACHE_TTL_MS = 5 * 60_000;
+const symbolCache = new Map();
+async function listBinanceSymbols(market) {
+    const now = Date.now();
+    const cached = symbolCache.get(market);
+    if (cached && now - cached.ts < SYMBOL_CACHE_TTL_MS)
+        return cached.symbols;
+    const data = await binanceGet(binanceExchangeInfoUrl(market));
+    if (!Array.isArray(data?.symbols)) {
+        throw new Error("Binance exchangeInfo: unexpected payload shape");
+    }
+    symbolCache.set(market, { ts: now, symbols: data.symbols });
+    return data.symbols;
+}
+async function binanceSymbols(args) {
+    const all = await listBinanceSymbols(args.market);
+    const limit = Math.max(1, Math.min(args.limit ?? 200, 2000));
+    const q = args.query?.toUpperCase();
+    const quote = args.quote_asset?.toUpperCase();
+    const base = args.base_asset?.toUpperCase();
+    const status = args.status?.toUpperCase();
+    const ct = args.contract_type?.toUpperCase();
+    const filtered = all.filter((s) => {
+        if (q && !s.symbol.toUpperCase().includes(q))
+            return false;
+        if (quote && s.quoteAsset?.toUpperCase() !== quote)
+            return false;
+        if (base && s.baseAsset?.toUpperCase() !== base)
+            return false;
+        if (status && s.status?.toUpperCase() !== status)
+            return false;
+        if (ct && s.contractType?.toUpperCase() !== ct)
+            return false;
+        return true;
+    });
+    const truncated = filtered.length > limit;
+    const slice = filtered.slice(0, limit).map((s) => ({
+        symbol: s.symbol,
+        base: s.baseAsset,
+        quote: s.quoteAsset,
+        status: s.status,
+        ...(s.contractType ? { contract_type: s.contractType } : {}),
+    }));
+    return {
+        market: args.market,
+        total_symbols: all.length,
+        matched: filtered.length,
+        returned: slice.length,
+        truncated,
+        symbols: slice,
+    };
+}
+// ─── MCP server wiring ────────────────────────────────────────────────────
+const asTextResult = (value) => ({
+    content: [{ type: "text", text: JSON.stringify(value, null, 2) }],
+});
+const ParamMapSchema = z.record(z.string(), z.union([z.string(), z.number(), z.boolean()]));
+const ParamGridSchema = z.record(z.string(), z.array(z.union([z.string(), z.number(), z.boolean()])).min(1));
+const MarketSchema = z.enum(["spot", "usdt_perp"]);
+const IntervalSchema = z.enum(BINANCE_INTERVALS);
+// ─── strategy() header overrides ──────────────────────────────────────────
+//
+// The pineforge-engine runtime accepts a fixed set of strategy(...) header
+// overrides via PINEFORGE_OVERRIDES. Enumerating them here (instead of an
+// opaque ParamMap) gives the agent the exact keys, value types, and enum
+// values it can pass — and lets list_strategy_overrides expose the same
+// catalog at runtime.
+const COMMISSION_TYPES = ["percent", "cash_per_order", "cash_per_contract"];
+const QTY_TYPES = ["fixed", "percent_of_equity", "cash"];
+const CLOSE_ENTRIES_RULES = ["ANY", "FIFO"];
+const STRATEGY_OVERRIDES_CATALOG = {
+    initial_capital: {
+        type: "number",
+        description: "Starting equity in account currency. Mirrors `strategy(initial_capital=...)`.",
+    },
+    pyramiding: {
+        type: "integer",
+        description: "Maximum same-direction entries before further entries are blocked. " +
+            "0 = single position. Mirrors `strategy(pyramiding=...)`.",
+    },
+    slippage: {
+        type: "integer",
+        description: "Per-fill slippage in ticks (mintick units). Mirrors `strategy(slippage=...)`.",
+    },
+    commission_value: {
+        type: "number",
+        description: "Commission magnitude. Units depend on commission_type. " +
+            "Mirrors `strategy(commission_value=...)`.",
+    },
+    commission_type: {
+        type: "enum",
+        enum: COMMISSION_TYPES,
+        description: "Commission units. 'percent' = percent of trade value, " +
+            "'cash_per_order' = fixed cash per order, " +
+            "'cash_per_contract' = fixed cash per contract. " +
+            "Mirrors `strategy(commission_type=...)`.",
+    },
+    default_qty_value: {
+        type: "number",
+        description: "Default order size, interpreted per default_qty_type. " +
+            "Mirrors `strategy(default_qty_value=...)`.",
+    },
+    default_qty_type: {
+        type: "enum",
+        enum: QTY_TYPES,
+        description: "Default order sizing mode. 'fixed' = contracts/shares, " +
+            "'percent_of_equity' = percent of current equity, " +
+            "'cash' = fixed cash amount per order. " +
+            "Mirrors `strategy(default_qty_type=...)`.",
+    },
+    process_orders_on_close: {
+        type: "boolean",
+        description: "When true, market orders submitted on a bar fill at that bar's close " +
+            "instead of waiting for the next bar's open. " +
+            "Mirrors `strategy(process_orders_on_close=...)`.",
+    },
+    close_entries_rule: {
+        type: "enum",
+        enum: CLOSE_ENTRIES_RULES,
+        description: "How `strategy.close(id)` resolves which entries to close. " +
+            "'FIFO' = first-in-first-out (default). " +
+            "'ANY' = match all entries with the same id. " +
+            "Mirrors `strategy(close_entries_rule=...)`.",
+    },
+};
+const StrategyOverridesSchema = z.object({
+    initial_capital: z.number().describe(STRATEGY_OVERRIDES_CATALOG.initial_capital.description).optional(),
+    pyramiding: z.number().int().min(0).describe(STRATEGY_OVERRIDES_CATALOG.pyramiding.description).optional(),
+    slippage: z.number().int().min(0).describe(STRATEGY_OVERRIDES_CATALOG.slippage.description).optional(),
+    commission_value: z.number().min(0).describe(STRATEGY_OVERRIDES_CATALOG.commission_value.description).optional(),
+    commission_type: z.enum(COMMISSION_TYPES).describe(STRATEGY_OVERRIDES_CATALOG.commission_type.description).optional(),
+    default_qty_value: z.number().describe(STRATEGY_OVERRIDES_CATALOG.default_qty_value.description).optional(),
+    default_qty_type: z.enum(QTY_TYPES).describe(STRATEGY_OVERRIDES_CATALOG.default_qty_type.description).optional(),
+    process_orders_on_close: z.boolean().describe(STRATEGY_OVERRIDES_CATALOG.process_orders_on_close.description).optional(),
+    close_entries_rule: z.enum(CLOSE_ENTRIES_RULES).describe(STRATEGY_OVERRIDES_CATALOG.close_entries_rule.description).optional(),
+}).strict();
+const StrategyOverridesGridSchema = z.object({
+    initial_capital: z.array(z.number()).min(1).optional(),
+    pyramiding: z.array(z.number().int().min(0)).min(1).optional(),
+    slippage: z.array(z.number().int().min(0)).min(1).optional(),
+    commission_value: z.array(z.number().min(0)).min(1).optional(),
+    commission_type: z.array(z.enum(COMMISSION_TYPES)).min(1).optional(),
+    default_qty_value: z.array(z.number()).min(1).optional(),
+    default_qty_type: z.array(z.enum(QTY_TYPES)).min(1).optional(),
+    process_orders_on_close: z.array(z.boolean()).min(1).optional(),
+    close_entries_rule: z.array(z.enum(CLOSE_ENTRIES_RULES)).min(1).optional(),
+}).strict();
+// ─── Runtime args (run_backtest_full args, NOT strategy() header) ─────────
+//
+// These are passed to pineforge-engine's run_backtest_full() rather than the
+// per-strategy override table. They control how the engine consumes input
+// bars (timeframe semantics, bar magnifier sub-bar sampling) and never
+// appear in the Pine source. The engine validates them and surfaces any
+// mismatch (e.g. script_tf finer than input_tf) through
+// strategy_get_last_error() — agents see that as
+// {"engine":"pineforge","error":"..."} on stdout, exit code 1.
+const MAGNIFIER_DISTS = [
+    "uniform", "cosine", "triangle", "endpoints", "front_loaded", "back_loaded",
+];
+const RUNTIME_ARGS_CATALOG = {
+    input_tf: {
+        type: "enum",
+        description: "Chart bar timeframe — the resolution of the OHLCV CSV being fed in. " +
+            "Use Pine timeframe strings: '1', '5', '15', '60', '240' (minutes), " +
+            "'D', '1D', 'W', '1W', 'M', '1M'. Empty / omitted = the engine " +
+            "auto-detects the timeframe from the gap between the first two bars. " +
+            "Set explicitly when the CSV's resolution is ambiguous, when fewer " +
+            "than 2 bars are present, or when registering request.security() " +
+            "evaluators that need the chart TF stated up front.",
+    },
+    script_tf: {
+        type: "enum",
+        description: "Strategy / script timeframe — the resolution at which the strategy's " +
+            "on_bar() runs. Empty / omitted = same as input_tf (no aggregation). " +
+            "Set to a coarser timeframe (e.g. input_tf='5', script_tf='60') to " +
+            "have the engine aggregate the input bars into higher-TF script bars " +
+            "before invoking the strategy. MUST be coarser than or equal to " +
+            "input_tf — finer values are rejected with the error " +
+            "'script timeframe must be coarser than or equal to input timeframe'. " +
+            "When the Pine source uses request.security() to pull a higher TF, " +
+            "leave script_tf empty (the security TF is encoded inside the Pine " +
+            "source, not here).",
+    },
+    bar_magnifier: {
+        type: "boolean",
+        description: "When true, the engine samples each input bar's price path at " +
+            "magnifier_samples sub-points and walks stop / limit / trailing " +
+            "orders against the sub-bars instead of the bar's OHLC corners. " +
+            "Improves intra-bar fill realism for strategies that hinge on " +
+            "wick fills, but multiplies engine work by ~magnifier_samples. " +
+            "Default false. Most useful when input_tf is coarse (15m+) and " +
+            "the strategy uses tight intra-bar exits.",
+    },
+    magnifier_samples: {
+        type: "integer",
+        description: "Sub-bar sample count when bar_magnifier=true. Minimum 2 " +
+            "(open + close); typical range 4–16. Higher values give smoother " +
+            "intra-bar fill simulation at linear cost. Ignored when " +
+            "bar_magnifier=false. Default 4.",
+    },
+    magnifier_dist: {
+        type: "enum",
+        enum: MAGNIFIER_DISTS,
+        description: "Distribution of sub-bar samples along the OHLC path when " +
+            "bar_magnifier=true. 'endpoints' (default) always includes the " +
+            "exact O/H/L/C corners and uniformly fills between them — best for " +
+            "TradingView parity. 'uniform' = equal spacing. 'cosine' = denser " +
+            "near segment endpoints. 'triangle' = denser near segment midpoints. " +
+            "'front_loaded' / 'back_loaded' = denser near open / close, " +
+            "simulating opening- or closing-impulse price action.",
+    },
+};
+const RuntimeArgsSchema = z.object({
+    input_tf: z.string().describe(RUNTIME_ARGS_CATALOG.input_tf.description).optional(),
+    script_tf: z.string().describe(RUNTIME_ARGS_CATALOG.script_tf.description).optional(),
+    bar_magnifier: z.boolean().describe(RUNTIME_ARGS_CATALOG.bar_magnifier.description).optional(),
+    magnifier_samples: z.number().int().min(2).describe(RUNTIME_ARGS_CATALOG.magnifier_samples.description).optional(),
+    magnifier_dist: z.enum(MAGNIFIER_DISTS).describe(RUNTIME_ARGS_CATALOG.magnifier_dist.description).optional(),
+}).strict();
+// ─── Server factory ────────────────────────────────────────────────────────
+/**
+ * Build the MCP server with the common tool surface registered against
+ * `runner`. When `opts.imageTools` is true the Docker image-management tools
+ * (`pull_engine_image`, `check_engine_image`) are added; otherwise a read-only
+ * `engine_info` tool is registered instead (the in-container/local product,
+ * where the engine is baked in and there is nothing to pull/check).
+ *
+ * The caller is responsible for connecting a transport.
+ */
+export function createServer(runner, opts) {
+    const server = new McpServer({ name: "pineforge-codegen-mcp", version: VERSION }, { capabilities: { tools: {} } });
+    // Mode-aware run-location clauses so tool copy stays accurate per backend
+    // without duplicating tool logic.
+    const transpileWhere = runner.mode === "docker"
+        ? "using the pineforge-codegen transpiler bundled in the pineforge-engine Docker image"
+        : "using the pineforge-codegen transpiler bundled in this image (in-process, no host Docker)";
+    const backtestWhere = runner.mode === "docker"
+        ? "via the pineforge-engine Docker image on the user's local machine. Fully local — " +
+            "transpile + backtest run in-container; nothing leaves the box, no API key."
+        : "via the bundled pineforge-engine (in-process, no host Docker daemon). Fully local — " +
+            "transpile + backtest run in this image; nothing leaves the box, no API key.";
+    const gridWhere = runner.mode === "docker"
+        ? "transpile the Pine source ONCE (locally, in-container)"
+        : "transpile the Pine source ONCE (in-process, no host Docker)";
+    const concurrencyHelp = runner.mode === "docker"
+        ? "Set concurrency > 1 to run backtests in parallel — each docker container " +
+            "has its own startup overhead, so 2-4 is usually plenty."
+        : "Set concurrency > 1 to run backtests in parallel — 2-4 is usually plenty.";
+    server.registerTool("transpile_pine", {
+        description: "Transpile PineScript v6 source to a C++ translation unit locally, " +
+            transpileWhere + ". " +
+            "No API key, no network — source never leaves the machine. Returns the " +
+            "generated C++ as text. Use backtest_pine if you also want to run the strategy.",
+        inputSchema: {
+            source: z.string().describe("PineScript v6 source (must include //@version=6)."),
+            image: z.string().optional().describe(`Docker image override. Defaults to ${DEFAULT_IMAGE}.`),
+        },
+    }, async ({ source, image }) => ({
+        content: [{
+                type: "text",
+                text: await runner.transpile(source, image),
+            }],
+    }));
+    server.registerTool("backtest_pine", {
+        description: "Transpile a PineScript v6 strategy and run it against an OHLCV CSV " +
+            backtestWhere + " " +
+            "Optional `inputs` overrides input.*() named values from the Pine source " +
+            "(keys = the second arg of input.*(...) calls, e.g. 'Fast Length'). " +
+            "Optional `overrides` overrides strategy(...) header fields " +
+            "(initial_capital, commission_value, default_qty_value, pyramiding, " +
+            "slippage, default_qty_type, commission_type, process_orders_on_close). " +
+            "Returns the parsed JSON report (summary, trades, applied_inputs, " +
+            "applied_overrides, elapsed_seconds). Use backtest_pine_grid for sweeps.",
+        inputSchema: {
+            source: z.string().describe("PineScript v6 source."),
+            ohlcv_csv_path: z.string().describe("Absolute or cwd-relative path to OHLCV CSV with header " +
+                "'timestamp,open,high,low,close,volume' (timestamp = UNIX ms UTC)."),
+            image: z.string().optional().describe(`Docker image override. Defaults to ${DEFAULT_IMAGE}.`),
+            inputs: ParamMapSchema.optional().describe("Map of Pine input.*() names → value (string/number/bool). " +
+                "Sent as PINEFORGE_INPUTS env var to the runtime."),
+            overrides: StrategyOverridesSchema.optional().describe("strategy(...) header overrides. Each key maps to a single argument " +
+                "of the Pine `strategy()` call; only the keys you set are applied. " +
+                "Sent as PINEFORGE_OVERRIDES env var. Call list_engine_params " +
+                "for the full catalog with types and enum values."),
+            runtime: RuntimeArgsSchema.optional().describe("Engine runtime args (NOT strategy() header) controlling timeframe " +
+                "semantics and intra-bar fill simulation. input_tf / script_tf set " +
+                "the chart and strategy timeframes — script_tf must be coarser " +
+                "than or equal to input_tf or the engine rejects the run. " +
+                "bar_magnifier + magnifier_samples + magnifier_dist enable sub-bar " +
+                "price-path sampling for tighter stop / limit fills. Each field is " +
+                "optional and only forwarded to the engine when set. Call " +
+                "list_engine_params for the full catalog."),
+        },
+    }, async ({ source, ohlcv_csv_path, image, inputs, overrides, runtime }) => asTextResult(await runBacktest(runner, { source, ohlcv_csv_path, image, inputs, overrides, runtime })));
+    server.registerTool("backtest_pine_grid", {
+        description: "Run a parameter sweep: " + gridWhere + ", " +
+            "then re-run the same compiled strategy against the OHLCV CSV across the " +
+            "cartesian product of `inputs` × `overrides` grids. Returns a ranked list " +
+            "of {inputs, overrides, summary, elapsed_seconds} entries sorted by `sort_by` " +
+            "descending, plus the top entry under `best`. Cap: max_combinations (default " +
+            "64). " + concurrencyHelp,
+        inputSchema: {
+            source: z.string().describe("PineScript v6 source."),
+            ohlcv_csv_path: z.string().describe("Path to OHLCV CSV (same format as backtest_pine)."),
+            image: z.string().optional().describe(`Docker image override. Defaults to ${DEFAULT_IMAGE}.`),
+            inputs: ParamGridSchema.optional().describe("Grid of input.*() names → list of values to sweep. " +
+                "Example: {\"Fast Length\": [8, 12, 19], \"Slow Length\": [21, 26, 39]}"),
+            overrides: StrategyOverridesGridSchema.optional().describe("Grid of strategy(...) header overrides → list of values, one axis " +
+                "per key. Example: {\"default_qty_value\": [1, 5], \"commission_value\": [0.04]}. " +
+                "Call list_engine_params for the full catalog with types and enum values."),
+            fixed_inputs: ParamMapSchema.optional().describe("Inputs applied to every combo (overridden by per-combo `inputs` keys)."),
+            fixed_overrides: StrategyOverridesSchema.optional().describe("Overrides applied to every combo (overridden by per-combo `overrides` keys)."),
+            runtime: RuntimeArgsSchema.optional().describe("Engine runtime args applied to every combo in the sweep. Same " +
+                "shape as backtest_pine.runtime — input_tf / script_tf / " +
+                "bar_magnifier / magnifier_samples / magnifier_dist. Currently " +
+                "fixed across the grid (not swept); add to the grid axes through " +
+                "future versions if you need to vary them."),
+            max_combinations: z.number().int().min(1).max(1024).optional()
+                .describe("Hard cap on combinations. Default 64."),
+            concurrency: z.number().int().min(1).max(8).optional()
+                .describe("Parallel backtests. Default 1."),
+            include_trades: z.boolean().optional()
+                .describe("Include the per-trade list in each result. Default false (saves tokens)."),
+            sort_by: z.enum(["net_pnl", "win_rate_pct", "max_drawdown", "total_trades"])
+                .optional().describe("summary.* field to rank by, descending. Default net_pnl."),
+        },
+    }, async (args) => asTextResult(await runBacktestGrid(runner, args)));
+    server.registerTool("fetch_binance_ohlcv", {
+        description: "Fetch OHLCV candles from Binance public API and write a backtest-ready " +
+            "CSV (header: timestamp,open,high,low,close,volume; timestamp = open time " +
+            "in UNIX ms UTC). Supports `spot` and `usdt_perp` (USDT-margined " +
+            "perpetual futures). Requests larger than 1000 bars are paginated " +
+            "automatically. The output path must " +
+            "live inside the MCP cwd unless PINEFORGE_ALLOW_ANYWHERE=1.",
+        inputSchema: {
+            symbol: z.string().min(2).describe("Binance symbol, e.g. 'BTCUSDT'. Use binance_symbols to validate."),
+            interval: IntervalSchema.describe("Kline interval. Spot supports 1s + 1m..1M; usdt_perp supports 1m..1M (no 1s)."),
+            market: MarketSchema.optional().describe("'spot' (default) or 'usdt_perp'."),
+            limit: z.number().int().min(1).max(100_000).optional()
+                .describe("Total bars to fetch. Default 1000. Paginated above 1000."),
+            start_time: z.number().int().nonnegative().optional()
+                .describe("UNIX ms UTC. If unset, derived from end_time/now and limit."),
+            end_time: z.number().int().nonnegative().optional()
+                .describe("UNIX ms UTC. Defaults to now."),
+            output_path: z.string().describe("Path to write the CSV (will create parent dirs as needed)."),
+        },
+    }, async ({ symbol, interval, market, limit, start_time, end_time, output_path }) => asTextResult(await fetchBinanceOhlcv({
+        symbol,
+        interval,
+        market: market ?? "spot",
+        limit: limit ?? 1000,
+        start_time,
+        end_time,
+        output_path,
+    })));
+    server.registerTool("binance_symbols", {
+        description: "List/validate symbols available on the Binance public API for OHLCV " +
+            "fetching. Filters: `query` (substring of the symbol), `quote_asset` " +
+            "(e.g. 'USDT'), `base_asset` (e.g. 'BTC'), `status` (e.g. 'TRADING'), " +
+            "`contract_type` (futures only, e.g. 'PERPETUAL'). Results are " +
+            "cached 5 min in process. Free.",
+        inputSchema: {
+            market: MarketSchema.describe("'spot' or 'usdt_perp'."),
+            query: z.string().optional().describe("Case-insensitive substring of the symbol."),
+            quote_asset: z.string().optional().describe("Filter by quote asset (e.g. 'USDT')."),
+            base_asset: z.string().optional().describe("Filter by base asset (e.g. 'BTC')."),
+            status: z.string().optional().describe("Filter by status. 'TRADING' returns active only."),
+            contract_type: z.string().optional().describe("Futures only. 'PERPETUAL' for usdt_perp swaps."),
+            limit: z.number().int().min(1).max(2000).optional()
+                .describe("Max symbols to return. Default 200."),
+        },
+    }, async (args) => asTextResult(await binanceSymbols(args)));
+    server.registerTool("list_engine_params", {
+        description: "Returns the full catalog of engine knobs accepted by backtest_pine / " +
+            "backtest_pine_grid in two groups: strategy_overrides (the 9 " +
+            "strategy(...) header fields the runtime reads via " +
+            "PINEFORGE_OVERRIDES — initial_capital, pyramiding, slippage, " +
+            "commission_value, commission_type, default_qty_value, " +
+            "default_qty_type, process_orders_on_close, close_entries_rule) and " +
+            "runtime_args (input_tf, script_tf, bar_magnifier, " +
+            "magnifier_samples, magnifier_dist — args to run_backtest_full, " +
+            "NOT part of the strategy() header). Each entry is " +
+            "{key, type, enum?, description}. Does not run the engine. " +
+            "Use this to discover what knobs the engine exposes " +
+            "before issuing a backtest.",
+        inputSchema: {},
+    }, async () => asTextResult({
+        strategy_overrides: Object.entries(STRATEGY_OVERRIDES_CATALOG).map(([key, spec]) => ({
+            key,
+            type: spec.type,
+            ...(spec.enum ? { enum: [...spec.enum] } : {}),
+            description: spec.description,
+        })),
+        runtime_args: Object.entries(RUNTIME_ARGS_CATALOG).map(([key, spec]) => ({
+            key,
+            type: spec.type,
+            ...(spec.enum ? { enum: [...spec.enum] } : {}),
+            description: spec.description,
+        })),
+    }));
+    if (opts.imageTools) {
+        server.registerTool("pull_engine_image", {
+            description: "Run `docker pull` for the pineforge-engine runtime image on the user's " +
+                "machine. Useful before the first backtest_pine call.",
+            inputSchema: {
+                image: z.string().optional().describe(`Image to pull. Defaults to ${DEFAULT_IMAGE}.`),
+            },
+        }, async ({ image }) => asTextResult(await runner.pullImage(image ?? DEFAULT_IMAGE)));
+        server.registerTool("check_engine_image", {
+            description: "Check whether the local pineforge-engine Docker image is up to date " +
+                "with the registry. Compares per-platform manifest digests via " +
+                "`docker manifest inspect --verbose` (no image layers downloaded). " +
+                "Returns up_to_date + recommend_pull. With auto_pull=true, runs " +
+                "`docker pull` in the same call when the local image is stale or " +
+                "missing. Note: this is independent of " +
+                "the MCP server's own version (`@pineforge/codegen-mcp`); the MCP " +
+                "version and the engine image version evolve separately.",
+            inputSchema: {
+                image: z.string().optional().describe(`Image to check. Defaults to ${DEFAULT_IMAGE}.`),
+                auto_pull: z.boolean().optional().describe("If true and the image is stale or missing, run `docker pull` in " +
+                    "the same call. Default false (report only)."),
+            },
+        }, async ({ image, auto_pull }) => asTextResult(await runner.checkImage(image ?? DEFAULT_IMAGE, auto_pull === true)));
+    }
+    else {
+        server.registerTool("engine_info", {
+            description: "Report the bundled backtest engine: mode, baked-in flag, and version. " +
+                "This image runs the engine in-process (no host Docker daemon needed).",
+            inputSchema: {},
+        }, async () => asTextResult(await runner.engineInfo()));
+    }
+    return server;
+}
+//# sourceMappingURL=server.js.map