From 40f52a4a4579209e282cb4af7c190afa563d551a Mon Sep 17 00:00:00 2001
From: akastijn <akastijn@alttd.com>
Date: Sat, 11 Jul 2026 19:14:37 +0200
Subject: [PATCH] Update `PlayerService` to adjust `PLAYER_STATE` during
 `ROUND_STATE.ENDED` phase

---
 .../com/alttd/hunger_games/services/PlayerService.java     | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/alttd/hunger_games/services/PlayerService.java b/src/main/java/com/alttd/hunger_games/services/PlayerService.java
index 3fd5676..31b1059 100644
--- a/src/main/java/com/alttd/hunger_games/services/PlayerService.java
+++ b/src/main/java/com/alttd/hunger_games/services/PlayerService.java
@@ -131,7 +131,12 @@ public class PlayerService implements RoundListener {
         if (optionalPlayerState.isPresent()) {
             PLAYER_STATE playerState = optionalPlayerState.get();
             return Optional.of(switch (playerState) {
-                case DISCONNECTED, SPECTATING -> PLAYER_STATE.SPECTATING;
+                case DISCONNECTED, SPECTATING -> {
+                    if (roundState.equals(ROUND_STATE.ENDED)) {
+                        yield PLAYER_STATE.REGISTERED;
+                    }
+                    yield PLAYER_STATE.SPECTATING;
+                }
                 case REGISTERED -> PLAYER_STATE.REGISTERED;
             });
         }