feat(dev): expose upgrade handler

unjs · Feb 25, 2024 · 5374429 · 5374429
1 parent 25a79c9
commit 5374429
Showing 1 changed file with 16 additions and 9 deletions.
diff --git a/src/dev/server.ts b/src/dev/server.ts
@@ -38,6 +38,7 @@ export interface NitroDevServer {
   app: App;
   close: () => Promise<void>;
   watcher?: FSWatcher;
+  upgrade: (req, socket, head) => void;
 }
 
 function initWorker(filename: string): Promise<NitroWorker> | null {
@@ -242,20 +243,25 @@ export function createDevServer(nitro: Nitro): NitroDevServer {
     })
   );
 
+  // Upgrade handler
+  const upgrade = (req, socket, head) => {
+    return proxy.proxy.ws(
+      req,
+      socket,
+      {
+        target: getWorkerAddress(),
+        xfwd: true,
+      },
+      head
+    );
+  };
+
   // Listen
   let listeners: Listener[] = [];
   const _listen: NitroDevServer["listen"] = async (port, opts?) => {
     const listener = await listen(toNodeListener(app), { port, ...opts });
     listener.server.on("upgrade", (req, sock, head) => {
-      proxy.proxy.ws(
-        req,
-        sock as any,
-        {
-          target: getWorkerAddress(),
-          xfwd: true,
-        },
-        head
-      );
+      upgrade(req, sock, head);
     });
     listeners.push(listener);
     return listener;
@@ -285,6 +291,7 @@ export function createDevServer(nitro: Nitro): NitroDevServer {
     app,
     close,
     watcher,
+    upgrade,
   };
 }