import { assertEquals } from "@std/assert";
import type { PayloadHook } from "@berry/api";
import { testExtension } from "@berry/api/testing.ts";
import misc from "./mod.ts";

type Model = Parameters<PayloadHook>[1];
const codex = { provider: "openai-codex" } as unknown as Model;
const other = { provider: "other" } as unknown as Model;

Deno.test("compact starts a manual compaction, guided by any arguments", async () => {
  const ext = testExtension(misc);
  assertEquals(await ext.command("compact"), "Compacting context\u2026");
  assertEquals(
    await ext.command("compact", "  keep the failing test names  "),
    "Compacting context\u2026",
  );
  assertEquals(ext.compactions, [undefined, "keep the failing test names"]);
});

Deno.test("oai-fast toggles fast mode for codex payloads only", async () => {
  const ext = testExtension(misc);
  assertEquals(await ext.payload({ a: 1 }, codex), { a: 1 });
  assertEquals(await ext.command("oai-fast", ""), "OpenAI Codex fast mode enabled.");
  assertEquals(await ext.payload({ a: 1 }, codex), { a: 1, service_tier: "priority" });
  assertEquals(await ext.payload({ a: 1 }, other), { a: 1 });
  assertEquals(await ext.payload("text", codex), "text");
  assertEquals(await ext.command("oai-fast", ""), "OpenAI Codex fast mode disabled.");
  assertEquals(await ext.payload({ a: 1 }, codex), { a: 1 });
});

Deno.test("oai-fast explicitly sets fast mode without toggling it", async () => {
  const ext = testExtension(misc);
  for (const mode of ["off", "on", "on", "off", "off"]) {
    const enabled = mode === "on";
    assertEquals(
      await ext.command("oai-fast", ` ${mode} `),
      `OpenAI Codex fast mode ${enabled ? "enabled" : "disabled"}.`,
    );
    assertEquals(
      await ext.payload({}, codex),
      enabled ? { service_tier: "priority" } : {},
    );
  }
});

Deno.test("oai-fast rejects invalid arguments without changing state", async () => {
  const ext = testExtension(misc);
  for (const mode of ["off", "on"]) {
    await ext.command("oai-fast", mode);
    for (const args of ["true", "on off", "onn"]) {
      assertEquals(await ext.command("oai-fast", args), "Usage: /oai-fast [on|off]");
      assertEquals(ext.state.oaiFast, mode === "on");
    }
  }
});