From 9fda0a2c4bea0f24a4d79e30e9d07851569f41fe Mon Sep 17 00:00:00 2001
From: Florent Chehab <perso@floflo.ch>
Date: Sun, 10 May 2020 23:10:54 +0200
Subject: [PATCH] fix(back): prevent crash

---
 scripts/server-backend.js | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/scripts/server-backend.js b/scripts/server-backend.js
index d673bc6..dec238a 100644
--- a/scripts/server-backend.js
+++ b/scripts/server-backend.js
@@ -194,6 +194,7 @@ function startBackendServer(port) {
     io.on("connection", function (socket) {
         var whiteboardId = null;
         socket.on("disconnect", function () {
+            if (infoByWhiteboard.has(whiteboardId)) {
             const whiteboardServerSideInfo = infoByWhiteboard.get(whiteboardId);
 
             if (socket && socket.id) {
@@ -201,11 +202,13 @@ function startBackendServer(port) {
             }
 
             whiteboardServerSideInfo.decrementNbConnectedUsers();
+
             if (whiteboardServerSideInfo.hasConnectedUser()) {
                 socket.compress(false).broadcast.emit("refreshUserBadges", null); //Removes old user Badges
             } else {
                 infoByWhiteboard.delete(whiteboardId);
             }
+            }
         });
 
         socket.on("drawToWhiteboard", function (content) {