From 40f31ba0a2aeb50c8e93f1f2ac8a7d608d97bf2a Mon Sep 17 00:00:00 2001
From: tk422 <not@gmail.com>
Date: Wed, 15 Nov 2023 18:01:59 -0500
Subject: [PATCH] Add map to down for Alertmanager

---
 Nagstamon/Config.py                                  |  1 +
 Nagstamon/QUI/__init__.py                            |  2 ++
 Nagstamon/Servers/Alertmanager/alertmanagerserver.py |  5 +++++
 Nagstamon/resources/qui/settings_server.ui           | 10 ++++++++++
 4 files changed, 18 insertions(+)

diff --git a/Nagstamon/Config.py b/Nagstamon/Config.py
index 3e86d896e..e780d5015 100644
--- a/Nagstamon/Config.py
+++ b/Nagstamon/Config.py
@@ -969,6 +969,7 @@ def __init__(self):
         self.alertmanager_filter = ''
         self.map_to_critical = 'critical,error'
         self.map_to_warning = 'warning,warn'
+        self.map_to_down = 'down'
         self.map_to_unknown = 'unknown'
         self.map_to_ok = 'ok'
 
diff --git a/Nagstamon/QUI/__init__.py b/Nagstamon/QUI/__init__.py
index 5bf09653e..7eb881802 100644
--- a/Nagstamon/QUI/__init__.py
+++ b/Nagstamon/QUI/__init__.py
@@ -5815,6 +5815,8 @@ def __init__(self, dialog):
             self.window.input_lineedit_map_to_warning: ['Alertmanager'],
             self.window.label_map_to_critical: ['Alertmanager'],
             self.window.input_lineedit_map_to_critical: ['Alertmanager'],
+            self.window.label_map_to_down: ['Alertmanager'],
+            self.window.input_lineedit_map_to_down: ['Alertmanager'],
             self.window.input_lineedit_notification_filter: ['IcingaDBWebNotifications'],
             self.window.label_notification_filter: ['IcingaDBWebNotifications'],
             self.window.input_lineedit_notification_lookback: ['IcingaDBWebNotifications'],
diff --git a/Nagstamon/Servers/Alertmanager/alertmanagerserver.py b/Nagstamon/Servers/Alertmanager/alertmanagerserver.py
index eb578fc6b..f65c3400d 100644
--- a/Nagstamon/Servers/Alertmanager/alertmanagerserver.py
+++ b/Nagstamon/Servers/Alertmanager/alertmanagerserver.py
@@ -47,6 +47,7 @@ class AlertmanagerServer(GenericServer):
     map_to_status_information = ''
     map_to_critical = ''
     map_to_warning = ''
+    map_to_down = ''
     map_to_unknown = ''
     map_to_ok = ''
     name = ''
@@ -96,6 +97,8 @@ def map_severity(self, the_severity):
             return "CRITICAL"
         if the_severity in self.map_to_warning.split(','):
             return "WARNING"
+        if the_severity in self.map_to_down.split(','):
+            return "DOWN"
         if the_severity in self.map_to_ok.split(','):
             return "OK"
         return the_severity.upper()
@@ -183,6 +186,8 @@ def _get_status(self):
                   self.map_to_critical)
         log.debug("severity config (map_to_warning): '%s'",
                   self.map_to_warning)
+        log.debug("severity config (map_to_down): '%s'",
+                  self.map_to_down)
         log.debug("severity config (map_to_ok): '%s'",
                   self.map_to_ok)
 
diff --git a/Nagstamon/resources/qui/settings_server.ui b/Nagstamon/resources/qui/settings_server.ui
index 170f57e2c..eedf4119c 100644
--- a/Nagstamon/resources/qui/settings_server.ui
+++ b/Nagstamon/resources/qui/settings_server.ui
@@ -737,6 +737,9 @@
        <widget class="QLineEdit" name="input_lineedit_map_to_critical"/>
       </item>
       <item row="19" column="2" colspan="2">
+       <widget class="QLineEdit" name="input_lineedit_map_to_down"/>
+      </item>
+      <item row="20" column="2" colspan="2">
        <widget class="QLineEdit" name="input_lineedit_map_to_unknown"/>
       </item>
       <item row="16" column="1">
@@ -761,6 +764,13 @@
        </widget>
       </item>
       <item row="19" column="1">
+       <widget class="QLabel" name="label_map_to_down">
+        <property name="text">
+         <string>Map to DOWN</string>
+        </property>
+       </widget>
+      </item>
+      <item row="20" column="1">
        <widget class="QLabel" name="label_map_to_unknown">
         <property name="text">
          <string>Map to UNKNOWN</string>