From 59ccc3a21576393388ac072d5b4b3769911fb386 Mon Sep 17 00:00:00 2001
From: myztillx <33730898+myztillx@users.noreply.github.com>
Date: Tue, 26 Dec 2023 23:21:57 -0500
Subject: [PATCH] Added Option for Grafana Alertmanager (#2498)

---------

Co-authored-by: shamoon <4887959+shamoon@users.noreply.github.com>
---
 src/widgets/grafana/component.jsx | 20 ++++++++++++++------
 src/widgets/grafana/widget.js     |  3 +++
 2 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/src/widgets/grafana/component.jsx b/src/widgets/grafana/component.jsx
index 32d2a98e5..ecedac8ee 100755
--- a/src/widgets/grafana/component.jsx
+++ b/src/widgets/grafana/component.jsx
@@ -10,12 +10,23 @@ export default function Component({ service }) {
   const { widget } = service;
   const { data: statsData, error: statsError } = useWidgetAPI(widget, "stats");
   const { data: alertsData, error: alertsError } = useWidgetAPI(widget, "alerts");
+  const { data: alertmanagerData, error: alertmanagerError } = useWidgetAPI(widget, "alertmanager");
 
-  if (statsError || alertsError) {
+  let alertsInt = 0;
+
+  if (alertsError || !alertsData || alertsData.length === 0) {
+    if (alertmanagerData) {
+      alertsInt = alertmanagerData.length;
+    }
+  } else {
+    alertsInt = alertsData.filter((a) => a.state === "alerting").length;
+  }
+
+  if (statsError || (alertsError && alertmanagerError)) {
     return <Container service={service} error={statsError ?? alertsError} />;
   }
 
-  if (!statsData || !alertsData) {
+  if (!statsData || (!alertsData && !alertmanagerData)) {
     return (
       <Container service={service}>
         <Block label="grafana.dashboards" />
@@ -31,10 +42,7 @@ export default function Component({ service }) {
       <Block label="grafana.dashboards" value={t("common.number", { value: statsData.dashboards })} />
       <Block label="grafana.datasources" value={t("common.number", { value: statsData.datasources })} />
       <Block label="grafana.totalalerts" value={t("common.number", { value: statsData.alerts })} />
-      <Block
-        label="grafana.alertstriggered"
-        value={t("common.number", { value: alertsData.filter((a) => a.state === "alerting").length })}
-      />
+      <Block label="grafana.alertstriggered" value={t("common.number", { value: alertsInt })} />
     </Container>
   );
 }
diff --git a/src/widgets/grafana/widget.js b/src/widgets/grafana/widget.js
index af3e56eaa..2805555f8 100755
--- a/src/widgets/grafana/widget.js
+++ b/src/widgets/grafana/widget.js
@@ -8,6 +8,9 @@ const widget = {
     alerts: {
       endpoint: "alerts",
     },
+    alertmanager: {
+      endpoint: "alertmanager/grafana/api/v2/alerts",
+    },
     stats: {
       endpoint: "admin/stats",
       validate: ["dashboards"],