Move source into src

2024-09-20 11:47:09 -07:00
parent 2ff3fba9bc
commit 698cd61a88
84 changed files with 0 additions and 0 deletions
--- a/src/Patches/HideoutLevelPatches.cs
+++ b/src/Patches/HideoutLevelPatches.cs
@@ -0,0 +1,87 @@
+using EFT.Hideout;
+using HarmonyLib;
+using SPT.Reflection.Patching;
+using System.Reflection;
+
+namespace UIFixes;
+
+public static class HideoutLevelPatches
+{
+    private static string CurrentArea;
+    private static ELevelType CurrentLevel = ELevelType.NotSet;
+
+    public static void Enable()
+    {
+        new SelectAreaPatch().Enable();
+        new ChangeLevelPatch().Enable();
+        new PickInitialLevelPatch().Enable();
+        new ClearLevelPatch().Enable();
+    }
+
+    public class SelectAreaPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(AreaScreenSubstrate), nameof(AreaScreenSubstrate.SelectArea));
+        }
+
+        [PatchPrefix]
+        public static void Prefix(AreaData areaData)
+        {
+            if (areaData.Template.Id != CurrentArea)
+            {
+                CurrentArea = areaData.Template.Id;
+                CurrentLevel = ELevelType.NotSet;
+            }
+        }
+    }
+
+    public class ChangeLevelPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(AreaScreenSubstrate), nameof(AreaScreenSubstrate.method_6));
+        }
+
+        [PatchPrefix]
+        public static void Prefix(ELevelType state)
+        {
+            CurrentLevel = state;
+        }
+    }
+
+    public class PickInitialLevelPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(AreaScreenSubstrate), nameof(AreaScreenSubstrate.method_3));
+        }
+
+        [PatchPrefix]
+        public static bool Prefix(ref ELevelType __result)
+        {
+            if (CurrentLevel != ELevelType.NotSet)
+            {
+                __result = CurrentLevel;
+                return false;
+            }
+
+            return true;
+        }
+    }
+
+    public class ClearLevelPatch : ModulePatch
+    {
+        protected override MethodBase GetTargetMethod()
+        {
+            return AccessTools.Method(typeof(HideoutScreenOverlay), nameof(HideoutScreenOverlay.ReturnToPreviousState));
+        }
+
+        [PatchPostfix]
+        public static void Postfix()
+        {
+            CurrentArea = null;
+            CurrentLevel = ELevelType.NotSet;
+        }
+    }
+}