CustomSpawnPoint: initial release

CustomSpawnPoint/SpawnPointManager.cs

@@ -0,0 +1,265 @@
+using ABI_RC.Core.InteractionSystem;
+using ABI_RC.Core.Player;
+using ABI_RC.Systems.GameEventSystem;
+using ABI.CCK.Components;
+using UnityEngine;
+using Object = UnityEngine.Object;
+using ABI_RC.Core;
+using Newtonsoft.Json;
+
+namespace NAK.CustomSpawnPoint
+{
+    internal static class SpawnPointManager
+    {
+        #region Fields
+
+        private static string currentWorldId = string.Empty;
+        private static SpawnPointData? currentSpawnPoint;
+
+        private static string requestedWorldId = string.Empty;
+        private static SpawnPointData? requestedSpawnPoint;
+
+        private static Dictionary<string, SpawnPointData> spawnPoints = new();
+        private static readonly string jsonFilePath = Path.Combine("UserData", "customspawnpoints.json");
+        
+        private static GameObject[] customSpawnPointsArray;
+        private static GameObject[] originalSpawnPointsArray;
+
+        #endregion Fields
+
+        #region Initialization
+
+        internal static void Init()
+        {
+            LoadSpawnpoints();
+            CVRGameEventSystem.World.OnLoad.AddListener(OnWorldLoaded);
+            CVRGameEventSystem.World.OnUnload.AddListener(OnWorldUnloaded);
+            MelonLoader.MelonCoroutines.Start(WaitMainMenuUi());
+        }
+
+        private static System.Collections.IEnumerator WaitMainMenuUi()
+        {
+            while (ViewManager.Instance == null)
+                yield return null;
+            while (ViewManager.Instance.gameMenuView == null)
+                yield return null;
+            while (ViewManager.Instance.gameMenuView.Listener == null)
+                yield return null;
+
+            ViewManager.Instance.OnUiConfirm.AddListener(OnClearSpawnpointConfirm);
+            ViewManager.Instance.gameMenuView.Listener.FinishLoad += (_) =>
+            {
+                ViewManager.Instance.gameMenuView.View._view.ExecuteScript(spawnpointJs);
+            };
+            ViewManager.Instance.gameMenuView.Listener.ReadyForBindings += () =>
+            {
+                // listen for setting the spawn point on our custom button
+                ViewManager.Instance.gameMenuView.View.BindCall("NAKCallSetSpawnpoint", SetSpawnPoint);
+            };
+            
+            // create our custom spawn point object
+            GameObject customSpawnPointObject = new("[CustomSpawnPoint]");
+            Object.DontDestroyOnLoad(customSpawnPointObject);
+
+            // add to array so we can easily replace worlds spawn points
+            customSpawnPointsArray = new[] { customSpawnPointObject };
+        }
+
+        #endregion Initialization
+        
+        #region Game Events
+        
+        private static void OnWorldLoaded(string worldId)
+        {
+            CVRWorld world = CVRWorld.Instance;
+            if (world == null) return;
+            
+            CustomSpawnPointMod.Logger.Msg("World loaded: " + worldId);
+
+            currentWorldId = worldId;
+            currentSpawnPoint = spawnPoints.GetValueOrDefault(currentWorldId);
+            originalSpawnPointsArray ??= world.spawns; // cache the original spawn points array, if null its fine
+
+            if (currentSpawnPoint.HasValue)
+            {
+                UpdateCustomSpawnPointTransform(currentSpawnPoint.Value);
+                world.spawns = customSpawnPointsArray; // set the custom spawn points array
+                
+                // CVRWorld.Awake already moved player, but OnWorldLoaded is invoked in OnEnable
+                RootLogic.Instance.SpawnOnWorldInstance();
+            }
+        }
+
+        private static void OnWorldUnloaded(string worldId)
+        {
+            ClearCurrentWorldState();
+        }
+
+        internal static void OnRequestWorldDetailsPage(string worldId)
+        {
+            CustomSpawnPointMod.Logger.Msg("Requesting world details page for world: " + worldId);
+            
+            requestedWorldId = worldId;
+            requestedSpawnPoint = spawnPoints.TryGetValue(requestedWorldId, out SpawnPointData spawnPoint) ? spawnPoint : null;
+
+            bool hasSpawnpoint = requestedSpawnPoint.HasValue;
+            UpdateMenuButtonState(hasSpawnpoint, worldId == currentWorldId && CVRWorld.Instance != null && CVRWorld.Instance.allowFlying);
+        }
+        
+        private static void OnClearSpawnpointConfirm(string id, string value, string data)
+        {
+            if (id != "nak_clear_spawnpoint") return;
+            if (value == "true") ClearSpawnPoint();
+        }
+
+        #endregion Game Events
+
+        #region Spawnpoint Management
+        
+        public static void SetSpawnPoint()
+            => SetSpawnPointForWorld(currentWorldId);
+        
+        public static void ClearSpawnPoint()
+            => ClearSpawnPointForWorld(currentWorldId);
+        
+        private static void SetSpawnPointForWorld(string worldId)
+        {
+            CustomSpawnPointMod.Logger.Msg("Setting spawn point for world: " + worldId);
+            
+            Vector3 playerPosition = PlayerSetup.Instance.GetPlayerPosition();
+            Quaternion playerRotation = PlayerSetup.Instance.GetPlayerRotation();
+
+            // Update or create the spawn point data
+            SpawnPointData spawnPoint = new()
+            {
+                Position = playerPosition,
+                Rotation = playerRotation.eulerAngles
+            };
+
+            spawnPoints[worldId] = spawnPoint;
+
+            // update the current world state if applicable
+            if (worldId == currentWorldId)
+            {
+                currentSpawnPoint = spawnPoint;
+                UpdateCustomSpawnPointTransform(spawnPoint);
+                // update the custom spawn points array
+                if (CVRWorld.Instance != null) CVRWorld.Instance.spawns = customSpawnPointsArray;
+                ViewManager.Instance.NotifyUser("(Local) Client", "Set custom spawnpoint", 2f);
+            }
+
+            SaveSpawnpoints();
+            UpdateMenuButtonState(true, worldId == currentWorldId);
+        }
+
+        private static void ClearSpawnPointForWorld(string worldId)
+        {
+            CustomSpawnPointMod.Logger.Msg("Clearing spawn point for world: " + worldId);
+            
+            if (spawnPoints.ContainsKey(worldId))
+                spawnPoints.Remove(worldId);
+
+            if (worldId == currentWorldId)
+            {
+                currentSpawnPoint = null;
+                
+                // restore the original spawn points array
+                if (CVRWorld.Instance != null) CVRWorld.Instance.spawns = originalSpawnPointsArray;
+                ViewManager.Instance.NotifyUser("(Local) Client", "Cleared custom spawnpoint", 2f);
+            }
+
+            SaveSpawnpoints();
+            UpdateMenuButtonState(false, worldId == currentWorldId);
+        }
+
+        private static void UpdateCustomSpawnPointTransform(SpawnPointData spawnPoint)
+        {
+            customSpawnPointsArray[0].transform.SetPositionAndRotation(spawnPoint.Position, Quaternion.Euler(spawnPoint.Rotation));
+        }
+
+        private static void UpdateMenuButtonState(bool hasSpawnpoint, bool isInWorld)
+        {
+            ViewManager.Instance.gameMenuView.View.TriggerEvent("NAKUpdateSpawnpointStatus", hasSpawnpoint.ToString(), isInWorld.ToString());
+        }
+
+        private static void ClearCurrentWorldState()
+        {
+            currentWorldId = string.Empty;
+            currentSpawnPoint = null;
+            originalSpawnPointsArray = null;
+        }
+
+        #endregion Spawnpoint Management
+
+        #region JSON Management
+
+        private static void LoadSpawnpoints()
+        {
+            if (File.Exists(jsonFilePath))
+            {
+                string json = File.ReadAllText(jsonFilePath);
+                spawnPoints = JsonConvert.DeserializeObject<Dictionary<string, SpawnPointData>>(json);
+            }
+            else
+            {
+                SaveSpawnpoints(); // create the file if it doesn't exist
+            }
+        }
+
+        private static void SaveSpawnpoints()
+        {
+            File.WriteAllText(jsonFilePath, JsonConvert.SerializeObject(spawnPoints, Formatting.Indented, 
+                new JsonSerializerSettings { ReferenceLoopHandling = ReferenceLoopHandling.Ignore } // death
+            ));
+        }
+
+        #endregion JSON Management
+        
+        #region Spawnpoint JS
+
+        private const string spawnpointJs = @"
+let hasSpawnpointForThisWorld = false;
+let spawnpointButton = null;
+
+// replace the screenshot button with ours
+spawnpointButton = document.querySelector('.action-btn.button.data-worldPreload.row2.col2.disabled');
+if (spawnpointButton) {
+    spawnpointButton.classList.remove('disabled');
+    spawnpointButton.innerHTML = '<img src=""gfx/respawn.svg""><span>Spawnpoint</span>';
+    spawnpointButton.setAttribute('onclick', 'onClickSpawnpointButton();');
+}
+
+function onClickSpawnpointButton(){
+    if (spawnpointButton.classList.contains('disabled')) {
+        return;
+    }
+
+    if (hasSpawnpointForThisWorld) {
+        uiConfirmShow(""Custom Spawnpoint"", ""Are you sure you want to clear your spawnpoint?"", ""nak_clear_spawnpoint"", """");
+    } else {
+        engine.call('NAKCallSetSpawnpoint');
+    }
+}
+
+engine.on('NAKUpdateSpawnpointStatus', function (hasSpawnpoint, isInWorld) {
+    hasSpawnpointForThisWorld = hasSpawnpoint === 'True';
+    if (spawnpointButton) {
+        spawnpointButton.classList.toggle('disabled', isInWorld !== 'True');
+    }
+});
+";
+
+        #endregion Spawnpoint JS
+    }
+
+    #region Serializable
+    
+    [Serializable]
+    public struct SpawnPointData
+    {
+        public Vector3 Position;
+        public Vector3 Rotation;
+    }
+    
+    #endregion Serializable
+}