From e639759ed5f2c216a504929f8fd5d5cb1b13566b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?David=20Majdand=C5=BEi=C4=87?= <dmajdandzic@neos.hr>
Date: Sat, 25 Mar 2023 18:12:26 +0100
Subject: [PATCH] Fix issue with websockets being closed

---
 main.js | 21 ++++++++++++---------
 1 file changed, 12 insertions(+), 9 deletions(-)

diff --git a/main.js b/main.js
index 1d8a798..c2bac10 100644
--- a/main.js
+++ b/main.js
@@ -1170,8 +1170,6 @@ class HTTPServer {
 }
 
 class WSServer {
-	// TODO: Implement center adding and removing...
-	// TODO: This will probably have to be reworked a little to accommodate centers.
 	clients = {};
 	unknownClients = [];
 
@@ -1241,17 +1239,22 @@ class WSServer {
 	}
 
 	removeClient(ws) {
-		// TODO: Fix this
-		for (let sessionId in this.clients) {
-			let index = this.clients[sessionId].indexOf(ws);
+		this.clients.client = this.removeFromArray(this.clients.client, ws);
+		this.clients.center = this.removeFromArray(this.clients.center, ws);
+	}
+
+	removeFromArray(array, element) {
+		for (let sessionId in array) {
+			let index = array[sessionId].indexOf(element);
 			if (index > -1) {
-				delete this.clients[sessionId][index];
+				delete array[sessionId][index];
 			}
-			this.clients[sessionId] = this.clients[sessionId].filter(Boolean);
-			if (this.clients[sessionId].length === 0) {
-				delete this.clients[sessionId];
+			array[sessionId] = array[sessionId].filter(Boolean);
+			if (array[sessionId].length === 0) {
+				delete array[sessionId];
 			}
 		}
+		return array;
 	}
 
 	onClientSessionStatusChange(sessionId, newStatus) {