import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";

export default function (pi: ExtensionAPI) {
  let enabled = false;

  pi.registerCommand("oai-fast", {
    description: "Toggle OpenAI Codex fast mode",
    handler: async (_args, ctx) => {
      enabled = !enabled;
      ctx.ui.setStatus(
        "oai-fast",
        enabled ? ctx.ui.theme.fg("accent", "OAI fast") : undefined,
      );
      ctx.ui.notify(`OpenAI Codex fast mode ${enabled ? "enabled" : "disabled"}.`, "info");
    },
  });

  pi.on("before_provider_request", (event, ctx) => {
    if (
      !enabled ||
      ctx.model?.provider !== "openai-codex" ||
      typeof event.payload !== "object" ||
      event.payload === null ||
      Array.isArray(event.payload)
    ) {
      return;
    }

    // The Codex backend maps the legacy priority tier to Fast mode.
    return { ...event.payload, service_tier: "priority" };
  });
}