From 7e0bce1dca7dbfbeaa1445ea2de7536b2cfb5eb9 Mon Sep 17 00:00:00 2001
From: Arthur Lu <learthurgo@gmail.com>
Date: Wed, 19 Apr 2023 02:46:05 +0000
Subject: [PATCH] add unit return to resources path

---
 main.js | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/main.js b/main.js
index 9a808ad..0d39efa 100644
--- a/main.js
+++ b/main.js
@@ -8,7 +8,7 @@ var api = require("./package.json");
 
 const {pveAPIToken, listenPort, domain} = require("./vars.js");
 const {checkAuth, requestPVE, handleResponse, getUsedResources, getDiskInfo} = require("./pveutils.js");
-const {init, getResourceMeta, getUser, approveResources, setUsedResources} = require("./db.js");
+const {init, getResourceMeta, getUser, approveResources, setUsedResources, getResourceUnits} = require("./db.js");
 
 const app = express();
 app.use(helmet());
@@ -49,6 +49,7 @@ app.get("/api/user/resources", async(req, res) => {
 	let used = await getUsedResources(req, getResourceMeta());
 	setUsedResources(req.cookies.username, used);
 	let user = await getUser(req.cookies.username);
+	user.units = getResourceUnits();
 	res.status(200).send(user);
 	res.end();
 	return;