107 lines
3.1 KiB
JavaScript
107 lines
3.1 KiB
JavaScript
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])) { // 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][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][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][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}; |