add websocket support

2022-02-18 20:33:01 -07:00
parent 2437f82e94
commit ebabaf5362
3 changed files with 19 additions and 7 deletions
--- a/index.js
+++ b/index.js
@@ -363,10 +363,17 @@ class Router {
            if (res.raw) {
                return res.raw
            }
-            return new Response(res.body, {
+
+            const resInit = {
                status: res.status || (res.body ? 200 : 204),
                headers: res.headers
-            })
+            }
+
+            if (res.webSocket) {
+                resInit.webSocket = res.webSocket
+            }
+
+            return new Response(res.body, resInit)
        } catch(err) {
            console.error(err)
            return new Response(this.debugMode ? err.stack : '', { status: 500 })