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

using OWML.Common;
using QSB.SectorSync;
using QSB.Syncs.Sectored.Transforms;
using QSB.Tools.ProbeLauncherTool;
using QSB.Utility;
using UnityEngine;

namespace QSB.Tools.ProbeTool.TransformSync
{
	public class PlayerProbeSync : SectoredTransformSync
	{
		protected override float DistanceLeeway => 10f;
		protected override bool UseInterpolation => true;
		protected override bool AllowDisabledAttachedObject => true;
		protected override bool IsPlayerObject => true;

		public static PlayerProbeSync LocalInstance { get; private set; }

		public override void OnStartAuthority() => LocalInstance = this;

		protected override Transform InitLocalTransform()
		{
			SectorSync.Init(Locator.GetProbe().GetSectorDetector(), TargetType.Probe);

			var body = Locator.GetProbe().transform;
			Player.ProbeBody = body.gameObject;

			if (Player.Body == null)
			{
				DebugLog.ToConsole($"Warning - Player.Body is null!", MessageType.Warning);
				return null;
			}

			var listener = Player.Body.AddComponent<ProbeListener>();
			listener.Init(Locator.GetProbe());

			var launcherListener = Player.Body.AddComponent<ProbeLauncherListener>();
			launcherListener.Init(Player.LocalProbeLauncher);

			return body;
		}

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

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

			var body = probe.gameObject.activeSelf
				? probe.InstantiateInactive()
				: Instantiate(probe);

			body.name = "RemoteProbeTransform";

			ProbeCreator.CreateProbe(body, Player);

			Player.ProbeBody = body.gameObject;

			return body;
		}

		protected override void GetFromAttached()
		{
			if (AttachedTransform.gameObject.activeInHierarchy)
			{
				base.GetFromAttached();
				return;
			}

			var probeOWRigidbody = Locator.GetProbe().GetOWRigidbody();
			if (probeOWRigidbody == null)
			{
				DebugLog.ToConsole($"Warning - Could not find OWRigidbody of local probe.", MessageType.Warning);
			}

			var probeLauncher = Player.LocalProbeLauncher;
			// TODO : make this sync to the *active* probe launcher's _launcherTransform
			var launcherTransform = probeLauncher._launcherTransform;
			probeOWRigidbody.SetPosition(launcherTransform.position);
			probeOWRigidbody.SetRotation(launcherTransform.rotation);

			base.GetFromAttached();

			var currentReferenceSector = ReferenceSector;
			var playerReferenceSector = Player.TransformSync.ReferenceSector;

			if (currentReferenceSector != playerReferenceSector)
			{
				SetReferenceSector(playerReferenceSector);
			}
		}

		protected override bool IsReady => AttachedTransform != null || Locator.GetProbe() != null;
	}
}