Catch when RF changes engine or tank config

KSP-RO · Sep 18, 2023 · bb24731 · bb24731
1 parent 86d19f8
commit bb24731
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/Source/Harmony/RealFuels.cs b/Source/Harmony/RealFuels.cs
@@ -49,6 +49,30 @@ internal static void Postfix_DrawSelectButton(RealFuels.ModuleEngineConfigsBase
         {
             RFECMPatcher.techNode = null;
         }
+
+        [HarmonyPostfix]
+        [HarmonyPatch("SetConfiguration", new Type[] { typeof(ConfigNode), typeof(bool) })]
+        internal static void Postfix_SetConfiguration()
+        {
+            if (HighLogic.LoadedSceneIsEditor && KerbalConstructionTime.KerbalConstructionTime.Instance != null)
+            {
+                KerbalConstructionTime.KerbalConstructionTime.Instance.IsEditorRecalcuationRequired = true;
+            }
+        }
+    }
+
+    [HarmonyPatch(typeof(RealFuels.Tanks.ModuleFuelTanks))]
+    internal class RFMFTPatcher
+    {
+        [HarmonyPostfix]
+        [HarmonyPatch("RaiseTankDefinitionChanged")]
+        internal static void Postfix_RaiseTankDefinitionChanged()
+        {
+            if (HighLogic.LoadedSceneIsEditor && KerbalConstructionTime.KerbalConstructionTime.Instance != null)
+            {
+                KerbalConstructionTime.KerbalConstructionTime.Instance.IsEditorRecalcuationRequired = true;
+            }
+        }
     }
 
     [HarmonyPatch(typeof(RealFuels.EntryCostManager))]