feat: support full bundle mode in ssr

packages/vite/rolldown.config.ts

@@ -147,7 +147,7 @@ const moduleRunnerConfig = defineConfig({
     '@vitejs/devtools/cli-commands',
     ...Object.keys(pkg.dependencies),
   ],
-  plugins: [bundleSizeLimit(54), enableSourceMapsInWatchModePlugin()],
+  plugins: [bundleSizeLimit(55), enableSourceMapsInWatchModePlugin()],
   output: {
     ...sharedNodeOptions.output,
     minify: {

packages/vite/src/module-runner/constants.ts

@@ -5,3 +5,8 @@ export const ssrDynamicImportKey = `__vite_ssr_dynamic_import__`
 export const ssrExportAllKey = `__vite_ssr_exportAll__`
 export const ssrExportNameKey = `__vite_ssr_exportName__`
 export const ssrImportMetaKey = `__vite_ssr_import_meta__`
+
+export const ssrRolldownRuntimeKey = `__rolldown_runtime__`
+export const ssrRolldownRuntimeDefineMethod = `__vite_ssr_defineRuntime__`
+export const ssrRolldownRuntimeCreateHotContextMethod = `__vite_ssr_createHotContext__`
+export const ssrRolldownRuntimeTransport = `__vite_ssr_transport__`

packages/vite/src/module-runner/createImportMeta.ts

@@ -14,12 +14,13 @@ const envProxy = new Proxy({} as any, {
 export function createDefaultImportMeta(
   modulePath: string,
 ): ModuleRunnerImportMeta {
-  const href = posixPathToFileHref(modulePath)
-  const filename = modulePath
-  const dirname = posixDirname(modulePath)
+  const isVirtual = modulePath.startsWith('data:application/javascript,')
+  const href = isVirtual ? modulePath : posixPathToFileHref(modulePath)
+  const filename = isVirtual ? undefined : modulePath
+  const dirname = isVirtual ? undefined : posixDirname(modulePath)
   return {
-    filename: isWindows ? toWindowsPath(filename) : filename,
-    dirname: isWindows ? toWindowsPath(dirname) : dirname,
+    filename: isWindows && filename ? toWindowsPath(filename) : filename,
+    dirname: isWindows && dirname ? toWindowsPath(dirname) : dirname,
     url: href,
     env: envProxy,
     resolve(_id: string, _parent?: string) {

packages/vite/src/module-runner/esmEvaluator.ts

@@ -9,6 +9,7 @@ import {
   ssrImportKey,
   ssrImportMetaKey,
   ssrModuleExportsKey,
+  ssrRolldownRuntimeKey,
 } from './constants'
 import type { ModuleEvaluator, ModuleRunnerContext } from './types'
 
@@ -28,6 +29,7 @@ export class ESModulesEvaluator implements ModuleEvaluator {
       ssrDynamicImportKey,
       ssrExportAllKey,
       ssrExportNameKey,
+      ssrRolldownRuntimeKey,
       // source map should already be inlined by Vite
       '"use strict";' + code,
     )
@@ -39,6 +41,7 @@ export class ESModulesEvaluator implements ModuleEvaluator {
       context[ssrDynamicImportKey],
       context[ssrExportAllKey],
       context[ssrExportNameKey],
+      context[ssrRolldownRuntimeKey],
     )
 
     Object.seal(context[ssrModuleExportsKey])

packages/vite/src/module-runner/index.ts

@@ -37,5 +37,9 @@ export {
   ssrImportKey,
   ssrImportMetaKey,
   ssrModuleExportsKey,
+  ssrRolldownRuntimeKey,
+  ssrRolldownRuntimeDefineMethod,
+  ssrRolldownRuntimeCreateHotContextMethod,
+  ssrRolldownRuntimeTransport,
 } from './constants'
 export type { InterceptorOptions } from './sourcemap/interceptor'

packages/vite/src/module-runner/runner.ts

@@ -1,4 +1,4 @@
-import type { ViteHotContext } from '#types/hot'
+import type { DevRuntime } from 'rolldown/experimental/runtime-types'
 import { HMRClient, HMRContext, type HMRLogger } from '../shared/hmr'
 import { cleanUrl, isPrimitive } from '../shared/utils'
 import { analyzeImportedModDifference } from '../shared/ssrTransform'
@@ -16,14 +16,18 @@ import type {
   ResolvedResult,
   SSRImportMetadata,
 } from './types'
-import { posixDirname, posixPathToFileHref, posixResolve } from './utils'
+import { posixDirname, posixJoin } from './utils'
 import {
   ssrDynamicImportKey,
   ssrExportAllKey,
   ssrExportNameKey,
   ssrImportKey,
   ssrImportMetaKey,
   ssrModuleExportsKey,
+  ssrRolldownRuntimeCreateHotContextMethod,
+  ssrRolldownRuntimeDefineMethod,
+  ssrRolldownRuntimeKey,
+  ssrRolldownRuntimeTransport,
 } from './constants'
 import { hmrLogger, silentConsole } from './hmrLogger'
 import { createHMRHandlerForRunner } from './hmrHandler'
@@ -47,6 +51,42 @@ export class ModuleRunner {
   >()
   private isBuiltin?: (id: string) => boolean
   private builtinsPromise?: Promise<void>
+  private rolldownDevRuntime?: DevRuntime
+
+  // We need the proxy because the runtime MUST be ready before the first import is processed.
+  // Because `context['__rolldown_runtime__']` is passed down before the modules are executed as a function argument.
+  private rolldownDevRuntimeProxy = new Proxy(
+    {},
+    {
+      get: (_, p, receiver) => {
+        // Special `__rolldown_runtime__.__vite_ssr_defineRuntime__` method only for the module runner,
+        // It's not available in the browser because it's a global there. We cannot have it as a global because
+        //   - It's possible to have multiple runners (`ssrLoadModule` has its own compat runner);
+        //   - We don't want to pollute Dev Server's global namespace.
+        if (p === ssrRolldownRuntimeDefineMethod) {
+          return (runtime: DevRuntime) => {
+            this.rolldownDevRuntime = runtime
+          }
+        }
+
+        if (p === ssrRolldownRuntimeCreateHotContextMethod) {
+          return this.closed
+            ? () => {}
+            : (url: string) => this.ensureModuleHotContext(url)
+        }
+
+        if (p === ssrRolldownRuntimeTransport) {
+          return this.closed ? undefined : this.transport
+        }
+
+        if (!this.rolldownDevRuntime) {
+          throw new Error(`__rolldown_runtime__ was not initialized.`)
+        }
+
+        return Reflect.get(this.rolldownDevRuntime, p, receiver)
+      },
+    },
+  ) as DevRuntime
 
   private closed = false
 
@@ -89,7 +129,7 @@ export class ModuleRunner {
    */
   public async import<T = any>(url: string): Promise<T> {
     const fetchedModule = await this.cachedModule(url)
-    return await this.cachedRequest(url, fetchedModule)
+    return await this.cachedRequest(fetchedModule.url, fetchedModule)
   }
 
   /**
@@ -353,7 +393,7 @@ export class ModuleRunner {
       // it's possible to provide an object with toString() method inside import()
       dep = String(dep)
       if (dep[0] === '.') {
-        dep = posixResolve(posixDirname(url), dep)
+        dep = posixJoin(posixDirname(url), dep)
       }
       return request(dep, { isDynamicImport: true })
     }
@@ -380,9 +420,10 @@ export class ModuleRunner {
     const createImportMeta =
       this.options.createImportMeta ?? createDefaultImportMeta
 
-    const modulePath = cleanUrl(file || moduleId)
+    const modulePath = file
+      ? cleanUrl(file)
+      : `data:application/javascript,${code};`
     // disambiguate the `<UNIT>:/` on windows: see nodejs/node#31710
-    const href = posixPathToFileHref(modulePath)
     const meta = await createImportMeta(modulePath)
     const exports = Object.create(null)
     Object.defineProperty(exports, Symbol.toStringTag, {
@@ -393,20 +434,19 @@ export class ModuleRunner {
 
     mod.exports = exports
 
-    let hotContext: ViteHotContext | undefined
     if (this.hmrClient) {
       Object.defineProperty(meta, 'hot', {
         enumerable: true,
         get: () => {
           if (!this.hmrClient) {
-            throw new Error(`[module runner] HMR client was closed.`)
+            return
           }
           this.debug?.('[module runner] creating hmr context for', mod.url)
-          hotContext ||= new HMRContext(this.hmrClient, mod.url)
-          return hotContext
+          this.ensureModuleHotContext(mod.url)
+          return this.moduleHotContexts.get(mod.url)
         },
         set: (value) => {
-          hotContext = value
+          this.moduleHotContexts.set(mod.url, value)
         },
       })
     }
@@ -423,14 +463,29 @@ export class ModuleRunner {
           get: getter,
         }),
       [ssrImportMetaKey]: meta,
+      [ssrRolldownRuntimeKey]: this.rolldownDevRuntimeProxy,
     }
 
-    this.debug?.('[module runner] executing', href)
+    this.debug?.('[module runner] executing', meta.href)
 
     await this.evaluator.runInlinedModule(context, code, mod)
 
     return exports
   }
+
+  private moduleHotContexts = new Map<string, HMRContext>()
+
+  private ensureModuleHotContext(url: string) {
+    if (!this.hmrClient) {
+      return
+    }
+
+    if (!this.moduleHotContexts.has(url)) {
+      const hotContext = new HMRContext(this.hmrClient, url)
+      this.moduleHotContexts.set(url, hotContext)
+    }
+    return this.moduleHotContexts.get(url)
+  }
 }
 
 function exportAll(exports: any, sourceModule: any) {

packages/vite/src/module-runner/types.ts

@@ -1,3 +1,4 @@
+import type { DevRuntime } from 'rolldown/experimental/runtime-types'
 import type { ViteHotContext } from '#types/hot'
 import type { HMRLogger } from '../shared/hmr'
 import type {
@@ -19,6 +20,7 @@ import type {
   ssrImportKey,
   ssrImportMetaKey,
   ssrModuleExportsKey,
+  ssrRolldownRuntimeKey,
 } from './constants'
 import type { InterceptorOptions } from './sourcemap/interceptor'
 
@@ -41,6 +43,10 @@ export interface ModuleRunnerContext {
   [ssrExportAllKey]: (obj: any) => void
   [ssrExportNameKey]: (name: string, getter: () => unknown) => void
   [ssrImportMetaKey]: ModuleRunnerImportMeta
+  /**
+   * @internal
+   */
+  [ssrRolldownRuntimeKey]: DevRuntime | undefined
 }
 
 export interface ModuleEvaluator {

packages/vite/src/module-runner/utils.ts

@@ -34,6 +34,7 @@ function encodePathChars(filepath: string) {
 
 export const posixDirname: (path: string) => string = pathe.dirname
 export const posixResolve: (...paths: string[]) => string = pathe.resolve
+export const posixJoin: (...paths: string[]) => string = pathe.join
 
 export function posixPathToFileHref(posixPath: string): string {
   let resolved = posixResolve(posixPath)

packages/vite/src/node/config.ts

@@ -131,6 +131,7 @@ import {
 } from './server/pluginContainer'
 import { nodeResolveWithVite } from './nodeResolve'
 import { FullBundleDevEnvironment } from './server/environments/fullBundleEnvironment'
+import { createFullBundleRunnableDevEnvironment } from './server/environments/fullBundleRunnableEnvironment'
 
 const debug = createDebugger('vite:config', { depth: 10 })
 const promisifiedRealpath = promisify(fs.realpath)
@@ -257,6 +258,13 @@ function defaultCreateClientDevEnvironment(
   })
 }
 
+function defaultCreateSSRDevEnvironment(name: string, config: ResolvedConfig) {
+  if (config.experimental.ssrBundledDev) {
+    return createFullBundleRunnableDevEnvironment(name, config)
+  }
+  return createRunnableDevEnvironment(name, config)
+}
+
 function defaultCreateDevEnvironment(name: string, config: ResolvedConfig) {
   return createRunnableDevEnvironment(name, config)
 }
@@ -585,6 +593,14 @@ export interface ExperimentalOptions {
    * @default false
    */
   bundledDev?: boolean
+  /**
+   * Enable full bundle mode in SSR.
+   *
+   * This is highly experimental.
+   *
+   * @experimental
+   */
+  ssrBundledDev?: boolean
 }
 
 export interface LegacyOptions {
@@ -835,6 +851,7 @@ const configDefaults = Object.freeze({
     hmrPartialAccept: false,
     enableNativePlugin: process.env._VITE_TEST_JS_PLUGIN ? false : 'v2',
     bundledDev: false,
+    ssrBundledDev: false,
   },
   future: {
     removePluginHookHandleHotUpdate: undefined,
@@ -897,7 +914,9 @@ export function resolveDevEnvironmentOptions(
       createEnvironment:
         environmentName === 'client'
           ? defaultCreateClientDevEnvironment
-          : defaultCreateDevEnvironment,
+          : environmentName === 'ssr'
+            ? defaultCreateSSRDevEnvironment
+            : defaultCreateDevEnvironment,
       recoverable: consumer === 'client',
       moduleRunnerTransform: consumer === 'server',
     },
@@ -1888,7 +1907,10 @@ export async function resolveConfig(
     cacheDir,
     command,
     mode,
-    isBundled: config.experimental?.bundledDev || isBuild,
+    isBundled:
+      config.experimental?.bundledDev ||
+      config.experimental?.ssrBundledDev ||
+      isBuild, // TODO: ssr shouldn't break client
     isWorker: false,
     mainConfig: null,
     bundleChain: [],

packages/vite/src/node/index.ts

@@ -49,6 +49,11 @@ export {
   type RunnableDevEnvironment,
   type RunnableDevEnvironmentContext,
 } from './server/environments/runnableEnvironment'
+export {
+  createFullBundleRunnableDevEnvironment,
+  isFullBundleRunnableDevEnvironment,
+  type FullBundleRunnableDevEnvironment,
+} from './server/environments/fullBundleRunnableEnvironment'
 export {
   createFetchableDevEnvironment,
   isFetchableDevEnvironment,