feat: rolldown vite full bundle mode

.github/workflows/ci.yml

@@ -124,6 +124,9 @@ jobs:
       - name: Test serve
         run: pnpm run test-serve
 
+      - name: Test full bundle mode serve
+        run: pnpm run test-full-bundle-mode
+
       - name: Test build
         run: pnpm run test-build
 

package.json

@@ -23,7 +23,8 @@
     "format": "prettier --write --cache .",
     "lint": "eslint --cache .",
     "typecheck": "tsc -p scripts --noEmit && pnpm -r --parallel run typecheck",
-    "test": "pnpm test-unit && pnpm test-serve && pnpm test-build",
+    "test": "pnpm test-unit && pnpm test-serve && pnpm run test-full-bundle-mode && pnpm test-build",
+    "test-full-bundle-mode": "VITE_TEST_FULL_BUNDLE_MODE=1 vitest run -c vitest.config.e2e.ts",
     "test-serve": "vitest run -c vitest.config.e2e.ts",
     "test-build": "VITE_TEST_BUILD=1 vitest run -c vitest.config.e2e.ts",
     "test-unit": "vitest run",
@@ -98,6 +99,7 @@
   "packageManager": "[email protected]",
   "pnpm": {
     "overrides": {
+      "vitest>vite": "npm:vite@^6.2.6",
       "vite": "workspace:rolldown-vite@*"
     },
     "patchedDependencies": {

packages/vite/index.cjs

@@ -13,6 +13,7 @@ Object.assign(module.exports, require('./dist/node-cjs/publicUtils.cjs'))
 const asyncFunctions = [
   'build',
   'createServer',
+  'createServerWithResolvedConfig',
   'preview',
   'transformWithEsbuild',
   'transformWithOxc',

packages/vite/src/client/client.ts

@@ -7,6 +7,7 @@ import {
 } from '../shared/moduleRunnerTransport'
 import { createHMRHandler } from '../shared/hmrHandler'
 import { ErrorOverlay, overlayId } from './overlay'
+import './hmrModuleRunner'
 import '@vite/env'
 
 // injected by the hmr plugin when served
@@ -20,6 +21,7 @@ declare const __HMR_BASE__: string
 declare const __HMR_TIMEOUT__: number
 declare const __HMR_ENABLE_OVERLAY__: boolean
 declare const __WS_TOKEN__: string
+declare const __FULL_BUNDLE_MODE__: boolean
 
 console.debug('[vite] connecting...')
 
@@ -140,20 +142,36 @@ const hmrClient = new HMRClient(
   },
   transport,
   async function importUpdatedModule({
+    url,
     acceptedPath,
     timestamp,
     explicitImportRequired,
     isWithinCircularImport,
   }) {
-    const [acceptedPathWithoutQuery, query] = acceptedPath.split(`?`)
-    const importPromise = import(
-      /* @vite-ignore */
-      base +
-        acceptedPathWithoutQuery.slice(1) +
-        `?${explicitImportRequired ? 'import&' : ''}t=${timestamp}${
-          query ? `&${query}` : ''
-        }`
-    )
+    function importModuleWithFullBundleMode() {
+      const importPromise = import(base + url)
+      return importPromise.then(() =>
+        // @ts-expect-error globalThis.__rolldown_runtime__
+        globalThis.__rolldown_runtime__.loadExports(acceptedPath),
+      )
+    }
+
+    function importModule() {
+      const [acceptedPathWithoutQuery, query] = acceptedPath.split(`?`)
+      const importPromise = import(
+        /* @vite-ignore */
+        base +
+          acceptedPathWithoutQuery.slice(1) +
+          `?${explicitImportRequired ? 'import&' : ''}t=${timestamp}${
+            query ? `&${query}` : ''
+          }`
+      )
+      return importPromise
+    }
+
+    const importPromise = __FULL_BUNDLE_MODE__
+      ? importModuleWithFullBundleMode()
+      : importModule()
     if (isWithinCircularImport) {
       importPromise.catch(() => {
         console.info(
@@ -165,6 +183,7 @@ const hmrClient = new HMRClient(
     }
     return await importPromise
   },
+  __FULL_BUNDLE_MODE__,
 )
 transport.connect!(createHMRHandler(handleMessage))
 
@@ -436,7 +455,7 @@ export function removeStyle(id: string): void {
 }
 
 export function createHotContext(ownerPath: string): ViteHotContext {
-  return new HMRContext(hmrClient, ownerPath)
+  return new HMRContext(hmrClient, ownerPath, __FULL_BUNDLE_MODE__)
 }
 
 /**

packages/vite/src/client/hmrModuleRunner.ts

@@ -0,0 +1,63 @@
+import { createHotContext } from './client'
+
+declare const __FULL_BUNDLE_MODE__: boolean
+
+if (__FULL_BUNDLE_MODE__) {
+  class DevRuntime {
+    modules: Record<string, { exports: any }> = {}
+
+    static getInstance() {
+      // @ts-expect-error __rolldown_runtime__
+      let instance = globalThis.__rolldown_runtime__
+      if (!instance) {
+        instance = new DevRuntime()
+        // @ts-expect-error __rolldown_runtime__
+        globalThis.__rolldown_runtime__ = instance
+      }
+      return instance
+    }
+
+    createModuleHotContext(moduleId: string) {
+      return createHotContext(moduleId)
+    }
+
+    applyUpdates(_boundaries: string[]) {
+      //
+    }
+
+    registerModule(
+      id: string,
+      module: { exports: Record<string, () => unknown> },
+    ) {
+      this.modules[id] = module
+    }
+
+    loadExports(id: string) {
+      const module = this.modules[id]
+      if (module) {
+        return module.exports
+      } else {
+        console.warn(`Module ${id} not found`)
+        return {}
+      }
+    }
+
+    // __esmMin
+    // @ts-expect-error need to add typing
+    createEsmInitializer = (fn, res) => () => (fn && (res = fn((fn = 0))), res)
+    // __commonJSMin
+    // @ts-expect-error need to add typing
+    createCjsInitializer = (cb, mod) => () => (
+      mod || cb((mod = { exports: {} }).exports, mod), mod.exports
+    )
+    // @ts-expect-error it is exits
+    __toESM = __toESM
+    // @ts-expect-error it is exits
+    __toCommonJS = __toCommonJS
+    // @ts-expect-error it is exits
+    __export = __export
+  }
+
+  // @ts-expect-error __rolldown_runtime__
+  globalThis.__rolldown_runtime__ ||= new DevRuntime()
+}

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

@@ -77,6 +77,7 @@ export class ModuleRunner {
         resolvedHmrLogger,
         this.transport,
         ({ acceptedPath }) => this.import(acceptedPath),
+        false,
       )
       if (!this.transport.connect) {
         throw new Error(
@@ -391,7 +392,7 @@ export class ModuleRunner {
             throw new Error(`[module runner] HMR client was closed.`)
           }
           this.debug?.('[module runner] creating hmr context for', mod.url)
-          hotContext ||= new HMRContext(this.hmrClient, mod.url)
+          hotContext ||= new HMRContext(this.hmrClient, mod.url, false)
           return hotContext
         },
         set: (value) => {

packages/vite/src/node/__tests__/plugins/import.spec.ts

@@ -75,10 +75,7 @@ describe('transformCjsImport', () => {
       ),
     ).toBe(
       'import __vite__cjsImport0_react from "./node_modules/.vite/deps/react.js"; ' +
-        'const react = ((m) => m?.__esModule ? m : {\n' +
-        '\t...typeof m === "object" && !Array.isArray(m) || typeof m === "function" ? m : {},\n' +
-        '\tdefault: m\n' +
-        '})(__vite__cjsImport0_react)',
+        'const react = ((m) => m?.__esModule ? m : { ...typeof m === "object" && !Array.isArray(m) || typeof m === "function" ? m : {}, default: m })(__vite__cjsImport0_react)',
     )
   })