quantum-space-buddies/QSB/Events/EventHandler.cs

using System.Collections;
using System.Collections.Generic;
using OWML.ModHelper.Events;
using QSB.Animation;
using QSB.Messaging;
using QSB.TransformSync;
using UnityEngine;
using UnityEngine.Networking;

namespace QSB.Events
{
    public class EventHandler : NetworkBehaviour
    {
        public static EventHandler LocalInstance;

        private MessageHandler<EventMessage> _eventHandler;

        private void Awake()
        {
            LocalInstance = this;

            _eventHandler = new MessageHandler<EventMessage>();
            _eventHandler.OnClientReceiveMessage += OnClientReceiveMessage;
            _eventHandler.OnServerReceiveMessage += OnServerReceiveMessage;
        }

        public void Send(EventType eventType)
        {
            StartCoroutine(SendEvent(eventType));
        }

        private IEnumerator SendEvent(EventType eventType)
        {
            yield return new WaitUntil(() => PlayerTransformSync.LocalInstance != null);
            var message = new EventMessage
            {
                EventType = (int)eventType,
                SenderId = PlayerTransformSync.LocalInstance.netId.Value
            };
            _eventHandler.SendToServer(message);
        }

        private void OnServerReceiveMessage(EventMessage message)
        {
            _eventHandler.SendToAll(message);
        }

        private void OnClientReceiveMessage(EventMessage message)
        {
            if (message.SenderId != PlayerTransformSync.LocalInstance.netId.Value)
            {
                switch ((EventType)message.EventType)
                {
                    case EventType.TurnOnFlashlight:
                        PlayerRegistry.GetPlayerFlashlight(message.SenderId).TurnOn();
                        PlayerRegistry.UpdateState(message.SenderId, State.Flashlight, true);
                        break;
                    case EventType.TurnOffFlashlight:
                        PlayerRegistry.GetPlayerFlashlight(message.SenderId).TurnOff();
                        PlayerRegistry.UpdateState(message.SenderId, State.Flashlight, false);
                        break;
                    case EventType.SuitUp:
                        PlayerRegistry.UpdateState(message.SenderId, State.Suit, true);
                        break;
                    case EventType.RemoveSuit:
                        PlayerRegistry.UpdateState(message.SenderId, State.Suit, false);
                        break;
                }
            }
        }
    }
}