TheKingsRace/Assets/Scripts/Network/GameNetPortal.cs

using System;
using MLAPI;
using MLAPI.Messaging;
using MLAPI.Serialization.Pooled;
using MLAPI.Transports;
using MLAPI.Transports.UNET;
using UnityEngine;

public class GameNetPortal : MonoBehaviour {

    public static GameNetPortal Instance => instance;
    private static GameNetPortal instance;

    public event Action OnNetworkReadied;

    public event Action<ConnectStatus> OnConnectionFinished;
    public event Action<ConnectStatus> OnDisconnectReasonReceived;

    public event Action<ulong, int> OnClientSceneChanged;

    public event Action OnUserDisconnectRequested;

    private void Awake()
    {
        if (instance != null && instance != this)
        {
            Destroy(gameObject);
            return;
        }

        instance = this;
        DontDestroyOnLoad(gameObject);
    }

    private void Start()
    {
        NetworkManager.Singleton.OnServerStarted += HandleNetworkReady;
        NetworkManager.Singleton.OnClientConnectedCallback += HandleClientConnected;

        RegisterClientMessageHandlers();
        RegisterServerMessageHandlers();

        SetOptions();
    }

    private void SetOptions() { 
        // Set the Use MLAPI Relay Option if it doesn't yet exist
        if (!PlayerPrefs.HasKey("UseMLAPIRelay")) {
            PlayerPrefs.SetInt("UseMLAPIRelay", 0);
        } else { 
            if (PlayerPrefs.GetInt("UseMLAPIRelay") == 0) {
                GameObject.FindGameObjectWithTag("NetworkManager").GetComponent<UNetTransport>().UseMLAPIRelay = false;
            } else {
                GameObject.FindGameObjectWithTag("NetworkManager").GetComponent<UNetTransport>().UseMLAPIRelay = true;
            }
        }

        // Set other options here
    }

    private void OnDestroy()
    {
        if (NetworkManager.Singleton != null)
        {
            NetworkManager.Singleton.OnServerStarted -= HandleNetworkReady;
            NetworkManager.Singleton.OnClientConnectedCallback -= HandleClientConnected;
        }

        UnregisterClientMessageHandlers();
        UnregisterServerMessageHandlers();
    }

    public void StartHost()
    {
        NetworkManager.Singleton.StartHost();
    }

    public void RequestDisconnect()
    {
        OnUserDisconnectRequested?.Invoke();
    }

    private void HandleClientConnected(ulong clientId)
    {
        if (clientId != NetworkManager.Singleton.LocalClientId) { return; }

        HandleNetworkReady();
    }

    private void HandleNetworkReady()
    {
        if (NetworkManager.Singleton.IsHost)
        {
            OnConnectionFinished?.Invoke(ConnectStatus.Success);
        }

        OnNetworkReadied?.Invoke();
    }

    #region Message Handlers

    private void RegisterClientMessageHandlers()
    {
        CustomMessagingManager.RegisterNamedMessageHandler("ServerToClientConnectResult", (senderClientId, stream) =>
        {
            using (var reader = PooledNetworkReader.Get(stream))
            {
                ConnectStatus status = (ConnectStatus)reader.ReadInt32();

                OnConnectionFinished?.Invoke(status);
            }
        });

        CustomMessagingManager.RegisterNamedMessageHandler("ServerToClientSetDisconnectReason", (senderClientId, stream) =>
        {
            using (var reader = PooledNetworkReader.Get(stream))
            {
                ConnectStatus status = (ConnectStatus)reader.ReadInt32();

                OnDisconnectReasonReceived?.Invoke(status);
            }
        });
    }

    private void RegisterServerMessageHandlers()
    {
        CustomMessagingManager.RegisterNamedMessageHandler("ClientToServerSceneChanged", (senderClientId, stream) =>
        {
            using (var reader = PooledNetworkReader.Get(stream))
            {
                int sceneIndex = reader.ReadInt32();

                OnClientSceneChanged?.Invoke(senderClientId, sceneIndex);
            }
        });
    }

    private void UnregisterClientMessageHandlers()
    {
        CustomMessagingManager.UnregisterNamedMessageHandler("ServerToClientConnectResult");
        CustomMessagingManager.UnregisterNamedMessageHandler("ServerToClientSetDisconnectReason");
    }

    private void UnregisterServerMessageHandlers()
    {
        CustomMessagingManager.UnregisterNamedMessageHandler("ClientToServerSceneChanged");
    }

    #endregion

    #region Message Senders

    public void ServerToClientConnectResult(ulong netId, ConnectStatus status)
    {
        using (var buffer = PooledNetworkBuffer.Get())
        {
            using (var writer = PooledNetworkWriter.Get(buffer))
            {
                writer.WriteInt32((int)status);
                CustomMessagingManager.SendNamedMessage("ServerToClientConnectResult", netId, buffer, NetworkChannel.Internal);
            }
        }
    }

    public void ServerToClientSetDisconnectReason(ulong netId, ConnectStatus status)
    {
        using (var buffer = PooledNetworkBuffer.Get())
        {
            using (var writer = PooledNetworkWriter.Get(buffer))
            {
                writer.WriteInt32((int)status);
                CustomMessagingManager.SendNamedMessage("ServerToClientSetDisconnectReason", netId, buffer, NetworkChannel.Internal);
            }
        }
    }

    public void ClientToServerSceneChanged(int newScene)
    {
        if (NetworkManager.Singleton.IsHost)
        {
            OnClientSceneChanged?.Invoke(NetworkManager.Singleton.ServerClientId, newScene);
        }
        else if (NetworkManager.Singleton.IsConnectedClient)
        {
            using (var buffer = PooledNetworkBuffer.Get())
            {
                using (var writer = PooledNetworkWriter.Get(buffer))
                {
                    writer.WriteInt32(newScene);
                    CustomMessagingManager.SendNamedMessage("ClientToServerSceneChanged", NetworkManager.Singleton.ServerClientId, buffer, NetworkChannel.Internal);
                }
            }
        }
    }

    #endregion
}