Add compression logic

src/compress.ts

@@ -0,0 +1,47 @@
+import { globby } from "globby";
+import { createReadStream, createWriteStream } from "node:fs";
+import { hrtime } from "node:process";
+import { promises as stream } from "node:stream";
+import { createBrotliCompress, createGzip } from "node:zlib";
+
+import { Logger } from "./logger.js";
+
+const filterFile = (file: string): boolean => {
+  return [".css", ".js", ".html", ".xml", ".cjs", ".mjs", ".svg", ".txt"].every((ext) =>
+    file.endsWith(ext),
+  );
+};
+
+export const gzip = async (dir: URL): Promise<void> => {
+  const start = hrtime.bigint();
+
+  const files = (await globby(dir.pathname)).filter(filterFile);
+  for (const file of files) {
+    if (filterFile(file)) continue;
+    const source = createReadStream(file);
+    const destination = createWriteStream(`${file}.gz`);
+    const gzip = createGzip({ level: 9 });
+    await stream.pipeline(source, gzip, destination);
+  }
+
+  const end = hrtime.bigint();
+
+  Logger.info(`finished gzip of ${files.length} files in ${(end - start) / 1000000n}m`);
+};
+
+export const brotli = async (dir: URL): Promise<void> => {
+  const start = hrtime.bigint();
+
+  const files = (await globby(dir.pathname)).filter(filterFile);
+  for (const file of files) {
+    if (filterFile(file)) continue;
+    const source = createReadStream(file);
+    const destination = createWriteStream(`${file}.br`);
+    const brotli = createBrotliCompress();
+    await stream.pipeline(source, brotli, destination);
+  }
+
+  const end = hrtime.bigint();
+
+  Logger.info(`finished brotli of ${files.length} files in ${(end - start) / 1000000n}m`);
+};

src/index.ts

@@ -1,12 +1,14 @@
 import type { AstroIntegration } from "astro";
 
+import { brotli, gzip } from "./compress.js";
 import { Logger } from "./logger.js";
 
 export const createCompressionPlugin = (): AstroIntegration => {
   return {
     name: "astro-compressor",
     hooks: {
       "astro:build:done": async ({ dir }) => {
+        await Promise.allSettled([gzip(dir), brotli(dir)]);
         Logger.success("Compression finished\n");
       },
     },