Ticket #3707: t3707_repair_cost.diff

binaries/data/mods/public/simulation/components/Repairable.js

@@ -13 +13 @@
 {
     this.builders = []; // builder entities
     this.buildMultiplier = 1; // Multiplier for the amount of work builders do.
+    this.costsToPay = { "food": 0, "wood": 0, "metal": 0, "stone": 0 };
     this.repairTimeRatio = +this.template.RepairTimeRatio;
 };
 
@@ -50 +51 @@
         this.buildMultiplier = Math.pow(numBuilders, 0.7) / numBuilders;
 };
 
-// TODO: should we have resource costs?
 Repairable.prototype.Repair = function(builderEnt, rate)
 {
     let cmpHealth = Engine.QueryInterface(this.entity, IID_Health);
     let cmpCost = Engine.QueryInterface(this.entity, IID_Cost);
-    if (!cmpHealth || !cmpCost)
+    let cmpPlayer = QueryOwnerInterface(this.entity);
+    if (!cmpHealth || !cmpCost || !cmpPlayer)
         return;
     let damage = cmpHealth.GetMaxHitpoints() - cmpHealth.GetHitpoints();
     if (damage <= 0)
@@ -67 +68 @@
     if (repairTime)
         work *= (cmpHealth.GetMaxHitpoints() / repairTime);
     let amount = Math.min(damage, work);
+
+    // Calculate resource costs for the added hitpoints
+    let newPrice = cmpCost.GetResourceCosts();
+    let repairCost = [];
+    let resourcesToSubtract = [];
+    for (let type in newPrice)
+    {
+        repairCost[type] = 0.5 * newPrice[type] * amount / cmpHealth.GetMaxHitpoints();
+        this.costsToPay[type] += repairCost[type];
+        resourcesToSubtract[type] = Math.floor(this.costsToPay[type]);
+        this.costsToPay[type] -= resourcesToSubtract[type];
+    };
+
+    // Substract resource
+    // TrySubtractResources should report error to player (if they ran out of resources)
+    // If ran out of resource, send a message to entities to stop repairing this building
+    if (!cmpPlayer.TrySubtractResources(resourcesToSubtract))
+    {
+
+        Engine.PostMessage(this.entity, MT_ConstructionFinished, { "entity": this.entity, "newentity": this.entity });
+        return;
+    };
+    
     cmpHealth.Increase(amount);
 
     // If we repaired all the damage, send a message to entities to stop repairing this building