oh-my-pi/packages/coding-agent/test/export-html-markdown.test.ts

import { describe, expect, test } from "bun:test";
import * as vm from "node:vm";
import { parseHTML } from "linkedom";
import { Marked } from "marked";

const [templateHtml, templateJs] = await Promise.all([
	Bun.file(new URL("../src/export/html/template.html", import.meta.url)).text(),
	Bun.file(new URL("../src/export/html/template.js", import.meta.url)).text(),
]);

interface MinimalMessageEntry {
	type: "message";
	id: string;
	parentId: string | null;
	timestamp: string;
	message: {
		role: "user" | "assistant";
		content: string | unknown[];
		timestamp: number;
	};
}

interface MinimalSession {
	header: {
		type: "session";
		version: number;
		id: string;
		timestamp: string;
		cwd: string;
	};
	entries: MinimalMessageEntry[];
	leafId: string;
}

function renderSession(session: MinimalSession) {
	const { document, window } = parseHTML(templateHtml);
	const sessionData = document.getElementById("session-data");
	if (!sessionData) throw new Error("Export template is missing session data");
	sessionData.textContent = Buffer.from(JSON.stringify(session)).toBase64();
	Object.defineProperty(window, "location", {
		value: new URL("https://example.test/export.html"),
		configurable: true,
	});
	Object.defineProperty(window, "matchMedia", {
		value: () => ({ matches: false, addEventListener() {}, removeEventListener() {} }),
		configurable: true,
	});

	const context = vm.createContext({
		window,
		document,
		marked: new Marked(),
		hljs: {
			getLanguage: () => false,
			highlight: () => ({ value: "" }),
			highlightAuto: () => ({ value: "" }),
		},
		URL,
		URLSearchParams,
		TextDecoder,
		Uint8Array,
		atob,
		navigator: { clipboard: null },
		localStorage: { getItem: () => null, setItem() {} },
		setTimeout: () => 0,
		clearTimeout() {},
	});
	vm.runInContext(templateJs, context);
	return document;
}

function createSession(entries: MinimalMessageEntry[], leafId: string, id: string): MinimalSession {
	return {
		header: {
			type: "session",
			version: 3,
			id,
			timestamp: "2026-01-01T00:00:00.000Z",
			cwd: "/tmp",
		},
		entries,
		leafId,
	};
}

function createDeepChainSession(depth: number): MinimalSession {
	const entries: MinimalMessageEntry[] = [
		{
			type: "message",
			id: "message-0",
			parentId: null,
			timestamp: "2026-01-01T00:00:00.000Z",
			message: {
				role: "user",
				content: "root",
				timestamp: 0,
			},
		},
	];
	for (let i = 1; i < depth; i++) {
		entries.push({
			type: "message",
			id: `message-${i}`,
			parentId: `message-${i - 1}`,
			timestamp: new Date(Date.UTC(2026, 0, 1, 0, 0, i)).toISOString(),
			message: {
				role: "assistant",
				content: [],
				timestamp: i,
			},
		});
	}
	return createSession(entries, "message-0", "deep-chain-test");
}

function renderMarkdown(source: string): Element {
	const document = renderSession(
		createSession(
			[
				{
					type: "message",
					id: "message-1",
					parentId: null,
					timestamp: "2026-01-01T00:00:00.000Z",
					message: {
						role: "user",
						content: source,
						timestamp: 0,
					},
				},
			],
			"message-1",
			"markdown-test",
		),
	);

	const rendered = document.querySelector(".markdown-content");
	if (!rendered) throw new Error("Export viewer did not render Markdown content");
	return rendered;
}

describe("HTML export Markdown", () => {
	test("renders inline Markdown in ordered, unordered, and nested list items", () => {
		const rendered = renderMarkdown("**outside**\n\n- **bold** and *italic* and `code`\n  1. **nested**");

		expect(rendered.querySelector("p strong")?.textContent).toBe("outside");
		expect(rendered.querySelector("ul > li > strong")?.textContent).toBe("bold");
		expect(rendered.querySelector("ul > li > em")?.textContent).toBe("italic");
		expect(rendered.querySelector("ul > li > code")?.textContent).toBe("code");
		expect(rendered.querySelector("ol > li > strong")?.textContent).toBe("nested");
	});

	test("renders a deep valid conversation tree without overflowing the call stack", () => {
		const document = renderSession(createDeepChainSession(30_000));

		expect(document.querySelectorAll(".tree-node").length).toBe(1);
		expect(document.querySelector(".tree-node.active")?.getAttribute("data-id")).toBe("message-0");
		expect(document.querySelector("#messages")?.textContent).toContain("root");
	});
});