From fa3bc34e23e757d57ca42e66c41dc8e712ae5547 Mon Sep 17 00:00:00 2001
From: Wei <vonweb@users.noreply.github.com>
Date: Thu, 7 Jan 2021 19:34:31 +0800
Subject: [PATCH] feat(proxy): add rewrite support for ws (#1407)

Co-authored-by: wangwei <wangwei@qdcares.cn>
---
 packages/vite/src/node/server/middlewares/proxy.ts | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/packages/vite/src/node/server/middlewares/proxy.ts b/packages/vite/src/node/server/middlewares/proxy.ts
index e5689961a73bb6..17d4b89c77a33c 100644
--- a/packages/vite/src/node/server/middlewares/proxy.ts
+++ b/packages/vite/src/node/server/middlewares/proxy.ts
@@ -58,6 +58,9 @@ export function proxyMiddleware({
           (opts.ws || opts.target?.toString().startsWith('ws:')) &&
           req.headers['sec-websocket-protocol'] !== HMR_HEADER
         ) {
+          if (opts.rewrite) {
+            req.url = opts.rewrite(url)
+          }
           proxy.ws(req, socket, head)
         }
       }