From f3eaa7b0d4ad7c71f1ef56b5d15083449ef52a86 Mon Sep 17 00:00:00 2001
From: Nitin Kumar <snitin315@gmail.com>
Date: Mon, 21 Feb 2022 09:01:04 +0530
Subject: [PATCH] fix: compatibility with `webpack-dev-server`

---
 src/index.js | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/index.js b/src/index.js
index 371dfdb..806c5ea 100644
--- a/src/index.js
+++ b/src/index.js
@@ -22,7 +22,7 @@ class ErrorOverlayPlugin {
         compiler.options.devServer.host
       sockOptions.sockPath =
         compiler.options.devServer.client?.webSocketURL?.pathname ||
-        compiler.options.devServer.webSocketServer?.options.path ||
+        (compiler.options.devServer.webSocketServer === "object" && compiler.options.devServer.webSocketServer.options?.path) ||
         '/ws'
       sockOptions.sockPort =
         compiler.options.devServer.client?.webSocketURL?.port ||
@@ -37,11 +37,12 @@ class ErrorOverlayPlugin {
       if (devServerEnabled) {
         const originalOnBeforeSetupMiddleware =
           options.devServer.onBeforeSetupMiddleware
-        options.devServer.onBeforeSetupMiddleware = (devServer) => {
+          options.devServer.setupMiddlewares = (middlewares, devServer) => {
           if (originalOnBeforeSetupMiddleware) {
             originalOnBeforeSetupMiddleware(devServer)
           }
-          devServer.app.use(errorOverlayMiddleware())
+          middlewares.unshift(errorOverlayMiddleware());
+          return middlewares;
         }
       }
     })