@nka212bg/backend-utils 0.1.40 → 0.1.41

package/package.json

@@ -1,7 +1,7 @@
 {
     "name": "@nka212bg/backend-utils",
     "author": "nka212bg",
-    "version": "0.1.40",
+    "version": "0.1.41",
     "main": "index.js",
     "dependencies": {
         "@maxmind/geoip2-node": "^5.0.0",

package/socket.js

@@ -1,13 +1,51 @@
 const WebSocket = require("ws");
 
-module.exports = (server) => {
-    const socket = new WebSocket.Server({ server, noServer: !server });
+const parseParams = (url) => {
+    try {
+        return Object.fromEntries(new URLSearchParams(decodeURIComponent(url || "").replace(/^(\/\?|\?|\/)/g, "")));
+    } catch (error) {
+        return {};
+    }
+};
+
+// options.authenticate (params, req) => boolean, rejects the connection with 401 during the handshake when falsy
+module.exports = (server, { authenticate } = {}) => {
+    const socket = new WebSocket.Server({
+        server,
+        noServer: !server,
+        verifyClient: authenticate && (({ req }) => !!authenticate(parseParams(req.url), req)),
+    });
     socket.on("connection", (ws, req) => {
-        try {
-            ws.params = Object.fromEntries(new URLSearchParams(decodeURIComponent(req.url || "").replace(/^(\/\?|\?|\/)/g, "")));
-        } catch (error) {}
+        ws.params = parseParams(req.url);
+        const userId = ws.params.userId;
+
+        // first tab of this user -> online
+        if (userId && !isConnected(userId, ws)) notifyWatchers(userId, 1);
+
+        // { watchStatuses: [userIds] } -> replies { statuses: { userId: 0 | 1 } }, then pushes { status: { userId, online } } on changes
+        ws.on("message", (raw) => {
+            try {
+                const { watchStatuses } = JSON.parse(raw);
+                if (!watchStatuses) return;
+                ws.watching = new Set(watchStatuses.map(String));
+                ws.send(JSON.stringify({ statuses: socket.getStatuses(watchStatuses) }));
+            } catch (error) {}
+        });
+
+        // last tab closed -> offline
+        ws.on("close", () => userId && !isConnected(userId, ws) && notifyWatchers(userId, 0));
     });
 
+    const isConnected = (userId, except) => {
+        for (const c of socket.clients) if (c !== except && c.params?.userId === userId) return true;
+        return false;
+    };
+
+    const notifyWatchers = (userId, online) => {
+        const data = JSON.stringify({ statuses: { [userId]: online } });
+        for (const c of socket.clients) if (c.watching?.has(userId)) c.send(data);
+    };
+
     socket.broadcast = (data, { skipId, skipToken } = {}) => {
         if (!data) return 0;
         if (typeof data === "object") data = JSON.stringify(data);
@@ -50,23 +88,19 @@ module.exports = (server) => {
 
     socket.getStatuses = (users) => {
         users = new Set(String(users).split(/[,"' ]/g).filter(Boolean));
-
         const usersObj = {};
-        for (const u of users) usersObj[u] = 0;
         
-        let remaining = users.size;
-        for (const client of socket.clients || []) {
-            if (!remaining) break;
-            if (!users.has(client.params.userId)) continue;
+        for (const u of users) usersObj[u] = 0;
 
-            usersObj[client.params.userId] = 1;
-            remaining--;
+        for (const client of socket.clients || []) {
+            if (!users.size) break;
+            const id = client.params?.userId;
+            if (users.delete(id)) usersObj[id] = 1; // delete() is true only the first time -> extra tabs are ignored
         }
 
         return usersObj;
     };
 
-
     return socket;
 };