import { rootDir } from '../tool-utils';
import { mergePotToPo } from '../utils/translation';
import { supportedLocales } from './utils/supportedLocales';
import { GettextExtractor, HtmlExtractors } from 'gettext-extractor';

const websiteAssetsDir = `${rootDir}/Assets/WebsiteAssets`;
const localesDir = `${websiteAssetsDir}/locales`;

const createPotFile = async (potFilePath: string) => {
	const extractor = new GettextExtractor();

	const htmlParser = extractor
		.createHtmlParser([
			HtmlExtractors.elementContent('[translate]'),
		]);

	htmlParser.parseFile(`${websiteAssetsDir}/templates/front.mustache`);
	htmlParser.parseFile(`${websiteAssetsDir}/templates/plans.mustache`);
	htmlParser.parseFile(`${websiteAssetsDir}/templates/partials/plan.mustache`);

	extractor.savePotFile(potFilePath);
};

const main = async () => {
	const potFilePath = `${websiteAssetsDir}/website.pot`;
	await createPotFile(potFilePath);

	for (const locale of supportedLocales) {
		if (locale.id === 'en_GB') continue; // English is the source language
		await mergePotToPo(potFilePath, `${localesDir}/${locale.id}.po`);
	}
};

main().catch((error) => {
	console.error(error);
	process.exit(1);
});