quantum-space-buddies/QSB/Player/PlayerInfo.cs

using OWML.Common;
using QSB.Animation.Player;
using QSB.API.Messages;
using QSB.Audio;
using QSB.ClientServerStateSync;
using QSB.HUD;
using QSB.Messaging;
using QSB.ModelShip;
using QSB.Patches;
using QSB.Player.Messages;
using QSB.Player.TransformSync;
using QSB.QuantumSync.WorldObjects;
using QSB.ShipSync;
using QSB.Tools;
using QSB.Utility;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;

namespace QSB.Player;

public partial class PlayerInfo
{
	/// <summary>
	/// the player transform sync's net id
	/// </summary>
	public uint PlayerId { get; }
	public string Name { get; set; }
	public PlayerHUDMarker HudMarker { get; set; }
	public PlayerMapMarker MapMarker { get; set; }
	public PlayerBox HUDBox { get; set; }
	public Transform MinimapPlayerMarker { get; set; }
	public PlayerTransformSync TransformSync { get; }
	public ClientState State { get; set; }
	public EyeState EyeState { get; set; }
	public bool IsDead { get; set; }
	public bool IsReady { get; set; }
	public bool IsInMoon { get; set; }
	public bool IsInShrine { get; set; }
	public bool IsInEyeShuttle { get; set; }
	public bool IsInShip { get; set; }
	public bool IsInCloak { get; set; }
	public IQSBQuantumObject EntangledObject { get; set; }
	public QSBPlayerAudioController AudioController { get; set; }
	public bool IsLocalPlayer => TransformSync.isLocalPlayer; // if TransformSync is ever null, i give permission for nebula to make fun of me about it for the rest of time - johncorby
	public ThrusterLightTracker ThrusterLightTracker;
	public bool FlyingShip => ShipManager.Instance.CurrentFlyer == PlayerId;
	public bool FlyingModelShip => ModelShipManager.Instance.CurrentFlyer == PlayerId;
	public RemotePlayerRulesetDetector RulesetDetector { get; set; }

	public PlayerInfo(PlayerTransformSync transformSync)
	{
		PlayerId = transformSync.netId;
		TransformSync = transformSync;
		AnimationSync = transformSync.GetComponent<AnimationSync>();
	}

	/// <summary>
	/// called on player transform sync uninit.
	/// (BOTH local and non-local)
	/// </summary>
	public void Reset()
	{
		if (AnimationSync != null)
		{
			AnimationSync.Reset();
		}

		EyeState = default;
		IsDead = default;
		IsReady = default;
		IsInMoon = default;
		IsInShrine = default;
		IsInEyeShuttle = default;
		EntangledObject = default;

		CurrentCharacterDialogueTree = default;

		InDreamWorld = default;
		AssignedSimulationLantern = default;

		Campfire = default;
		HeldItem = default;
		FlashlightActive = default;
		SuitedUp = default;
		LocalProbeLauncherEquipped = default;
		SignalscopeEquipped = default;
		TranslatorEquipped = default;
		ProbeActive = default;
		ProbeLauncherEquipped = default;
		IsTranslating = default;
	}

	public void UpdateObjectsFromStates()
	{
		FlashLight.UpdateState(FlashlightActive && Visible);
		Translator.ChangeEquipState(TranslatorEquipped);
		ProbeLauncherTool.ChangeEquipState(LocalProbeLauncherEquipped);
		Signalscope.ChangeEquipState(SignalscopeEquipped);
		AnimationSync.SetSuitState(SuitedUp);
	}

	public void UpdateStatesFromObjects()
	{
		if (Locator.GetFlashlight() == null || Locator.GetPlayerBody() == null)
		{
			FlashlightActive = false;
			SuitedUp = false;
		}
		else
		{
			FlashlightActive = Locator.GetFlashlight()._flashlightOn;
			SuitedUp = Locator.GetPlayerBody().GetComponent<PlayerSpacesuit>().IsWearingSuit()
				|| QSBSceneManager.CurrentScene == OWScene.EyeOfTheUniverse;
		}

		new PlayerInformationMessage().Send();
	}

	private QSBTool GetToolByType(ToolType type)
	{
		if (CameraBody == null)
		{
			DebugLog.ToConsole($"Warning - Tried to GetToolByType({type}) on player {PlayerId}, but CameraBody was null.", MessageType.Warning);
			return null;
		}

		var tools = CameraBody.GetComponentsInChildren<QSBTool>();

		if (tools == null || tools.Length == 0)
		{
			DebugLog.ToConsole($"Warning - Couldn't find any QSBTools for player {PlayerId}.", MessageType.Warning);
			return null;
		}

		var tool = tools.FirstOrDefault(x => x.Type == type);

		if (tool == null)
		{
			DebugLog.ToConsole($"Warning - No tool found on player {PlayerId} matching ToolType {type}.", MessageType.Warning);
		}

		return tool;
	}

	public void SetVisible(bool visible, float seconds = 0)
	{
		if (IsLocalPlayer)
		{
			return;
		}

		if (!_ditheringAnimator)
		{
			DebugLog.ToConsole($"Warning - {PlayerId}.DitheringAnimator is null!", MessageType.Warning);
			return;
		}

		_ditheringAnimator.SetVisible(visible, seconds);
		if (!visible)
		{
			FlashLight.UpdateState(false);
		}
		else
		{
			FlashLight.UpdateState(FlashlightActive);
		}
	}

	public void Die()
	{
		IsDead = true;
		SetVisible(false, 1);
		HUDBox.OnDeath();
	}

	public void Revive()
	{
		IsDead = false;
		SetVisible(true, 1);
		HUDBox.OnRespawn();
	}

	// internal for RequestStateResyncMessage
	internal readonly Dictionary<string, object> _customData = new();

	public void SetCustomData<T>(string key, T data)
	{
		_customData[key] = data;

		if (!QSBPatch.Remote && typeof(T).IsSerializable)
		{
			new AddonCustomDataSyncMessage(PlayerId, key, data).Send();
		}
	}

	public T GetCustomData<T>(string key)
	{
		if (!_customData.TryGetValue(key, out var value))
		{
			return default;
		}

		return (T)value;
	}

	public override string ToString() => $"{PlayerId}:{GetType().Name} ({Name})";
}