quantum-space-buddies/QSB/Player/PlayerInfoParts/Tools.cs

using OWML.Common;
using QSB.CampfireSync.WorldObjects;
using QSB.ItemSync.WorldObjects.Items;
using QSB.RoastingSync;
using QSB.Tools;
using QSB.Tools.FlashlightTool;
using QSB.Tools.ProbeLauncherTool;
using QSB.Tools.ProbeLauncherTool.WorldObjects;
using QSB.Tools.ProbeTool;
using QSB.Utility;
using UnityEngine;

namespace QSB.Player;

public partial class PlayerInfo
{
	public GameObject ProbeBody { get; set; }
	public QSBSurveyorProbe Probe { get; set; }
	public QSBFlashlight FlashLight => CameraBody == null ? null : CameraBody.GetComponentInChildren<QSBFlashlight>();
	public QSBTool Signalscope => GetToolByType(ToolType.Signalscope);
	public QSBTool Translator => GetToolByType(ToolType.Translator);
	public QSBProbeLauncherTool ProbeLauncherTool => (QSBProbeLauncherTool)GetToolByType(ToolType.ProbeLauncher);
	private Transform _handPivot;
	public Transform HandPivot
	{
		get
		{
			if (_handPivot == null)
			{
				_handPivot = Body.transform.Find(
					// TODO : kill me for my sins
					"REMOTE_Traveller_HEA_Player_v2/" +
					"Traveller_Rig_v01:Traveller_Trajectory_Jnt/" +
					"Traveller_Rig_v01:Traveller_ROOT_Jnt/" +
					"Traveller_Rig_v01:Traveller_Spine_01_Jnt/" +
					"Traveller_Rig_v01:Traveller_Spine_02_Jnt/" +
					"Traveller_Rig_v01:Traveller_Spine_Top_Jnt/" +
					"Traveller_Rig_v01:Traveller_RT_Arm_Clavicle_Jnt/" +
					"Traveller_Rig_v01:Traveller_RT_Arm_Shoulder_Jnt/" +
					"Traveller_Rig_v01:Traveller_RT_Arm_Elbow_Jnt/" +
					"Traveller_Rig_v01:Traveller_RT_Arm_Wrist_Jnt/" +
					"REMOTE_ItemCarryTool/" +
					"HandPivot"
				);
			}

			return _handPivot;
		}
	}
	public Transform ItemSocket => GetSocket("REMOTE_ItemSocket");
	public Transform ScrollSocket => GetSocket("REMOTE_ScrollSocket");
	public Transform SharedStoneSocket => GetSocket("REMOTE_SharedStoneSocket");
	public Transform WarpCoreSocket => GetSocket("REMOTE_WarpCoreSocket");
	public Transform VesselCoreSocket => GetSocket("REMOTE_VesselCoreSocket");
	public Transform SimpleLanternSocket => GetSocket("REMOTE_SimpleLanternSocket");
	public Transform DreamLanternSocket => GetSocket("REMOTE_DreamLanternSocket");
	public Transform SlideReelSocket => GetSocket("REMOTE_SlideReelSocket");
	public Transform VisionTorchSocket => GetSocket("REMOTE_VisionTorchSocket");

	private Transform GetSocket(string name)
	{
		var handSocket = HandPivot.Find(name);
		if (handSocket != null)
		{
			return handSocket;
		}

		var cameraSocket = CameraBody.transform.Find("REMOTE_ItemCarryTool").Find(name);
		if (cameraSocket != null)
		{
			DebugLog.ToConsole($"Warning - Could not find hand socket for socket name {name}, defaulting to camera socket.", MessageType.Warning);
			return cameraSocket;
		}

		DebugLog.ToConsole($"Error - Could not find hand socket or camera socket for socket name {name}.", MessageType.Error);
		return null;
	}

	public QSBMarshmallow Marshmallow { get; set; }
	public QSBCampfire Campfire { get; set; }
	public IQSBItem HeldItem { get; set; }
	public bool FlashlightActive { get; set; }
	public bool SuitedUp { get; set; }
	public bool LocalProbeLauncherEquipped { get; set; }
	public bool SignalscopeEquipped { get; set; }
	public bool TranslatorEquipped { get; set; }
	public bool ProbeActive { get; set; }
	public GameObject RoastingStick { get; set; }
	public QSBProbeLauncher ProbeLauncherEquipped { get; set; }
	public bool IsTranslating { get; set; }
}