rsbuild-plugin-compress 1.0.0

package/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 yeser
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

package/compress.mjs

@@ -0,0 +1,59 @@
+import { Worker } from "node:worker_threads";
+import * as os from "node:os";
+import * as path from "node:path";
+import * as fs from "node:fs";
+
+//#region compress.ts
+const pluginCompress = (options = {}) => {
+	const { enableGzip = true, enableBrotli = true, test = /\.(js|css|html|svg)$/, threshold = 1024 } = options;
+	const maxWorkers = Math.max(1, os.cpus().length - 1);
+	const queue = [];
+	return {
+		name: "rsbuild-plugin-compress",
+		setup(api) {
+			if (api.context.action !== "build") {
+				api.logger.info(`rsbuild-plugin-compress: ${api.context.action} mode, skip compression.`);
+				return;
+			} else api.logger.info(`rsbuild-plugin-compress: ${api.context.action} mode, start compression.`);
+			api.modifyRsbuildConfig((config) => {});
+			api.modifyRspackConfig((config) => {
+				if (!config.plugins) config.plugins = [];
+				config.plugins.push({ apply(compiler) {
+					compiler.hooks.afterEmit.tapPromise("rsbuild-plugin-compress", async (compilation) => {
+						const outDir = compiler.outputPath;
+						for (const filename of Object.keys(compilation.assets)) {
+							if (!test.test(filename)) continue;
+							const asset = compilation.getAsset(filename);
+							if (!asset) continue;
+							const source = asset?.source?.source?.()?.toString("utf-8");
+							if (!source || source.length < threshold) continue;
+							const worker = new Worker(path.resolve(__dirname, "./compress.worker.js"), { workerData: {
+								source,
+								enableGzip,
+								enableBrotli
+							} });
+							const job = new Promise((resolve, reject) => {
+								worker.on("message", async (result) => {
+									const filePath = path.join(outDir, filename);
+									if (result.gzip) await fs.promises.writeFile(`${filePath}.gz`, result.gzip);
+									if (result.brotli) await fs.promises.writeFile(`${filePath}.br`, result.brotli);
+									resolve();
+								});
+								worker.on("error", reject);
+							});
+							queue.push(job);
+							while (queue.length >= maxWorkers) {
+								await Promise.race(queue);
+								queue.splice(0, 1);
+							}
+						}
+						await Promise.all(queue);
+					});
+				} });
+			});
+		}
+	};
+};
+
+//#endregion
+export { pluginCompress };

package/compress.worker.js

@@ -0,0 +1,28 @@
+// build-plugins/compress.worker.ts
+import { parentPort, workerData } from 'node:worker_threads';
+import { promisify } from 'node:util';
+import zlib from 'node:zlib';
+
+const gzip = promisify(zlib.gzip);
+const brotliCompress = promisify(zlib.brotliCompress);
+
+async function run() {
+    const { source, enableGzip, enableBrotli } = workerData;
+    const result = {};
+
+    if (enableGzip) {
+        result.gzip = await gzip(source, { level: 9 });
+    }
+
+    if (enableBrotli) {
+        result.brotli = await brotliCompress(source, {
+            params: {
+                [zlib.constants.BROTLI_PARAM_QUALITY]: 11,
+            },
+        });
+    }
+
+    parentPort?.postMessage(result);
+}
+
+run();

package/package.json

@@ -0,0 +1,30 @@
+{
+  "name": "rsbuild-plugin-compress",
+  "version": "1.0.0",
+  "description": "",
+  "license": "ISC",
+  "author": "",
+  "type": "module",
+  "scripts": {
+    "build": "tsdown"
+  },
+  "devDependencies": {
+    "@rsbuild/core": "^1.7.3",
+    "@rspack/core": "^1.7.6",
+    "@types/node": "^25.3.2",
+    "tsdown": "^0.20.3",
+    "typescript": "^5.9.3"
+  },
+  "exports": {
+    ".": "./compress.mjs",
+    "./package.json": "./package.json"
+  },
+  "files": [
+    "compress.mjs",
+    "compress.worker.js"
+  ],
+  "publishConfig": {
+    "access": "public",
+    "registry": "https://registry.npmjs.org"
+  }
+}