extension.js

'use strict';

let nodecg;
let elapsedTime;
let remainingTime;
let isCooldownActive;
let passengers;
let dayTotal;
let threshold;
let duration;

let cdTimer = null;

function Train(extensionApi) {
	nodecg = extensionApi;

	// These properties are transient and are not persisted
	elapsedTime = nodecg.Replicant('elapsedTime', {defaultValue: 0, persistent: false});
	remainingTime = nodecg.Replicant('remainingTime', {defaultValue: 0, persistent: false});
	isCooldownActive = nodecg.Replicant('isCooldownActive', {defaultValue: false, persistent: false});

	// These properties are persisted to disk
	passengers = nodecg.Replicant('passengers', {defaultValue: 0});
	dayTotal = nodecg.Replicant('dayTotal', {defaultValue: 0});
	threshold = nodecg.Replicant('threshold', {defaultValue: 10});
	duration = nodecg.Replicant('duration', {defaultValue: 300});

	nodecg.listenFor('startCooldown', this.startCooldown.bind(this));
	nodecg.listenFor('endCooldown', this.endCooldown.bind(this));
	nodecg.listenFor('resetCooldown', this.resetCooldown.bind(this));

	// temporary workaround for some bundles until I figure out how to make this better
	nodecg.listenFor('getPassengers', (data, cb) => {
		cb(passengers.value);
	});

	nodecg.listenFor('getDayTotal', (data, cb) => {
		cb(dayTotal.value);
	});
}

Train.prototype.startCooldown = function () {
	_killTimer(); // Kill any existing cooldown timer

	elapsedTime.value = 0;
	remainingTime.value = duration.value;
	isCooldownActive.value = true;
	cdTimer = setInterval(this.tickCooldown.bind(this), 1000);

	nodecg.sendMessage('cooldownStart');
};

Train.prototype.tickCooldown = function () {
	elapsedTime.value++;
	remainingTime.value--;
	nodecg.sendMessage('cooldownTick', {
		remainingTime: remainingTime.value,
		elapsedTime: elapsedTime.value
	});

	if (remainingTime.value <= 0) {
		remainingTime.value = 0; // force to zero if we somehow went negative
		this.endCooldown();
	}
};

Train.prototype.resetCooldown = function () {
	elapsedTime.value = 0;
	remainingTime.value = duration.value;
};

Train.prototype.endCooldown = function () {
	_killTimer();
	passengers.value = 0;
	isCooldownActive.value = false;
	nodecg.sendMessage('cooldownEnd');
};

Train.prototype.addPassenger = function () {
	passengers.value++;
	dayTotal.value++;

	const train = {
		passengers: passengers.value,
		dayTotal: dayTotal.value,
		threshold: threshold.value,
		isHype: passengers.value >= threshold.value
	};

	if (train.isHype && nodecg.bundleConfig.resetAfterThreshold) {
		passengers.value = 0;
		this.endCooldown();
	} else if (nodecg.bundleConfig.autoStartCooldown) {
		this.startCooldown();
	}

	return train;
};

function _killTimer() {
	if (cdTimer !== null) {
		clearInterval(cdTimer);
		cdTimer = null;
	}
}

module.exports = function (extensionApi) {
	return new Train(extensionApi);
};