#!/usr/bin/env node
"use strict";

const assert = require("node:assert");
const fs = require("node:fs/promises");
const path = require("node:path");
const { chromium } = require("playwright");

const ROOT = path.resolve(__dirname, "..");
const BASE_URL = (process.env.MNOTE_UI_BASE_URL || process.env.MNOTE_WEB_SMOKE_BASE_URL || "http://127.0.0.1:3000").replace(/\/+$/, "");
const UI_TIMEOUT_MS = Number(process.env.MNOTE_SMOKE_UI_TIMEOUT_MS || 30_000);
const OUT_DIR = path.join(ROOT, "tmp", "task426-mnote-web-main-no-reload-smoke");

function cssString(value) {
  return String(value).replace(/\\/g, "\\\\").replace(/"/g, '\\"');
}

async function quickLogin(page) {
  await page.goto(`${BASE_URL}/auth`, { waitUntil: "domcontentloaded", timeout: UI_TIMEOUT_MS });
  const quickLoginButton = page.getByRole("button", { name: "测试账号快速登录" });
  if (await quickLoginButton.count()) {
    await quickLoginButton.click({ timeout: UI_TIMEOUT_MS });
    await page.waitForURL((url) => url.pathname === "/", { timeout: UI_TIMEOUT_MS });
  }
}

async function readCreatedState(page, documentId) {
  return page.evaluate((id) => {
    const pageRows = Array.from(document.querySelectorAll(`#sidebar-tree-root .tree-row[data-shell-mode="page"][data-node-id="${CSS.escape(id)}"]`));
    const fileRows = Array.from(document.querySelectorAll("#sidebar-file-tree-root .tree-row[data-shell-mode=\"filetree\"]"))
      .filter((row) => row.getAttribute("data-doc-id") === id || row.getAttribute("data-document-id") === id)
      .map((row) => row.getAttribute("data-row-id"));
    return {
      url: location.href,
      title: document.title,
      localApplied: document.documentElement.getAttribute("data-mnote-tree-local-command-applied") || "",
      pageRows: pageRows.length,
      fileRows,
      nodeCount: document.querySelectorAll("li[data-node-id]").length,
    };
  }, documentId);
}

async function main() {
  await fs.mkdir(OUT_DIR, { recursive: true });
  const browser = await chromium.launch({ headless: process.env.HEADFUL !== "1" });
  const context = await browser.newContext({ viewport: { width: 1440, height: 960 } });
  const page = await context.newPage();
  const requests = [];
  page.on("response", (response) => {
    const url = response.url();
    if (url.includes("/api/tree/commands") || url.includes("/api/sidebar") || url.includes("/api/tree/events")) {
      requests.push({ url, status: response.status() });
    }
  });

  try {
    await quickLogin(page);
    await page.goto(BASE_URL, { waitUntil: "domcontentloaded", timeout: UI_TIMEOUT_MS });
    const beforeCount = await page.locator("li[data-node-id]").count();

    await page.getByRole("button", { name: "新建页面" }).click({ timeout: UI_TIMEOUT_MS });
    await page.waitForURL((url) => url.pathname.startsWith("/documents/"), { timeout: UI_TIMEOUT_MS });
    const createdDocumentId = new URL(page.url()).pathname.split("/").filter(Boolean).pop();
    assert(createdDocumentId, "新建后 URL 缺少 documentId");
    await page.waitForFunction(
      (id) => {
        const pageRow = document.querySelector(`#sidebar-tree-root .tree-row[data-shell-mode="page"][data-node-id="${CSS.escape(id)}"]`);
        const fileRows = Array.from(document.querySelectorAll("#sidebar-file-tree-root .tree-row[data-shell-mode=\"filetree\"]"))
          .filter((row) => row.getAttribute("data-doc-id") === id || row.getAttribute("data-document-id") === id);
        return pageRow && fileRows.some((row) => row.getAttribute("data-row-id") === `doc:${id}`) && fileRows.some((row) => row.getAttribute("data-row-id") === `index:${id}`);
      },
      createdDocumentId,
      { timeout: UI_TIMEOUT_MS },
    );
    const afterCreate = await readCreatedState(page, createdDocumentId);
    assert.equal(afterCreate.localApplied, "create", "新建页面应由主文档壳本地 apply");
    assert.equal(afterCreate.pageRows, 1, "Page Tree 应立即出现新页面");
    assert.deepEqual(afterCreate.fileRows.sort(), [`doc:${createdDocumentId}`, `index:${createdDocumentId}`].sort(), "File Tree 应立即出现新页面 doc/index 行");

    const rowSelector = `#sidebar-tree-root .tree-row[data-shell-mode="page"][data-node-id="${cssString(createdDocumentId)}"]`;
    await page.locator(rowSelector).scrollIntoViewIfNeeded({ timeout: UI_TIMEOUT_MS });
    page.once("dialog", async (dialog) => {
      await dialog.accept();
    });
    await page.locator(rowSelector).click({ button: "right", timeout: UI_TIMEOUT_MS });
    await page.getByRole("menuitem", { name: /删除/ }).click({ timeout: UI_TIMEOUT_MS });
    await page.waitForFunction(
      (id) => {
        const pageRow = document.querySelector(`#sidebar-tree-root .tree-row[data-shell-mode="page"][data-node-id="${CSS.escape(id)}"]`);
        const fileRows = Array.from(document.querySelectorAll("#sidebar-file-tree-root .tree-row[data-shell-mode=\"filetree\"]"))
          .filter((row) => row.getAttribute("data-doc-id") === id || row.getAttribute("data-document-id") === id);
        return !pageRow && fileRows.length === 0;
      },
      createdDocumentId,
      { timeout: UI_TIMEOUT_MS },
    );
    const afterDelete = await readCreatedState(page, createdDocumentId);
    assert.equal(afterDelete.localApplied, "remove", "删除页面应由主文档壳本地 apply");
    assert.equal(afterDelete.pageRows, 0, "Page Tree 应立即移除删除页面");
    assert.equal(afterDelete.fileRows.length, 0, "File Tree 应立即移除删除页面");

    const result = {
      ok: true,
      baseUrl: BASE_URL,
      beforeCount,
      createdDocumentId,
      afterCreate,
      afterDelete,
      requests,
    };
    await fs.writeFile(path.join(OUT_DIR, "result.json"), `${JSON.stringify(result, null, 2)}\n`, "utf8");
    console.log(JSON.stringify(result, null, 2));
  } finally {
    await browser.close();
  }
}

main().catch(async (error) => {
  await fs.mkdir(OUT_DIR, { recursive: true });
  const result = {
    ok: false,
    baseUrl: BASE_URL,
    error: error && error.stack ? error.stack : String(error),
  };
  await fs.writeFile(path.join(OUT_DIR, "result.json"), `${JSON.stringify(result, null, 2)}\n`, "utf8");
  console.error(result.error);
  process.exit(1);
});