0ad/binaries/data/mods/public/simulation/components/Upkeep.js

function Upkeep() {}

Upkeep.prototype.Schema =
	"<a:help>Controls the resource upkeep of an entity.</a:help>" +
	"<element name='Rates' a:help='Upkeep Rates'>" +
		Resources.BuildSchema("nonNegativeDecimal") +
	"</element>" +
	"<element name='Interval' a:help='Number of milliseconds must pass for the player to pay the next upkeep.'>" +
		"<ref name='nonNegativeDecimal'/>" +
	"</element>";

Upkeep.prototype.Init = function()
{
	this.upkeepInterval = +this.template.Interval;
	this.CheckTimer();
};

/**
 * @return {number} - The interval between resource subtractions, in ms.
 */
Upkeep.prototype.GetInterval = function()
{
	return this.upkeepInterval;
};

/**
 * @return {Object} - The upkeep rates in the form of { "resourceName": {number} }.
 */
Upkeep.prototype.GetRates = function()
{
	return this.rates;
};

/**
 * @return {boolean} - Whether this entity has at least one non-zero amount of resources to pay.
 */
Upkeep.prototype.ComputeRates = function()
{
	this.rates = {};
	let hasUpkeep = false;
	for (let resource in this.template.Rates)
	{
		let rate = ApplyValueModificationsToEntity("Upkeep/Rates/" + resource, +this.template.Rates[resource], this.entity);
		if (rate)
		{
			this.rates[resource] = rate;
			hasUpkeep = true;
		}
	}

	return hasUpkeep;
};

/**
 * Try to subtract the needed resources.
 * Data and lateness are unused.
 */
Upkeep.prototype.Pay = function(data, lateness)
{
	let cmpPlayer = QueryOwnerInterface(this.entity);
	if (!cmpPlayer)
		return;

	if (!cmpPlayer.TrySubtractResources(this.rates))
		this.HandleInsufficientUpkeep();
	else
		this.HandleSufficientUpkeep();
};

/**
 * E.g. take a hitpoint, reduce CP.
 */
Upkeep.prototype.HandleInsufficientUpkeep = function()
{
	if (this.unpayed)
		return;

	let cmpIdentity = Engine.QueryInterface(this.entity, IID_Identity);
	if (cmpIdentity)
		cmpIdentity.SetControllable(false);
	this.unpayed = true;
};

/**
 * Reset to the previous stage.
 */
Upkeep.prototype.HandleSufficientUpkeep = function()
{
	if (!this.unpayed)
		return;

	let cmpIdentity = Engine.QueryInterface(this.entity, IID_Identity);
	if (cmpIdentity)
		cmpIdentity.SetControllable(true);
	delete this.unpayed;
};

Upkeep.prototype.OnValueModification = function(msg)
{
	if (msg.component != "Upkeep")
		return;

	this.CheckTimer();
};

/**
 * Recalculate the interval and update the timer accordingly.
 */
Upkeep.prototype.CheckTimer = function()
{
	if (!this.ComputeRates())
	{
		if (!this.timer)
			return;

		let cmpTimer = Engine.QueryInterface(SYSTEM_ENTITY, IID_Timer);
		cmpTimer.CancelTimer(this.timer);
		delete this.timer;
		return;
	}

	let oldUpkeepInterval = this.upkeepInterval;
	this.upkeepInterval = ApplyValueModificationsToEntity("Upkeep/Interval", +this.template.Interval, this.entity);
	if (this.upkeepInterval < 0)
	{
		let cmpTimer = Engine.QueryInterface(SYSTEM_ENTITY, IID_Timer);
		cmpTimer.CancelTimer(this.timer);
		delete this.timer;
		return;
	}

	if (this.timer)
	{
		if (this.upkeepInterval == oldUpkeepInterval)
			return;

		// If the timer wasn't invalidated before (interval <= 0), just update it.
		if (oldUpkeepInterval > 0)
		{
			let cmpTimer = Engine.QueryInterface(SYSTEM_ENTITY, IID_Timer);
			cmpTimer.UpdateRepeatTime(this.timer, this.upkeepInterval);
			return;
		}
	}

	let cmpTimer = Engine.QueryInterface(SYSTEM_ENTITY, IID_Timer);
	this.timer = cmpTimer.SetInterval(this.entity, IID_Upkeep, "Pay", this.upkeepInterval, this.upkeepInterval, undefined);
};

Engine.RegisterComponentType(IID_Upkeep, "Upkeep", Upkeep);