feat(vite): auto-register server consumer environments as services (#3928)

Author: schiller-manuel

examples/vite-rsc/vite.config.ts

@@ -6,17 +6,14 @@ import react from "@vitejs/plugin-react";
 
 export default defineConfig({
   plugins: [
-    nitro({
-      experimental: {
-        vite: {
-          services: {
-            ssr: { entry: "./app/framework/entry.ssr.tsx" },
-            rsc: { entry: "./app/framework/entry.rsc.tsx" },
-          },
-        },
+    nitro(),
+    rsc({
+      serverHandler: false,
+      entries: {
+        ssr: "./app/framework/entry.ssr.tsx",
+        rsc: "./app/framework/entry.rsc.tsx",
       },
     }),
-    rsc({ serverHandler: false }),
     react(),
   ],
 

src/build/vite/plugin.ts

@@ -16,6 +16,7 @@ import {
   getEnvRunner,
   createNitroEnvironment,
   createServiceEnvironments,
+  createServiceEnvironment,
 } from "./env.ts";
 import { configureViteDevServer } from "./dev.ts";
 import { runtimeDir } from "nitro/meta";
@@ -115,6 +116,54 @@ function nitroEnv(ctx: NitroPluginContext): VitePlugin {
         );
         config.build!.emptyOutDir = false;
         config.build!.outDir = useNitro(ctx).options.output.publicDir;
+        return;
+      }
+
+      // Skip if already registered as a service
+      if (name === "nitro" || ctx.services[name]) {
+        return;
+      }
+
+      // Auto-register server consumer environments as services
+      const entry = getEntry(
+        config.build?.rolldownOptions?.input ||
+          config.build?.rollupOptions?.input
+      );
+      if (typeof entry !== "string") {
+        return;
+      }
+
+      // Resolve and register as a service
+      const resolvedEntry =
+        resolveModulePath(entry, {
+          from: [ctx.nitro!.options.rootDir, ...ctx.nitro!.options.scanDirs],
+          extensions: DEFAULT_EXTENSIONS,
+          suffixes: ["", "/index"],
+          try: true,
+        }) || entry;
+
+      ctx.services[name] = { entry: resolvedEntry };
+      debug(
+        `[env]  Auto-detected service "${name}" with entry: ${resolvedEntry}`
+      );
+
+      // Return service environment configuration to merge
+      return createServiceEnvironment(ctx, name, { entry: resolvedEntry });
+    },
+
+    configResolved() {
+      // Setup default SSR renderer after all environments are configured
+      if (
+        !ctx.nitro!.options.renderer?.handler &&
+        !ctx.nitro!.options.renderer?.template &&
+        ctx.services.ssr?.entry
+      ) {
+        ctx.nitro!.options.renderer ??= {};
+        ctx.nitro!.options.renderer.handler = resolve(
+          runtimeDir,
+          "internal/vite/ssr-renderer"
+        );
+        ctx.nitro!.routing.sync();
       }
     },
   };
@@ -369,20 +418,6 @@ async function setupNitroContext(
     ctx.nitro.options.serverEntry = false;
   }
 
-  // Default SSR renderer
-  if (
-    !ctx.nitro.options.renderer?.handler &&
-    !ctx.nitro.options.renderer?.template &&
-    ctx.services.ssr?.entry
-  ) {
-    ctx.nitro.options.renderer ??= {};
-    ctx.nitro.options.renderer.handler = resolve(
-      runtimeDir,
-      "internal/vite/ssr-renderer"
-    );
-    ctx.nitro!.routing.sync();
-  }
-
   // Determine default Vite dist directory
   const publicDistDir = (ctx._publicDistDir =
     userConfig.build?.outDir ||