/** biome-ignore-all lint/suspicious/noConsole: Biome v2 migration */
import fs from "node:fs";

import path from "node:path";
import { fileURLToPath } from "node:url";

const __filename = fileURLToPath(import.meta.url);
const __dirname = path.dirname(__filename);

const NO_WRITE_KEY = "--no-write";
const dontWrite = process.argv.includes(NO_WRITE_KEY);

const KNOWN_SUFFIXES = ["_zero", "_one", "_two", "_few", "_many", "_other"];

const REPO_TRANSLATIONS_INFO_URL =
	"https://github.com/sendou-ink/sendou.ink#translations";

const MD = {
	inlineCode: (s: string) => `\`${s}\``,
	strong: (s: string) => `**${s}**`,
	h2: (s: string) => `## ${s}`,
	li: (s: string) => `- ${s}`,
	ticked: (s: string) => `- [x] ${s}`,
	unticked: (s: string) => `- [ ] ${s}`,
};

const otherLanguageTranslationPath = (code?: string, fileName?: string) =>
	path.join(
		...[__dirname, "..", "locales", code, fileName].filter(
			(val): val is string => !!val,
		),
	);

const allOtherLanguages = fs
	.readdirSync(otherLanguageTranslationPath())
	.filter((lang) => lang !== "en");

const missingTranslations: Record<
	string,
	Record<string, Array<string>>
> = Object.fromEntries(allOtherLanguages.map((lang) => [lang, {}]));

const totalTranslationCounts: Record<string, number> = {};

const fileNames: string[] = fs.readdirSync(otherLanguageTranslationPath("en"));

for (const file of fileNames) {
	const englishContent = JSON.parse(
		fs.readFileSync(otherLanguageTranslationPath("en", file), "utf8").trim(),
	) as Record<string, string>;
	const key = file.replace(".json", "");
	const englishContentKeys = getKeysWithoutSuffix(englishContent, "en", file);

	if (file !== "gear.json" && file !== "weapons.json") {
		totalTranslationCounts[key] = englishContentKeys.length;
	}

	for (const lang of allOtherLanguages) {
		// .DS_STORE
		if (lang.startsWith(".")) continue;

		try {
			const otherRawContent = fs
				.readFileSync(otherLanguageTranslationPath(lang, file), "utf8")
				.trim();
			let otherLanguageContent: Record<string, string>;
			try {
				otherLanguageContent = JSON.parse(otherRawContent);
			} catch {
				throw new Error(`failed to parse ${lang}/${file}`);
			}

			const otherLanguageContentKeys = getKeysWithoutSuffix(
				otherLanguageContent,
				lang,
				file,
			);

			validateVariables({
				english: englishContent,
				other: otherLanguageContent,
				lang,
				file,
			});
			validateNoDuplicateKeys({
				otherRawContent,
				file,
				lang,
			});

			const missingKeys = englishContentKeys.filter(
				(key) => !otherLanguageContentKeys.includes(key),
			);

			if (key === "weapons" || key === "gear") {
				if (missingKeys.length > 0) {
					throw new Error(`missing keys in ${lang}/${file}`);
				}
			} else {
				missingTranslations[lang][key] = missingKeys;
			}
		} catch (e) {
			if ((e as { code: string }).code !== "ENOENT") throw e;

			missingTranslations[lang][key] = englishContentKeys;
		}
	}
}

console.info("no issues found inside translation files");

if (dontWrite) {
	process.exit(0);
}

const markdown = createTranslationProgessMarkdown({
	missingTranslations,
	totalTranslationCounts,
});

const translationProgressPath = path.join(
	__dirname,
	"..",
	"translation-progress.md",
);

fs.writeFileSync(translationProgressPath, markdown);

function validateVariables({
	english,
	other,
	lang,
	file,
}: {
	english: Record<string, string>;
	other: Record<string, string>;
	lang: string;
	file: string;
}) {
	for (const [key, value] of Object.entries(english)) {
		const otherValue = other[key];
		if (!otherValue) continue;

		const englishMatches = value.match(/{{(.*?)}}/g);
		const otherMatches = otherValue.match(/{{(.*?)}}/g);

		if (!englishMatches && !otherMatches) continue;

		for (const englishVar of englishMatches ?? []) {
			if (!otherMatches?.includes(englishVar)) {
				throw new Error(
					`variable mismatch in ${lang}/${file}: ${englishVar} is missing in ${otherValue}`,
				);
			}
		}
	}
}

function validateNoDuplicateKeys({
	otherRawContent,
	lang,
	file,
}: {
	otherRawContent: string;
	lang: string;
	file: string;
}) {
	const keys = new Set<string>();
	const duplicateKeys = new Set<string>();
	for (const line of otherRawContent.split("\n")) {
		const key = line.trim().split(":")[0];
		if (!key) continue;

		if (keys.has(key)) {
			duplicateKeys.add(key);
		}
		keys.add(key);
	}

	if (duplicateKeys.size > 0) {
		throw new Error(
			`duplicate key(s) in ${lang}/${file}: ${Array.from(duplicateKeys).join(
				", ",
			)}`,
		);
	}
}

// get keys while respecting different plural/context key suffixes in different languages.
function getKeysWithoutSuffix(
	translations: Record<string, string>,
	lang: string,
	file: string,
): string[] {
	const foundSuffixKeys = new Set<string>();
	const keys = [];

	for (const [key, value] of Object.entries(translations)) {
		if (value === "") {
			continue; // Consider key missing if untranslated
		}

		const suffix = KNOWN_SUFFIXES.find((sfx) => key.endsWith(sfx));
		if (!suffix) {
			if (foundSuffixKeys.has(key)) {
				throw new Error(
					`Found same key with and without suffixes in ${lang}/${file}: ${key}`,
				);
			}
			keys.push(key);
			continue;
		}

		const baseKey = key.replace(suffix, "");

		if (foundSuffixKeys.has(baseKey)) {
			// Already found this key with a suffix. Duplicates are handled elsewhere.
			continue;
		}

		if (keys.includes(baseKey)) {
			throw new Error(
				`Found same key with and without suffixes in ${lang}/${file}: ${baseKey}`,
			);
		}

		keys.push(baseKey);
		foundSuffixKeys.add(baseKey);
	}

	return keys;
}

type StatusProps = {
	totalCount: number;
	missingCount: number;
	percentage?: boolean;
};
function MDCompletionStatus({
	totalCount,
	missingCount,
	percentage,
}: StatusProps) {
	const circle =
		missingCount === 0 ? "🟢" : missingCount === totalCount ? "🔴" : "🟡";

	const nonMissingCount = totalCount - missingCount;

	if (!percentage) {
		return `${circle} ${nonMissingCount}/${totalCount}`;
	}

	const percent =
		totalCount === 0 ? 100 : Math.floor((nonMissingCount / totalCount) * 100);

	return `${circle} ${percent}%`;
}

function MDOverviewTable({
	totalTranslationCounts,
}: {
	totalTranslationCounts: Record<string, number>;
}) {
	const totalKeysCount = Object.values(totalTranslationCounts).reduce(
		(a, b) => a + b,
		0,
	);
	const relevantFiles = fileNames.filter(
		(name) => name !== "weapons.json" && name !== "gear.json",
	);

	const rows = [];

	rows.push(
		`| Language | Total | ${relevantFiles.map(MD.inlineCode).join(" | ")} |`,
	);

	rows.push(`| :-- | :-: | ${relevantFiles.map(() => ":-:").join(" | ")} |`);

	for (const [lang, missingKeysObj] of Object.entries(missingTranslations)) {
		const cells = [];

		cells.push(MD.strong(lang));

		const totalAmountOfMissingKeys = Object.values(missingKeysObj).reduce(
			(a, b) => a + b.length,
			0,
		);

		cells.push(
			MD.strong(
				MDCompletionStatus({
					totalCount: totalKeysCount,
					missingCount: totalAmountOfMissingKeys,
					percentage: true,
				}),
			),
		);

		for (const file of relevantFiles) {
			const fileKey = file.replace(".json", "");
			const missingKeysInFile = missingKeysObj[fileKey];
			if (!missingKeysInFile) {
				return "";
			}

			cells.push(
				MDCompletionStatus({
					totalCount: totalTranslationCounts[fileKey],
					missingCount: missingKeysInFile.length,
				}),
			);
		}

		rows.push(`| ${cells.join(" | ")} |`);
	}

	return rows.join("\n");
}

function MDDetailsList({
	summary,
	content,
}: {
	summary: string;
	content: string[];
}) {
	return `<details><summary>${summary}</summary><ul>${content
		.map((c) => `<li>${c}</li>`)
		.join("")}</ul></details>`;
}

function MDMissingKeysList({
	missingTranslations,
}: {
	missingTranslations: Record<string, Record<string, Array<string>>>;
}) {
	const blocks = [];

	for (const [lang, missingKeysObj] of Object.entries(missingTranslations)) {
		const parts = [];

		parts.push(MD.h2(lang));

		for (const [fileKey, missingKeys] of Object.entries(missingKeysObj)) {
			const noneMissing = missingKeys.length === 0;
			const checkbox = noneMissing ? MD.ticked : MD.unticked;
			const fileEntry = checkbox(MD.inlineCode(`${fileKey}.json`));

			const allMissing = missingKeys.length === totalTranslationCounts[fileKey];

			const keysLabel = allMissing
				? "All keys"
				: missingKeys.length === 1
					? "1 key"
					: `${missingKeys.length} keys`;

			const details = noneMissing
				? ""
				: MDDetailsList({
						summary: `${keysLabel} missing`,
						content: allMissing
							? [
									`Create a fresh copy of ${MD.inlineCode(
										`en/${fileKey}.json`,
									)} to get started.`,
								]
							: missingKeys,
					});

			parts.push(`${fileEntry} ${details}`);
		}

		blocks.push(parts.join("\n"));
	}

	return blocks.join("\n\n");
}

function createTranslationProgessMarkdown({
	missingTranslations,
	totalTranslationCounts,
}: {
	missingTranslations: Record<string, Record<string, Array<string>>>;
	totalTranslationCounts: Record<string, number>;
}) {
	return `
> 🤖 This issue is fully automated, it should always be up-to-date.

# Translation Progress

If you want to contribute by adding missing translations, make sure to read the [project description](${REPO_TRANSLATIONS_INFO_URL}) on how to do this 💚

## Overview

Key: 🟢 = Done, 🟡 = In progress, 🔴 = Not started

${MDOverviewTable({ totalTranslationCounts })}

## Missing Keys
  
${MDMissingKeysList({ missingTranslations })}`;
}