import * as path from 'path';
import * as fs from 'fs';
import readdirp, { EntryInfo } from 'readdirp';

const inputFolder = path.resolve(process.argv[2]);

console.log(`Clearing private definitions in ${inputFolder}`);

function filterTypeScriptFiles(content: string) {
	var leadingPrivate = /^.*@private/gi;
	if (leadingPrivate.test(content)) {
		return { shouldDelete: true };
	}

	let blockCommentPrivate = /\/\*\*([^](?!\*\/))*@module([^](?!\*\/))*@private[^]*?\*\//g;
	if (blockCommentPrivate.test(content)) {
		return { shouldDelete: true };
	}

	let newContent = content;
	newContent = newContent.replace(/\/\/[\/\s]*@private[^]*?\/\/[\/\s]*?@endprivate/gm, '');

	if (newContent !== content) {
		return { shouldReplace: true, newContent: newContent };
	}

	return { shouldReplace: false, shouldDelete: false };
}

readdirp(inputFolder, {
	fileFilter: ['*.d.ts'],
	directoryFilter: function (di) {
		return !di.path.includes('node_modules');
	},
})
	.on('data', (entry: EntryInfo) => {
		const { fullPath } = entry;
		const content = fs.readFileSync(fullPath, 'utf8');
		const { shouldDelete, shouldReplace, newContent } = filterTypeScriptFiles(content);

		if (shouldDelete) {
			console.log('[Delete]', fullPath);
			fs.unlinkSync(fullPath);
		} else if (shouldReplace) {
			console.log('[Cleared]', fullPath);
			try {
				fs.writeFileSync(fullPath, newContent || '', 'utf8');
			} catch (error) {
				console.log('ERROR writing file: ' + fullPath, error);
				process.exit(1);
			}
		}
	})
	.on('warn', (error: Error) => console.error('non-fatal error', error))
	.on('error', (error: Error) => {
		console.error('fatal error', error);
		process.exit(1);
	})
	.on('end', () => console.log('done'));