Ticket #3707: t3707_repair_cost2.diff

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

@@ -1 +1 @@
 function Repairable() {}
 
+var g_CostMultiplier = 0.5
+
 Repairable.prototype.Schema =
     "<a:help>Deals with repairable structures and units.</a:help>" +
     "<a:example>" +
@@ -13 +15 @@
 {
     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 +53 @@
         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)
@@ -64 +67 @@
     // Calculate the amount of hitpoints that will be added (using diminishing rate when several builders)
     let work = rate * this.buildMultiplier * this.GetRepairRate();
     let amount = Math.min(damage, work);
-    cmpHealth.Increase(amount);
 
-    // If we repaired all the damage, send a message to entities to stop repairing this building
-    if (amount >= damage)
+    // Calculate resource costs for the added hitpoints
+    let newPrice = cmpCost.GetResourceCosts();
+    let resourcesToSubtract = {};
+    for (let type in newPrice)
+    {
+        this.costsToPay[type] += g_CostMultiplier * newPrice[type] * amount / cmpHealth.GetMaxHitpoints();
+        resourcesToSubtract[type] = Math.floor(this.costsToPay[type]);
+        this.costsToPay[type] -= resourcesToSubtract[type];
+    }
+
+    // TrySubtractResources should report error to player (if they ran out of resources) 
+    let affordable = cmpPlayer.TrySubtractResources(resourcesToSubtract)
+    if (affordable) 
+        cmpHealth.Increase(amount);
+
+    // If we repaired all the damage or ran out of resource, send a message to entities to stop repairing this building
+    if (amount >= damage || !affordable)
         Engine.PostMessage(this.entity, MT_ConstructionFinished, { "entity": this.entity, "newentity": this.entity });
 };