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;
}