From 4c23ac5c5ee09a879ba442db103e48c2baeae399 Mon Sep 17 00:00:00 2001
From: Arthur Lu <root@tronnet.net>
Date: Mon, 9 Jan 2023 22:24:04 -0800
Subject: [PATCH] add logic for disk and drive icon

---
 scripts/config.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/config.js b/scripts/config.js
index 3ac1063..7b02c4f 100644
--- a/scripts/config.js
+++ b/scripts/config.js
@@ -32,7 +32,7 @@ async function populateForm (node, type, vmid) {
 	let diskTypes;
 	if (type === "lxc") {
 		addResourceLine("resources", "images/resources/swap.svg", "Swap", {type: "number", value: config.data.swap, min: 0, step: 1}, "GiB"); // TODO add max from quota API
-		addDiskLine("disks", "rootfs", "images/resources/disk.svg", "Root FS", config.data.rootfs);
+		addDiskLine("disks", "rootfs", "images/resources/drive.svg", "Root FS", config.data.rootfs);
 		diskPrefixes = ["mp"];
 		diskTypes = ["MP"];
 	}
@@ -47,7 +47,7 @@ async function populateForm (node, type, vmid) {
 
 		Object.keys(config.data).forEach(element => {
 			if (element.startsWith(prefix)) {
-				addDiskLine("disks", element, "images/resources/disk.svg", `${type} ${element.replace(prefix, "")}`, config.data[element]);
+				addDiskLine("disks", element, config.data[element].includes("media=cdrom") ? "images/resources/disk.svg" : "images/resources/drive.svg", `${type} ${element.replace(prefix, "")}`, config.data[element]);
 			}
 		})