quantum-space-buddies/QSB/EyeOfTheUniverse/InstrumentSync/Patches/QuantumInstrumentPatches.cs

using HarmonyLib;
using QSB.Events;
using QSB.EyeOfTheUniverse.InstrumentSync.WorldObjects;
using QSB.Patches;
using QSB.WorldSync;
using UnityEngine;

namespace QSB.EyeOfTheUniverse.InstrumentSync.Patches
{
	internal class QuantumInstrumentPatches : QSBPatch
	{
		public override QSBPatchTypes Type => QSBPatchTypes.OnClientConnect;

		[HarmonyPostfix]
		[HarmonyPatch(typeof(QuantumInstrument), nameof(QuantumInstrument.OnPressInteract))]
		public static void OnPressInteract(QuantumInstrument __instance)
			=> QSBEventManager.FireEvent(EventNames.QSBGatherInstrument, QSBWorldSync.GetWorldFromUnity<QSBQuantumInstrument>(__instance));

		[HarmonyPrefix]
		[HarmonyPatch(typeof(QuantumInstrument), nameof(QuantumInstrument.Update))]
		public static bool Update(QuantumInstrument __instance)
		{
			if (__instance._gatherWithScope && !__instance._waitToFlickerOut)
			{
				__instance._scopeGatherPrompt.SetVisibility(false);
				if (Locator.GetToolModeSwapper().GetSignalScope().InZoomMode()
					&& Vector3.Angle(__instance.transform.position - Locator.GetPlayerCamera().transform.position, Locator.GetPlayerCamera().transform.forward) < 1f)
				{
					__instance._scopeGatherPrompt.SetVisibility(true);
					if (OWInput.IsNewlyPressed(InputLibrary.interact, InputMode.All))
					{
						__instance.Gather();
						QSBEventManager.FireEvent(EventNames.QSBGatherInstrument, QSBWorldSync.GetWorldFromUnity<QSBQuantumInstrument>(__instance));
						Locator.GetPromptManager().RemoveScreenPrompt(__instance._scopeGatherPrompt);
					}
				}
			}

			if (__instance._waitToFlickerOut && Time.time > __instance._flickerOutTime)
			{
				__instance.FinishGather();
			}

			return false;
		}
	}
}