import assert from "node:assert/strict";
import { readFile, access } from "node:fs/promises";
import { test } from "node:test";

const site = new URL("../", import.meta.url);
const routes = JSON.parse(await readFile(new URL(".generated/routes.json", site), "utf8"));
const old = JSON.parse(await readFile(new URL("migration-anchors.json", site), "utf8"));

test("every migrated route and heading anchor survives the cutover", async () => {
  const failures = [];
  for (const [path, headings] of Object.entries(old)) {
    const route = path === "." ? "/" : `/${path}`;
    if (!routes.includes(route)) {
      failures.push(`Missing route: ${route}`);
      continue;
    }
    const html = await readFile(new URL(`dist/${path === "." ? "" : `${path}/`}index.html`, site), "utf8");
    for (const heading of headings) {
      const escaped = heading.id.replaceAll("&", "&").replaceAll('"', """);
      if (!html.includes(`id="${escaped}"`) && !html.includes(`id='${escaped}'`))
        failures.push(`${route}#${heading.id}`);
    }
  }
  assert.deepEqual(failures, []);
});

test("converted guides render admonitions, preserve content, and publish local images", async () => {
  const quickStart = await readFile(new URL("dist/getting_started/quick_start/index.html", site), "utf8");
  assert.ok(quickStart.includes("Need more control?"));
  assert.ok(!quickStart.includes("!!! tip"));
  const home = await readFile(new URL("dist/index.html", site), "utf8");
  assert.ok(home.includes("Welcome to FastVideo"));
  assert.ok(home.includes("/FastVideo/assets/logos/logo.svg"));
  assert.ok(home.includes("FastVideo is an inference and post-training framework for video diffusion models."));
  assert.ok(!home.includes("About FastVideo and performance overview"));
  await readFile(new URL("dist/assets/logos/logo.svg", site));
});

test("Python API and generated examples contain the actual rendered source documentation", async () => {
  const api = await readFile(new URL("dist/api/fastvideo/configs/pipelines/base/index.html", site), "utf8");
  assert.ok(api.includes('id="fastvideo.configs.pipelines.base.PipelineConfig"'));
  assert.ok(api.includes("from_pretrained"));
  assert.ok(api.includes("python-reference"));
  const examples = routes.filter((route) => route.startsWith("/inference/examples/"));
  assert.ok(examples.length > 5);
  const example = await readFile(new URL("dist/inference/examples/basic/index.html", site), "utf8");
  assert.ok(example.includes("VideoGenerator"));
});

test("generated examples show their page title only once", async () => {
  const failures = [];
  for (const route of routes.filter((item) => item.includes("/examples/") || item.startsWith("/examples/"))) {
    const html = await readFile(new URL(`dist${route}/index.html`, site), "utf8");
    const headings = [...html.matchAll(/<h([1-6])\b[^>]*>([\s\S]*?)<\/h\1>/g)].map((match) => ({
      level: Number(match[1]),
      text: match[2]
        .replace(/<[^>]+>/g, "")
        .replace(/[^\p{L}\p{N}]/gu, "")
        .toLowerCase(),
    }));
    const title = headings.find((heading) => heading.level === 1)?.text;
    if (!title || headings.filter((heading) => heading.text === title).length !== 1) failures.push(route);
  }
  assert.deepEqual(failures, []);
});

test("the deployment artifact contains no retired site scripts or styles", async () => {
  for (const name of ["custom.js", "custom.css", "copy-page.js", "cookbook.js", "cookbook-runbooks.css"]) {
    await assert.rejects(access(new URL(`dist/assets/${name}`, site)));
  }
  const home = await readFile(new URL("dist/index.html", site), "utf8");
  assert.ok(home.includes("home-featured-model"));
  const family = await readFile(new URL("dist/cookbook/wan/index.html", site), "utf8");
  assert.ok(!family.includes("legacy-cookbook"));
  assert.ok(family.includes("recipe-card"));
});