quantum-space-buddies/QSB/TransformSync/PlayerProbeSync.cs

using OWML.Common;
using QSB.Player;
using QSB.Tools;
using QSB.Utility;
using System.Linq;
using UnityEngine;

namespace QSB.TransformSync
{
	public class PlayerProbeSync : TransformSync
	{
		private Transform _disabledSocket;

		protected void Start()
		{
			var lowestBound = QSBPlayerManager.GetSyncObjects<PlayerTransformSync>().Where(x => x.NetId.Value < NetId.Value).OrderBy(x => x.NetId.Value).Last();
			NetIdentity.SetRootIdentity(lowestBound.NetIdentity);
		}

		private Transform GetProbe()
			=> Locator.GetProbe().transform.Find("CameraPivot").Find("Geometry");

		protected override Transform InitLocalTransform()
		{
			var body = GetProbe();

			SetSocket(Player.Camera.transform);
			Player.ProbeBody = body.gameObject;

			return body;
		}

		protected override Transform InitRemoteTransform()
		{
			var probe = GetProbe();

			if (probe == null)
			{
				DebugLog.ToConsole("Error - Probe is null!", MessageType.Error);
				return default;
			}

			var body = probe.InstantiateInactive();
			body.name = "RemoteProbeTransform";

			Destroy(body.GetComponentInChildren<ProbeAnimatorController>());

			PlayerToolsManager.CreateProbe(body, Player);

			QSBCore.Helper.Events.Unity.RunWhen(() => (Player.ProbeLauncher != null), () => SetSocket(Player.ProbeLauncher.ToolGameObject.transform));
			Player.ProbeBody = body.gameObject;

			return body;
		}

		private void SetSocket(Transform socket)
		{
			_disabledSocket = socket;
		}

		protected override void UpdateTransform()
		{
			base.UpdateTransform();
			if (Player == null)
			{
				DebugLog.ToConsole($"Player is null for {AttachedNetId}!", MessageType.Error);
				return;
			}
			if (_disabledSocket == null)
			{
				DebugLog.ToConsole($"DisabledSocket is null for {AttachedNetId}! (ProbeLauncher null? : {Player.ProbeLauncher == null})", MessageType.Error);
				return;
			}
			if (Player.GetState(State.ProbeActive) || ReferenceSector?.Sector == null)
			{
				return;
			}
			if (HasAuthority)
			{
				transform.position = ReferenceSector.Transform.InverseTransformPoint(_disabledSocket.position);
				return;
			}
			if (SyncedTransform.position == Vector3.zero)
			{
				return;
			}
			SyncedTransform.localPosition = ReferenceSector.Transform.InverseTransformPoint(_disabledSocket.position);
		}

		public override bool IsReady => Locator.GetProbe() != null
			&& Player != null
			&& QSBPlayerManager.PlayerExists(Player.PlayerId)
			&& Player.IsReady
			&& NetId.Value != uint.MaxValue
			&& NetId.Value != 0U;
	}
}