quantum-space-buddies/QSB/MessagesCore/MessageHandler.cs

using OWML.Common;
using QSB.EventsCore;
using QSB.QuantumUNET;
using QSB.Utility;
using System;
using System.Linq;
using UnityEngine.Networking;

namespace QSB.Messaging
{
	// Extend this to create new message handlers.
	public class MessageHandler<T> where T : MessageBase, new()
	{
		public event Action<T> OnClientReceiveMessage;
		public event Action<T> OnServerReceiveMessage;

		private readonly EventType _eventType;

		public MessageHandler(EventType eventType)
		{
			_eventType = eventType + MsgType.Highest + 1;
			if (QSBNetworkManager.Instance.IsReady)
			{
				Init();
			}
			else
			{
				QSBNetworkManager.Instance.OnNetworkManagerReady += Init;
			}
		}

		private void Init()
		{
			var eventName = Enum.GetName(typeof(EventType), _eventType - 1 - MsgType.Highest).ToUpper();
			if (QSBNetworkServer.handlers.Keys.Contains((short)_eventType))
			{
				DebugLog.ToConsole($"Warning - NetworkServer already contains a handler for EventType {_eventType}", MessageType.Warning);
				QSBNetworkServer.handlers.Remove((short)_eventType);
			}
			QSBNetworkServer.RegisterHandler((short)_eventType, OnServerReceiveMessageHandler);
			QSBNetworkManager.singleton.client.RegisterHandler((short)_eventType, OnClientReceiveMessageHandler);
		}

		public void SendToAll(T message)
		{
			if (!QSBNetworkManager.Instance.IsReady)
			{
				return;
			}
			QSBNetworkServer.SendToAll((short)_eventType, message);
		}

		public void SendToServer(T message)
		{
			if (!QSBNetworkManager.Instance.IsReady)
			{
				return;
			}
			QSBNetworkManager.singleton.client.Send((short)_eventType, message);
		}

		private void OnClientReceiveMessageHandler(QSBNetworkMessage netMsg)
		{
			var message = netMsg.ReadMessage<T>();
			OnClientReceiveMessage?.Invoke(message);
		}

		private void OnServerReceiveMessageHandler(QSBNetworkMessage netMsg)
		{
			var message = netMsg.ReadMessage<T>();
			OnServerReceiveMessage?.Invoke(message);
		}

	}
}