refactor(rsbuild-ssr): use server.setup

Author: chenjiahan

rsbuild/ssr/rsbuild.config.ts

@@ -1,21 +1,16 @@
-import {
-  defineConfig,
-  logger,
-  type RequestHandler,
-  type SetupMiddlewaresContext,
-} from '@rsbuild/core';
+import { defineConfig, logger, type RequestHandler, type RsbuildDevServer } from '@rsbuild/core';
 import { pluginReact } from '@rsbuild/plugin-react';
 
 export const serverRender =
-  (serverContext: SetupMiddlewaresContext): RequestHandler =>
+  (server: RsbuildDevServer): RequestHandler =>
   async (_req, res, _next) => {
-    const indexModule = await serverContext.environments.node.loadBundle<{
+    const indexModule = await server.environments.node.loadBundle<{
       render: () => string;
     }>('index');
 
     const markup = indexModule.render();
 
-    const template = await serverContext.environments.web.getTransformedHtml('index');
+    const template = await server.environments.web.getTransformedHtml('index');
 
     const html = template.replace('<!--app-content-->', markup);
 
@@ -27,12 +22,16 @@ export const serverRender =
 
 export default defineConfig({
   plugins: [pluginReact()],
-  dev: {
-    setupMiddlewares: [
-      ({ unshift }, serverContext) => {
-        const serverRenderMiddleware = serverRender(serverContext);
+  server: {
+    setup: [
+      ({ action, server }) => {
+        if (action !== 'dev') {
+          return;
+        }
 
-        unshift(async (req, res, next) => {
+        const serverRenderMiddleware = serverRender(server);
+
+        server.middlewares.use(async (req, res, next) => {
           if (req.method === 'GET' && req.url === '/') {
             try {
               await serverRenderMiddleware(req, res, next);