import { describe, it, before, after } from "node:test";
import assert from "node:assert/strict";
import { chromium, type Browser, type Page } from "playwright";
import { collectPage, type RawNode, type RawChild } from "../src/capture/walker.js";

function isText(c: RawChild): c is { text: string } {
  return (c as { text?: string }).text !== undefined;
}

function findByTag(root: RawNode, tag: string): RawNode | null {
  if (root.tag === tag) return root;
  for (const c of root.children) {
    if (isText(c)) continue;
    const hit = findByTag(c, tag);
    if (hit) return hit;
  }
  return null;
}

function textRun(node: RawNode): string {
  let out = "";
  for (const c of node.children) out += isText(c) ? c.text : textRun(c);
  return out;
}

describe("walker whitespace-only text nodes", () => {
  let browser: Browser;
  let page: Page;
  before(async () => {
    browser = await chromium.launch();
    page = await browser.newPage();
  });
  after(async () => {
    await browser.close();
  });

  const capture = async (html: string) => {
    await page.setContent(html);
    // tsx/esbuild wraps functions with a __name() helper for stack traces; the
    // serialized collectPage carries those calls, so shim it (same as capture.ts).
    await page.evaluate("globalThis.__name = globalThis.__name || ((fn) => fn);");
    return page.evaluate(collectPage);
  };

  it("keeps the lone space that is the only child of an inline element", async () => {
    // Real case (ooni.com): the space between "of" and "the" lives alone inside
    // <strong>; dropping it fuses the adjacent text runs ("ofthe").
    const snap = await capture("<p>Creator of<strong> </strong><em><strong>the world's</strong></em></p>");
    const p = findByTag(snap.root, "p")!;
    const strong = p.children.find((c) => !isText(c) && c.tag === "strong") as RawNode;
    assert.deepEqual(strong.children, [{ text: " " }]);
    assert.equal(textRun(p), "Creator of the world's");
  });

  it("still keeps the single space between inline elements", async () => {
    const snap = await capture("<p><em>a</em> <em>b</em></p>");
    const p = findByTag(snap.root, "p")!;
    assert.equal(textRun(p), "a b");
  });

  it("does not emit a space inside an empty block container", async () => {
    const snap = await capture("<main><section>\n   \n</section></main>");
    const section = findByTag(snap.root, "section")!;
    assert.deepEqual(section.children, []);
  });
});

function findByClass(root: RawNode, cls: string): RawNode | null {
  if ((root.attrs?.class ?? "").split(/\s+/).includes(cls)) return root;
  for (const c of root.children) {
    if (isText(c)) continue;
    const hit = findByClass(c, cls);
    if (hit) return hit;
  }
  return null;
}

describe("walker off-screen visibility", () => {
  let browser: Browser;
  let page: Page;
  before(async () => {
    browser = await chromium.launch();
    page = await browser.newPage();
    await page.setViewportSize({ width: 375, height: 768 });
  });
  after(async () => {
    await browser.close();
  });

  const capture = async (html: string) => {
    await page.setContent(html);
    await page.evaluate("globalThis.__name = globalThis.__name || ((fn) => fn);");
    return page.evaluate(collectPage);
  };

  it("marks a fixed off-screen-left drawer's un-hidden inner content invisible", async () => {
    // Real case (ooni.com): a slide-in search drawer is a position:fixed box parked at
    // x:-375 (right edge at 0) with visibility:hidden; its inner content sets
    // visibility:visible (so getComputedStyle reports it visible) but the whole box is
    // still off the left edge. The drawer must not leak into the clone's DOM text.
    const snap = await capture(`
      <div class="drawer" style="position:fixed;left:0;top:0;width:375px;height:768px;
           transform:translateX(-100%);visibility:hidden;">
        <div class="inner" style="visibility:visible;width:343px;">
          <h2 class="drawer-title">Popular Searches</h2>
        </div>
      </div>
      <p class="onscreen">Hello</p>`);
    const title = findByClass(snap.root, "drawer-title")!;
    assert.equal(title.visible, false, "off-screen-left drawer title should be invisible");
    // The genuinely on-screen paragraph is still visible (sanity that the gate isn't over-broad).
    const onscreen = findByClass(snap.root, "onscreen")!;
    assert.equal(onscreen.visible, true, "on-screen content stays visible");
  });

  it("keeps a partially-overlapping negative-margin decoration visible", async () => {
    // A decoration pulled left by a negative margin so it straddles the left edge
    // (x:-40, width:200 -> right edge +160) still paints and must stay visible.
    const snap = await capture(`
      <div style="overflow:hidden;width:375px;">
        <div class="deco" style="width:200px;height:40px;margin-left:-40px;background:red;">peek</div>
      </div>`);
    const deco = findByClass(snap.root, "deco")!;
    assert.ok(deco.bbox.x < 0, "decoration starts off-screen left");
    assert.ok(deco.bbox.x + deco.bbox.width > 0, "but its right edge is on-screen");
    assert.equal(deco.visible, true, "partially-overlapping decoration stays visible");
  });

  it("keeps normal in-flow content below the fold visible", async () => {
    // The page scrolls vertically, so content below the viewport is reachable and must
    // NOT be marked invisible (only position:fixed boxes are pinned).
    const snap = await capture(`
      <div style="height:2000px;"></div>
      <p class="belowfold" style="height:40px;">Below the fold</p>`);
    const below = findByClass(snap.root, "belowfold")!;
    assert.ok(below.bbox.y > 768, "content is below the 768px viewport");
    assert.equal(below.visible, true, "below-fold in-flow content stays visible");
  });

  it("marks a computed-visibility:hidden subtree invisible", async () => {
    // A subtree whose computed visibility is actually hidden (and does not set
    // visibility:visible on any descendant) is not painted.
    const snap = await capture(`
      <div style="visibility:hidden;width:300px;">
        <span class="hidden-child">secret</span>
      </div>`);
    const child = findByClass(snap.root, "hidden-child")!;
    assert.equal(child.visible, false, "computed-hidden child is invisible");
  });

  it("rejects a fixed box parked entirely below the viewport", async () => {
    // A position:fixed banner pinned below the viewport bottom never scrolls into view.
    const snap = await capture(`
      <div class="fixedlow" style="position:fixed;left:0;top:900px;width:375px;height:60px;">
        <span>hidden banner</span>
      </div>`);
    const fixed = findByClass(snap.root, "fixedlow")!;
    assert.equal(fixed.visible, false, "fixed box below the viewport is invisible");
  });
});

describe("walker font-metric probe tagging (fix 4)", () => {
  let browser: Browser;
  let page: Page;
  before(async () => {
    browser = await chromium.launch();
    page = await browser.newPage();
    await page.setViewportSize({ width: 375, height: 768 });
  });
  after(async () => {
    await browser.close();
  });

  const capture = async (html: string) => {
    await page.setContent(html);
    await page.evaluate("globalThis.__name = globalThis.__name || ((fn) => fn);");
    return page.evaluate(collectPage);
  };

  it("tags a far-off-screen, non-painting measurement scratch node as a probe", async () => {
    // The classic font-metric probe pattern (WordPress/typography libs): absolutely positioned,
    // parked ~100000px off-screen, visibility:hidden, holding measurement text.
    const snap = await capture(`
      <p class="real">Real content</p>
      <div class="probe" style="position:absolute;top:-99999px;left:-99999px;
           visibility:hidden;white-space:nowrap;">Mgy</div>`);
    const probe = findByClass(snap.root, "probe")!;
    assert.equal(probe.probe, true, "far-off-screen hidden scratch node is a probe");
    const real = findByClass(snap.root, "real")!;
    assert.ok(!real.probe, "real content is not a probe");
  });

  it("does NOT tag a near-off-screen hidden drawer (real content) as a probe", async () => {
    // A slide-in drawer parked just off the left edge (x:-375, visibility:hidden) is real
    // content that a controller can reveal — it must NOT be mistaken for a measurement probe.
    const snap = await capture(`
      <div class="drawer" style="position:fixed;left:0;top:0;width:375px;height:768px;
           transform:translateX(-100%);visibility:hidden;">
        <h2 class="dtitle">Menu</h2>
      </div>`);
    const drawer = findByClass(snap.root, "drawer")!;
    assert.ok(!drawer.probe, "a near-off-screen drawer is not a probe");
  });

  it("does NOT tag an sr-only (visible, on-screen-adjacent) accessibility label as a probe", async () => {
    // Screen-reader-only text stays visibility:visible so AT can read it; even parked far off
    // via left:-9999px it must survive (and 9999px is under the 10000px probe threshold anyway).
    const snap = await capture(`
      <a href="/"><span class="sr" style="position:absolute;left:-9999px;">Skip to content</span>Home</a>`);
    const sr = findByClass(snap.root, "sr")!;
    assert.ok(!sr.probe, "sr-only accessible text is not a probe");
  });
});

describe("walker sizing probe: circular authored-height guard", () => {
  let browser: Browser;
  let page: Page;
  before(async () => {
    browser = await chromium.launch();
    page = await browser.newPage({ viewport: { width: 1280, height: 720 } });
  });
  after(async () => {
    await browser.close();
  });

  const capture = async (html: string) => {
    await page.setContent(html);
    await page.evaluate("globalThis.__name = globalThis.__name || ((fn) => fn);");
    return page.evaluate(collectPage);
  };

  it("keeps an explicit 100vh height (circular hero/fill-child pair)", async () => {
    // A hero authored `height:100vh` with a `height:100%` fill child: setting the hero to
    // `height:auto` still reproduces its box because the child pins it back, so the raw probe
    // would read hAuto:true and the authored 100vh would be dropped — hero collapses to 0.
    const snap = await capture(`
      <style>
        .hero { height: 100vh; display: flex; }
        .fill { height: 100%; width: 100%; }
      </style>
      <section class="hero"><div class="fill"><p>content</p></div></section>`);
    const hero = findByClass(snap.root, "hero")!;
    assert.ok(hero.sizing, "hero was probed");
    assert.equal(hero.sizing!.hAuto, false, "explicit 100vh is not content-sized");
    assert.equal(hero.sizing!.hFill, false, "explicit 100vh is authored, not a parent fill");
    // Box actually equals the viewport height (720), proving the circular reproduction.
    assert.ok(Math.abs(hero.bbox.height - 720) <= 1, "hero rendered at 100vh");
  });

  it("keeps an explicit px height whose fill child reproduces it", async () => {
    const snap = await capture(`
      <style>
        .section { height: 400px; display: flex; }
        .fill { height: 100%; width: 100%; }
      </style>
      <div class="section"><div class="fill"><p>content</p></div></div>`);
    const section = findByClass(snap.root, "section")!;
    assert.ok(section.sizing, "section was probed");
    assert.equal(section.sizing!.hAuto, false, "explicit 400px is not content-sized");
    assert.equal(section.sizing!.hFill, false, "explicit 400px is authored, not a fill");
    assert.ok(Math.abs(section.bbox.height - 400) <= 1, "section rendered at 400px");
  });

  it("keeps an explicit height authored via inline style", async () => {
    const snap = await capture(`
      <style>.fill { height: 100%; width: 100%; }</style>
      <div class="box" style="height: 300px; display: flex;">
        <div class="fill"><p>content</p></div>
      </div>`);
    const box = findByClass(snap.root, "box")!;
    assert.ok(box.sizing, "box was probed");
    assert.equal(box.sizing!.hAuto, false, "inline explicit height is kept");
    assert.equal(box.sizing!.hFill, false, "inline explicit height is not a fill");
  });

  it("resolves the mutual parent/child pair without disturbing the fill child", async () => {
    // The child is a GENUINE fill (height:100%) and must keep hFill:true / hAuto:false so the
    // generator emits h-full for it; only the parent's circular verdict is corrected.
    const snap = await capture(`
      <style>
        .outer { height: 500px; display: flex; }
        .inner { height: 100%; width: 100%; }
      </style>
      <div class="outer"><div class="inner"><p>content</p></div></div>`);
    const outer = findByClass(snap.root, "outer")!;
    const inner = findByClass(snap.root, "inner")!;
    assert.equal(outer.sizing!.hAuto, false, "parent explicit height kept");
    assert.equal(outer.sizing!.hFill, false, "parent is authored, not a fill");
    // The child authors only `height:100%` (a fill), so the explicit-height guard must NOT fire on
    // it — hFill stays true so the generator can still emit h-full for the genuine fill child.
    assert.equal(inner.sizing!.hFill, true, "child still fills the definite parent");
  });

  it("still detects a genuinely content-sized (auto) height as hAuto", async () => {
    // No authored height anywhere: the box is content-sized and must stay droppable.
    const snap = await capture(`
      <style>.wrap { display: block; }</style>
      <div class="wrap"><p>just some flowing text content</p></div>`);
    const wrap = findByClass(snap.root, "wrap")!;
    assert.ok(wrap.sizing, "wrap was probed");
    assert.equal(wrap.sizing!.hAuto, true, "content-sized height is still auto");
  });

  it("does not treat a percentage or zero authored height as explicit", async () => {
    // height:100% is the FILL case (handled by hFill), and height:0 is not definite; neither
    // should trip the explicit-height override.
    const snap = await capture(`
      <style>
        .pct-parent { height: 300px; }
        .pct { height: 100%; }
      </style>
      <div class="pct-parent"><div class="pct"><p>x</p></div></div>`);
    const pct = findByClass(snap.root, "pct")!;
    assert.ok(pct.sizing, "pct was probed");
    // A true fill child: hFill true, hAuto false — untouched by the explicit-height guard.
    assert.equal(pct.sizing!.hFill, true, "percentage height stays a fill");
    assert.equal(pct.sizing!.hAuto, false, "percentage fill is not content-sized");
  });
});