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) : base(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);
		}
	}
}