quantum-space-buddies/QSB/Anglerfish/Messages/AnglerChangeStateMessage.cs

using QSB.Anglerfish.WorldObjects;
using QSB.Messaging;
using QSB.Player;
using QuantumUNET.Transport;
using UnityEngine;

namespace QSB.Anglerfish.Messages
{
	public class AnglerChangeStateMessage : QSBEnumWorldObjectMessage<QSBAngler, AnglerfishController.AnglerState>
	{
		private uint TargetId;
		private Vector3 LocalDisturbancePos;

		public AnglerChangeStateMessage(QSBAngler qsbAngler)
		{
			Value = qsbAngler.AttachedObject._currentState;
			TargetId = TargetToId(qsbAngler.TargetTransform);
			LocalDisturbancePos = qsbAngler.AttachedObject._localDisturbancePos;
		}

		public AnglerChangeStateMessage() { }

		public override void Serialize(QNetworkWriter writer)
		{
			base.Serialize(writer);
			writer.Write(TargetId);
			writer.Write(LocalDisturbancePos);
		}

		public override void Deserialize(QNetworkReader reader)
		{
			base.Deserialize(reader);
			TargetId = reader.ReadUInt32();
			LocalDisturbancePos = reader.ReadVector3();
		}

		public override void OnReceiveRemote()
		{
			WorldObject.TargetTransform = IdToTarget(TargetId);
			WorldObject.AttachedObject._localDisturbancePos = LocalDisturbancePos;
			WorldObject.AttachedObject.ChangeState(Value);
		}

		private static uint TargetToId(Transform transform)
		{
			if (transform == null)
			{
				return uint.MaxValue;
			}

			if (transform == Locator.GetShipTransform())
			{
				return uint.MaxValue - 1;
			}

			return QSBPlayerManager.LocalPlayerId;
		}

		private static Transform IdToTarget(uint id)
		{
			if (id == uint.MaxValue)
			{
				return null;
			}

			if (id == uint.MaxValue - 1)
			{
				return Locator.GetShipTransform();
			}

			return QSBPlayerManager.GetPlayer(id).Body.transform;
		}
	}
}