ProxmoxAAS-API/db.js

const fs = require("fs");

filename = "localdb.json";

let db = {};

/**
 * called at app startup, can be used to initialize any variables needed for database access
 */
function init () {
	try {
		db = JSON.parse(fs.readFileSync(filename));
	}
	catch {
		fs.writeFileSync(filename, JSON.stringify(db));
	}
}

/**
 * user requests additional resources specified in k-v pairs
 * @param {string} user user's proxmox username in the form username@authrealm
 * @param {Object} resources k-v pairs with resource name as keys and resource ammount as values
 * @returns {boolean} whether the user is approved to allocate requested resources
 */
function requestResources (user, resources) {
	let approved = true;
	Object.keys(resources).forEach((element) => {
		if(!(element in db[user].available)) { // if the resource does not exist in the user's entry, assume the user is not allowed to use it
			approved = false;
		}
		else if (db[user].available[element] - resources[element] < 0) {
			approved = false;
		}
	});
	return approved;
}

/**
 * user allocates additional resources specified in k-v pairs
 * @param {string} user user's proxmox username in the form username@authrealm
 * @param {Object} resources k-v pairs with resource name as keys and resource ammount as values
 * @returns {boolean} true if resources were successfully allocated, false otherwise
 */
function allocateResources (user, resources) {
	let newdb = {};
	Object.assign(newdb, db);
	Object.keys(resources).forEach((element) => {
		if(typeof(resources[element]) === "number" && isFinite(resources[element])) {
			newdb[user].available[element] -= resources[element];
		}
		else {
			return false;
		}
	});
	try {
		fs.writeFileSync(filename, JSON.stringify(newdb));
		Object.assign(db, newdb);
		return true;
	}
	catch {
		fs.writeFileSync(filename, JSON.stringify(db))
		return false;
	}
}

/**
 * user releases allocated resources specified in k-v pairs
 * @param {string} user user's proxmox username in the form username@authrealm
 * @param {Object} resources k-v pairs with resource name as keys and resource ammount as values
 * @returns {boolean} true if resources were successfully deallocated, false otherwise
 */
function releaseResources (user, resources) {
	let newdb = {};
	Object.assign(newdb, db);
	Object.keys(resources).forEach((element) => {
		if(typeof(resources[element]) === "number" && isFinite(resources[element]) && resources[element]) {
			newdb[user].available[element] += resources[element];
		}
		else {
			return false;
		}
	});
	try {
		fs.writeFileSync(filename, JSON.stringify(newdb));
		Object.assign(db, newdb);
		return true;
	}
	catch {
		fs.writeFileSync(filename, JSON.stringify(db))
		return false;
	}
}

/**
 * return a read only copy of the user resources
 * @param {string} user user's proxmox username in the form username@authrealm
 * @returns {Object} user's remaining resources as k-v pairs with resource name as keys and resource ammount as values
 */
function getResources (user) {
	let returnVal = {};
	if(user in db) {
		Object.assign(returnVal, db[user]);
	}
	return returnVal;
}

module.exports = {init, requestResources, allocateResources, releaseResources, getResources};