import { expect, test, type Page } from "@playwright/test";
import { mockAgentApi } from "./support/mock-agent-api";

test("temporary Agent analysis windows replace and restore the map workspace", async ({ page }) => {
  await prepareMainFrame(page);
  await page.goto("/", { waitUntil: "domcontentloaded" });

  const frame = page.getByTestId("workbench-main-frame");
  const mapWindow = page.locator("#workspace-map");
  const testButton = page.getByRole("button", { name: "测试打开 Agent 多图表窗口" }).first();

  await expect(frame).toBeVisible();
  await expect(mapWindow).toBeVisible();
  await expect(testButton).toBeVisible();

  await testButton.click();
  await expect(page.getByRole("heading", { name: "北辰分区压力异常诊断", exact: true })).toBeVisible();
  await expect(mapWindow).toBeHidden();

  await testButton.click();
  await expect(page.getByRole("heading", { name: "供水服务分区平衡分析", exact: true })).toBeVisible();
  await expect(page.getByText("2 / 6 个临时分析窗")).toBeVisible();

  await page.getByRole("button", { name: "最小化供水服务分区平衡分析" }).click();
  await expect(page.getByRole("button", { name: "恢复供水服务分区平衡分析" })).toBeVisible();
  await page.getByRole("button", { name: "恢复供水服务分区平衡分析" }).click();
  await expect(page.getByRole("heading", { name: "供水服务分区平衡分析", exact: true })).toBeVisible();

  await page.getByRole("button", { name: "关闭并销毁供水服务分区平衡分析" }).click();
  await page.getByRole("button", { name: "关闭并销毁北辰分区压力异常诊断" }).click();
  await expect(mapWindow).toBeVisible();
  await expect(page.getByRole("heading", { name: "北辰分区压力异常诊断", exact: true })).toHaveCount(0);
  await expect(page.getByText("0 / 6 个临时分析窗")).toBeVisible();
});

test("the map document floats and minimizes to its taskbar entry", async ({ page }) => {
  await prepareMainFrame(page);
  await page.goto("/", { waitUntil: "domcontentloaded" });

  const workspace = page.locator("#main-workspace");
  const mapWindow = page.locator("#workspace-map");
  const dockedBox = await workspace.boundingBox();
  expect(dockedBox).not.toBeNull();

  await page.getByRole("button", { name: "浮动供水管网地图" }).click();
  const floatingBox = await mapWindow.boundingBox();
  expect(floatingBox).not.toBeNull();
  expect(floatingBox!.width).toBeLessThan(dockedBox!.width);
  expect(floatingBox!.height).toBeLessThan(dockedBox!.height);
  await expect(page.getByRole("button", { name: "最大化供水管网地图" })).toBeVisible();

  await page.getByRole("button", { name: "最小化供水管网地图" }).click();
  await expect(mapWindow).toBeHidden();
  const restoreTask = page.getByRole("button", { name: "恢复供水管网地图" });
  await expect(restoreTask).toBeVisible();

  await restoreTask.click();
  await expect(mapWindow).toBeVisible();
  await expect(page.getByRole("button", { name: "最大化供水管网地图" })).toBeVisible();
});

test("the test control cycles through three trusted document profiles", async ({ page }) => {
  await prepareMainFrame(page);
  await page.goto("/", { waitUntil: "domcontentloaded" });
  const testButton = page.getByRole("button", { name: "测试打开 Agent 多图表窗口" }).first();

  await testButton.click();
  await testButton.click();
  await testButton.click();

  await expect(page.getByRole("heading", { name: "北辰分区压力异常诊断", exact: true })).toBeAttached();
  await expect(page.getByRole("heading", { name: "供水服务分区平衡分析", exact: true })).toBeAttached();
  await expect(page.getByRole("heading", { name: "低压事件调度处置流程", exact: true })).toBeVisible();
  await expect(page.getByText("3 / 6 个临时分析窗")).toBeVisible();
});

test("temporary documents survive a mobile breakpoint round trip", async ({ page }) => {
  await prepareMainFrame(page);
  await page.goto("/", { waitUntil: "domcontentloaded" });
  await page.getByRole("button", { name: "测试打开 Agent 多图表窗口" }).first().click();

  const analysisHeading = page.locator("h2", { hasText: "北辰分区压力异常诊断" });
  await expect(analysisHeading).toBeVisible();

  await page.setViewportSize({ width: 390, height: 844 });
  await expect(page.locator("#workspace-map")).toBeVisible();
  await expect(analysisHeading).toBeAttached();
  await expect(analysisHeading).toBeHidden();

  await page.setViewportSize({ width: 1440, height: 900 });
  await expect(analysisHeading).toBeVisible();
  await expect(page.locator("#workspace-map")).toBeHidden();
});

test("layout presets arrange transient windows and keep the map visible", async ({ page }) => {
  await prepareMainFrame(page);
  await page.goto("/", { waitUntil: "domcontentloaded" });
  const testButton = page.getByRole("button", { name: "测试打开 Agent 多图表窗口" }).first();

  await testButton.click();
  await testButton.click();
  await testButton.click();
  await page.getByRole("button", { name: "设置窗口布局，当前自由排列" }).click();
  await page.getByRole("menuitemradio", { name: /主次/ }).click();

  const workspace = page.locator("#main-workspace");
  await expect(workspace).toHaveAttribute("data-layout-mode", "primary");
  await expect(page.locator("#workspace-map")).toBeVisible();
  await expect(page.locator("section[aria-label='低压事件调度处置流程']")).toBeVisible();
  await expect(page.locator("section[aria-label='北辰分区压力异常诊断']")).toBeHidden();

  const workspaceBox = await workspace.boundingBox();
  const primaryBox = await page.locator("section[aria-label='低压事件调度处置流程']").boundingBox();
  expect(workspaceBox).not.toBeNull();
  expect(primaryBox).not.toBeNull();
  expect(primaryBox!.width).toBeGreaterThan(workspaceBox!.width * 0.6);

  await testButton.click();
  await expect(page.locator("section[aria-label='北辰分区压力异常诊断']").last()).toBeVisible();
  const newPrimaryBox = await page.locator("section[aria-label='北辰分区压力异常诊断']").last().boundingBox();
  expect(newPrimaryBox).not.toBeNull();
  expect(newPrimaryBox!.width).toBeGreaterThan(workspaceBox!.width * 0.6);
});

test("dragging a small window onto the primary window swaps their complete slots", async ({ page }) => {
  await prepareMainFrame(page);
  await page.goto("/", { waitUntil: "domcontentloaded" });
  await page.getByRole("button", { name: "测试打开 Agent 多图表窗口" }).first().click();
  await page.getByRole("button", { name: "设置窗口布局，当前自由排列" }).click();
  await page.getByRole("menuitemradio", { name: /主次/ }).click();

  const mapWindow = page.locator("#workspace-map");
  const analysisWindow = page.locator("section[aria-label='北辰分区压力异常诊断']");
  const mapBefore = await mapWindow.boundingBox();
  const analysisBefore = await analysisWindow.boundingBox();
  expect(mapBefore).not.toBeNull();
  expect(analysisBefore).not.toBeNull();
  expect(mapBefore!.width).toBeLessThan(analysisBefore!.width);

  await page.mouse.move(mapBefore!.x + 72, mapBefore!.y + 22);
  await page.mouse.down();
  await page.mouse.move(
    analysisBefore!.x + analysisBefore!.width / 2,
    analysisBefore!.y + analysisBefore!.height / 2,
    { steps: 10 }
  );
  await expect(page.getByText("释放以交换")).toBeVisible();
  await page.mouse.up();

  const mapAfter = await mapWindow.boundingBox();
  const analysisAfter = await analysisWindow.boundingBox();
  expect(mapAfter).not.toBeNull();
  expect(analysisAfter).not.toBeNull();
  expect(mapAfter!.width).toBeGreaterThan(analysisAfter!.width);
  expect(Math.abs(mapAfter!.x - analysisBefore!.x)).toBeLessThanOrEqual(2);
  expect(Math.abs(analysisAfter!.x - mapBefore!.x)).toBeLessThanOrEqual(2);
  await expect(page.getByText("供水管网地图与北辰分区压力异常诊断已交换位置")).toBeAttached();
});

test("dropping on blank workspace exits auto layout and Escape cancels a drag", async ({ page }) => {
  await prepareMainFrame(page);
  await page.goto("/", { waitUntil: "domcontentloaded" });
  await page.getByRole("button", { name: "测试打开 Agent 多图表窗口" }).first().click();
  await page.getByRole("button", { name: "设置窗口布局，当前自由排列" }).click();
  await page.getByRole("menuitemradio", { name: /双栏/ }).click();

  const workspace = page.locator("#main-workspace");
  const analysisWindow = page.locator("section[aria-label='北辰分区压力异常诊断']");
  const before = await analysisWindow.boundingBox();
  const workspaceBox = await workspace.boundingBox();
  expect(before).not.toBeNull();
  expect(workspaceBox).not.toBeNull();

  await page.mouse.move(before!.x + 80, before!.y + 22);
  await page.mouse.down();
  await page.mouse.move(workspaceBox!.x + 4, workspaceBox!.y + 4, { steps: 8 });
  await page.keyboard.press("Escape");
  await expect(workspace).toHaveAttribute("data-layout-mode", "split");
  const afterCancel = await analysisWindow.boundingBox();
  expect(Math.abs(afterCancel!.x - before!.x)).toBeLessThanOrEqual(2);

  await page.mouse.move(before!.x + 80, before!.y + 22);
  await page.mouse.down();
  await page.mouse.move(workspaceBox!.x + 4, workspaceBox!.y + 4, { steps: 8 });
  await page.mouse.up();
  await expect(workspace).toHaveAttribute("data-layout-mode", "free");
  await expect(page.getByText("北辰分区压力异常诊断已转为自由排列")).toBeAttached();
});

async function prepareMainFrame(page: Page) {
  await mockAgentApi(page);
  await page.route("**/runtime-config.js", async (route) => {
    await route.fulfill({
      contentType: "application/javascript",
      body: `globalThis.__TJWATER_CONFIG__ = {
        TJWATER_AUTH_MODE: "disabled",
        TJWATER_MAPBOX_ACCESS_TOKEN: "",
        TJWATER_MAP_URL: "https://workspace-map.invalid/geoserver",
        TJWATER_GEOSERVER_WORKSPACE: "tjwater",
        TJWATER_AGENT_API_BASE_URL: "http://127.0.0.1:8787",
        TJWATER_ENABLE_DEV_PANEL: "true",
        TJWATER_ENABLE_MSW: "false"
      };`
    });
  });
  await page.route("https://workspace-map.invalid/**", async (route) => {
    await route.fulfill({ status: 204 });
  });
}