From baf14e3178f0dc11ff646cf35757c091edf97fe0 Mon Sep 17 00:00:00 2001
From: jaywcjlove <398188662@qq.com>
Date: Tue, 20 Sep 2022 13:45:56 +0800
Subject: [PATCH] fix: Fix devServer setupMiddlewares issue.

---
 core/src/scripts/start.ts             | 4 ++++
 example/react-component-tsx/.kktrc.ts | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/core/src/scripts/start.ts b/core/src/scripts/start.ts
index 492256b5..6a5fb8a8 100644
--- a/core/src/scripts/start.ts
+++ b/core/src/scripts/start.ts
@@ -99,6 +99,7 @@ export default async function start(argvs: StartArgs) {
       }
       delete serverConf.onAfterSetupMiddleware;
       delete serverConf.onBeforeSetupMiddleware;
+      const setupMiddlewares = overrideDevServerConfig.setupMiddlewares;
       serverConf.setupMiddlewares = (middlewares, devServer) => {
         // Keep `evalSourceMapMiddleware`
         // middlewares before `redirectServedPath` otherwise will not have any effect
@@ -122,6 +123,9 @@ export default async function start(argvs: StartArgs) {
         // it used the same host and port.
         // https://github.com/facebook/create-react-app/issues/2272#issuecomment-302832432
         devServer.app.use(noopServiceWorkerMiddleware(paths.publicUrlOrPath));
+        if (setupMiddlewares) {
+          return setupMiddlewares(middlewares, devServer);
+        }
         return middlewares;
       };
       return serverConf;
diff --git a/example/react-component-tsx/.kktrc.ts b/example/react-component-tsx/.kktrc.ts
index 952d20fc..2701fe10 100644
--- a/example/react-component-tsx/.kktrc.ts
+++ b/example/react-component-tsx/.kktrc.ts
@@ -31,5 +31,9 @@ export default (conf: Configuration, env: 'development' | 'production', options:
  */
 export const devServer = (config: WebpackDevServer.Configuration) => {
   // Return your customised Webpack Development Server config.
+  // setupMiddlewares Example
+  config.setupMiddlewares = (middlewares, server) => {
+    return middlewares;
+  };
   return config;
 };