#!/usr/bin/env bun
/**
 * External browser dataset benchmark runner (#10333).
 *
 * Runs committed Mind2Web/WebArena-style fixtures through the real
 * plugin-browser BROWSER command router and writes an evidence artifact.
 */

import { mkdir, writeFile } from "node:fs/promises";
import path from "node:path";
import { fileURLToPath } from "node:url";
import {
  EXTERNAL_WEB_DATASET_TASKS,
  NoopPolicy,
  OraclePolicy,
  runBenchmarkSuite,
  WrongPolicy,
} from "../src/benchmark/index.ts";

const here = path.dirname(fileURLToPath(import.meta.url));
const repoRoot = path.resolve(here, "../../..");

function parseArgs(argv) {
  const opts = { policy: "oracle", out: null };
  for (let i = 0; i < argv.length; i++) {
    const arg = argv[i];
    if (arg === "--policy") opts.policy = argv[++i];
    else if (arg === "--out") opts.out = argv[++i];
  }
  return opts;
}

function makePolicy(name) {
  if (name === "noop") return new NoopPolicy();
  if (name === "wrong") return new WrongPolicy();
  return new OraclePolicy();
}

async function main() {
  const opts = parseArgs(process.argv.slice(2));
  const outPath =
    opts.out ??
    path.join(
      repoRoot,
      "test-results/evidence/10333-browser-external-dataset",
      `external-dataset-${opts.policy}-run.json`,
    );

  const report = await runBenchmarkSuite({
    benchmarkName: "external-web-dataset",
    tasks: EXTERNAL_WEB_DATASET_TASKS,
    seeds: [0],
    policy: makePolicy(opts.policy),
  });
  const stamped = { generatedAt: new Date().toISOString(), ...report };

  await mkdir(path.dirname(outPath), { recursive: true });
  await writeFile(outPath, `${JSON.stringify(stamped, null, 2)}\n`);

  console.log(
    `\nExternal dataset benchmark - engine=${report.engine} policy=${report.policy}`,
  );
  console.log("-".repeat(64));
  for (const task of report.summary.byTask) {
    const mark = task.solved === task.total ? "PASS" : "FAIL";
    console.log(
      `  ${mark} ${task.taskId.padEnd(28)} ${task.solved}/${task.total}`,
    );
  }
  console.log("-".repeat(64));
  console.log(
    `  TOTAL solved ${report.summary.solved}/${report.summary.total} ` +
      `(success rate ${(report.summary.successRate * 100).toFixed(1)}%)`,
  );
  console.log(`\n  artifact -> ${path.relative(repoRoot, outPath)}\n`);
}

main().catch((error) => {
  console.error(error);
  process.exit(1);
});