WIP web socket server experiments

apps/webapp/package.json

@@ -8,6 +8,7 @@
     "build:db:seed": "esbuild --platform=node --bundle --minify --format=cjs ./prisma/seed.ts --outdir=prisma",
     "build:remix": "remix build",
     "build:server": "esbuild --platform=node --format=cjs ./server.ts --outdir=build",
+    "build:socket-server": "esbuild --platform=node --format=cjs ./socket-server.ts --outdir=build",
     "dev": "cross-env PORT=3030 remix dev",
     "dev:manual": "cross-env PORT=3030 remix dev -c \"node ./build/server.js\"",
     "format": "prettier --write .",
@@ -132,6 +133,7 @@
     "tiny-invariant": "^1.2.0",
     "ulid": "^2.3.0",
     "ulidx": "^2.2.1",
+    "ws": "^8.11.0",
     "zod": "3.22.3",
     "zod-error": "1.5.0",
     "zod-validation-error": "^1.5.0"
@@ -176,6 +178,7 @@
     "@types/simple-oauth2": "^5.0.4",
     "@types/slug": "^5.0.3",
     "@types/tar": "^6.1.4",
+    "@types/ws": "^8.5.3",
     "@typescript-eslint/eslint-plugin": "^5.59.6",
     "@typescript-eslint/parser": "^5.59.6",
     "autoprefixer": "^10.4.13",

apps/webapp/server.ts

@@ -55,7 +55,31 @@ app.all(
 
 const port = process.env.REMIX_APP_PORT || process.env.PORT || 3000;
 
+import { SocketServer } from "./socket-server";
+import { WebSocketServer } from "ws";
+
 if (process.env.HTTP_SERVER_DISABLED !== "true") {
+  const wss = new WebSocketServer({ noServer: true });
+
+  wss.on("connection", (ws, req) => {
+    const apiKey = req.headers.authorization;
+
+    if (!apiKey || typeof apiKey !== "string") {
+      console.log("Invalid API Key, closing the connection");
+
+      ws.close(1008, "Invalid API Key");
+      return;
+    }
+
+    const keyPart = apiKey.split(" ")[1];
+
+    console.log("Initialization the TriggerServer now...");
+
+    const triggerServer = new SocketServer(ws);
+
+    console.log("TriggerServer initialized, sending the API Key...");
+  });
+
   const server = app.listen(port, () => {
     // require the built app so we're ready when the first request comes in
     require(BUILD_DIR);
@@ -73,6 +97,33 @@ if (process.env.HTTP_SERVER_DISABLED !== "true") {
       }
     });
   });
+
+  server.on("upgrade", async (req, socket, head) => {
+    console.log(
+      `Attemping to upgrade connection at url ${req.url} with headers: ${JSON.stringify(
+        req.headers
+      )}`
+    );
+
+    const url = new URL(req.url ?? "", "http://localhost");
+
+    // Only upgrade the connecting if the path is `/ws`
+    if (url.pathname !== "/ws") {
+      socket.destroy(
+        new Error(
+          "Cannot connect because of invalid path: Please include `/ws` in the path of your upgrade request."
+        )
+      );
+      return;
+    }
+
+    console.log(`Client connected, upgrading their connection...`);
+
+    // Handle the WebSocket connection
+    wss.handleUpgrade(req, socket, head, (ws) => {
+      wss.emit("connection", ws, req);
+    });
+  });
 } else {
   require(BUILD_DIR);
   console.log(`✅ app ready (skipping http server)`);

apps/webapp/socket-server.ts

@@ -0,0 +1,5 @@
+import { WebSocket } from "ws";
+
+export class SocketServer {
+  constructor(private socket: WebSocket) {}
+}