From 17ebdd8226ecaff930b666b7a16682805358fd31 Mon Sep 17 00:00:00 2001
From: shamoon <4887959+shamoon@users.noreply.github.com>
Date: Fri, 3 Feb 2023 01:12:52 -0800
Subject: [PATCH] Unifi widgets support sites other than default

---
 src/components/widgets/unifi_console/unifi_console.jsx | 3 +--
 src/utils/config/service-helpers.js                    | 4 ++++
 src/widgets/unifi/component.jsx                        | 2 +-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/components/widgets/unifi_console/unifi_console.jsx b/src/components/widgets/unifi_console/unifi_console.jsx
index 156e2866..13c90bd4 100644
--- a/src/components/widgets/unifi_console/unifi_console.jsx
+++ b/src/components/widgets/unifi_console/unifi_console.jsx
@@ -20,7 +20,6 @@ export default function Widget({ options }) {
             <BiError className="w-8 h-8 text-theme-800 dark:text-theme-200" />
             <div className="flex flex-col ml-3 text-left">
               <span className="text-theme-800 dark:text-theme-200 text-sm">{t("widget.api_error")}</span>
-              <span className="text-theme-800 dark:text-theme-200 text-xs">-</span>
             </div>
           </div>
         </div>
@@ -28,7 +27,7 @@ export default function Widget({ options }) {
     );
   }
 
-  const defaultSite = statsData?.data?.find(s => s.name === "default");
+  const defaultSite = options.site ? statsData?.data.find(s => s.desc === options.site) : statsData?.data?.find(s => s.name === "default");
 
   if (!defaultSite) {
     return (
diff --git a/src/utils/config/service-helpers.js b/src/utils/config/service-helpers.js
index 352367d2..e2e3055e 100644
--- a/src/utils/config/service-helpers.js
+++ b/src/utils/config/service-helpers.js
@@ -233,6 +233,7 @@ export function cleanServiceGroups(groups) {
           currency, // coinmarketcap widget
           symbols,
           defaultinterval,
+          site, // unifi widget
           namespace, // kubernetes widget
           app,
           podSelector,
@@ -256,6 +257,9 @@ export function cleanServiceGroups(groups) {
           if (server) cleanedService.widget.server = server;
           if (container) cleanedService.widget.container = container;
         }
+        if (type === "unifi") {
+          if (site) cleanedService.widget.site = site;
+        }
         if (type === "kubernetes") {
           if (namespace) cleanedService.widget.namespace = namespace;
           if (app) cleanedService.widget.app = app;
diff --git a/src/widgets/unifi/component.jsx b/src/widgets/unifi/component.jsx
index 25ba8ea2..ff318516 100644
--- a/src/widgets/unifi/component.jsx
+++ b/src/widgets/unifi/component.jsx
@@ -15,7 +15,7 @@ export default function Component({ service }) {
         return <Container error={statsError} />;
     }
 
-    const defaultSite = statsData?.data?.find(s => s.name === "default");
+    const defaultSite = widget.site ? statsData?.data.find(s => s.desc === widget.site) : statsData?.data?.find(s => s.name === "default");
 
     if (!defaultSite) {
         return (