209 lines
8.1 KiB
JavaScript
209 lines
8.1 KiB
JavaScript
import axios from "axios";
|
|
import { pveAPI, pveAPIToken } from "./db.js";
|
|
|
|
/**
|
|
* Send HTTP request to proxmox API. Allows requests to be made with user cookie credentials or an API token for controlled priviledge elevation.
|
|
* @param {string} path HTTP path, prepended with the proxmox API base path.
|
|
* @param {string} method HTTP method.
|
|
* @param {Object} cookies user cookies for authorization if an API token is not used. Safest option for authentication.
|
|
* @param {string} body body parameters and data to be sent. Optional.
|
|
* @param {string} token proxmox API token to be used for controled priviledge elevation, allows user requests to perform admin actions safely. Optional
|
|
* @returns {Obejct} HTTP response object or HTTP error object
|
|
*/
|
|
export async function requestPVE (path, method, cookies, body = null, token = null) {
|
|
const url = `${pveAPI}${path}`;
|
|
const content = {
|
|
method,
|
|
mode: "cors",
|
|
credentials: "include",
|
|
headers: {
|
|
"Content-Type": "application/x-www-form-urlencoded"
|
|
}
|
|
};
|
|
|
|
if (token) {
|
|
content.headers.Authorization = `PVEAPIToken=${token.user}@${token.realm}!${token.id}=${token.uuid}`;
|
|
}
|
|
else if (cookies) {
|
|
content.headers.CSRFPreventionToken = cookies.CSRFPreventionToken;
|
|
content.headers.Cookie = `PVEAuthCookie=${cookies.PVEAuthCookie}; CSRFPreventionToken=${cookies.CSRFPreventionToken}`;
|
|
}
|
|
|
|
if (body) {
|
|
content.data = JSON.parse(body);
|
|
}
|
|
|
|
try {
|
|
const response = await axios.request(url, content);
|
|
return response;
|
|
}
|
|
catch (error) {
|
|
return error.response;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Handle various proxmox API responses. Handles sync and async responses.
|
|
* In sync responses, responses are completed when the response arrives. Method returns the response directly.
|
|
* In async responses, proxmox sends responses with a UPID to track process completion. Method returns the status of the proxmox process once it completes.
|
|
* @param {string} node response originates from.
|
|
* @param {Object} result response from proxmox.
|
|
* @param {Object} res response object of ProxmoxAAS API call.
|
|
*/
|
|
export async function handleResponse (node, result, res) {
|
|
const waitFor = delay => new Promise(resolve => setTimeout(resolve, delay));
|
|
if (result.data.data && typeof (result.data.data) === "string" && result.data.data.startsWith("UPID:")) {
|
|
const upid = result.data.data;
|
|
let taskStatus = await requestPVE(`/nodes/${node}/tasks/${upid}/status`, "GET", null, null, pveAPIToken);
|
|
while (taskStatus.data.data.status !== "stopped") {
|
|
await waitFor(1000);
|
|
taskStatus = await requestPVE(`/nodes/${node}/tasks/${upid}/status`, "GET", null, null, pveAPIToken);
|
|
}
|
|
if (taskStatus.data.data.exitstatus === "OK") {
|
|
const result = taskStatus.data.data;
|
|
const taskLog = await requestPVE(`/nodes/${node}/tasks/${upid}/log`, "GET", null, null, pveAPIToken);
|
|
result.log = taskLog.data.data;
|
|
res.status(200).send(result);
|
|
res.end();
|
|
}
|
|
else {
|
|
const result = taskStatus.data.data;
|
|
const taskLog = await requestPVE(`/nodes/${node}/tasks/${upid}/log`, "GET", null, null, pveAPIToken);
|
|
result.log = taskLog.data.data;
|
|
res.status(500).send(result);
|
|
res.end();
|
|
}
|
|
}
|
|
else {
|
|
res.status(result.status).send(result.data);
|
|
res.end();
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get the amount of resources used by specified user.
|
|
* @param {Object} req ProxmoxAAS API request object.
|
|
* @param {Object} resourceMeta data about application resources, to indicate which resources are tracked.
|
|
* @returns {Object} k-v pairs of resource name and used amounts
|
|
*/
|
|
export async function getUsedResources (req, resourceMeta) {
|
|
const response = await requestPVE("/cluster/resources", "GET", req.cookies);
|
|
const used = {};
|
|
const diskprefixes = [];
|
|
for (const resourceName of Object.keys(resourceMeta)) {
|
|
if (resourceMeta[resourceName].type === "storage") {
|
|
used[resourceName] = 0;
|
|
for (const diskPrefix of resourceMeta[resourceName].disks) {
|
|
diskprefixes.push(diskPrefix);
|
|
}
|
|
}
|
|
else if (resourceMeta[resourceName].type === "list") {
|
|
used[resourceName] = [];
|
|
}
|
|
else {
|
|
used[resourceName] = 0;
|
|
}
|
|
}
|
|
for (const instance of response.data.data) {
|
|
if (instance.type === "lxc" || instance.type === "qemu") {
|
|
let config = await requestPVE(`/nodes/${instance.node}/${instance.type}/${instance.vmid}/config`, "GET", req.cookies);
|
|
config = config.data.data;
|
|
for (const key of Object.keys(config)) {
|
|
if (Object.keys(used).includes(key) && resourceMeta[key].type === "numeric") {
|
|
used[key] += Number(config[key]);
|
|
}
|
|
else if (diskprefixes.some(prefix => key.startsWith(prefix))) {
|
|
const diskInfo = await getDiskInfo(instance.node, instance.type, instance.vmid, key);
|
|
if (diskInfo) { // only count if disk exists
|
|
used[diskInfo.storage] += Number(diskInfo.size);
|
|
}
|
|
}
|
|
else if (key.startsWith("net") && config[key].includes("rate=")) { // only count net instances with a rate limit
|
|
used.network += Number(config[key].split("rate=")[1].split(",")[0]);
|
|
}
|
|
else if (key.startsWith("hostpci")) {
|
|
const deviceInfo = await getDeviceInfo(instance.node, instance.type, instance.vmid, config[key].split(",")[0]);
|
|
if (deviceInfo) { // only count if device exists
|
|
used.pci.push(deviceInfo.device_name);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return used;
|
|
}
|
|
|
|
/**
|
|
* Get meta data for a specific disk. Adds info that is not normally available in a instance's config.
|
|
* @param {string} node containing the query disk.
|
|
* @param {string} type of instance with query disk.
|
|
* @param {string} vmid of instance with query disk
|
|
* @param {string} disk name of the query disk, ie. sata0.
|
|
* @returns {Objetc} k-v pairs of specific disk data, including storage and size of unused disks.
|
|
*/
|
|
export async function getDiskInfo (node, type, vmid, disk) {
|
|
try {
|
|
const config = await requestPVE(`/nodes/${node}/${type}/${vmid}/config`, "GET", null, null, pveAPIToken);
|
|
const storageID = config.data.data[disk].split(":")[0];
|
|
const volID = config.data.data[disk].split(",")[0];
|
|
const volInfo = await requestPVE(`/nodes/${node}/storage/${storageID}/content/${volID}`, "GET", null, null, pveAPIToken);
|
|
volInfo.data.data.storage = storageID;
|
|
return volInfo.data.data;
|
|
}
|
|
catch {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get meta data for a specific pci device. Adds info that is not normally available in a instance's config.
|
|
* @param {string} node containing the query device.
|
|
* @param {string} type of instance with query device.
|
|
* @param {string} vmid of instance with query device.
|
|
* @param {string} qid pci bus id number of the query device, ie. 89ab:cd:ef.0.
|
|
* @returns {Object} k-v pairs of specific device data, including device name and manufacturer.
|
|
*/
|
|
export async function getDeviceInfo (node, type, vmid, qid) {
|
|
try {
|
|
const result = (await requestPVE(`/nodes/${node}/hardware/pci`, "GET", null, null, pveAPIToken)).data.data;
|
|
const deviceData = [];
|
|
result.forEach((element) => {
|
|
if (element.id.startsWith(qid)) {
|
|
deviceData.push(element);
|
|
}
|
|
});
|
|
deviceData.sort((a, b) => {
|
|
return a.id < b.id;
|
|
});
|
|
const device = deviceData[0];
|
|
device.subfn = structuredClone(deviceData.slice(1));
|
|
return device;
|
|
}
|
|
catch {
|
|
return null;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get available devices on specific node.
|
|
* @param {string} node to get devices from.
|
|
* @param {Object} cookies user authentication, unused since the API token is required.
|
|
* @returns {Array.<Object>} array of k-v pairs of specific device data, including device name and manufacturer, which are available on the specified node.
|
|
*/
|
|
export async function getNodeAvailDevices (node, cookies) {
|
|
// get node pci devices
|
|
let nodeAvailPci = (await requestPVE(`/nodes/${node}/hardware/pci`, "GET", cookies, null, pveAPIToken)).data.data;
|
|
// for each node container, get its config and remove devices which are already used
|
|
const vms = (await requestPVE(`/nodes/${node}/qemu`, "GET", cookies, null, pveAPIToken)).data.data;
|
|
for (const vm of vms) {
|
|
const config = (await requestPVE(`/nodes/${node}/qemu/${vm.vmid}/config`, "GET", cookies, null, pveAPIToken)).data.data;
|
|
Object.keys(config).forEach((key) => {
|
|
if (key.startsWith("hostpci")) {
|
|
const deviceID = config[key].split(",")[0];
|
|
nodeAvailPci = nodeAvailPci.filter(element => !element.id.includes(deviceID));
|
|
}
|
|
});
|
|
}
|
|
return nodeAvailPci;
|
|
}
|