quantum-space-buddies/QSB/OrbSync/TransformSync/NomaiOrbTransformSync.cs

using OWML.Common;
using QSB.AuthoritySync;
using QSB.OrbSync.WorldObjects;
using QSB.Syncs.Unsectored.Transforms;
using QSB.Utility;
using QSB.WorldSync;
using System.Collections.Generic;
using UnityEngine;

namespace QSB.OrbSync.TransformSync
{
	public class NomaiOrbTransformSync : UnsectoredTransformSync
	{
		/// <summary>
		/// normally prints error when attached object is null.
		/// this overrides it so that doesn't happen, since the orb can be destroyed.
		/// </summary>
		protected override bool CheckValid() => _attachedBody && base.CheckValid();
		protected override bool UseInterpolation => true;
		protected override float DistanceLeeway => 1f;

		protected override Transform InitLocalTransform() => _qsbOrb.AttachedObject.transform;
		protected override Transform InitRemoteTransform() => _qsbOrb.AttachedObject.transform;

		private OWRigidbody _attachedBody;
		private QSBOrb _qsbOrb;
		private static readonly List<NomaiOrbTransformSync> _instances = new();

		public override void OnStartClient()
		{
			_instances.Add(this);
			base.OnStartClient();
		}

		public override void OnStopClient()
		{
			_instances.Remove(this);
			base.OnStopClient();
		}

		protected override void Init()
		{
			var index = _instances.IndexOf(this);
			if (!OrbManager.Orbs.TryGet(index, out var orb))
			{
				DebugLog.ToConsole($"Error - No orb at index {index}.", MessageType.Error);
				return;
			}

			_qsbOrb = orb.GetWorldObject<QSBOrb>();
			_qsbOrb.TransformSync = this;

			base.Init();
			_attachedBody = AttachedTransform.GetAttachedOWRigidbody();
			SetReferenceTransform(_attachedBody.GetOrigParent());

			if (QSBCore.IsHost)
			{
				netIdentity.RegisterAuthQueue();
			}

			_attachedBody.OnUnsuspendOWRigidbody += OnUnsuspend;
			_attachedBody.OnSuspendOWRigidbody += OnSuspend;
			netIdentity.SendAuthQueueMessage(_attachedBody.IsSuspended() ? AuthQueueAction.Remove : AuthQueueAction.Add);
		}

		protected override void Uninit()
		{
			if (QSBCore.IsHost)
			{
				netIdentity.UnregisterAuthQueue();
			}

			_attachedBody.OnUnsuspendOWRigidbody -= OnUnsuspend;
			_attachedBody.OnSuspendOWRigidbody -= OnSuspend;
			_attachedBody = null;

			base.Uninit();
		}

		private void OnUnsuspend(OWRigidbody suspendedBody) => netIdentity.SendAuthQueueMessage(AuthQueueAction.Add);
		private void OnSuspend(OWRigidbody suspendedBody) => netIdentity.SendAuthQueueMessage(AuthQueueAction.Remove);
	}
}