import { execFileSync } from "node:child_process";
import { mkdirSync, rmSync, writeFileSync } from "node:fs";
import { join } from "node:path";
import { getCreateEnvFileCommand } from "@dokploy/server/utils/builders/compose";
import { afterEach, describe, expect, it } from "vitest";

// Regression coverage for https://github.com/Dokploy/dokploy/issues/4694 —
// values must survive Docker Compose's own `.env` parsing, not just base64 decode.
const appName = `env-file-literals-${process.pid}`;
const projectPath = join(process.cwd(), ".docker", "compose", appName);
const codePath = join(projectPath, "code");

afterEach(() => {
	try {
		execFileSync("docker", ["compose", "down", "--remove-orphans"], {
			cwd: codePath,
			stdio: "ignore",
		});
	} catch {
		// Project may not have been created (e.g. an earlier assertion failed).
	}
	rmSync(projectPath, { force: true, recursive: true });
});

const cases: Record<string, string> = {
	PASSWORD: "pa$$word",
	SPECIAL: '!"#$%&/()=?',
	NESTED_JSON: '{"nested":{"a":1}}',
	MAIL_PASSWORD: "abc#de",
	TRAILING_BACKSLASH: "trailing\\",
	QUOTE_INSIDE: 'she said "hi"',
	APOSTROPHE: "it's a test",
	UNICODE: "héllo wörld 日本語 🚀",
	MULTILINE_PEM: "-----BEGIN KEY-----\nabc123\n-----END KEY-----",
};

// How each value must be typed in the UI so the (unchanged) dotenv input
// parser resolves it to the raw string in `cases` above.
const inputEncoding: Record<string, string> = {
	PASSWORD: "pa$$word",
	SPECIAL: `'!"#$%&/()=?'`,
	NESTED_JSON: '{"nested":{"a":1}}',
	MAIL_PASSWORD: `"abc#de"`,
	TRAILING_BACKSLASH: "trailing\\",
	QUOTE_INSIDE: 'she said "hi"',
	APOSTROPHE: "it's a test",
	UNICODE: "héllo wörld 日本語 🚀",
	MULTILINE_PEM: '"-----BEGIN KEY-----\nabc123\n-----END KEY-----"',
};

describe("getCreateEnvFileCommand", () => {
	it("writes special environment values that Docker Compose reads back literally", () => {
		mkdirSync(codePath, { recursive: true });

		const serviceEnv = Object.entries(inputEncoding)
			.map(([key, value]) => `${key}=${value}`)
			.join("\n");

		const command = getCreateEnvFileCommand({
			appName,
			composePath: "docker-compose.yml",
			env: serviceEnv,
			randomize: false,
			suffix: "",
			serverId: null,
			environment: { project: { env: "" }, env: "" },
		} as Parameters<typeof getCreateEnvFileCommand>[0]);

		execFileSync("bash", ["-c", command]);

		const composeFile = `services:\n  test:\n    image: busybox\n    environment:\n${Object.keys(
			cases,
		)
			.map((key) => `      - ${key}=\${${key}}`)
			.join("\n")}\n`;
		writeFileSync(join(codePath, "docker-compose.yml"), composeFile);

		const dumpScript = `for k in ${Object.keys(cases).join(" ")}; do printf '%s\\0' "$k"; eval "printf '%s\\0' \\"\\$$k\\""; done`;

		const out = execFileSync(
			"docker",
			["compose", "run", "--rm", "-T", "test", "sh", "-c", dumpScript],
			{ cwd: codePath, encoding: "utf8" },
		);

		const parts = out.split("\0");
		const actual: Record<string, string> = {};
		for (let i = 0; i < parts.length - 1; i += 2) {
			actual[parts[i] as string] = parts[i + 1] as string;
		}

		for (const [key, value] of Object.entries(cases)) {
			expect(actual[key], key).toBe(value);
		}
	}, 60000);
});