implement addAtBContract

MegaMek · Sep 25, 2024 · dcf55ed · dcf55ed
1 parent a79000f
commit dcf55ed
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/MekHQ/src/mekhq/campaign/market/contractMarket/CamOpsContractMarket.java b/MekHQ/src/mekhq/campaign/market/contractMarket/CamOpsContractMarket.java
@@ -36,6 +36,12 @@ public CamOpsContractMarket() {
 
     @Override
     public AtBContract addAtBContract(Campaign campaign) {
+        Optional<AtBContract> c = generateContract(campaign);
+        if (c.isPresent()) {
+            AtBContract atbContract = c.get();
+            contracts.add(atbContract);
+            return atbContract;
+        }
         return null;
     }
 
@@ -58,8 +64,7 @@ public void generateContractOffers(Campaign campaign, boolean newCampaign) {
             rollNegotiation(negotiationSkill, ratingMod + contractMods.offersMod) - BASE_NEGOTIATION_TARGET);
 
         for (int i = 0; i < numOffers; i++) {
-            Optional<AtBContract> c = generateContract(campaign);
-            c.ifPresent(contract -> contracts.add(contract));
+            addAtBContract(campaign);
         }
         updateReport(campaign);
     }