quantum-space-buddies/QSB/QSBSceneManager.cs

using OWML.Common;
using QSB.Utility;
using QSB.WorldSync;
using System;

namespace QSB
{
	public static class QSBSceneManager
	{
		public static OWScene CurrentScene => LoadManager.GetCurrentScene();

		public static bool IsInUniverse => InUniverse(CurrentScene);

		public static event Action<OWScene, OWScene, bool> OnSceneLoaded;
		public static event Action<OWScene, OWScene> OnUniverseSceneLoaded;

		static QSBSceneManager()
		{
			LoadManager.OnCompleteSceneLoad += OnCompleteSceneLoad;
			DebugLog.DebugWrite("Scene Manager ready.", MessageType.Success);
		}

		private static void OnCompleteSceneLoad(OWScene oldScene, OWScene newScene)
		{
			DebugLog.DebugWrite($"COMPLETE SCENE LOAD ({oldScene} -> {newScene})", MessageType.Info);
			QSBWorldSync.RemoveWorldObjects();
			var universe = InUniverse(newScene);
			if (QSBCore.IsInMultiplayer && universe)
			{
				// So objects have time to be deleted, made, whatever
				Delay.RunNextFrame(() => QSBWorldSync.BuildWorldObjects(newScene).Forget());
			}

			OnSceneLoaded?.SafeInvoke(oldScene, newScene, universe);
			if (universe)
			{
				OnUniverseSceneLoaded?.SafeInvoke(oldScene, newScene);
			}

			if (newScene == OWScene.TitleScreen && QSBCore.IsInMultiplayer)
			{
				QSBNetworkManager.singleton.StopHost();
			}
		}

		private static bool InUniverse(OWScene scene) =>
			scene is OWScene.SolarSystem or OWScene.EyeOfTheUniverse;
	}
}