NeoFlow Hub
Self-Hosted Workspace Platform

1import { WebSocketServer, WebSocket } from "ws";
2import { verifyJwt } from "@hub/auth";
3
4const wss = new WebSocketServer({ port: 3100 });
5const rooms = new Map<string, Set<WebSocket>>(); // channelId → connected clients
6
7wss.on("connection", async (socket, req) => {
8  const token = new URL(req.url!, "ws://hub").searchParams.get("token");
9  const user = await verifyJwt(token ?? "");
10  if (!user) return socket.close(4001, "Unauthorized");
11
12  socket.on("message", async (raw) => {
13    const msg = JSON.parse(raw.toString());
14
15    if (msg.type === "JOIN_CHANNEL") {
16      const room = rooms.get(msg.channelId) ?? new Set();
17      room.add(socket);
18      rooms.set(msg.channelId, room);
19      broadcastPresence(msg.channelId, user.id, "ONLINE");
20    }
21
22    if (msg.type === "SEND_MESSAGE") {
23      const saved = await db.messages.create({ data: { ...msg, userId: user.id } });
24      rooms.get(msg.channelId)?.forEach(client =>
25        client.send(JSON.stringify({ type: "NEW_MESSAGE", data: saved }))
26      );
27    }
28  });
29
30  socket.on("close", () => {
31    rooms.forEach(room => room.delete(socket));
32    broadcastPresence("*", user.id, "OFFLINE");
33  });
34});