// @ts-check

import { spawn } from 'node:child_process';
import { readdirSync, readFileSync, writeFileSync } from 'node:fs';
import * as path from 'node:path';
import pLimit from 'p-limit';
import { tsconfigResolverSync } from 'tsconfig-resolver';

function checkExamples() {
	let examples = readdirSync('./examples', { withFileTypes: true });
	examples = examples.filter((dirent) => dirent.isDirectory());

	console.log(`Running astro check on ${examples.length} examples...`);

	// Run astro check in parallel with 5 at most
	const checkPromises = [];
	const limit = pLimit(5);

	for (const example of examples) {
		checkPromises.push(
			limit(() =>
				new Promise((resolve) => {
					const originalConfig = prepareExample(example.name);
					let data = '';
					const child = spawn('node', ['../../packages/astro/astro.js', 'check'], {
						cwd: path.join('./examples', example.name),
						env: { ...process.env, FORCE_COLOR: 'true' },
					});

					child.stdout.on('data', function (buffer) {
						data += buffer.toString();
					});

					child.on('exit', (code) => {
						if (code !== 0) {
							console.error(data);
						}
						if (originalConfig) {
							resetExample(example.name, originalConfig);
						}
						resolve(code);
					});
				})
			)
		)
	}

	Promise.all(checkPromises).then((codes) => {
		if (codes.some((code) => code !== 0)) {
			process.exit(1);
		}

		console.log("No errors found!");
	});
}

/**
 * @param {string} examplePath
 */
function prepareExample(examplePath) {
	const tsconfigPath = path.join('./examples/', examplePath, 'tsconfig.json');
	const tsconfig = tsconfigResolverSync({ filePath: tsconfigPath, cache: false });
	let originalConfig = undefined;

	if (tsconfig.exists) {
		tsconfig.config.extends = 'astro/tsconfigs/strictest';
		originalConfig = readFileSync(tsconfigPath).toString();

		if (!tsconfig.config.compilerOptions) {
			tsconfig.config.compilerOptions = {};
		}

		tsconfig.config.compilerOptions = Object.assign(tsconfig.config.compilerOptions, {
			types: tsconfig.config.compilerOptions.types ?? [], // Speeds up tests
		});
	}

	writeFileSync(tsconfigPath, JSON.stringify(tsconfig.config));

	return originalConfig;
}

/**
 * @param {string} examplePath
 * @param {string} originalConfig
 */
function resetExample(examplePath, originalConfig) {
	const tsconfigPath = path.join('./examples/', examplePath, 'tsconfig.json');
	writeFileSync(tsconfigPath, originalConfig);
}

checkExamples();