RizzeBattleRoyale/Assets/TPSBR/Scripts/Core/Scene.cs

using System.Collections;
using System.Collections.Generic;
using UnityEngine;
using Fusion;
using TPSBR.UI;

namespace TPSBR
{
	[System.Serializable]
	public class SceneContext
	{
		// Player

		[HideInInspector]
		public string           PeerUserID;
		[HideInInspector]
		public PlayerData       PlayerData;

		// General

		public SceneAudio       Audio;
		public SceneUI          UI;
		public ObjectCache      ObjectCache;
		public SceneInput       Input;
		public SceneCamera      Camera;
		public NetworkGame      NetworkGame;

		[HideInInspector]
		public GlobalSettings   Settings;
		[HideInInspector]
		public RuntimeSettings  RuntimeSettings;

		// Gameplay

		public Announcer        Announcer;

		[HideInInspector]
		public bool             IsVisible;
		[HideInInspector]
		public bool             HasInput;

		[HideInInspector]
		public NetworkRunner    Runner;
		[HideInInspector]
		public GameplayMode     GameplayMode;
		[HideInInspector]
		public PlayerRef        LocalPlayerRef;
		[HideInInspector]
		public PlayerRef        ObservedPlayerRef;
		[HideInInspector]
		public Agent            ObservedAgent;
		[HideInInspector]
		public Transform        WaitingAgentTransform;

		[HideInInspector]
		public SceneMap         Map;

		// Server Gameplay

		public Backfill         Backfill;

		// Menu

		public PlayerPreview    PlayerPreview;
		public Matchmaking      Matchmaking;
		public Matchmaker       Matchmaker;
	}

	public class Scene : CoreBehaviour
	{
		// PUBLIC MEMBERS

		public bool         ContextReady { get; private set; }
		public bool         IsActive     { get; private set; }
		public SceneContext Context      => _context;

		// PRIVATE MEMBERS

		[SerializeField]
		private bool _selfInitialize;
		[SerializeField]
		private SceneContext _context;

		private bool _isInitialized;
		private List<SceneService> _services = new List<SceneService>();

		// PUBLIC METHODS

		public void PrepareContext()
		{
			OnPrepareContext(_context);
		}

		public void Initialize()
		{
			if (_isInitialized == true)
				return;

			if (ContextReady == false)
			{
				OnPrepareContext(_context);
			}

			OnInitialize();

			_isInitialized = true;
		}

		public IEnumerator Activate()
		{
			if (_isInitialized == false)
				yield break;

			yield return OnActivate();

			IsActive = true;
		}

		public void Deactivate()
		{
			if (IsActive == false)
				return;

			OnDeactivate();

			IsActive = false;
		}

		public void Deinitialize()
		{
			if (_isInitialized == false)
				return;

			Deactivate();

			OnDeinitialize();

			ContextReady = false;
			_isInitialized = false;
		}

		public T GetService<T>() where T : SceneService
		{
			for (int i = 0, count = _services.Count; i < count; i++)
			{
				if (_services[i] is T service)
					return service;
			}

			return null;
		}

		public void Quit()
		{
			Deinitialize();

#if UNITY_EDITOR
			UnityEditor.EditorApplication.isPlaying = false;
#else
			Application.Quit();
#endif
		}

		// MONOBEHAVIOUR

		protected void Awake()
		{
			if (_selfInitialize == true)
			{
				Initialize();
			}
		}

		protected IEnumerator Start()
		{
			if (_isInitialized == false)
				yield break;

			if (_selfInitialize == true && IsActive == false)
			{
				// UI cannot be initialized in Awake, Canvas elements need to Awake first
				AddService(_context.UI);

				yield return Activate();
			}
		}

		protected virtual void Update()
		{
			if (IsActive == false)
				return;

			OnTick();
		}

		protected virtual void LateUpdate()
		{
			if (IsActive == false)
				return;

			OnLateTick();
		}

		protected void OnDestroy()
		{
			Deinitialize();
		}

		protected void OnApplicationQuit()
		{
			Deinitialize();
		}

		// PROTECTED METHODS

		protected virtual void OnPrepareContext(SceneContext context)
		{
			context.PlayerData      = Global.PlayerService.PlayerData;
			context.Settings        = Global.Settings;
			context.RuntimeSettings = Global.RuntimeSettings;

			context.HasInput        = true;
			context.IsVisible       = true;

			ContextReady = true;
		}

		protected virtual void OnInitialize()
		{
			CollectServices();
		}

		protected virtual IEnumerator OnActivate()
		{
			for (int i = 0; i < _services.Count; i++)
			{
				_services[i].Activate();
			}

			yield break;
		}

		protected virtual void OnTick()
		{
			for (int i = 0, count = _services.Count; i < count; i++)
			{
				_services[i].Tick();
			}
		}

		protected virtual void OnLateTick()
		{
			for (int i = 0, count = _services.Count; i < count; i++)
			{
				_services[i].LateTick();
			}
		}

		protected virtual void OnDeactivate()
		{
			for (int i = 0; i < _services.Count; i++)
			{
				_services[i].Deactivate();
			}
		}

		protected virtual void OnDeinitialize()
		{
			for (int i = 0; i < _services.Count; i++)
			{
				_services[i].Deinitialize();
			}

			_services.Clear();
		}

		protected virtual void CollectServices()
		{
			var services = GetComponentsInChildren<SceneService>(true);

			foreach (var service in services)
			{
				AddService(service);
			}
		}

		protected void AddService(SceneService service)
		{
			if (service == null)
			{
				Debug.LogError($"Missing service");
				return;
			}

			if (_services.Contains(service) == true)
			{
				Debug.LogError($"Service {service.gameObject.name} already added.");
				return;
			}

			service.Initialize(this, Context);

			_services.Add(service);
		}
	}
}