Fix config purchase not checking for unlock credit

KSP-RO · Jun 11, 2024 · 9905219 · 9905219
1 parent dfe9cad
commit 9905219
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/Source/ModuleROHeatshield.cs b/Source/ModuleROHeatshield.cs
@@ -353,14 +353,13 @@ public bool IsConfigUnlocked(string configName)
 
         public bool PurchaseConfig(PartUpgradeHandler.Upgrade upgd)
         {
-            if (Funding.CanAfford(upgd.entryCost))
-            {
-                PartUpgradeManager.Handler.SetUnlocked(upgd.name, true);
-                GameEvents.OnPartUpgradePurchased.Fire(upgd);
-                return true;
-            }
+            CurrencyModifierQuery cmq = CurrencyModifierQuery.RunQuery(TransactionReasons.RnDPartPurchase, -upgd.entryCost, 0, 0);
+            if (!cmq.CanAfford())
+                return false;
 
-            return false;
+            PartUpgradeManager.Handler.SetUnlocked(upgd.name, true);
+            GameEvents.OnPartUpgradePurchased.Fire(upgd);
+            return true;
         }
 
         private void UpdatePresetsList(string[] presetNames)