jelito/Assets/jelycho/Code/Main/JelychoGame.cs

using System;
using System.Collections.Generic;
using RebootKit.Engine.Console;
using RebootKit.Engine.Main;
using RebootKit.Engine.Simulation;
using RebootReality.jelycho.Player;
using RebootReality.jelycho.Player.HUD;
using Unity.Collections;
using Unity.Netcode;
using UnityEngine;
using UnityEngine.AddressableAssets;
using UnityEngine.Assertions;
using Logger = RebootKit.Engine.Foundation.Logger;

namespace RebootReality.jelycho.Main {
    class PlayerState : INetworkSerializable {
        public ulong ClientID;

        public PlayerController Controller;
        public PlayerActor Actor;

        public void NetworkSerialize<T>(BufferSerializer<T> serializer) where T : IReaderWriter { }
    }

    public class JelychoGame : Game {
        static readonly Logger s_Logger = new Logger(nameof(JelychoGame));

        [Header("Player")]
        [SerializeField] PlayerController m_PlayerControllerPrefab;
        [SerializeField] AssetReferenceGameObject m_PlayerActorPrefab;

        readonly List<PlayerState> m_PlayerStates = new List<PlayerState>();

        void Awake() { }

        public override void OnDestroy() {
            base.OnDestroy();
        }

        protected override void OnPlayerBecameReady(ulong clientID) {
            s_Logger.Info($"Player {clientID} became ready");

            if (IsServer) {
                PlayerActor actor =
                    RR.SpawnActor(m_PlayerActorPrefab, Vector3.zero, Quaternion.identity) as PlayerActor;

                PlayerState playerState = GetPlayerState(clientID);
                Assert.IsNotNull(playerState);
                playerState.Actor = actor;
                playerState.Controller.PossessActor(actor);
            }
        }

        public override void OnNetworkSpawn() {
            base.OnNetworkSpawn();

            NetworkManager.Singleton.OnClientConnectedCallback += OnClientConnected;
            NetworkManager.Singleton.OnClientDisconnectCallback += OnClientStopped;
        }

        public override void OnNetworkDespawn() {
            base.OnNetworkDespawn();

            NetworkManager.Singleton.OnClientConnectedCallback -= OnClientConnected;
            NetworkManager.Singleton.OnClientDisconnectCallback -= OnClientStopped;
        }

        void OnClientConnected(ulong clientID) {
            s_Logger.Info($"Client {clientID} connected");

            if (IsServer) {
                s_Logger.Info($"Creating player state for client {clientID}");

                PlayerController controller = Instantiate(m_PlayerControllerPrefab);
                controller.name = $"PlayerController_{clientID}";
                controller.NetworkObject.SpawnAsPlayerObject(clientID);

                m_PlayerStates.Add(new PlayerState {
                    ClientID = clientID,
                    Controller = controller,
                });
            }
        }

        void OnClientStopped(ulong clientID) {
            s_Logger.Info($"Client {clientID} disconnected");

            if (IsServer) {
                for (int i = m_PlayerStates.Count - 1; i >= 0; i--) {
                    if (m_PlayerStates[i].ClientID == clientID) {
                        s_Logger.Info($"Removing player state for client {clientID}");
                        m_PlayerStates.RemoveAtSwapBack(i);
                        break;
                    }
                }
            }
        }

        PlayerState GetPlayerState(ulong clientID) {
            foreach (PlayerState state in m_PlayerStates) {
                if (state.ClientID == clientID) {
                    return state;
                }
            }

            s_Logger.Error($"Player state for client {clientID} not found");
            return null;
        }
    }

    public static class JelychoConsoleCommands {
        [RCCMD("worlds", "Lists all worlds")]
        public static void Worlds(string[] args) {
            if (args.Length != 1) {
                RR.WriteToConsole($"Usage: {args[0]}");
                return;
            }

            RR.WriteToConsole("Worlds:");
            foreach (WorldConfigAsset worldConfigAsset in RR.WorldConfigsAssets) {
                RR.WriteToConsole($"  {worldConfigAsset.Config.name}");
            }
        }

        [RCCMD("connect_dev", "Connects to a local server for development purposes")]
        public static void ConnectToLocalServer(string[] args) {
            RR.WriteToConsole("Connecting to server...");
            RR.Connect();
        }

        [RCCMD("connect", "Connects to a server with given Steam ID. Usage: connect <steam_id>")]
        public static void ConnectToServer(string[] args) {
            if (args.Length < 1 || args.Length > 2) {
                RR.WriteToConsole($"Usage: {args[0]} <steam_id>");
                return;
            }

            RR.WriteToConsole("Connecting to server...");

            if (args.Length != 2) {
                RR.WriteToConsole($"Usage: {args[0]} <steam_id>");
                return;
            }

            if (ulong.TryParse(args[1], out ulong steamID)) {
                RR.ConnectWithSteamID(steamID);
            } else {
                RR.WriteToConsole($"Invalid Steam ID: {args[1]}");
            }
        }

        [RCCMD("disconnect", "Disconnects from the current server")]
        public static void DisconnectFromServer(string[] args) {
            if (args.Length != 1) {
                RR.WriteToConsole($"Usage: {args[0]}");
                return;
            }

            RR.Disconnect();
        }

        [RCCMD("version", "Displays the current game version")]
        public static void GameVersion(string[] args) {
            RR.WriteToConsole($"Game version: {Application.version}");
        }

        [RCCMD("nq", "Quits the game")]
        public static void Nq(string[] args) {
            Application.Quit();
        }
    }
}