import { expect, test } from "vitest";
import { load, dump } from "js-yaml";
import { generateRandomHash } from "@/server/utils/docker/compose";
import type { ComposeSpecification } from "@/server/utils/docker/types";
import { addPrefixToSecretsRoot } from "@/server/utils/docker/compose/secrets";

test("Generate random hash with 8 characters", () => {
	const hash = generateRandomHash();

	expect(hash).toBeDefined();
	expect(hash.length).toBe(8);
});

const composeFileSecretsRoot = `
version: "3.8"

services:
  web:
    image: nginx:latest

secrets:
  db_password:
    file: ./db_password.txt
`;

test("Add prefix to secrets in root property", () => {
	const composeData = load(composeFileSecretsRoot) as ComposeSpecification;
	const prefix = generateRandomHash();

	if (!composeData?.secrets) {
		return;
	}
	const secrets = addPrefixToSecretsRoot(composeData.secrets, prefix);
	expect(secrets).toBeDefined();
	if (secrets) {
		for (const secretKey of Object.keys(secrets)) {
			expect(secretKey).toContain(`-${prefix}`);
			expect(secrets[secretKey]).toBeDefined();
		}
	}
});

const composeFileSecretsRoot1 = `
version: "3.8"

services:
  api:
    image: myapi:latest

secrets:
  api_key:
    file: ./api_key.txt
`;

test("Add prefix to secrets in root property (Test 1)", () => {
	const composeData = load(composeFileSecretsRoot1) as ComposeSpecification;
	const prefix = generateRandomHash();

	if (!composeData?.secrets) {
		return;
	}
	const secrets = addPrefixToSecretsRoot(composeData.secrets, prefix);
	expect(secrets).toBeDefined();

	if (secrets) {
		for (const secretKey of Object.keys(secrets)) {
			expect(secretKey).toContain(`-${prefix}`);
			expect(secrets[secretKey]).toBeDefined();
		}
	}
});

const composeFileSecretsRoot2 = `
version: "3.8"

services:
  frontend:
    image: nginx:latest

secrets:
  frontend_secret:
    file: ./frontend_secret.txt
  db_password:
    external: true
`;

test("Add prefix to secrets in root property (Test 2)", () => {
	const composeData = load(composeFileSecretsRoot2) as ComposeSpecification;
	const prefix = generateRandomHash();

	if (!composeData?.secrets) {
		return;
	}
	const secrets = addPrefixToSecretsRoot(composeData.secrets, prefix);
	expect(secrets).toBeDefined();

	if (secrets) {
		for (const secretKey of Object.keys(secrets)) {
			expect(secretKey).toContain(`-${prefix}`);
			expect(secrets[secretKey]).toBeDefined();
		}
	}
});