import fs from "fs";
import path from "path";

import webpack from "webpack";

/**
 * Webpack plugin that reads all JSON rule files from `src/rules/`,
 * assigns sequential IDs to each rule, and emits a single `rules.json`
 * asset used by `declarativeNetRequest`.
 */
class CreateRulesPlugin {
	/**
	 * @param {import("webpack").Compiler} compiler
	 * @returns {void}
	 */
	apply(compiler) {
		compiler.hooks.compilation.tap("CreateRulesPlugin", (compilation) => {
			compilation.hooks.processAssets.tap(
				{
					name: "CreateRulesPlugin",
					stage: compiler.webpack.Compilation.PROCESS_ASSETS_STAGE_ADDITIONAL
				},
				(_assets) => {
					const ruleFileNames = fs.readdirSync(path.resolve(SOURCE_DIR, "rules"));
					const rules = [];
					let ruleId = 1;

					for (const ruleFileName of ruleFileNames) {
						const rule = JSON.parse(fs.readFileSync(path.resolve("src/rules", ruleFileName)));

						if (rule instanceof Array) {
							for (const currentRule of rule) {
								currentRule.id = ruleId++;
								rules.push(currentRule);
							}
						} else {
							rule.id = ruleId++;
							rules.push(rule);
						}
					}

					compilation.emitAsset(
						"rules.json",
						new webpack.sources.RawSource(
							JSON.stringify(
								rules,
								undefined,
								process.env.NODE_ENV === "production"
									? undefined
									: "\t"
							)
						)
					);
				}
			);
		});
	}
}

export default CreateRulesPlugin;