Better lobby system

Room/Room.cs

@@ -10,9 +10,10 @@ public class Room
     private readonly IRoomManager _roomManager;
     private readonly string _roomId;
     private readonly List<ConnectionInstance> _connections = new();
-    private ConnectionInstance _host;
+    private ConnectionInstance? _host;
     private readonly Chat _chat;
-    private readonly Game _game = new();
+    private Game? _game;
+    private RoomState _state = RoomState.LOBBY;
 
     public Room(IRoomManager roomManager, string roomId)
     {
@@ -24,14 +25,13 @@ public class Room
     public async Task InstantiateConnection(WebSocket socket, string name)
     {
         var connection = AddConnection(socket, name);
-        if (IsEmpty()) _host = connection;
-        _game.AddPlayerToGame(connection);
+        _game?.AddPlayerToGame(connection);
         await HandleConnection(connection);
     }
 
     private async Task HandleConnection(ConnectionInstance connection)
     {
-        BroadcastGameState();
+        BroadcastState();
         var webSocketResponse = await WebsocketManager.ReceiveAsync(connection.Socket);
         while (!webSocketResponse.Result!.CloseStatus.HasValue)
         {
@@ -40,6 +40,7 @@ public class Room
             {
                 case "GAME":
                 {
+                    if (_state != RoomState.GAME) break;
                     var gameInput = JsonSerializer.Deserialize<ActionDTO>(message.Payload);
                     _game.HandleAction(connection.ConnectionId, gameInput);
                     break;
@@ -54,9 +55,17 @@ public class Room
                     _chat.SendChatMessage(connection, cleanedMessage);
                     break;
                 }
+                case "LOBBY":
+                {
+                    if (connection.ConnectionId == _host?.ConnectionId)
+                    {
+                        ChangeLobbyState(RoomState.GAME);
+                    }
+                    break;
+                }
             }
 
-            BroadcastGameState();
+            BroadcastState();
             webSocketResponse = await WebsocketManager.ReceiveAsync(connection.Socket);
         }
         WebsocketManager.CloseAsync(connection.Socket, webSocketResponse.Result);
@@ -67,6 +76,7 @@ public class Room
     {
         var connectionId = Guid.NewGuid().ToString();
         var connection = new ConnectionInstance(name, connectionId, socket);
+        if (IsEmpty()) _host = connection;
         _connections.Add(connection);
         return connection;
     }
@@ -78,6 +88,10 @@ public class Room
 
     private void BroadcastGameState()
     {
+        if (_game == null)
+        {
+            return;
+        }
         foreach (var connection in _connections)
         {
             var gameState = new GameState(_game, connection.ConnectionId);
@@ -86,10 +100,50 @@ public class Room
         }
     }
 
+    private void BroadcastState()
+    {
+        switch (_state)
+        {
+            case RoomState.LOBBY:
+                var message = new MessageDTO("LOBBY", JsonSerializer.Serialize("a"));
+                WebsocketManager.BroadcastAsync(GetWebsockets(), JsonSerializer.Serialize(message));
+                break;
+            case RoomState.GAME:
+                BroadcastGameState();
+                break;
+            default:
+                //
+                break;
+        }
+    }
+
+    private void ChangeLobbyState(RoomState targetState)
+    {
+        switch (targetState)
+        {
+            case RoomState.LOBBY:
+                _state = RoomState.LOBBY;
+                _game = null;
+                break;
+            case RoomState.GAME:
+            {
+                _state = RoomState.GAME;
+                _game = new Game(this);
+                foreach (var connectionInstance in _connections)
+                {
+                    _game.AddPlayerToGame(connectionInstance);
+                }
+                break;
+            }
+            default:
+                throw new ArgumentOutOfRangeException(nameof(targetState), targetState, null);
+        }
+    }
+
     private void HandleDisconnect(string socketId)
     {
         RemoveConnection(socketId);
-        _game.RemovePlayer(socketId);
+        _game?.RemovePlayer(socketId);
         if (IsEmpty())
         {
             _roomManager.RemoveRoom(_roomId);
@@ -112,6 +166,19 @@ public class Room
     
     private static string StripHTML(string input)
     {
-        return Regex.Replace(input, "<.*?>", String.Empty);
+        return Regex.Replace(input, "<.*?>", string.Empty);
     }
+
+    public void EndGame(Player player)
+    {
+        var message = new MessageDTO("END", JsonSerializer.Serialize(new PlayerDTO(player)));
+        WebsocketManager.BroadcastAsync(GetWebsockets(), JsonSerializer.Serialize(message));
+        ChangeLobbyState(RoomState.LOBBY);
+    }
+}
+
+public enum RoomState
+{
+    LOBBY,
+    GAME
 }