import path from "node:path";
import { expect, test, type Page } from "../support/fixtures";
import { seedWorkspace } from "../support/helpers/seed-client";
import { openAgentRoute } from "../support/helpers/mock-agent";
import type { DaemonClient } from "@getpaseo/client/internal/daemon-client";
import { connectDaemonClient } from "../support/helpers/daemon-client-loader";

// This fixture catalog advertises Ultrafast on Sol; another model offers Fast only.
// The real daemon discovers both through the same Codex model/list boundary.
test.use({
  e2eDaemonConfig: {
    version: 1,
    agents: {
      providers: {
        codex: {
          enabled: true,
          command: [process.execPath, path.resolve("e2e/fixtures/catalog-codex.mjs")],
        },
        claude: { enabled: false },
        copilot: { enabled: false },
        opencode: { enabled: false },
        pi: { enabled: false },
        omp: { enabled: false },
      },
    },
  },
});

// Rendered light-theme colors: palette yellow 400 for Fast and Ultrafast, muted foreground for Normal.
const ACTIVE_SPEED_COLOR = "rgb(251, 191, 36)";
const NORMAL_SPEED_COLOR = "rgb(113, 113, 122)";

async function openSpeedSelector(page: Page): Promise<void> {
  const speed = page.getByRole("button", { name: /^(Select speed|Speed: .+)$/ });
  const features = page.getByRole("button", { name: "Open agent features", exact: true });
  const modelSettings = page.getByRole("button", { name: /^Select model \(/ });
  await expect(speed.or(features).or(modelSettings).filter({ visible: true }).first()).toBeVisible({
    timeout: 30_000,
  });
  if (!(await speed.isVisible())) {
    if (await features.isVisible()) await features.click();
    else await modelSettings.click();
  }
  await expect(speed).toBeVisible({ timeout: 30_000 });
  await speed.click();
  await expect(page.getByText("Normal", { exact: true }).last()).toBeVisible();
}

async function selectSpeed(page: Page, label: string): Promise<void> {
  await page.getByRole("button", { name: label, exact: true }).last().click();
  const toolbarTrigger = page.getByRole("button", { name: `Speed: ${label}`, exact: true });
  const sheetTrigger = page.getByRole("button", { name: "Select speed", exact: true });
  await expect(toolbarTrigger.or(sheetTrigger)).toBeVisible();
  if (await toolbarTrigger.isVisible()) await expect(toolbarTrigger).toHaveText("");
  else await expect(sheetTrigger).toContainText(label);
}

async function expectSpeedIconColor(page: Page, color: string): Promise<void> {
  const trigger = page
    .getByRole("button", { name: /^(Speed: .+|Select speed)$/ })
    .filter({ visible: true });
  await expect(trigger.locator("svg").first()).toHaveCSS("stroke", color);
}

async function expectSpeedChoices(page: Page, choices: string[]): Promise<void> {
  for (const choice of ["Normal", "Fast", "Ultrafast"]) {
    const option = page.getByText(choice, { exact: true }).last();
    if (choices.includes(choice)) {
      await expect(option).toBeVisible();
      await expect(option).toBeInViewport({ ratio: 1 });
    } else await expect(option).not.toBeVisible();
  }
}

async function switchAgentModel(agentId: string, modelId: string): Promise<void> {
  const client = await connectDaemonClient<
    Pick<DaemonClient, "setAgentModel" | "close" | "connect">
  >({
    clientIdPrefix: "codex-speed-model",
  });
  try {
    await client.setAgentModel(agentId, modelId);
  } finally {
    await client.close();
  }
}

for (const viewport of [
  { name: "desktop", width: 1280, height: 850 },
  { name: "compact", width: 390, height: 844 },
]) {
  test(`Codex catalog drives speed choices on ${viewport.name}`, async ({ page }, testInfo) => {
    test.setTimeout(180_000);
    await page.setViewportSize(viewport);
    const workspace = await seedWorkspace({ repoPrefix: "codex-speed-" });
    try {
      const agent = await workspace.client.createAgent({
        provider: "codex",
        cwd: workspace.repoPath,
        workspaceId: workspace.workspaceId,
        title: "Speed settings",
        model: "gpt-6.1-sol",
        modeId: "full-access",
        initialPrompt: "Get ready to work on this project.",
      });
      await openAgentRoute(page, { workspaceId: workspace.workspaceId, agentId: agent.id });
      if (viewport.name === "desktop") {
        const trigger = page.getByRole("button", { name: "Speed: Normal", exact: true });
        await expect(trigger).toBeVisible({ timeout: 30_000 });
        await expect(trigger).toHaveText("");
        await page.screenshot({ path: testInfo.outputPath("desktop-normal-trigger.png") });
      }
      await openSpeedSelector(page);
      await selectSpeed(page, "Fast");
      await expectSpeedIconColor(page, ACTIVE_SPEED_COLOR);
      await page.screenshot({ path: testInfo.outputPath(`${viewport.name}-fast-selected.png`) });
      await openSpeedSelector(page);
      await expectSpeedChoices(page, ["Normal", "Fast", "Ultrafast"]);
      await page.screenshot({ path: testInfo.outputPath(`${viewport.name}-speed-options.png`) });
      await selectSpeed(page, "Ultrafast");
      await expectSpeedIconColor(page, ACTIVE_SPEED_COLOR);
      await openSpeedSelector(page);
      await expectSpeedChoices(page, ["Normal", "Fast", "Ultrafast"]);
      await page.screenshot({
        path: testInfo.outputPath(`${viewport.name}-ultrafast-selected.png`),
      });
      await selectSpeed(page, "Normal");
      await expectSpeedIconColor(page, NORMAL_SPEED_COLOR);
      if (viewport.name === "desktop") {
        await switchAgentModel(agent.id, "gpt-6-sol");
        await openSpeedSelector(page);
        await expectSpeedChoices(page, ["Normal", "Fast"]);
        await page.screenshot({ path: testInfo.outputPath("desktop-fast-only.png") });
      }
    } finally {
      await workspace.cleanup();
    }
  });
}