const pClient = require('prom-client');
const KEYS = ['live', 'flexibytes', 'aggregate'];

// Initial Setup
let gauges = {};
pClient.collectDefaultMetrics({ timeout: 60000 });

KEYS.map(key => {
	gauges[`${key}_used`] = new pClient.Gauge({
		name: `act_fup_${key}_usage_bytes`,
		help: `ACT ${key} usage in bytes (precision GB)`,
	});

	gauges[`${key}_total`] = new pClient.Gauge({
		name: `act_fup_${key}_total_usage_bytes`,
		help: `ACT ${key} usage in bytes (precision GB)`,
	});
});

module.exports = {
	format: function(metrics) {
		KEYS.map(key => {
			gauges[`${key}_used`].set(metrics[key].usedBytes);
			gauges[`${key}_total`].set(metrics[key].totalBytes);
		});

		return pClient.register.metrics();
	},
	contentType: pClient.register.contentType,
};