Harmony-patch ROHS config can afford query to factor in unlock credit.

Also change other Harmony patches to use the same format.
KSP-RO · Aug 19, 2024 · a764bb3 · a764bb3
1 parent 12339d7
commit a764bb3
Show file tree

Hide file tree

Showing 4 changed files with 52 additions and 4 deletions.
diff --git a/Source/RP0/Harmony/CustomBarnKit.cs b/Source/RP0/Harmony/CustomBarnKit.cs
@@ -1,12 +1,21 @@
 using HarmonyLib;
+using System;
 using System.Reflection;
 
 namespace RP0.Harmony
 {
     [HarmonyPatch]
     internal class PatchCustomBarnKit_LoadUpgradesPrices
     {
-        static MethodBase TargetMethod() => AccessTools.TypeByName("CustomBarnKit.CustomBarnKit").GetMethod("LoadUpgradesPrices", AccessTools.all);
+        internal static readonly Type _type = AccessTools.TypeByName("CustomBarnKit.CustomBarnKit");
+
+        internal static MethodBase TargetMethod() => AccessTools.Method(_type, "LoadUpgradesPrices");
+
+        [HarmonyPrepare]
+        internal static bool Prepare()
+        {
+            return _type != null;
+        }
 
         [HarmonyPrefix]
         internal static void Prefix_LoadUpgradesPrices(ref bool ___varLoaded, out bool __state)

diff --git a/Source/RP0/Harmony/KSCSwitcher.cs b/Source/RP0/Harmony/KSCSwitcher.cs
@@ -1,19 +1,20 @@
 using HarmonyLib;
+using System;
 using System.Reflection;
 
 namespace RP0.Harmony
 {
     [HarmonyPatch]
     internal class PatchKSCSwitcher
     {
-        internal static readonly System.Type KSCSwitcherType = AccessTools.TypeByName("regexKSP.KSCSwitcher");
+        internal static readonly System.Type _type = AccessTools.TypeByName("regexKSP.KSCSwitcher");
 
-        internal static MethodBase TargetMethod() => KSCSwitcherType == null ? null : AccessTools.Method(KSCSwitcherType, "SetSite", new System.Type[] { typeof(ConfigNode) });
+        internal static MethodBase TargetMethod() => AccessTools.Method(_type, "SetSite", new Type[] { typeof(ConfigNode) });
 
         [HarmonyPrepare]
         internal static bool Prepare()
         {
-            return KSCSwitcherType != null;
+            return _type != null;
         }
 
         [HarmonyPostfix]

diff --git a/Source/RP0/Harmony/ROHeatshields.cs b/Source/RP0/Harmony/ROHeatshields.cs
@@ -0,0 +1,37 @@
+using HarmonyLib;
+using System;
+using System.Reflection;
+
+namespace RP0.Harmony
+{
+    [HarmonyPatch]
+    internal class ROHSPartModulePatcher
+    {
+        internal static readonly Type _type = AccessTools.TypeByName("ROHeatshields.ModuleROHeatshield");
+        internal static MethodInfo _mi;
+
+        internal static MethodBase TargetMethod() => _mi;
+
+        [HarmonyPrepare]
+        internal static bool Prepare()
+        {
+            if (_type != null)
+            {
+                _mi ??= AccessTools.Method(_type, "CanAffordEntryCost", new Type[] { typeof(float) });
+            }
+            return _mi != null;
+        }
+
+        [HarmonyPrefix]
+        internal static bool Prefix_CanAffordEntryCost(float cost, ref bool __result)
+        {
+            var cmq = CurrencyModifierQueryRP0.RunQuery(TransactionReasonsRP0.PartOrUpgradeUnlock, -cost, 0d, 0d);
+            double postCMQcost = -cmq.GetTotal(CurrencyRP0.Funds, false);
+            double credit = UnlockCreditHandler.Instance.TotalCredit;
+            cmq.AddPostDelta(CurrencyRP0.Funds, credit, true);
+            __result = cmq.CanAfford();
+
+            return false;
+        }
+    }
+}
diff --git a/Source/RP0/RP0.csproj b/Source/RP0/RP0.csproj
@@ -50,6 +50,7 @@
     <Compile Include="Harmony\FlightInputHandler.cs" />
     <Compile Include="Harmony\ModuleRCS.cs" />
     <Compile Include="Harmony\RealAntennas.cs" />
+    <Compile Include="Harmony\ROHeatshields.cs" />
     <Compile Include="ModIntegrations\KSCSwitcherInterop.cs" />
     <Compile Include="ModIntegrations\TFInterop.cs" />
     <Compile Include="Programs\ProgramModifier.cs" />