pi-ext

import { mkdirSync, mkdtempSync, rmSync, writeFileSync } from "node:fs";
import { tmpdir } from "node:os";
import path from "node:path";
import { afterEach, describe, expect, it } from "vitest";
import {
	createPiSession,
	type PiSession,
} from "../../../test/fixture/create-pi-session.ts";
import { OPEN_BROWSER_SETTING } from "../implementation.ts";
import { resolveSetting } from "../src/pi-ext-settings.ts";

type Case = {
	user?: unknown;
	project?: unknown;
	env?: string;
	untrusted?: boolean;
};

describe("strata.openBrowser resolution", () => {
	let session: PiSession | undefined;
	let root: string | undefined;
	afterEach(async () => {
		await session?.dispose();
		session = undefined;
		if (root) rmSync(root, { recursive: true, force: true });
		root = undefined;
	});

	async function resolve(input: Case) {
		root = mkdtempSync(path.join(tmpdir(), "pi-ext-strata-settings-"));
		if ("project" in input) {
			const dir = path.join(root, ".pi");
			mkdirSync(dir, { recursive: true });
			writeFileSync(
				path.join(dir, "settings.json"),
				JSON.stringify({ strata: { openBrowser: input.project } }),
			);
		}
		session = await createPiSession({
			cwd: root,
			env: { STRATA_NO_OPEN: input.env },
			...("user" in input
				? { userSettings: { strata: { openBrowser: input.user } } }
				: {}),
		});
		if (input.untrusted)
			session.session.settingsManager.setProjectTrusted(false);
		const ctx = session.session.extensionRunner.createContext();
		return resolveSetting(
			{ getFlag: () => undefined },
			ctx,
			OPEN_BROWSER_SETTING,
		);
	}

	it.each([
		["default on", {}, true, "default"],
		["user false", { user: false }, false, "user"],
		[
			"project true over user false",
			{ user: false, project: true },
			true,
			"project",
		],
		[
			"project false over user true",
			{ user: true, project: false },
			false,
			"project",
		],
		["env 1 over project true", { project: true, env: "1" }, false, "env"],
		["env true over user true", { user: true, env: "true" }, false, "env"],
		["env 0 enables over user false", { user: false, env: "0" }, true, "env"],
		[
			"env false enables over project false",
			{ project: false, env: "false" },
			true,
			"env",
		],
		[
			"untrusted project ignored",
			{ user: false, project: true, untrusted: true },
			false,
			"user",
		],
	] satisfies [string, Case, boolean, string][])(
		"%s",
		async (_name, input, value, source) => {
			expect(await resolve(input)).toEqual({ ok: true, value, source });
		},
	);

	it.each([
		["string project", { user: true, project: "false" }, "project settings"],
		["numeric user", { user: 0 }, "user settings"],
		[
			"empty env",
			{ user: false, env: "" },
			"environment variable STRATA_NO_OPEN",
		],
		[
			"word env",
			{ user: false, env: "yes" },
			"environment variable STRATA_NO_OPEN",
		],
	] satisfies [string, Case, string][])(
		"rejects %s without falling through",
		async (_name, input, from) => {
			expect(await resolve(input)).toMatchObject({
				ok: false,
				error: `Invalid strata.openBrowser from ${from}`,
			});
		},
	);
});