quantum-space-buddies/QSB/TriggerSync/WorldObjects/QSBTrigger.cs

using Cysharp.Threading.Tasks;
using QSB.Messaging;
using QSB.Player;
using QSB.TriggerSync.Messages;
using QSB.Utility;
using QSB.WorldSync;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using UnityEngine;

namespace QSB.TriggerSync.WorldObjects
{
	public interface IQSBTrigger : IWorldObject
	{
		List<PlayerInfo> Occupants { get; }

		void Enter(PlayerInfo player);

		void Exit(PlayerInfo player);
	}

	public abstract class QSBTrigger<TO> : WorldObject<OWTriggerVolume>, IQSBTrigger
	{
		public TO TriggerOwner { get; init; }

		public List<PlayerInfo> Occupants { get; } = new();

		protected virtual string CompareTag => "PlayerDetector";

		public override async UniTask Init(CancellationToken ct)
		{
			AttachedObject.OnEntry += OnEnterEvent;
			AttachedObject.OnExit += OnExitEvent;

			QSBPlayerManager.OnRemovePlayer += OnPlayerLeave;

			Delay.RunWhen(() => QSBWorldSync.AllObjectsReady, () =>
			{
				if (AttachedObject._trackedObjects != null && AttachedObject._trackedObjects.Any(x => x.CompareTag(CompareTag)))
				{
					((IQSBTrigger)this).SendMessage(new TriggerMessage(true));
				}
			});
		}

		public override void OnRemoval()
		{
			AttachedObject.OnEntry -= OnEnterEvent;
			AttachedObject.OnExit -= OnExitEvent;

			QSBPlayerManager.OnRemovePlayer -= OnPlayerLeave;
		}

		public override void SendInitialState(uint to) =>
			((IQSBTrigger)this).SendMessage(new TriggerInitialStateMessage(Occupants) { To = to });

		protected void OnEnterEvent(GameObject hitObj)
		{
			if (hitObj.CompareTag(CompareTag))
			{
				((IQSBTrigger)this).SendMessage(new TriggerMessage(true));
			}
		}

		protected void OnExitEvent(GameObject hitObj)
		{
			if (hitObj.CompareTag(CompareTag))
			{
				((IQSBTrigger)this).SendMessage(new TriggerMessage(false));
			}
		}

		private void OnPlayerLeave(PlayerInfo player)
		{
			if (Occupants.Contains(player))
			{
				Exit(player);
			}
		}

		public void Enter(PlayerInfo player)
		{
			if (!Occupants.SafeAdd(player))
			{
				return;
			}

			OnEnter(player);
		}

		public void Exit(PlayerInfo player)
		{
			if (!Occupants.QuickRemove(player))
			{
				return;
			}

			OnExit(player);
		}

		/// <summary>
		/// called when a player enters this trigger
		/// </summary>
		protected virtual void OnEnter(PlayerInfo player) { }

		/// <summary>
		/// called when a player exits this trigger or leaves the game
		/// </summary>
		protected virtual void OnExit(PlayerInfo player) { }
	}
}