quantum-space-buddies/QSB/Tools/ProbeTool/QSBSurveyorProbe.cs

using QSB.Player;
using QSB.Utility;
using UnityEngine;

namespace QSB.Tools.ProbeTool;

[UsedInUnityProject]
public class QSBSurveyorProbe : MonoBehaviour, ILightSource
{
	public delegate void SurveyorProbeEvent();
	public delegate void RetrieveEvent(float retrieveLength);

	[SerializeField]
	private OWLight2[] _illuminationCheckLights;

	public event SurveyorProbeEvent OnLaunchProbe;
	public event SurveyorProbeEvent OnAnchorProbe;
	public event SurveyorProbeEvent OnUnanchorProbe;
	public event SurveyorProbeEvent OnRetrieveProbe;
	public event SurveyorProbeEvent OnProbeDestroyed;
	public event SurveyorProbeEvent OnTakeSnapshot;
	public event RetrieveEvent OnStartRetrieveProbe;

	private GameObject _detectorObj;
	private RulesetDetector _rulesetDetector;
	private SingularityWarpEffect _warpEffect;
	private QSBProbeCamera _forwardCam;
	private QSBProbeCamera _reverseCam;
	private QSBProbeCamera _rotatingCam;
	private bool _isRetrieving;
	private PlayerInfo _owner;
	private bool _anchored;
	private LightSourceVolume _lightSourceVol;

	public RulesetDetector GetRulesetDetector()
		=> _rulesetDetector;

	public QSBProbeCamera GetForwardCamera()
	{
		return _forwardCam;
	}

	public QSBProbeCamera GetReverseCamera()
	{
		return _reverseCam;
	}

	public QSBProbeCamera GetRotatingCamera()
	{
		return _rotatingCam;
	}

	private void Awake()
	{
		_detectorObj = GetComponentInChildren<RulesetDetector>().gameObject;
		_rulesetDetector = _detectorObj.GetComponent<RulesetDetector>();
		_warpEffect = GetComponentInChildren<SingularityWarpEffect>();
		_warpEffect.OnWarpComplete += OnWarpComplete;
		_isRetrieving = false;

		var probeCameras = GetComponentsInChildren<QSBProbeCamera>();
		for (var i = 0; i < probeCameras.Length; i++)
		{
			if (probeCameras[i].GetID() == ProbeCamera.ID.Forward)
			{
				_forwardCam = probeCameras[i];
			}
			else if (probeCameras[i].GetID() == ProbeCamera.ID.Reverse)
			{
				_reverseCam = probeCameras[i];
			}
			else if (probeCameras[i].GetID() == ProbeCamera.ID.Rotating)
			{
				_rotatingCam = probeCameras[i];
			}
		}
	}

	private void Start()
	{
		_lightSourceVol = this.GetRequiredComponentInChildren<LightSourceVolume>();
		_lightSourceVol.LinkLightSource(this);

		if (_owner == null)
		{
			// i dont *think* this is possible, but honestly i dont fucking know anymore
			DebugLog.DebugWrite($"Warning - QSBProbe ran Start() without an assigned owner!", OWML.Common.MessageType.Warning);
			_lightSourceVol.SetVolumeActivation(false);
			gameObject.SetActive(false);
		}

		if (_owner.ProbeActive)
		{
			_lightSourceVol.SetVolumeActivation(true);
			gameObject.SetActive(true);
		}
		else
		{
			_lightSourceVol.SetVolumeActivation(false);
			gameObject.SetActive(false);
		}
	}

	protected void OnDestroy() => _warpEffect.OnWarpComplete -= OnWarpComplete;

	public void SetOwner(PlayerInfo player)
	{
		if (_owner != null)
		{
			DebugLog.ToConsole($"Warning - Trying to set owner of probe that already has an owner!", OWML.Common.MessageType.Warning);
		}

		_owner = player;
	}

	private void OnWarpComplete() => Deactivate();

	public bool IsRetrieving()
		=> IsLaunched() && _isRetrieving;

	public bool IsLaunched()
		=> gameObject.activeSelf;

	public bool IsAnchored()
		=> IsLaunched() && _anchored;

	public void HandleEvent(ProbeEvent probeEvent)
	{
		if (_owner == null)
		{
			DebugLog.ToConsole($"Error - Trying to handle event on probe with no owner.", OWML.Common.MessageType.Error);
			return;
		}

		switch (probeEvent)
		{
			case ProbeEvent.Launch:
				_anchored = false;
				_owner.ProbeActive = true;
				gameObject.SetActive(true);
				_lightSourceVol.SetVolumeActivation(true);
				transform.position = _owner.ProbeLauncherTool.transform.position;
				transform.rotation = _owner.ProbeLauncherTool.transform.rotation;

				if (OnLaunchProbe == null)
				{
					DebugLog.ToConsole($"Warning - OnLaunchProbe is null!", OWML.Common.MessageType.Warning);
					break;
				}

				OnLaunchProbe();
				break;
			case ProbeEvent.Anchor:
				_anchored = true;
				_owner.ProbeActive = true; // just in case it was missed
				if (OnAnchorProbe == null)
				{
					DebugLog.ToConsole($"Warning - OnAnchorProbe is null!", OWML.Common.MessageType.Warning);
					break;
				}

				OnAnchorProbe();
				break;
			case ProbeEvent.Unanchor:
				_anchored = false;
				_owner.ProbeActive = true; // just in case it was missed
				_forwardCam.SetSandLevelController(null);
				_reverseCam.SetSandLevelController(null);
				_rotatingCam.SetSandLevelController(null);
				OnUnanchorProbe();
				break;
			case ProbeEvent.Retrieve:
				_anchored = false;
				_owner.ProbeActive = false;
				if (OnRetrieveProbe == null)
				{
					DebugLog.ToConsole($"Warning - OnRetrieveProbe is null!", OWML.Common.MessageType.Warning);
					break;
				}

				OnRetrieveProbe();
				break;
			case ProbeEvent.Destroy:
				_anchored = false;
				Destroy(gameObject);
				_owner.ProbeActive = false;
				if (OnProbeDestroyed == null)
				{
					DebugLog.ToConsole($"Warning - OnProbeDestroyed is null!", OWML.Common.MessageType.Warning);
					break;
				}

				OnProbeDestroyed();
				break;
			case ProbeEvent.Invalid:
			default:
				DebugLog.ToConsole($"Warning - Unknown/Invalid probe event.", OWML.Common.MessageType.Warning);
				break;
		}
	}

	private void Deactivate()
	{
		transform.localScale = Vector3.one;
		_lightSourceVol.SetVolumeActivation(false);
		gameObject.SetActive(false);
		_isRetrieving = false;
		_forwardCam.SetSandLevelController(null);
		_reverseCam.SetSandLevelController(null);
		_rotatingCam.SetSandLevelController(null);
	}

	public void OnStartRetrieve(float duration)
	{
		if (!_isRetrieving)
		{
			_isRetrieving = true;
			_warpEffect.WarpObjectOut(duration);

			if (OnStartRetrieveProbe == null)
			{
				DebugLog.ToConsole($"Warning - OnStartRetrieveProbe is null!", OWML.Common.MessageType.Warning);
				return;
			}

			OnStartRetrieveProbe(duration);
		}
		else
		{
			DebugLog.DebugWrite($"Warning - Tried to retrieve probe (owner: {_owner}) that was already retrieving?", OWML.Common.MessageType.Warning);
		}
	}

	public bool CheckIlluminationAtPoint(Vector3 point, float buffer = 0f, float maxDistance = float.PositiveInfinity)
	{
		for (var i = 0; i < _illuminationCheckLights.Length; i++)
		{
			if (_illuminationCheckLights[i].CheckIlluminationAtPoint(point, buffer, maxDistance))
			{
				return true;
			}
		}

		return false;
	}

	public LightSourceType GetLightSourceType() => LightSourceType.PROBE;
	public OWLight2[] GetLights() => _illuminationCheckLights;
	public Vector3 GetLightSourcePosition() => _lightSourceVol.transform.position;

	public void TakeSnapshot() => OnTakeSnapshot?.Invoke();
}