feat(server): support headers configurable

vitejs · Nov 26, 2021 · e5e8cd2 · e5e8cd2
1 parent 494e358
commit e5e8cd2
Show file tree

Hide file tree

Showing 8 changed files with 47 additions and 14 deletions.
diff --git a/packages/playground/cli/vite.config.js b/packages/playground/cli/vite.config.js
@@ -2,7 +2,10 @@ const { defineConfig } = require('vite')
 
 module.exports = defineConfig({
   server: {
-    host: 'localhost'
+    host: 'localhost',
+    headers: {
+      'Cache-Control': 'no-store'
+    }
   },
   build: {
     //speed up build

diff --git a/packages/vite/src/node/__tests__/config.spec.ts b/packages/vite/src/node/__tests__/config.spec.ts
@@ -172,6 +172,9 @@ describe('preview config', () => {
     host: true,
     open: true,
     https: true,
+    headers: {
+      'Cache-Control': 'no-store'
+    },
     proxy: { '/foo': 'http://localhost:4567' },
     cors: false
   })

diff --git a/packages/vite/src/node/http.ts b/packages/vite/src/node/http.ts
@@ -1,6 +1,6 @@
 import fs, { promises as fsp } from 'fs'
 import path from 'path'
-import { Server as HttpServer } from 'http'
+import { OutgoingHttpHeaders as HttpServerHeaders, Server as HttpServer } from 'http'
 import { ServerOptions as HttpsServerOptions } from 'https'
 import { isObject } from './utils'
 import { ProxyOptions } from './server/middlewares/proxy'
@@ -61,7 +61,11 @@ export interface CommonServerOptions {
    * Set to `true` to allow all methods from any origin, or configure separately
    * using an object.
    */
-  cors?: CorsOptions | boolean
+  cors?: CorsOptions | boolean,
+  /**
+   * Specify server response headers.
+   */
+  headers?: HttpServerHeaders
 }
 
 /**

diff --git a/packages/vite/src/node/index.ts b/packages/vite/src/node/index.ts
@@ -66,6 +66,7 @@ export type { ResolveOptions, InternalResolveOptions } from './plugins/resolve'
 export type { WebSocketServer } from './server/ws'
 export type { PluginContainer } from './server/pluginContainer'
 export type { ModuleGraph, ModuleNode, ResolvedUrl } from './server/moduleGraph'
+export type { SendOptions } from './server/send';
 export type { ProxyOptions } from './server/middlewares/proxy'
 export type {
   TransformOptions,

diff --git a/packages/vite/src/node/preview.ts b/packages/vite/src/node/preview.ts
@@ -36,7 +36,8 @@ export function resolvePreviewOptions(
     https: preview?.https ?? server.https,
     open: preview?.open ?? server.open,
     proxy: preview?.proxy ?? server.proxy,
-    cors: preview?.cors ?? server.cors
+    cors: preview?.cors ?? server.cors,
+    headers: preview?.headers ?? server.headers
   }
 }
 

diff --git a/packages/vite/src/node/server/middlewares/indexHtml.ts b/packages/vite/src/node/server/middlewares/indexHtml.ts
@@ -180,7 +180,7 @@ export function indexHtmlMiddleware(
         try {
           let html = fs.readFileSync(filename, 'utf-8')
           html = await server.transformIndexHtml(url, html, req.originalUrl)
-          return send(req, res, html, 'html')
+          return send(req, res, html, 'html', { headers: server.config.server.headers })
         } catch (e) {
           return next(e)
         }

diff --git a/packages/vite/src/node/server/middlewares/transform.ts b/packages/vite/src/node/server/middlewares/transform.ts
@@ -108,7 +108,7 @@ export function transformMiddleware(
         const map = (await moduleGraph.getModuleByUrl(originalUrl))
           ?.transformResult?.map
         if (map) {
-          return send(req, res, JSON.stringify(map), 'json')
+          return send(req, res, JSON.stringify(map), 'json', { headers: server.config.server.headers })
         } else {
           return next()
         }
@@ -180,10 +180,13 @@ export function transformMiddleware(
             res,
             result.code,
             type,
-            result.etag,
-            // allow browser to cache npm deps!
-            isDep ? 'max-age=31536000,immutable' : 'no-cache',
-            result.map
+            {
+              etag: result.etag,
+              // allow browser to cache npm deps!
+              cacheControl: isDep ? 'max-age=31536000,immutable' : 'no-cache',
+              headers: server.config.server.headers,
+              map: result.map
+            }
           )
         }
       }

diff --git a/packages/vite/src/node/server/send.ts b/packages/vite/src/node/server/send.ts
@@ -1,4 +1,4 @@
-import { IncomingMessage, ServerResponse } from 'http'
+import { IncomingMessage, OutgoingHttpHeaders, ServerResponse } from 'http'
 import getEtag from 'etag'
 import { SourceMap } from 'rollup'
 
@@ -11,15 +11,27 @@ const alias: Record<string, string | undefined> = {
   json: 'application/json'
 }
 
+export interface SendOptions {
+  etag?: string,
+  cacheControl?: string,
+  headers?: OutgoingHttpHeaders,
+  map?: SourceMap | null
+}
+
 export function send(
   req: IncomingMessage,
   res: ServerResponse,
   content: string | Buffer,
   type: string,
-  etag = getEtag(content, { weak: true }),
-  cacheControl = 'no-cache',
-  map?: SourceMap | null
+  options: SendOptions
 ): void {
+  const {
+    etag = getEtag(content, { weak: true }),
+    cacheControl = 'no-cache',
+    headers,
+    map
+  } = options
+
   if (res.writableEnded) {
     return
   }
@@ -33,6 +45,12 @@ export function send(
   res.setHeader('Cache-Control', cacheControl)
   res.setHeader('Etag', etag)
 
+  if (headers) {
+    for (const name in headers) {
+      res.setHeader(name, headers[name]!)
+    }
+  }
+
   // inject source map reference
   if (map && map.mappings) {
     if (isDebug) {