From 1bdf5b4c07432ba0419d993495d932c1eb714ce1 Mon Sep 17 00:00:00 2001
From: Ben Allfree <ben@benallfree.com>
Date: Fri, 30 Dec 2022 22:40:02 -0800
Subject: [PATCH] Websocket Svelte proxy support

---
 packages/proxy/src/index.ts | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

diff --git a/packages/proxy/src/index.ts b/packages/proxy/src/index.ts
index 4e0962fb..a44ca8a6 100644
--- a/packages/proxy/src/index.ts
+++ b/packages/proxy/src/index.ts
@@ -43,6 +43,25 @@ const server = createServer(options, async (req, res) => {
   }
 })
 
+//
+// Listen to the `upgrade` event and proxy the
+// WebSocket requests as well.
+//
+server.on('upgrade', function (req, socket, head) {
+  console.log(req.headers.host)
+  const { host } = req.headers
+
+  try {
+    if (host === process.env.PUBLIC_APP_DOMAIN) {
+      proxy.ws(req, socket, head, { target: `http://localhost:5173` })
+    } else {
+      proxy.ws(req, socket, head, { target: `http://localhost:3000` })
+    }
+  } catch (e) {
+    console.error(`Got an error ${e}`)
+  }
+})
+
 server.listen(443)
 
 const httpServer = http.createServer((req, res) => {