From 730f1c5ec10bacb28b47fa6b137ff1557e76b46e Mon Sep 17 00:00:00 2001
From: Michael Shamoon <4887959+shamoon@users.noreply.github.com>
Date: Sat, 7 Jan 2023 10:00:25 -0800
Subject: [PATCH] Support api key + username / pass for truenas widget

---
 src/widgets/truenas/widget.js | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/src/widgets/truenas/widget.js b/src/widgets/truenas/widget.js
index 4c479e9f..3139019b 100644
--- a/src/widgets/truenas/widget.js
+++ b/src/widgets/truenas/widget.js
@@ -1,9 +1,31 @@
 import { jsonArrayFilter } from "utils/proxy/api-helpers";
+import credentialedProxyHandler from "utils/proxy/handlers/credentialed";
 import genericProxyHandler from "utils/proxy/handlers/generic";
+import getServiceWidget from "utils/config/service-helpers";
 
 const widget = {
   api: "{url}/api/v2.0/{endpoint}",
-  proxyHandler: genericProxyHandler,
+  proxyHandler: async (req, res, map) => { // choose proxy handler based on widget settings
+    const { group, service } = req.query;
+  
+    if (group && service) {
+      const widgetOpts = await getServiceWidget(group, service);
+      let handler;
+      if (widgetOpts.username && widgetOpts.password) {
+        handler = genericProxyHandler;
+      } else if (widgetOpts.key) {
+        handler = credentialedProxyHandler;
+      }
+
+      if (handler) {
+        return handler(req, res, map);
+      } 
+        
+      return res.status(500).json({ error: "Username / password or API key required" });
+    }
+    
+    return res.status(500).json({ error: "Error parsing widget request" });
+  },
 
   mappings: {
     alerts: {