feat: full bundle dev env

packages/vite/src/client/client.ts

@@ -1,3 +1,4 @@
+/// <reference types="rolldown/experimental/runtime-types" />
 import type { ErrorPayload, HotPayload } from 'types/hmrPayload'
 import type { ViteHotContext } from 'types/hot'
 import { HMRClient, HMRContext } from '../shared/hmr'
@@ -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...')
 
@@ -37,6 +39,7 @@ const directSocketHost = __HMR_DIRECT_TARGET__
 const base = __BASE__ || '/'
 const hmrTimeout = __HMR_TIMEOUT__
 const wsToken = __WS_TOKEN__
+const isFullBundleMode = __FULL_BUNDLE_MODE__
 
 const transport = normalizeModuleRunnerTransport(
   (() => {
@@ -140,32 +143,53 @@ const hmrClient = new HMRClient(
     debug: (...msg) => console.debug('[vite]', ...msg),
   },
   transport,
-  async function importUpdatedModule({
-    acceptedPath,
-    timestamp,
-    explicitImportRequired,
-    isWithinCircularImport,
-  }) {
-    const [acceptedPathWithoutQuery, query] = acceptedPath.split(`?`)
-    const importPromise = import(
-      /* @vite-ignore */
-      base +
-        acceptedPathWithoutQuery.slice(1) +
-        `?${explicitImportRequired ? 'import&' : ''}t=${timestamp}${
-          query ? `&${query}` : ''
-        }`
-    )
-    if (isWithinCircularImport) {
-      importPromise.catch(() => {
-        console.info(
-          `[hmr] ${acceptedPath} failed to apply HMR as it's within a circular import. Reloading page to reset the execution order. ` +
-            `To debug and break the circular import, you can run \`vite --debug hmr\` to log the circular dependency path if a file change triggered it.`,
+  isFullBundleMode
+    ? async function importUpdatedModule({
+        url,
+        acceptedPath,
+        isWithinCircularImport,
+      }) {
+        const importPromise = import(base + url!).then(() =>
+          // @ts-expect-error globalThis.__rolldown_runtime__
+          globalThis.__rolldown_runtime__.loadExports(acceptedPath),
         )
-        pageReload()
-      })
-    }
-    return await importPromise
-  },
+        if (isWithinCircularImport) {
+          importPromise.catch(() => {
+            console.info(
+              `[hmr] ${acceptedPath} failed to apply HMR as it's within a circular import. Reloading page to reset the execution order. ` +
+                `To debug and break the circular import, you can run \`vite --debug hmr\` to log the circular dependency path if a file change triggered it.`,
+            )
+            pageReload()
+          })
+        }
+        return await importPromise
+      }
+    : async function importUpdatedModule({
+        acceptedPath,
+        timestamp,
+        explicitImportRequired,
+        isWithinCircularImport,
+      }) {
+        const [acceptedPathWithoutQuery, query] = acceptedPath.split(`?`)
+        const importPromise = import(
+          /* @vite-ignore */
+          base +
+            acceptedPathWithoutQuery.slice(1) +
+            `?${explicitImportRequired ? 'import&' : ''}t=${timestamp}${
+              query ? `&${query}` : ''
+            }`
+        )
+        if (isWithinCircularImport) {
+          importPromise.catch(() => {
+            console.info(
+              `[hmr] ${acceptedPath} failed to apply HMR as it's within a circular import. Reloading page to reset the execution order. ` +
+                `To debug and break the circular import, you can run \`vite --debug hmr\` to log the circular dependency path if a file change triggered it.`,
+            )
+            pageReload()
+          })
+        }
+        return await importPromise
+      },
 )
 transport.connect!(createHMRHandler(handleMessage))
 
@@ -570,3 +594,42 @@ export function injectQuery(url: string, queryToInject: string): string {
 }
 
 export { ErrorOverlay }
+
+if (isFullBundleMode && typeof DevRuntime !== 'undefined') {
+  class ViteDevRuntime extends DevRuntime {
+    override createModuleHotContext(moduleId: string) {
+      const ctx = createHotContext(moduleId)
+      // @ts-expect-error TODO: support CSS
+      ctx._internal = {
+        updateStyle,
+        removeStyle,
+      }
+      // @ts-expect-error TODO: support this function (used by plugin-react)
+      ctx.getExports = async () =>
+        // @ts-expect-error __rolldown_runtime__ / ctx.ownerPath
+        __rolldown_runtime__.loadExports(ctx.ownerPath)
+      return ctx
+    }
+
+    override applyUpdates(_boundaries: string[]): void {
+      // TODO: how should this be handled?
+      // noop, handled in the HMR client
+    }
+  }
+
+  // TODO: make this more performant
+  const wrappedSocket = {
+    readyState: WebSocket.OPEN,
+    send(data: string) {
+      const d = JSON.parse(data)
+      transport.send({
+        type: 'custom',
+        event: 'vite:module-loaded',
+        data: { modules: d.modules },
+      })
+    },
+  }
+  ;(globalThis as any).__rolldown_runtime__ ??= new ViteDevRuntime(
+    wrappedSocket,
+  )
+}

packages/vite/src/node/build.ts

@@ -407,6 +407,7 @@ export function resolveBuildEnvironmentOptions(
   raw: BuildEnvironmentOptions,
   logger: Logger,
   consumer: 'client' | 'server' | undefined,
+  isFullBundledDev: boolean,
 ): ResolvedBuildEnvironmentOptions {
   const deprecatedPolyfillModulePreload = raw.polyfillModulePreload
   const { polyfillModulePreload, ...rest } = raw
@@ -427,7 +428,7 @@ export function resolveBuildEnvironmentOptions(
     {
       ...buildEnvironmentOptionsDefaults,
       cssCodeSplit: !raw.lib,
-      minify: consumer === 'server' ? false : 'oxc',
+      minify: consumer === 'server' || isFullBundledDev ? false : 'oxc',
       rollupOptions: {},
       rolldownOptions: undefined,
       ssr: consumer === 'server',
@@ -488,10 +489,11 @@ export async function resolveBuildPlugins(config: ResolvedConfig): Promise<{
   pre: Plugin[]
   post: Plugin[]
 }> {
+  const isBuild = config.command === 'build'
   return {
     pre: [
       completeSystemWrapPlugin(),
-      ...(!config.isWorker ? [prepareOutDirPlugin()] : []),
+      ...(isBuild && !config.isWorker ? [prepareOutDirPlugin()] : []),
       perEnvironmentPlugin(
         'vite:rollup-options-plugins',
         async (environment) =>
@@ -504,11 +506,11 @@ export async function resolveBuildPlugins(config: ResolvedConfig): Promise<{
       ...(config.isWorker ? [webWorkerPostPlugin(config)] : []),
     ],
     post: [
-      ...buildImportAnalysisPlugin(config),
+      ...(isBuild ? buildImportAnalysisPlugin(config) : []),
       ...(config.nativePluginEnabledLevel >= 1 ? [] : [buildOxcPlugin()]),
       ...(config.build.minify === 'esbuild' ? [buildEsbuildPlugin()] : []),
-      terserPlugin(config),
-      ...(!config.isWorker
+      ...(isBuild ? [terserPlugin(config)] : []),
+      ...(isBuild && !config.isWorker
         ? [
             manifestPlugin(config),
             ssrManifestPlugin(),
@@ -549,10 +551,10 @@ function resolveConfigToBuild(
   )
 }
 
-function resolveRolldownOptions(
+export function resolveRolldownOptions(
   environment: Environment,
   chunkMetadataMap: ChunkMetadataMap,
-) {
+): RolldownOptions {
   const { root, packageCache, build: options } = environment.config
   const libOptions = options.lib
   const { logger } = environment
@@ -876,7 +878,7 @@ async function buildEnvironment(
   }
 }
 
-function enhanceRollupError(e: RollupError) {
+export function enhanceRollupError(e: RollupError): void {
   const stackOnly = extractStack(e)
 
   let msg = colors.red((e.plugin ? `[${e.plugin}] ` : '') + e.message)
@@ -1042,7 +1044,7 @@ const dynamicImportWarningIgnoreList = [
   `statically analyzed`,
 ]
 
-function clearLine() {
+export function clearLine(): void {
   const tty = process.stdout.isTTY && !process.env.CI
   if (tty) {
     process.stdout.clearLine(0)