quantum-space-buddies/QSB/GeyserSync/GeyserManager.cs

using QSB.Patches;
using QSB.WorldSync;
using UnityEngine;

namespace QSB.GeyserSync
{
	public class GeyserManager : MonoBehaviour
	{
		public static GeyserManager Instance { get; private set; }

		private void Awake()
		{
			Instance = this;
			QSBSceneManager.OnSceneLoaded += OnSceneLoaded;
			QSBPatchManager.OnPatchType += OnPatchType;
		}

		private void OnDestroy()
		{
			QSBSceneManager.OnSceneLoaded -= OnSceneLoaded;
			QSBPatchManager.OnPatchType -= OnPatchType;
		}

		private void OnSceneLoaded(OWScene scene, bool isInUniverse)
		{
			var geyserControllers = Resources.FindObjectsOfTypeAll<GeyserController>();
			for (var id = 0; id < geyserControllers.Length; id++)
			{
				var qsbGeyser = WorldRegistry.GetObject<QSBGeyser>(id) ?? new QSBGeyser();
				qsbGeyser.Init(geyserControllers[id], id);
				WorldRegistry.AddObject(qsbGeyser);
			}
		}

		public void OnPatchType(QSBPatchTypes type)
		{
			if (type != QSBPatchTypes.OnNonServerClientConnect)
			{
				return;
			}
			QSB.Helper.HarmonyHelper.EmptyMethod<GeyserController>("Update");
		}
	}
}