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

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

namespace QSB.OrbSync.TransformSync;

public class NomaiOrbTransformSync : UnsectoredTransformSync, ILinkedNetworkBehaviour<QSBOrb>
{
	/// <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() => AttachedTransform && base.CheckValid();

	protected override bool UseInterpolation => true;
	protected override float DistanceChangeThreshold => 1f;

	public QSBOrb WorldObject { get; private set; }
	public void LinkTo(IWorldObject worldObject) => WorldObject = (QSBOrb)worldObject;

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


	private static readonly List<NomaiOrbTransformSync> _instances = new();

	public override void OnStartClient()
	{
		if (QSBCore.IsHost)
		{
			netIdentity.RegisterAuthQueue();
		}

		base.OnStartClient();
	}

	public override void OnStopClient()
	{
		if (QSBCore.IsHost)
		{
			netIdentity.UnregisterAuthQueue();
		}

		base.OnStopClient();
	}

	protected override void Init()
	{
		base.Init();
		var body = AttachedTransform.GetAttachedOWRigidbody();
		SetReferenceTransform(body.GetOrigParent());

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

	protected override void Uninit()
	{
		base.Uninit();

		// this is null sometimes on here, but not on other similar transforms syncs (like anglers)
		// idk why, but whatever
		if (AttachedTransform)
		{
			var body = AttachedTransform.GetAttachedOWRigidbody();
			if (body)
			{
				body.OnUnsuspendOWRigidbody -= OnUnsuspend;
				body.OnSuspendOWRigidbody -= OnSuspend;
			}
		}
	}

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

	protected override void ApplyToAttached()
	{
		base.ApplyToAttached();

		WorldObject.AttachedObject.SetTargetPosition(AttachedTransform.position);
	}
}