import { responsesFixture } from "./support/responses-fixture";
import { expect, test } from "@playwright/test";
import { mockAgentApi } from "./support/mock-agent-api";
import { mockScadaApi } from "./support/mock-scada-api";
import { mockWorkbenchContext } from "./support/mock-workbench-context";

test("unified chart and map commands send success only after rendering, and final answer replaces process text", async ({ page }, testInfo) => {
  test.setTimeout(60000);
  await mockWorkbenchContext(page);
  await page.route("**/runtime-config.js", (route) => route.fulfill({ contentType: "application/javascript", body: 'globalThis.__TJWATER_CONFIG__ = { TJWATER_AUTH_MODE:"disabled", TJWATER_SERVER_API_BASE_URL:"https://tjwater-api.invalid", TJWATER_AGENT_URL:"http://127.0.0.1:8787", TJWATER_MAP_URL:"https://workbench-map.invalid/geoserver", TJWATER_GEOSERVER_WORKSPACE:"tjwater_next", TJWATER_WORKBENCH_RUNTIME_MODE:"auto", TJWATER_ENABLE_MSW:"false", TJWATER_ENABLE_DEV_PANEL:"true" };' }));
  await mockScadaApi(page, [{ device_id: "D1", device_type: "pressure", node_id: "J1", link_id: null, api_query_id: null, measurement_unit: "m", transmission_mode: "test", transmission_frequency: "1min", reliability: 1, x: null, y: null, longitude: 121.48, latitude: 31.1 }]);
  await mockAgentApi(page);
  const receipts: Array<{ actionId: string; status: string; output: unknown }> = [];
  const chart = { title: "监测压力", chart_type: "line", x_data: ["t1", "t2"], series: [{ name: "D1 scada_raw", data: [20, 21] }] };
  await page.route("**/api/v1/frontend-action-registry", (route) => route.fulfill({ headers: { "Access-Control-Allow-Origin": "*" }, json: { schema_version: "frontend-action-registry@2", actions: ["open_chart", "close_chart", "highlight_features", "clear_highlights", "zoom_to_map", "locate_features"].map((id) => ({ id, version: "frontend-action@2" })) } }));
  await page.route("**/api/v1/frontend-actions/*/result", async (route) => {
    if (route.request().method() === "OPTIONS") { await route.fulfill({ status: 204, headers: { "Access-Control-Allow-Origin": "*", "Access-Control-Allow-Headers": "*", "Access-Control-Allow-Methods": "POST, OPTIONS" } }); return; }
    receipts.push(route.request().postDataJSON());
    await route.fulfill({ headers: { "Access-Control-Allow-Origin": "*" }, json: { accepted: true } });
  });
  let responseNumber = 0;
  await page.route("**/api/v1/responses", async (route) => {
    responseNumber += 1;
    const now = Date.now();
    const action = (command: Record<string, unknown>, actionId: string) => ({ version: "frontend-action@2", actionId, toolCallId: actionId, sessionId: "migrated-session", command, issuedAt: now, expiresAt: now + 15000 });
    const commands = responseNumber === 2
      ? [{ action: "clear_highlights" }, { action: "close_chart", chart_id: "chat-pressure" }]
      : [{ action: "highlight_features", layer: "scada", feature_ids: ["D1"], effect: "pulse" },
          { action: "open_chart", chart_id: "chat-pressure", chart, evidence_refs: [] }];
    const chunks = [
      { type: "start", messageId: "assistant-migrated" },
      { type: "data-progress", data: { session_id: "migrated-session", id: "business-data", phase: "data", status: "completed", title: "获取业务数据" } },
      { type: "text-start", id: "text" }, { type: "text-delta", id: "text", delta: "正在整理过程说明" }, { type: "text-end", id: "text" },
      ...commands.map((command, index) => ({ type: "data-frontend_action", data: action(command, `command-${responseNumber}-${index}`) })),
      ...(responseNumber === 1 ? [{ type: "data-frontend_action", data: action({ action: "locate_features", layer: "scada", feature_ids: ["D1", "missing"] }, "locate-action") }] : []),
      ...(responseNumber === 1 ? [{ type: "data-frontend_action", data: action({ action: "zoom_to_map", center: [121.48, 31.1], zoom: 14.5 }, "zoom-action") }] : []),
      { type: "data-final_answer", data: { session_id: "migrated-session", content: "最终答复：压力曲线已核实。" } },
      { type: "finish", finishReason: "stop" },
    ];
    await route.fulfill({ headers: { "Access-Control-Allow-Origin": "*", "Content-Type": "text/event-stream", "Cache-Control": "no-cache" }, body: responsesFixture(chunks) });
  });
  await page.goto("/", { waitUntil: "domcontentloaded" });
  await page.getByRole("button", { name: /展开 Agent 助手面板/ }).click();
  const prompt = page.getByRole("textbox", { name: "询问供水问题，或描述需要处理的任务" }).filter({ visible: true }).first();
  await page.waitForFunction(() => Boolean(window.__waterNetworkMap?.getLayer("scada-pressure-hit")));
  await prompt.fill("查询历史并展示压力曲线");
  await page.getByRole("button", { name: "发送 Agent 指令" }).filter({ visible: true }).first().click();
  await expect(page.getByText("最终答复：压力曲线已核实。", { exact: true })).toBeVisible();
  await expect(page.getByText("正在整理过程说明", { exact: true })).toBeHidden();
  const display = page.getByLabel("对话展示", { exact: true });
  await expect(display.locator('[data-investigation-chart="chat-pressure"] svg[role="img"]')).toBeVisible();
  await expect.poll(() => receipts.length).toBe(4);

  expect(receipts.find((receipt) => receipt.actionId === "locate-action")?.output).toEqual({ located_ids: ["D1"], missing_ids: ["missing"], fitted: true });
  expect(receipts.every((receipt) => receipt.status === "succeeded")).toBe(true);

  const sourceSize = (source: string) => page.evaluate(async (source) => {
    const data = await (window.__waterNetworkMap?.getSource(source) as {
      getData: () => Promise<{ features: unknown[] }>
    } | undefined)?.getData();
    return data?.features.length ?? 0;
  }, source);
  await expect.poll(() => sourceSize("chat-presentation-highlights")).toBe(1);
  await page.getByRole("button", { name: "切换地图 Dev Panel" }).click();
  const tools = page.getByTestId("investigation-dev-tools");
  await tools.getByLabel("调查展示 JSONL", { exact: true }).fill([
    { record_type: "presentation", command: { action: "highlight_features", layer: "scada", feature_ids: ["D1"], effect: "pulse" } },
    { record_type: "presentation", command: { action: "open_chart", chart_id: "dev-pressure", chart, evidence_refs: [] } },
  ].map((record) => JSON.stringify(record)).join("\n"));
  await tools.getByRole("button", { name: "执行本地预览", exact: true }).click();
  await expect(page.getByLabel("本地预览展示", { exact: true }).locator('[data-investigation-chart="dev-pressure"]')).toBeVisible();
  await expect.poll(() => sourceSize("preview-presentation-highlights")).toBe(1);
  await page.screenshot({ path: testInfo.outputPath("unified-chat-dev.png") });
  await page.getByRole("button", { name: "切换地图 Dev Panel" }).click();
  const send = async (text: string) => {
    await prompt.fill(text);
    await page.getByRole("button", { name: "发送 Agent 指令" }).filter({ visible: true }).first().click();
  };
  await send("清除本轮对话展示");
  await expect.poll(() => receipts.length).toBe(6);
  await expect(display.locator('[data-investigation-chart="chat-pressure"]')).toHaveCount(0);
  await expect.poll(() => sourceSize("chat-presentation-highlights")).toBe(0);
  await expect.poll(() => sourceSize("preview-presentation-highlights")).toBe(1);
  await send("重新展示压力曲线");
  await expect.poll(() => receipts.length).toBe(8);
  await expect(display.locator('[data-investigation-chart="chat-pressure"]')).toBeVisible();
  await page.getByRole("button", { name: "新建 Agent 对话", exact: true }).click();
  await expect(display).toContainText("当前没有展示结果");
  await expect(display.getByRole("combobox", { name: "切换展示动作" })).toBeVisible();
  await expect.poll(() => sourceSize("chat-presentation-highlights")).toBe(0);
  await expect.poll(() => sourceSize("preview-presentation-highlights")).toBe(1);
  await expect(page.locator('[data-investigation-chart="dev-pressure"]')).toBeVisible();
});