better-tee/Assets/Code/Server/ServerMainController.cs

using System;
using System.Linq;
using System.Collections.Generic;
using UnityEngine;
using Mirror;

namespace BetterTee.Server
{

    public class ServerMainController : MonoBehaviour 
    {
        [Header("Status")]
        public Dictionary<NetworkConnection, ConnectedPlayer> players;
        public Dictionary<NetworkConnection, ConnectedViewer> viewers;
        public GamePhase gamePhase = GamePhase.UNINTIALIZED;
        public GameSettings gameSettings = null;
        public int? currentActNumber = null;
        public string lobbyPassword = null;

        public bool CanStartGame {
            get {
                return (
                    gamePhase == GamePhase.LOBBY &&
                    gameSettings != null &&
                    players.Count >= gameSettings.minimumPlayers
                );
            }
        }

        public ActSettings CurrentActSettings {
            get {
                if(gameSettings == null) return null;
                if(currentActNumber == null) return null;
                return gameSettings.acts[currentActNumber.Value - 1];
            }
        }

        [Header("Constants")]
        public const int MAX_CONNECTIONS = 32;
        public const int PASSWORD_LENGTH = 4;
        
        #region Unity Methods

        protected void Start() {
            ServerStart();
        }

        protected void OnDestroy() {
            if(NetworkServer.active) NetworkServer.Shutdown();
        }

        #endregion
        
        public void ServerStart() {
            gamePhase = GamePhase.LOBBY;
            players = new Dictionary<NetworkConnection, ConnectedPlayer>();
            viewers = new Dictionary<NetworkConnection, ConnectedViewer>();
            Transport.activeTransport = GetComponent<TelepathyTransport>();

            #region Password Creation
            char[] charList = "ABCDEFGHIJKLMNOPQRSTUVWXYZ".ToCharArray();
            lobbyPassword = "";
            for(int i = 0; i < PASSWORD_LENGTH; i++) {
                lobbyPassword += charList[UnityEngine.Random.Range(0, charList.Length)];
            }
            Debug.Log("The server password is: {}");
            #endregion

            #region Client Messages
            NetworkServer.RegisterHandler<NetMsg.Client.PlayerJoin>(OnPlayerJoin);
            NetworkServer.RegisterHandler<NetMsg.Client.ActResultsMsg>(OnActResults);
            #endregion
            #region Viewer Messages
            NetworkServer.RegisterHandler<NetMsg.Viewer.Settings>(OnGameSettings);
            NetworkServer.RegisterHandler<NetMsg.Viewer.ViewerLink>(OnViewerLink);
            NetworkServer.RegisterHandler<NetMsg.Viewer.GameStart>(OnGameStart);
            #endregion
            #region Other Messages
            NetworkServer.RegisterHandler<ConnectMessage>(OnConnect);
            NetworkServer.RegisterHandler<DisconnectMessage>(OnDisconnect);
            #endregion

            NetworkServer.Listen(MAX_CONNECTIONS);
        }

        public void SendLobbyUpdate() {
            SendToAllRegistered<NetMsg.Server.LobbyStatusChange>(new NetMsg.Server.LobbyStatusChange {
                players = players.Values.ToList().ConvertAll<ConnectedPlayerData>(player => player.Data).ToArray(),
                viewers = viewers.Values.ToList().ConvertAll<ConnectedViewerData>(viewer => viewer.Data).ToArray(),
                canStart = (gameSettings != null && players.Count >= gameSettings.minimumPlayers)
            });
        }

        public void GameStart() {
            gamePhase = GamePhase.ACTS;
            currentActNumber = 1;
            //TODO?
        }

        public void ActInit() {
            SendToAllRegistered<NetMsg.Server.ActInit>(new NetMsg.Server.ActInit {
                settings = CurrentActSettings
            });
        }

        #region SendToAll Methods

        public void SendToAllPlayers<T>(T message, int channelId = 0)
        where T: IMessageBase {
            foreach(NetworkConnection connection in players.Keys) {
                connection.Send<T>(message, channelId);
            }
        }

        public void SendToAllViewers<T>(T message, int channelId = 0)
        where T: IMessageBase {
            foreach(NetworkConnection connection in viewers.Keys) {
                connection.Send<T>(message, channelId);
            }
        }

        public void SendToAllRegistered<T>(T message, int channelId = 0)
        where T: IMessageBase {
            SendToAllPlayers<T>(message, channelId);
            SendToAllViewers<T>(message, channelId);
        }

        #endregion

        #region Network Events

        protected void OnConnect(NetworkConnection connection, ConnectMessage message) {}

        protected void OnDisconnect(NetworkConnection connection, DisconnectMessage message) {
            //How to handle disconnections?
            if(gamePhase == GamePhase.LOBBY) {
                try {
                    players.Remove(connection);
                }
                catch(KeyNotFoundException) {}

                try {
                    viewers.Remove(connection);
                }
                catch(KeyNotFoundException) {}

                SendLobbyUpdate();
            }
            else {
                Debug.LogWarning("Disconnections after the lobby phase aren't handled");
            }
        }

        protected void OnPlayerJoin(NetworkConnection connection, NetMsg.Client.PlayerJoin message) {
            if(gameSettings == null) {
                connection.Send<NetMsg.Server.Error.MissingGameSettings>(new NetMsg.Server.Error.MissingGameSettings());
                connection.Disconnect();
                return;
            }
            if(message.gamePassword != lobbyPassword) {
                connection.Send<NetMsg.Server.Error.InvalidPassword>(new NetMsg.Server.Error.InvalidPassword());
                connection.Disconnect();
                return;
            }
            if(gamePhase != GamePhase.LOBBY) {
                connection.Send<NetMsg.Server.Error.GameAlreadyStarted>(new NetMsg.Server.Error.GameAlreadyStarted());
                connection.Disconnect();
                return;
            }
            if(players.Count >= gameSettings.maximumPlayers) {
                connection.Send<NetMsg.Server.Error.MaxPlayersCapReached>(new NetMsg.Server.Error.MaxPlayersCapReached());
                connection.Disconnect();
                return;
            }

            ConnectedPlayer newPlayer = new ConnectedPlayer {
                name = message.playerName,
                id = players.Count
            };
            players.Add(connection, newPlayer);

            SendLobbyUpdate();
        }        

        protected void OnViewerLink(NetworkConnection connection, NetMsg.Viewer.ViewerLink message) {
            if(message.gamePassword != lobbyPassword) {
                connection.Send<NetMsg.Server.Error.InvalidPassword>(new NetMsg.Server.Error.InvalidPassword());
                connection.Disconnect();
                return;
            }

            ConnectedViewer newViewer = new ConnectedViewer {
                name = message.viewerName,
                id = viewers.Count
            };
            viewers.Add(connection, newViewer);

            SendLobbyUpdate();
        }

        protected void OnGameSettings(NetworkConnection connection, NetMsg.Viewer.Settings message) {
            if(gameSettings != null) {
                Debug.LogWarning("gameSettings were overwritten.");
            }

            gameSettings = message.settings;

            SendLobbyUpdate();
        }

        protected void OnGameStart(NetworkConnection connection, NetMsg.Viewer.GameStart message) {
            if(gameSettings == null) {
                connection.Send<NetMsg.Server.Error.NoSettings>(new NetMsg.Server.Error.NoSettings());
                connection.Disconnect();
                return;
            }
            if(players.Count < gameSettings.minimumPlayers) {
                connection.Send<NetMsg.Server.Error.NotEnoughPlayers>(new NetMsg.Server.Error.NotEnoughPlayers());
                connection.Disconnect();
                return;
            }

            GameStart();
        }
        
        protected void OnActResults(NetworkConnection connection, NetMsg.Client.ActResultsMsg message) {}

        #endregion
    }

}