quantum-space-buddies/QSB/ItemSync/WorldObjects/Items/QSBItem.cs

using QSB.ItemSync.WorldObjects.Sockets;
using QSB.Player;
using QSB.SectorSync.WorldObjects;
using QSB.Utility;
using QSB.WorldSync;
using UnityEngine;

namespace QSB.ItemSync.WorldObjects.Items
{
	public class QSBItem : WorldObject<OWItem>
	{
		public QSBItemSocket InitialSocket { get; private set; }
		public Transform InitialParent { get; private set; }
		public Vector3 InitialPosition { get; private set; }
		public Quaternion InitialRotation { get; private set; }
		public QSBSector InitialSector { get; private set; }

		public override void Init()
		{
			if (AttachedObject == null)
			{
				DebugLog.ToConsole($"Error - AttachedObject is null! Type:{GetType().Name}", OWML.Common.MessageType.Error);
				return;
			}

			StartDelayedReady();
			QSBCore.UnityEvents.RunWhen(() => QSBWorldSync.AllObjectsAdded, () =>
			{
				FinishDelayedReady();

				InitialParent = AttachedObject.transform.parent;
				InitialPosition = AttachedObject.transform.localPosition;
				InitialRotation = AttachedObject.transform.localRotation;
				var initialSector = AttachedObject.GetSector();
				if (initialSector != null)
				{
					InitialSector = initialSector.GetWorldObject<QSBSector>();
				}

				if (InitialParent == null)
				{
					DebugLog.ToConsole($"Warning - InitialParent of {AttachedObject.name} is null!", OWML.Common.MessageType.Warning);
				}

				if (InitialParent?.GetComponent<OWItemSocket>() != null)
				{
					var qsbObj = InitialParent.GetComponent<OWItemSocket>().GetWorldObject<QSBItemSocket>();
					InitialSocket = qsbObj;
				}
			});

			QSBPlayerManager.OnRemovePlayer += OnPlayerLeave;
		}

		public override void OnRemoval() => QSBPlayerManager.OnRemovePlayer -= OnPlayerLeave;

		private void OnPlayerLeave(PlayerInfo player)
		{
			if (player.HeldItem != this)
			{
				return;
			}

			if (InitialSocket != null)
			{
				InitialSocket.PlaceIntoSocket(this);
				return;
			}

			AttachedObject.transform.parent = InitialParent;
			AttachedObject.transform.localPosition = InitialPosition;
			AttachedObject.transform.localRotation = InitialRotation;
			AttachedObject.transform.localScale = Vector3.one;
			AttachedObject.SetSector(InitialSector?.AttachedObject);
			AttachedObject.SetColliderActivation(true);
		}

		public override void SendInitialState(uint to)
		{
			// todo SendResyncInfo
		}

		public ItemType GetItemType()
			=> AttachedObject.GetItemType();

		public void PickUpItem(Transform holdTransform)
			=> AttachedObject.PickUpItem(holdTransform);

		public void DropItem(Vector3 position, Vector3 normal, Sector sector)
		{
			AttachedObject.transform.SetParent(sector.transform);
			AttachedObject.transform.localScale = Vector3.one;
			var localDropNormal = AttachedObject._localDropNormal;
			var lhs = Quaternion.FromToRotation(AttachedObject.transform.TransformDirection(localDropNormal), normal);
			AttachedObject.transform.rotation = lhs * AttachedObject.transform.rotation;
			var localDropOffset = AttachedObject._localDropOffset;
			AttachedObject.transform.position = sector.transform.TransformPoint(position) + AttachedObject.transform.TransformDirection(localDropOffset);
			AttachedObject.SetSector(sector);
			AttachedObject.SetColliderActivation(true);
		}

		public void OnCompleteUnsocket()
			=> AttachedObject.OnCompleteUnsocket();
	}
}