feat: support compression

tinywaves · tinywaves · commit 9fa7fd0323b9 · 2024-12-18T01:23:11.000+08:00
diff --git a/fixtures/css.css b/fixtures/css.css
@@ -1,3 +1,4 @@
 html {
   background-color: #000;
+  color: #fff;
 }
diff --git a/fixtures/html.html b/fixtures/html.html
@@ -7,6 +7,6 @@
     <link rel="stylesheet" href="css.css" />
   </head>
   <body>
-    html
+    html - 中文
   </body>
 </html>
diff --git a/src/cli.ts b/src/cli.ts
@@ -34,6 +34,12 @@ const cliOptions = [
     defaultValue: false,
     usage: `htttp-server -dc`,
   },
+  {
+    option: '--compress',
+    description: 'Enable compress',
+    defaultValue: false,
+    usage: `htttp-server --compress`,
+  },
 ];
 
 cliOptions.forEach(({ option, description, defaultValue }) => program.option(option, description, defaultValue.toString()));
@@ -52,5 +58,6 @@ const server = new Server({
   dataPosition: opts.data,
   cors: JSON.parse(opts.cors),
   cache: !JSON.parse(opts.disableCache),
+  compress: JSON.parse(opts.compress),
 });
 server.start();
diff --git a/src/index.ts b/src/index.ts
@@ -4,6 +4,8 @@ import * as fs from 'node:fs/promises';
 import { createReadStream, readFileSync, statSync } from 'node:fs';
 import * as path from 'node:path';
 import { Buffer } from 'node:buffer';
+import { createBrotliCompress, createDeflate, createGzip } from 'node:zlib';
+import type { Gzip } from 'node:zlib';
 import chalk from 'chalk';
 import * as ejs from 'ejs';
 import mine from 'mime';
@@ -17,6 +19,7 @@ export default class Server {
   data: { [k: string]: ResourceItem[] } = {};
   cors: boolean = false;
   cache: boolean = true;
+  compress: boolean = false;
 
   constructor(options?: ServerOptions) {
     if (options?.port) {
@@ -40,6 +43,9 @@ export default class Server {
     if (!options?.cache) {
       this.cache = false;
     }
+    if (options?.compress) {
+      this.compress = true;
+    }
   }
 
   start() {
@@ -57,7 +63,7 @@ export default class Server {
           if (stat.isDirectory()) {
             this.processDirectory(wholePath, res, requestUrl);
           } else {
-            this.processFile(wholePath, res);
+            this.processFile(wholePath, res, req);
           }
         }
       } catch {
@@ -92,11 +98,20 @@ export default class Server {
     res.end(html);
   }
 
-  private async processFile(file: string, res: Res) {
+  private async processFile(file: string, res: Res, req: Req) {
     if (this.cache) {
       this.processCache(res, file); // The homepage will not be cached.
     }
     res.setHeader('Content-Type', `${mine.getType(file) ?? 'text/plain'};charset=utf-8`);
+    if (this.compress) {
+      const compressedStreamResponse = this.processCompress(req);
+      if (compressedStreamResponse) {
+        const [compressedStream, encoding] = compressedStreamResponse;
+        res.setHeader('Content-Encoding', encoding);
+        createReadStream(file).pipe(compressedStream).pipe(res);
+        return;
+      }
+    }
     createReadStream(file).pipe(res);
   }
 
@@ -244,4 +259,15 @@ export default class Server {
       return res.end();
     }
   }
+
+  private processCompress(req: Req): [Gzip, string] | undefined {
+    const acceptEncoding = req.headers['accept-encoding'];
+    if (acceptEncoding?.includes('gzip')) {
+      return [createGzip(), 'gzip'];
+    } else if (acceptEncoding?.includes('deflate')) {
+      return [createDeflate(), 'deflate'];
+    } else if (acceptEncoding?.includes('br')) {
+      return [createBrotliCompress(), 'br'];
+    }
+  }
 }
diff --git a/src/types.ts b/src/types.ts
@@ -6,6 +6,7 @@ export interface ServerOptions {
   dataPosition?: string;
   cors?: boolean;
   cache?: boolean;
+  compress?: boolean;
 }
 
 export type Res = http.ServerResponse<http.IncomingMessage> & {

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
`1`	`1`	`html {`
`2`	`2`	`background-color: #000;`
	`3`	`+ color: #fff;`
`3`	`4`	`}`
Original file line number	Diff line number	Diff line change
`@@ -6,6 +6,7 @@ export interface ServerOptions {`
`6`	`6`	`dataPosition?: string;`
`7`	`7`	`cors?: boolean;`
`8`	`8`	`cache?: boolean;`
	`9`	`+ compress?: boolean;`
`9`	`10`	`}`
`10`	`11`
`11`	`12`	`export type Res = http.ServerResponse<http.IncomingMessage> & {`