feat: enable cache

tinywaves · tinywaves · commit af0c440d03ff · 2024-12-17T22:16:30.000+08:00
diff --git a/fixtures/html.html b/fixtures/html.html
@@ -4,6 +4,7 @@
     <meta charset="UTF-8" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>html</title>
+    <link rel="stylesheet" href="css.css" />
   </head>
   <body>
     html
diff --git a/src/cli.ts b/src/cli.ts
@@ -28,6 +28,12 @@ const cliOptions = [
     defaultValue: false,
     usage: `htttp-server --cors`,
   },
+  {
+    option: '-dc, --disable-cache',
+    description: 'Disable cache',
+    defaultValue: false,
+    usage: `htttp-server -dc`,
+  },
 ];
 
 cliOptions.forEach(({ option, description, defaultValue }) => program.option(option, description, defaultValue.toString()));
@@ -45,5 +51,6 @@ const server = new Server({
   baseDir: opts.directory,
   dataPosition: opts.data,
   cors: JSON.parse(opts.cors),
+  cache: !JSON.parse(opts.disableCache),
 });
 server.start();
diff --git a/src/index.ts b/src/index.ts
@@ -16,6 +16,7 @@ export default class Server {
   baseDir: string = DEFAULT_BASE_DIR;
   data: { [k: string]: ResourceItem[] } = {};
   cors: boolean = false;
+  cache: boolean = true;
 
   constructor(options?: ServerOptions) {
     if (options?.port) {
@@ -36,6 +37,9 @@ export default class Server {
     if (options?.cors) {
       this.cors = true;
     }
+    if (!options?.cache) {
+      this.cache = false;
+    }
   }
 
   start() {
@@ -89,6 +93,9 @@ export default class Server {
   }
 
   private async processFile(file: string, res: Res) {
+    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`);
     createReadStream(file).pipe(res);
   }
@@ -208,4 +215,23 @@ export default class Server {
       }
     }
   }
+
+  private async processCache(res: Res, file: string) {
+    // This method is not recommended for production use. Because this time is server time, not client time, but browsers use client time to adjust if the cache is expired.
+    // res.setHeader('Expires', new Date(Date.now() + 10 * 1000).toUTCString());
+
+    // res.setHeader('Cache-Control', 'no-cache'); <===> res.setHeader('Cache-Control', 'max-age=0');
+    // res.setHeader('Cache-Control', 'no-store');
+    // no-cache: The browser will send a request to the server to check if the cache is expired, but the browser has cached the response.
+    // no-store: The browser will send a request to the server to check if the cache is expired, and the browser has not cached the response.
+
+    const stat = await fs.stat(file);
+    res.setHeader('Last-Modified', stat.mtime.toUTCString());
+    res.setHeader('Cache-Control', 'max-age=3600');
+    const ifModifiedSince = res.req?.headers['if-modified-since'];
+    if (ifModifiedSince && ifModifiedSince === stat.mtime.toUTCString()) {
+      res.statusCode = 304;
+      return res.end();
+    }
+  }
 }
diff --git a/src/types.ts b/src/types.ts
@@ -5,6 +5,7 @@ export interface ServerOptions {
   baseDir?: string;
   dataPosition?: string;
   cors?: boolean;
+  cache?: boolean;
 }
 
 export type Res = http.ServerResponse<http.IncomingMessage> & {

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@ export interface ServerOptions {`
`5`	`5`	`baseDir?: string;`
`6`	`6`	`dataPosition?: string;`
`7`	`7`	`cors?: boolean;`
	`8`	`+ cache?: boolean;`
`8`	`9`	`}`
`9`	`10`
`10`	`11`	`export type Res = http.ServerResponse<http.IncomingMessage> & {`