import { beforeAll, describe, expect, test } from "bun:test";
import { mkdtemp } from "node:fs/promises";
import { tmpdir } from "node:os";
import { join } from "node:path";
import type { Hono } from "hono";
import { openBucket } from "../src/bucket/reader.ts";
import { syncAll } from "../src/bucket/sync.ts";
import { fileAt, objectBytes } from "../src/mirror/queries.ts";
import { openMirror } from "../src/mirror/store.ts";
import { createApp } from "../src/server/app.ts";
import { openKeprDb } from "../src/server/db.ts";
import { type Fixture, makeKohBucket } from "./fixture.ts";

let fx: Fixture;
let app: Hono;
let mirrorDir: string;
const get = (path: string) => app.request(path);
const body = async (path: string) => (await get(path)).text();

beforeAll(async () => {
	fx = await makeKohBucket();
	const secretRepo = await makeKohBucket("hidden");
	const dataDir = await mkdtemp(join(tmpdir(), "kepr-pages-"));
	mirrorDir = join(dataDir, "mirror");
	const kepr = openKeprDb(dataDir);
	for (const bucketDir of [fx.bucketDir, secretRepo.bucketDir]) {
		await syncAll({ bucket: openBucket({ kind: "folder", path: bucketDir }), kepr, mirrorDir });
	}
	kepr.run(
		"UPDATE repos SET public = 1, install1 = 'nix profile add x', homepage = 'https://demo.example' WHERE name = 'demo'",
	);
	app = createApp(
		{
			domain: "kepr.example",
			dataDir,
			adminPasswordHash: "$argon2id$x",
			bucket: { kind: "folder", path: fx.bucketDir },
		},
		kepr,
	);
});

const [head, , second, first] = [0, 1, 2, 3].map((i) => () => fx.saveIds[i] as string);

describe("the visibility gate", () => {
	test("a private repo and an unknown one are the same 404, everywhere", async () => {
		for (const suffix of ["", "/files/src/main.ts", "/raw/src/main.ts"]) {
			const hidden = await get(`/hidden${suffix}`);
			const missing = await get(`/nowhere${suffix}`);
			expect(hidden.status).toBe(404);
			expect(missing.status).toBe(404);
			expect((await hidden.text()).replaceAll("/hidden", "/nowhere")).toBe(await missing.text());
		}
	});
});

describe("the repo page", () => {
	test("renders header, timeline, history and README at HEAD with a short cache", async () => {
		const response = await get("/demo");
		expect(response.status).toBe(200);
		expect(response.headers.get("cache-control")).toBe("public, max-age=60");
		const page = await response.text();
		expect(page).toContain('<h1 class="repo-name">demo</h1>');
		expect(page).toContain("A small repo for <strong>Kepr's</strong> tests.");
		expect(page).toContain(
			'<meta property="og:description" content="A small repo for Kepr&#39;s tests.">',
		);
		expect(page).toContain('href="https://demo.example"');
		expect(page).toContain('<span class="lang-badge">TypeScript</span>');
		expect(page).toContain("3 saves");
		expect(page).toContain('data-copy="nix profile add x"');
		expect(page.match(/class="dot d\d"/g)?.length).toBe(3);
		expect(page).toContain("density = activity · position = time");
		const history = [...page.matchAll(/class="history-id">([0-9a-f]+)</g)].map((m) => m[1]);
		expect(history).toEqual([head(), second(), first()]);
	});

	test("checkpoints never appear", async () => {
		expect(await body("/demo")).not.toContain(fx.saveIds[1] as string);
	});

	test("the README points relative links and images at raw files and drops unsafe links", async () => {
		const page = await body("/demo");
		expect(page).toContain('<a href="/demo/raw/docs/guide.md">the guide</a>');
		expect(page).toContain('<img src="/demo/raw/docs/logo.png" alt="logo">');
		expect(page).toContain('<a href="https://example.com">the site</a>');
		expect(page).not.toContain('href="javascript:');
		expect(page).toContain("&lt;script&gt;alert");
		expect(page).not.toContain("<script>alert");
	});
});

describe("the file viewer", () => {
	test("shows a file at HEAD and at an old save, with line anchors", async () => {
		const now = await get("/demo/files/src/main.ts");
		expect(now.headers.get("cache-control")).toBe("public, max-age=60");
		const nowPage = await now.text();
		expect(nowPage).toContain("hello");
		expect(nowPage).toContain(
			'<span class="line" id="L2"><a class="ln" href="#L2">2</a><span class="lc">',
		);
		expect(nowPage).toContain('class="tk-');

		const then = await get(`/demo/files/src/main.ts?at=${second().slice(0, 6)}`);
		expect(then.headers.get("cache-control")).toBe("public, max-age=31536000");
		const thenPage = await then.text();
		expect(thenPage).toContain("&#39;hi&#39;");
		expect(thenPage).not.toContain("hello");
		expect(thenPage).toContain(`/demo/raw/src/main.ts?at=${second()}`);
	});

	test("a file absent at that save, a bad save or an unsafe path is a 404", async () => {
		expect((await get(`/demo/files/src/main.ts?at=${first()}`)).status).toBe(404);
		expect((await get("/demo/files/src/main.ts?at=zzzzzz")).status).toBe(404);
		expect((await get("/demo/files/src/main.ts?at=abc")).status).toBe(404);
		expect((await get("/demo/files/src%2F..%2FREADME.md")).status).toBe(404);
		expect((await get("/demo/files/nothing.txt")).status).toBe(404);
	});

	test("Markdown renders, with a source toggle", async () => {
		const rendered = await body("/demo/files/docs/guide.md");
		expect(rendered).toContain("<h1>Guide</h1>");
		const source = await body("/demo/files/docs/guide.md?view=source");
		expect(source).toContain('id="L1"');
		expect(source).not.toContain("<h1>Guide</h1>");
	});

	test("an image displays inline from its raw file", async () => {
		expect(await body("/demo/files/docs/logo.png")).toContain('<img src="/demo/raw/docs/logo.png"');
	});
});

describe("raw files", () => {
	test("are the exact object bytes, text as text/plain", async () => {
		const mirror = openMirror(mirrorDir, "demo");
		for (const [path, type] of [
			["src/main.ts", "text/plain; charset=utf-8"],
			["docs/logo.png", "image/png"],
			["README.md", "text/plain; charset=utf-8"],
		] as const) {
			const response = await get(`/demo/raw/${path}`);
			const file = fileAt(mirror, head(), path);
			expect(response.headers.get("content-type")).toBe(type);
			expect(new Uint8Array(await response.arrayBuffer())).toEqual(
				objectBytes(mirror, file?.hash as string) as Uint8Array<ArrayBuffer>,
			);
		}
		expect(await (await get(`/demo/raw/src/main.ts?at=${second()}`)).text()).toContain("'hi'");
	});
});

test("static assets are served and the CSP allows Bunny Fonts but no inline script", async () => {
	const css = await get("/static/css/kepr.css");
	expect(css.status).toBe(200);
	expect(css.headers.get("content-type")).toContain("text/css");
	expect((await get("/static/../package.json")).status).toBe(404);
	const csp = (await get("/demo")).headers.get("content-security-policy") ?? "";
	expect(csp).toContain("style-src 'self' https://fonts.bunny.net");
	expect(csp).toContain("font-src https://fonts.bunny.net");
	expect(csp).not.toContain("unsafe-inline");
});