feat: make memory files middleware work

Author: underfin

packages/vite/src/node/build.ts

@@ -555,8 +555,8 @@ function resolveConfigToBuild(
   return resolveConfig(
     inlineConfig,
     command,
-    'production',
-    'production',
+    command === 'serve' ? 'development' : 'production',
+    command === 'serve' ? 'development' : 'production',
     false,
     patchConfig,
     patchPlugins,
@@ -655,10 +655,11 @@ async function buildEnvironment(
       '.css': 'js',
     },
     experimental: {
-      hmr: server ? {
-        host: server._currentServerHost!,
-        port: server._currentServerPort!,
-      } : false,
+      hmr: true
+      // hmr: server ? {
+      //   host: server._currentServerHost!,
+      //   port: server._currentServerPort!,
+      // } : false,
     }
   }
 

packages/vite/src/node/server/index.ts

@@ -54,7 +54,6 @@ import {
 import { initPublicFiles } from '../publicDir'
 import { getEnvFilesForMode } from '../env'
 import type { RequiredExceptFor } from '../typeUtils'
-import { cleanUrl } from '../../shared/utils'
 import type { WebSocketServer } from './ws'
 import { createWebSocketServer } from './ws'
 import { baseMiddleware } from './middlewares/base'
@@ -81,6 +80,7 @@ import { openBrowser as _openBrowser } from './openBrowser'
 import { searchForPackageRoot, searchForWorkspaceRoot } from './searchRoot'
 import type { DevEnvironment } from './environment'
 import { hostCheckMiddleware } from './middlewares/hostCheck'
+import { memoryFilesMiddleware } from './middlewares/memoryFiles'
 
 export interface ServerOptions extends CommonServerOptions {
   /**
@@ -913,19 +913,10 @@ export async function _createServer(
 
   // serve static files
   // middlewares.use(serveRawFsMiddleware(server))
-  // middlewares.use(serveStaticMiddleware(server))
+  // middlewares.use(c(server))
 
   // serve memory output dist files
-  middlewares.use((req, res, next) => {
-    const cleanedUrl = cleanUrl(req.url!)
-    const file = server.memoryFiles[cleanedUrl]
-    if (
-      file
-    ) {
-      return res.end(file)
-    }
-    next()
-  });
+  middlewares.use(memoryFilesMiddleware(server, false));
 
   // html fallback
   if (config.appType === 'spa' || config.appType === 'mpa') {
@@ -940,6 +931,7 @@ export async function _createServer(
   if (config.appType === 'spa' || config.appType === 'mpa') {
     // transform index.html
     // middlewares.use(indexHtmlMiddleware(root, server))
+    middlewares.use(memoryFilesMiddleware(server, true));
 
     // handle 404s
     middlewares.use(notFoundMiddleware())

packages/vite/src/node/server/middlewares/indexHtml.ts

@@ -437,6 +437,7 @@ import { cleanUrl,
 //   }
 // }
 
+// TODO(underfin): it only used for preview
 export function indexHtmlMiddleware(
   root: string,
   server: ViteDevServer | PreviewServer,

packages/vite/src/node/server/middlewares/memoryFiles.ts

@@ -0,0 +1,29 @@
+import type { Connect } from 'dep-types/connect'
+import { cleanUrl } from '../../../shared/utils'
+import type { ViteDevServer } from '..'
+
+export function memoryFilesMiddleware(server: ViteDevServer, handleHtml: boolean): Connect.NextHandleFunction {
+  return function viteMemoryFilesMiddleware(req, res, next)  {
+      const cleanedUrl = cleanUrl(req.url!).slice(1) // remove first /
+      if (cleanedUrl.endsWith('.html') && !handleHtml) { 
+        return next()
+      }
+      const file = server.memoryFiles[cleanedUrl]
+      if (
+        file
+      ) {
+        if (cleanedUrl.endsWith('.js')) {
+          res.setHeader('Content-Type', 'text/javascript')
+        }
+        const headers = server.config.server.headers
+        if (headers) {
+          for (const name in headers) {
+            res.setHeader(name, headers[name]!)
+          }
+        }
+        
+        return res.end(file)
+      }
+      next()
+    }
+}