import { describe, expect, it } from "bun:test";
import {
	describeInflight,
	describeScreenshot,
	type InflightOp,
	imageFormatForPath,
} from "@oh-my-pi/pi-coding-agent/tools/browser/tab-worker";

describe("browser op tracking — timeout diagnostics", () => {
	it("labels a screenshot op by its distinguishing argument", () => {
		expect(describeScreenshot({ selector: ".wb-paper-popover" })).toBe(
			'tab.screenshot({ selector: ".wb-paper-popover" })',
		);
		expect(describeScreenshot({ fullPage: true })).toBe("tab.screenshot({ fullPage: true })");
		expect(describeScreenshot()).toBe("tab.screenshot()");
		expect(describeScreenshot({})).toBe("tab.screenshot()");
	});

	it("names every still-running helper so a cell timeout is attributable", () => {
		const now = Date.now();
		// Inserted newest-first to prove the summary sorts by start time, not insertion order.
		const inflight = new Map<number, InflightOp>([
			[1, { label: "tab.observe()", startedAt: now - 1_000 }],
			[0, { label: 'tab.screenshot({ selector: ".x" })', startedAt: now - 3_000 }],
		]);

		const summary = describeInflight(inflight);

		// Oldest op (most likely the culprit) is listed first.
		expect(summary.indexOf("tab.screenshot")).toBeLessThan(summary.indexOf("tab.observe"));
		// Each op carries an elapsed-seconds annotation.
		expect(summary).toMatch(/tab\.screenshot\(\{ selector: "\.x" \}\) \(\d+\.\d+s\)/);
		expect(summary).toMatch(/tab\.observe\(\) \(\d+\.\d+s\)/);
	});

	it("returns an empty summary when nothing is in flight", () => {
		expect(describeInflight(new Map())).toBe("");
	});
});

describe("imageFormatForPath — explicit save capture format", () => {
	it("maps the save path's extension to the matching capture format", () => {
		expect(imageFormatForPath("/tmp/shot.webp")).toBe("webp");
		expect(imageFormatForPath("/tmp/shot.WEBP")).toBe("webp");
		expect(imageFormatForPath("/tmp/shot.jpg")).toBe("jpeg");
		expect(imageFormatForPath("/tmp/shot.jpeg")).toBe("jpeg");
		expect(imageFormatForPath("/tmp/shot.png")).toBe("png");
	});

	it("falls back to png for unknown or missing extensions", () => {
		expect(imageFormatForPath("/tmp/shot")).toBe("png");
		expect(imageFormatForPath("/tmp/shot.txt")).toBe("png");
		// A dotted directory must not leak its "extension" into an extensionless basename.
		expect(imageFormatForPath("/tmp/v1.2/shot")).toBe("png");
	});
});