quantum-space-buddies/QSB/Tools/ProbeTool/Messages/ProbeStartRetrieveEvent.cs

using QSB.Events;
using QSB.Messaging;
using QSB.Player;

namespace QSB.Tools.ProbeTool.Messages
{
	internal class ProbeStartRetrieveEvent : QSBEvent<FloatMessage>
	{
		public override bool RequireWorldObjectsReady => true;

		public override void SetupListener()
			=> GlobalMessenger<float>.AddListener(EventNames.QSBProbeStartRetrieve, Handler);

		public override void CloseListener()
			=> GlobalMessenger<float>.RemoveListener(EventNames.QSBProbeStartRetrieve, Handler);

		private void Handler(float duration) => SendEvent(CreateMessage(duration));

		private FloatMessage CreateMessage(float duration) => new()
		{
			AboutId = LocalPlayerId,
			Value = duration
		};

		public override void OnReceiveRemote(bool server, FloatMessage message)
		{
			var player = QSBPlayerManager.GetPlayer(message.AboutId);
			if (!player.IsReady || player.Probe == null)
			{
				return;
			}

			var probe = player.Probe;
			probe.OnStartRetrieve(message.Value);
		}
	}
}