some reorgnaizing, fix contract difficulty persistence

ajkroeg · Jul 1, 2022 · ba7025b · ba7025b
1 parent c08d699
commit ba7025b
Show file tree

Hide file tree

Showing 5 changed files with 195 additions and 146 deletions.
diff --git a/MapRandomizer/MapRandomizer/MapRandomizer.csproj b/MapRandomizer/MapRandomizer/MapRandomizer.csproj
@@ -67,7 +67,7 @@
     </Compile>
     <Compile Include="source\Logger.cs" />
     <Compile Include="source\ModInit.cs" />
-    <Compile Include="source\Patches\ContractPatch.cs" />
+    <Compile Include="source\Patches\DifficultyPersistancePatches.cs" />
     <Compile Include="source\Patches\MapRandomizer.cs" />
     <Compile Include="source\ModState.cs" />
   </ItemGroup>

diff --git a/MapRandomizer/MapRandomizer/Properties/AssemblyInfo.cs b/MapRandomizer/MapRandomizer/Properties/AssemblyInfo.cs
@@ -28,5 +28,5 @@
 //      Build Number
 //      Revision
 //
-[assembly: AssemblyVersion("1.0.0.8")]
-[assembly: AssemblyFileVersion("1.0.0.8")]
+[assembly: AssemblyVersion("1.0.1.0")]
+[assembly: AssemblyFileVersion("1.0.1.0")]
diff --git a/MapRandomizer/MapRandomizer/source/Patches/ContractPatch.cs b/MapRandomizer/MapRandomizer/source/Patches/ContractPatch.cs
diff --git a/MapRandomizer/MapRandomizer/source/Patches/DifficultyPersistancePatches.cs b/MapRandomizer/MapRandomizer/source/Patches/DifficultyPersistancePatches.cs
@@ -0,0 +1,66 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using BattleTech;
+using BattleTech.Save;
+using BattleTech.Save.Test;
+using Harmony;
+using MapRandomizer.Patches;
+using UnityEngine;
+
+namespace MapRandomizer.source.Patches
+{
+    internal class DifficultyPersistancePatches
+    {
+        [HarmonyPatch(typeof(SimGameState), "Dehydrate",
+            new Type[] {typeof(SimGameSave), typeof(SerializableReferenceContainer)})]
+        public static class SGS_Dehydrate_Patch//MR_DIFFICULTY_
+        {
+            public static void Prefix(SimGameState __instance)
+            {
+                foreach (var diffOverride in ModState.SavedDiffOverrides)
+                {
+                    __instance.CompanyStats.AddStatistic(diffOverride.Key, diffOverride.Value);
+                }
+            }
+        }
+
+        [HarmonyPatch(typeof(SimGameState), "Rehydrate", new Type[] {typeof(GameInstanceSave)})]
+        public static class SGS_Rehydrate_Patch
+        {
+           public static void Postfix(SimGameState __instance) 
+           {
+               foreach (var statistic in __instance.CompanyStats)
+               {
+                   if (statistic.Key.StartsWith("MR_DIFFICULTY_"))
+                   {
+                       ModState.SavedDiffOverrides.Add(statistic.Key, statistic.Value.Value<int>());
+                   }
+               }
+
+               foreach (var statOverride in ModState.SavedDiffOverrides)
+               {
+                   __instance.CompanyStats.RemoveStatistic(statOverride.Key);
+               }
+           }
+        }
+
+        [HarmonyPatch(typeof(Contract), "CompleteContract", new Type[] {typeof(MissionResult), typeof(bool)})]
+        public static class Contract_CompleteContract_Patch
+        {
+            public static void Prefix(Contract __instance, MissionResult result, bool isGoodFaithEffort)
+            {
+                var sim = UnityGameInstance.BattleTechGame.Simulation;
+                if (sim == null) return;
+                var baseDiff = sim.CurSystem.Def.GetDifficulty(sim.SimGameMode) + Mathf.FloorToInt(sim.GlobalDifficulty);
+                var quid = __instance.GenerateContractQuasiGUID(__instance.Override.employerTeam.FactionValue, __instance.Override.targetTeam.FactionValue, baseDiff, __instance.ContractBiome, sim.CurSystem);
+                if (ModState.SavedDiffOverrides.ContainsKey(quid))
+                {
+                    ModState.SavedDiffOverrides.Remove(quid);
+                }
+            }
+        }
+    }
+}