#!/usr/bin/env npx tsx

import assert from "node:assert";
import { createE2ETestContext, type TestDaemonContext } from "../helpers/test-daemon.ts";

interface E2EContext extends TestDaemonContext {
  paseo: (
    args: string[],
    opts?: { timeout?: number; cwd?: string },
  ) => Promise<{
    exitCode: number;
    stdout: string;
    stderr: string;
  }>;
}

const schema = JSON.stringify({
  type: "object",
  properties: {
    summary: { type: "string" },
  },
  required: ["summary"],
  additionalProperties: false,
});

const impossibleSchema = JSON.stringify({
  type: "object",
  properties: {
    summary: { type: "string" },
  },
  required: ["summary"],
  allOf: [
    { properties: { summary: { type: "string" } } },
    { properties: { summary: { type: "number" } } },
  ],
});

const OPEN_CODE_STRUCTURED_MODEL = "opencode/gpt-5-nano";

let ctx: E2EContext;

async function setup(): Promise<void> {
  ctx = await createE2ETestContext({ timeout: 180000 });
}

async function cleanup(): Promise<void> {
  if (ctx) {
    await ctx.stop();
  }
}

async function runProviderCase(input: {
  provider: "claude" | "codex" | "opencode";
  mode: string;
  model: string;
}): Promise<void> {
  const result = await ctx.paseo(
    [
      "run",
      "--provider",
      input.provider,
      "--mode",
      input.mode,
      "--model",
      input.model,
      "--output-schema",
      schema,
      `Return valid JSON with a short summary for provider ${input.provider}.`,
    ],
    { timeout: 180000 },
  );

  assert.strictEqual(
    result.exitCode,
    0,
    `expected ${input.provider} run to succeed, got ${result.exitCode}\nstdout:\n${result.stdout}\nstderr:\n${result.stderr}`,
  );

  const parsed = JSON.parse(result.stdout.trim()) as { summary?: unknown };
  assert.strictEqual(
    typeof parsed.summary,
    "string",
    `expected ${input.provider} output to contain string summary, got: ${result.stdout}`,
  );
  assert(parsed.summary && parsed.summary.length > 0, "summary must not be empty");
}

async function test_all_providers_return_structured_output(): Promise<void> {
  await runProviderCase({
    provider: "claude",
    mode: "bypassPermissions",
    model: "haiku",
  });
  await runProviderCase({
    provider: "codex",
    mode: "full-access",
    model: "gpt-5.3-codex",
  });
  await runProviderCase({
    provider: "opencode",
    mode: "default",
    model: OPEN_CODE_STRUCTURED_MODEL,
  });
}

async function test_schema_validation_is_enforced(): Promise<void> {
  const result = await ctx.paseo(
    [
      "run",
      "--provider",
      "claude",
      "--mode",
      "bypassPermissions",
      "--model",
      "haiku",
      "--output-schema",
      impossibleSchema,
      'Return exactly {"summary":"ok"} and nothing else.',
    ],
    { timeout: 180000 },
  );

  assert.notStrictEqual(result.exitCode, 0, "expected impossible schema to fail");
  const output = `${result.stdout}\n${result.stderr}`;
  assert(
    output.includes("OUTPUT_SCHEMA_FAILED"),
    `expected OUTPUT_SCHEMA_FAILED for impossible schema\nstdout:\n${result.stdout}\nstderr:\n${result.stderr}`,
  );
}

async function main(): Promise<void> {
  try {
    await setup();
    await test_all_providers_return_structured_output();
    await test_schema_validation_is_enforced();
  } catch (error) {
    console.error(error);
    process.exitCode = 1;
  } finally {
    await cleanup();
  }
}

main();