From a2f2080959b9054570dbeb4749d2648352f9ca47 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?P=C5=91cze=20Bence?= <bence@pocze.ch>
Date: Sun, 4 Apr 2021 17:43:37 +0200
Subject: [PATCH] MAPG-219 send 'end_round' when all members are finished

---
 multi/index.js | 50 ++++++++++++++++++++++++++++----------------------
 1 file changed, 28 insertions(+), 22 deletions(-)

diff --git a/multi/index.js b/multi/index.js
index 76708a7..91e9f39 100644
--- a/multi/index.js
+++ b/multi/index.js
@@ -97,28 +97,15 @@ class MultiGame {
         room.updated = new Date();
 
         var round = room.rounds[room.currentRound];
-
-        clearTimeout(round.timeoutHandler);
-        round.timeout = round.timeout - (new Date() - round.timeoutStarted);
-        if (round.timeout > MultiGame.ROUND_TIMEOUT_DIVIDER * MultiGame.ROUND_TIMEOUT_MINIMUM) {
-            round.timeout = round.timeout / MultiGame.ROUND_TIMEOUT_DIVIDER;
-        } else if (round.timeout > MultiGame.ROUND_TIMEOUT_MINIMUM) {
-            round.timeout = MultiGame.ROUND_TIMEOUT_MINIMUM;
-        }
-        round.timeoutStarted = new Date();
-        var self = this;
-        round.timeoutHandler = setTimeout(function () {
-            self._endRoundTimeout(room, round);
-        }, round.timeout + MultiGame.ROUND_TIMEOUT_OFFSET);
-
         var member = room.members.get(token);
         var allResults = this._collectResultsInRound(room, round);
 
-        this._broadcastTimeout(room, round);
         this._broadcastGuess(room, member.userName, guessPosition, distance, score);
 
         round.results.set(token, { guessPosition: guessPosition, distance: distance, score: score });
 
+        this._setNewTimeout(room, round);
+
         return allResults;
     }
 
@@ -139,7 +126,7 @@ class MultiGame {
         round.timeoutStarted = new Date();
         var self = this;
         round.timeoutHandler = setTimeout(function () {
-            self._endRoundTimeout(room, round);
+            self._endRound(room, round);
         }, round.timeout + MultiGame.ROUND_TIMEOUT_OFFSET);
 
         var data = {};
@@ -152,16 +139,35 @@ class MultiGame {
         });
     }
 
-    _endRoundTimeout(room, round) {
+    _setNewTimeout(room, round) {
         clearTimeout(round.timeoutHandler);
 
+        if (room.members.size === round.results.size) {
+            round.timeout = 0;
+            round.timeoutStarted = new Date();
+
+            this._endRound(room, round);
+        } else {
+            round.timeout = round.timeout - (new Date() - round.timeoutStarted);
+            if (round.timeout > MultiGame.ROUND_TIMEOUT_DIVIDER * MultiGame.ROUND_TIMEOUT_MINIMUM) {
+                round.timeout = round.timeout / MultiGame.ROUND_TIMEOUT_DIVIDER;
+            } else if (round.timeout > MultiGame.ROUND_TIMEOUT_MINIMUM) {
+                round.timeout = MultiGame.ROUND_TIMEOUT_MINIMUM;
+            }
+            round.timeoutStarted = new Date();
+            var self = this;
+            round.timeoutHandler = setTimeout(function () {
+                self._endRound(room, round);
+            }, round.timeout + MultiGame.ROUND_TIMEOUT_OFFSET);
+
+            this._broadcastTimeout(room, round);
+        }
+    }
+
+    _endRound(room, round) {
         var data = { position: round.place.position, allResults: this._collectResultsInRound(room, round) };
         var self = this;
-        room.members.forEach(function (member, token) {
-            if (round.results.has(token)) {
-                return;
-            }
-
+        room.members.forEach(function (member) {
             self._sendToMember(member, 'end_round', data);
         });
     }