fix: include headers during WebSocket upgrade (bis)

Related: b1ee41f

Author: darrachequesne

lib/server.ts

@@ -326,6 +326,10 @@ export class Server extends EventEmitter<
   ): Promise<Response> {
     const id = generateId();
 
+    if (this.opts.editHandshakeHeaders) {
+      await this.opts.editHandshakeHeaders(responseHeaders, req, server);
+    }
+
     let isUpgrade = req.headers.has("upgrade");
     let transport: Transport;
     if (isUpgrade) {
@@ -360,10 +364,6 @@ export class Server extends EventEmitter<
       return new Response(null);
     }
 
-    if (this.opts.editHandshakeHeaders) {
-      await this.opts.editHandshakeHeaders(responseHeaders, req, server);
-    }
-
     const promise = (transport as Polling).onRequest(req, responseHeaders);
 
     this.emitReserved("connection", socket, req, server);