From 5edc5874c2c7cdbfa212b4b846527ae874c736a5 Mon Sep 17 00:00:00 2001
From: Mihaela Dumitru <mihdmt@gmail.com>
Date: Tue, 10 Oct 2023 12:24:02 +0300
Subject: [PATCH] add user limit (#14)

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

diff --git a/src/index.ts b/src/index.ts
index 9f05c40..c4ca3da 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -2,7 +2,7 @@ import debug from 'debug';
 import dotenv from 'dotenv';
 import express from 'express';
 import http from 'http';
-import socketIO from 'socket.io';
+import socketIO, { Socket } from 'socket.io';
 import * as prometheus from 'socket.io-prometheus-metrics';
 
 const serverDebug = debug('server');
@@ -15,6 +15,8 @@ dotenv.config(
 
 const app = express();
 const port = process.env.PORT || 80; // default port to listen
+const users: Socket[] = [];
+const userLimit = Number(process.env.USER_LIMIT) || Infinity;
 
 app.get('/', (req, res) => {
     res.send('Excalidraw backend is up :)');
@@ -52,6 +54,22 @@ io.on('connection', socket => {
     socket.on('join-room', roomID => {
         serverDebug(`${socket.id} has joined ${roomID} for url ${socket.conn.request.url}`);
         socket.join(roomID);
+
+        users.push(socket);
+        socket.on('close', () => {
+            users.splice(users.indexOf(socket), 1);
+        });
+
+        const socketsCount = Object.keys(io.in(roomID).sockets).length;
+
+        if (socketsCount > userLimit) {
+            users.forEach((user: Socket) => {
+                user.disconnect();
+            });
+
+            return;
+        }
+
         if (io.sockets.adapter.rooms[roomID].length <= 1) {
             io.to(`${socket.id}`).emit('first-in-room');
         } else {