mediabunny/scripts/add-import-extensions.ts

import * as fs from 'fs';
import * as path from 'path';

// .js extensions are technically required in compliant ECMAScript, and Webpack needs them, so we add them here.

const walkDir = (dir: string) => {
	const files: string[] = [];
	const items = fs.readdirSync(dir);

	for (const item of items) {
		const fullPath = path.join(dir, item);
		const stat = fs.statSync(fullPath);

		if (stat.isDirectory()) {
			files.push(...walkDir(fullPath));
		} else if (item.endsWith('.js') || item.endsWith('.ts')) {
			files.push(fullPath);
		}
	}

	return files;
};

const fixFile = (filePath: string) => {
	const content = fs.readFileSync(filePath, 'utf8');
	// We only match relative imports
	let fixed = content.replace(
		/(\s+from\s+['"])(\.[^'"]*)(['"])/g, // This matches static imports
		'$1$2.js$3',
	);
	fixed = fixed.replace(
		/(import\s*\(\s*['"])(\.[^'"]*)(['"]\s*\))/g, // This matches dynamic imports
		'$1$2.js$3',
	);

	if (content !== fixed) {
		fs.writeFileSync(filePath, fixed);
	}
};

const jsFiles = walkDir('dist');
jsFiles.forEach(fixFile);