fix: $ vite preview 404 handling

vitejs · Apr 11, 2022 · 209c337 · 209c337
1 parent eb57627
commit 209c337
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/packages/vite/src/node/preview.ts b/packages/vite/src/node/preview.ts
@@ -1,4 +1,5 @@
 import path from 'path'
+import fs from 'fs'
 import sirv from 'sirv'
 import connect from 'connect'
 import compression from './server/middlewares/compression'
@@ -90,11 +91,20 @@ export async function preview(
     config.base,
     sirv(distDir, {
       etag: true,
-      dev: true,
-      single: true
+      dev: true
     })
   )
 
+  app.use(config.base, (_, res, next) => {
+    const file = path.join(distDir, './404.html')
+    if (fs.existsSync(file)) {
+      res.statusCode = 404
+      res.end(fs.readFileSync(file))
+    } else {
+      next()
+    }
+  })
+
   const options = config.preview
   const hostname = resolveHostname(options.host)
   const port = options.port ?? 4173