quantum-space-buddies/QSB/ShipSync/Messages/FlyShipMessage.cs

using QSB.AuthoritySync;
using QSB.Messaging;
using QSB.Player;
using QSB.Player.TransformSync;
using QSB.ShipSync.TransformSync;
using QSB.WorldSync;
using UnityEngine;

namespace QSB.ShipSync.Messages
{
	internal class FlyShipMessage : QSBMessage<bool>
	{
		static FlyShipMessage()
		{
			GlobalMessenger<OWRigidbody>.AddListener(OWEvents.EnterFlightConsole, _ => Handler(true));
			GlobalMessenger.AddListener(OWEvents.ExitFlightConsole, () => Handler(false));
		}

		private static void Handler(bool flying)
		{
			if (PlayerTransformSync.LocalInstance)
			{
				new FlyShipMessage(flying).Send();
			}
		}

		private FlyShipMessage(bool flying) => Data = flying;

		public override bool ShouldReceive => QSBWorldSync.AllObjectsReady;

		public override void OnReceiveLocal() => SetCurrentFlyer(From, Data);

		public override void OnReceiveRemote()
		{
			SetCurrentFlyer(From, Data);
			var shipCockpitController = GameObject.Find("ShipCockpitController").GetComponent<ShipCockpitController>();
			if (Data)
			{
				shipCockpitController._interactVolume.DisableInteraction();
			}
			else
			{
				shipCockpitController._interactVolume.EnableInteraction();
			}
		}

		private static void SetCurrentFlyer(uint id, bool isFlying)
		{
			ShipManager.Instance.CurrentFlyer = isFlying
				? id
				: uint.MaxValue;

			if (QSBCore.IsHost)
			{
				ShipTransformSync.LocalInstance.netIdentity.SetAuthority(isFlying
					? id
					: QSBPlayerManager.LocalPlayerId);
			}
		}
	}
}