switch to nx monorepo

2026-05-20 00:14:10 +08:00 · 2024-01-09 18:58:26 +01:00
parent e1bb2632b8
commit 093e634c0b
273 changed files with 11419 additions and 31 deletions
@@ -0,0 +1,43 @@
+import { WebSocketServer } from 'ws';
+
+export class CollabWsAdapter {
+  private readonly wss: WebSocketServer;
+
+  constructor() {
+    this.wss = new WebSocketServer({ noServer: true });
+  }
+
+  handleUpgrade(path: string, httpServer) {
+    httpServer.on('upgrade', (request, socket, head) => {
+      try {
+        const baseUrl = 'ws://' + request.headers.host + '/';
+        const pathname = new URL(request.url, baseUrl).pathname;
+
+        if (pathname === path) {
+          this.wss.handleUpgrade(request, socket, head, (ws) => {
+            this.wss.emit('connection', ws, request);
+          });
+        } else if (pathname === '/socket.io/') {
+          return;
+        } else {
+          socket.destroy();
+        }
+      } catch (err) {
+        socket.end('HTTP/1.1 400\r\n' + err.message);
+      }
+    });
+
+    return this.wss;
+  }
+
+  public destroy() {
+    try {
+      this.wss.clients.forEach((client) => {
+        client.terminate();
+      });
+      this.wss.close();
+    } catch (err) {
+      console.error(err);
+    }
+  }
+}