import { Router } from "express";
export const router = Router({ mergeParams: true }); ;

/**
 * GET - get available pcie devices for the given node and user
 * request:
 * - node: string - vm host node id
 * responses:
 * - 200: PVE PCI Device Object
 * - 401: {auth: false}
 * - 401: {auth: false, path: string}
 * - 500: {error: string}
 */
router.get("/", async (req, res) => {
	const params = {
		node: req.params.node,
		type: req.params.type,
		vmid: req.params.vmid,
	};

	// check auth
	const auth = await global.utils.checkAuth(req.cookies, res);
	if (!auth) {
		return;
	}

	// get instance config for pool membership
	const instance = await global.pve.getInstance(params.node, params.vmid);

	// ensure that requested instance type is vmid
	if (instance.type !== "VM") {
		res.status(400).send({ auth: true, error: `actual instance type is ${instance.type} but must be VM` });
	}
	else if (params.type !== "qemu") {
		res.status(400).send({ auth: true, error: `requested instance type is ${params.type} but must be qemu` });
	}

	// get pool and pool allowed nodes
	const pool = await global.access.getPool(instance.pool, req.cookies);
	const poolNodes = pool.pool["nodes-allowed"];
	if (poolNodes[params.node] !== true) { // user does not have access to the node
		res.status(401).send({ auth: false, path: params.node });
		res.end();
		return;
	}

	// get remaining user resources
	const poolAvailPci = (await global.utils.getPoolResources(req, instance.pool)).pci.nodes[params.node]; // we assume that the node list is used. TODO support global lists
	if (poolAvailPci === undefined) { // user has no available devices on this node, so send an empty list
		res.status(200).send([]);
		res.end();
	}
	else {
		// get node avail devices
		const node = await global.pve.getNode(params.node);
		let availableDevices = [];
		// get each device and filter out only thise which are not reserved
		for (const device of Object.values(node.devices)) {
			if (device.reserved === false) {
				availableDevices.push(device);
			}
		}
		//further filter out only devices which the user has access to
		availableDevices = availableDevices.filter(nodeAvail => poolAvailPci.some((userAvail) => {
			return nodeAvail.device_name && nodeAvail.device_name.includes(userAvail.match) && userAvail.avail > 0;
		}));

		res.status(200).send(availableDevices);
		res.end();
	}
});

/**
 * GET - get instance pcie device data
 * request:
 * - node: string - vm host node id
 * - type: string - vm type (lxc, qemu)
 * - vmid: number - vm id number to destroy
 * - hostpci: string - hostpci number
 * responses:
 * - 200: PVE PCI Device Object
 * - 401: {auth: false, path: string}
 * - 500: {error: string}
 */
router.get("/:hostpci", async (req, res) => {
	req.params = Object.assign({}, req.routeparams, req.params);
	const params = {
		node: req.params.node,
		type: req.params.type,
		vmid: req.params.vmid,
		hostpci: req.params.hostpci
	};

	// check auth for specific instance
	const vmpath = `/nodes/${params.node}/${params.type}/${params.vmid}`;
	const auth = await global.utils.checkAuth(req.cookies, res, vmpath);
	if (!auth) {
		return;
	}

	// get device
	const device = await global.pve.getDevice(params.node, params.vmid, params.hostpci);
	if (!device) {
		res.status(500).send({ error: `Could not find ${params.hostpci}=${device} in ${params.node}.` });
		res.end();
		return;
	}
	res.status(200).send(device);
	res.end();
});

/**
 * POST - modify existing instance pci device
 * request:
 * - node: string - vm host node id
 * - type: string - vm type (lxc, qemu)
 * - vmid: number - vm id number to destroy
 * - hostpci: string - hostpci number
 * - device: string - new device id
 * - pcie: Boolean - whether to use pci express or pci
 * response:
 * - 200: PVE Task Object
 * - 401: {auth: false, path: string}
 * - 500: {request: Object, error: string}
 * - 500: PVE Task Object
 */
router.post("/:hostpci/modify", async (req, res) => {
	req.params = Object.assign({}, req.routeparams, req.params);
	const params = {
		node: req.params.node,
		type: req.params.type,
		vmid: req.params.vmid,
		hostpci: req.params.hostpci,
		device: req.body.device,
		pcie: req.body.pcie
	};

	// check if type is qemu
	if (params.type !== "qemu") {
		res.status(500).send({ error: "Type must be qemu (vm)." });
		res.end();
		return;
	}

	// check auth for specific instance
	const vmpath = `/nodes/${params.node}/${params.type}/${params.vmid}`;
	const auth = await global.utils.checkAuth(req.cookies, res, vmpath);
	if (!auth) {
		return;
	}

	// get instance config for pool membership
	const instance = await global.pve.getInstance(params.node, params.vmid);

	// force all functions
	params.device = params.device.split(".")[0];

	// device must exist to be modified
	const existingDevice = await global.pve.getDevice(params.node, params.vmid, params.hostpci);
	if (!existingDevice) {
		res.status(500).send({ error: `No device in ${params.hostpci}.` });
		res.end();
		return;
	}

	// only check user and node availability if base id is different, we do the split in case of existing partial-function hostpci
	const userObj = global.utils.getUserObjFromUsername(req.cookies.username);
	if (existingDevice.device_bus.split(".")[0] !== params.device) {
		// setup request
		const node = await global.pve.getNode(params.node);
		const requestedDevice = node.devices[`${params.device}`];
		const request = { pci: requestedDevice.device_name };
		if (!requestedDevice) {
			res.status(500).send({ request, error: `Could not fulfil request for ${params.device}.` });
			res.end();
			return;
		}
		// check resource approval
		const { approved } = await global.utils.approveResources(req, userObj, params.node, instance.pool, request);
		if (!approved) {
			res.status(500).send({ request, error: `Could not fulfil request for ${requestedDevice.device_name}.` });
			res.end();
			return;
		}
		// check node availability
		if (!Object.values(node.devices).some(element => element.device_bus.split(".")[0] === params.device && element.reserved === false)) {
			res.status(500).send({ error: `Device ${params.device} is already in use on ${params.node}.` });
			res.end();
			return;
		}
	}
	
	// setup action
	const action = {};
	action[`${params.hostpci}`] = `${params.device},pcie=${params.pcie}`;

	// commit action
	const result = await global.pve.requestPVE(`${vmpath}/config`, "POST", { root: true }, action);
	await global.pve.handleResponse(params.node, result, res);
	await global.pve.syncNode(params.node);
});

/**
 * POST - add new instance pci device
 * request:
 * - node: string - vm host node id
 * - type: string - vm type (lxc, qemu)
 * - vmid: number - vm id number to destroy
 * - device: string - new device id
 * - pcie: Boolean - whether to use pci express or pci
 * response:
 * - 200: PVE Task Object
 * - 401: {auth: false, path: string}
 * - 500: {request: Object, error: string}
 * - 500: PVE Task Object
 */
router.post("/:hostpci/create", async (req, res) => {
	req.params = Object.assign({}, req.routeparams, req.params);
	const params = {
		node: req.params.node,
		type: req.params.type,
		vmid: req.params.vmid,
		hostpci: req.params.hostpci,
		device: req.body.device,
		pcie: req.body.pcie
	};

	// check if type is qemu
	if (params.type !== "qemu") {
		res.status(500).send({ error: "Type must be qemu (vm)." });
		res.end();
		return;
	}

	// check auth for specific instance
	const vmpath = `/nodes/${params.node}/${params.type}/${params.vmid}`;
	const auth = await global.utils.checkAuth(req.cookies, res, vmpath);
	if (!auth) {
		return;
	}

	// get instance config for pool membership
	const instance = await global.pve.getInstance(params.node, params.vmid);

	// force all functions
	params.device = params.device.split(".")[0];

	// device must not exist to be added
	const existingDevice = await global.pve.getDevice(params.node, params.vmid, params.hostpci);
	if (existingDevice) {
		res.status(500).send({ error: `Existing device in ${params.hostpci}.` });
		res.end();
		return;
	}

	// setup request
	const node = await global.pve.getNode(params.node);
	const requestedDevice = node.devices[`${params.device}`];
	const request = { pci: requestedDevice.device_name };

	// check resource approval
	const userObj = global.utils.getUserObjFromUsername(req.cookies.username);
	const { approved } = await global.utils.approveResources(req, userObj, params.node, instance.pool, request);
	if (!approved) {
		res.status(500).send({ request, error: `Could not fulfil request for ${requestedDevice.device_name}.` });
		res.end();
		return;
	}

	// check node availability
	if (!Object.values(node.devices).some(element => element.device_bus.split(".")[0] === params.device && element.reserved === false)) {
		res.status(500).send({ error: `Device ${params.device} is already in use on ${params.node}.` });
		res.end();
		return;
	}

	// setup action
	const action = {};
	action[`${params.hostpci}`] = `${params.device},pcie=${params.pcie}`;

	// commit action
	const result = await global.pve.requestPVE(`${vmpath}/config`, "POST", { root: true }, action);
	await global.pve.handleResponse(params.node, result, res);
	await global.pve.syncNode(params.node);
});

/**
 * DELETE - delete instance pci device
 * request:
 * - node: string - vm host node id
 * - type: string - vm type (lxc, qemu)
 * - vmid: number - vm id number to destroy
 * - hostpci: string - hostpci number
 * response:
 * - 200: PVE Task Object
 * - 401: {auth: false, path: string}
 * - 500: {request: Object, error: string}
 * - 500: PVE Task Object
 */
router.delete("/:hostpci/delete", async (req, res) => {
	req.params = Object.assign({}, req.routeparams, req.params);
	const params = {
		node: req.params.node,
		type: req.params.type,
		vmid: req.params.vmid,
		hostpci: req.params.hostpci
	};

	// check if type is qemu
	if (params.type !== "qemu") {
		res.status(500).send({ error: "Type must be qemu (vm)." });
		res.end();
		return;
	}

	// check auth for specific instance
	const vmpath = `/nodes/${params.node}/${params.type}/${params.vmid}`;
	const auth = await global.utils.checkAuth(req.cookies, res, vmpath);
	if (!auth) {
		return;
	}

	// check device is in instance config
	const device = global.pve.getDevice(params.node, params.vmid, params.hostpci);
	if (!device) {
		res.status(500).send({ error: `Could not find ${params.hostpci} in ${params.vmid}.` });
		res.end();
		return;
	}

	// setup action
	const action = { delete: `${params.hostpci}` };

	// commit action, need to use root user here because proxmox api only allows root to modify hostpci for whatever reason
	const result = await global.pve.requestPVE(`${vmpath}/config`, "POST", { root: true }, action);
	await global.pve.handleResponse(params.node, result, res);
	await global.pve.syncNode(params.node);
});