From a36a82e28404a331ddb18413ca0377d2abe39a4c Mon Sep 17 00:00:00 2001
From: Yannik Enss <me@adnidor.de>
Date: Sat, 17 Apr 2021 14:20:54 +0200
Subject: [PATCH] add service-only mode

---
 service.py                  | 42 +++++++++++++++++++++++++++++++++++--
 services_only_template.html |  7 +++++++
 2 files changed, 47 insertions(+), 2 deletions(-)
 create mode 100644 services_only_template.html

diff --git a/service.py b/service.py
index e5b424c..7ad4af8 100755
--- a/service.py
+++ b/service.py
@@ -8,6 +8,7 @@ import datetime
 import sys
 urllib3.disable_warnings()
 
+
 class StatusPage:
     def do_api_calls(self):
         #services
@@ -19,8 +20,14 @@ class StatusPage:
         requestbody = {
             "attrs": [ "name", "state", "last_check_result", "host_name", "display_name" ],
             "joins": [ "host.name", "host.state", "host.last_check_result", "host.vars", "host.display_name" ],
-            "filter": self.config['filters']['services'],
         }
+        if "filters" in self.config:
+            requestbody["filter"] = self.config['filters']['services']
+        elif "services" in self.config:
+            requestbody["services"] = self.config["services"]["list"]
+        else:
+            raise Exception("One of 'filters' and 'services' required")
+
         r = requests.get(request_url,
             headers=headers,
             data=json.dumps(requestbody),
@@ -102,6 +109,34 @@ class StatusPage:
             html_output = html_output + self.render_services_per_host(host)
         return html_output
 
+    def render_service_only_details(self):
+        services_hostname_template = """
+                <div id="{0}" class="card-header d-flex justify-content-between align-items-center">
+                    <h4><a href="#{0}">{1}</a></h4>
+                    <span class="badge badge-{2}">{3}</span>
+                </div>"""
+
+        with open("services_only_template.html", "r") as f:
+            htmlTemplate = f.read()
+
+        htmlOutput = ""
+
+        for service in sorted(self.services['results'], key=lambda x: self.prettify(x['attrs']['display_name']).lower()):
+            if service['attrs']['state'] == 0:
+                card_header = services_hostname_template.format(service['attrs']['name'], self.prettify(service['attrs']['display_name']), 'success', 'OK')
+            elif service['attrs']['state'] == 1:
+                card_header = services_hostname_template.format(service['attrs']['name'], self.prettify(service['attrs']['display_name']), 'warning', 'WARNING')
+            else:
+                card_header = services_hostname_template.format(service['attrs']['name'], self.prettify(service['attrs']['display_name']), 'danger', 'CRITICAL')
+
+            htmlOutput += htmlTemplate.format(
+                card_header = card_header,
+                services_operational = "",
+                services_warning = "",
+                services_critical = ""
+            )
+        return htmlOutput
+
 
     def render_index_html(self, service_details):
         with open("template.html", "r") as f:
@@ -131,5 +166,8 @@ class StatusPage:
 if __name__ == "__main__":
     page = StatusPage()
     page.do_api_calls()
-    service_details = page.render_service_details()
+    if page.config['output']['services_only']:
+        service_details = page.render_service_only_details()
+    else:
+        service_details = page.render_service_details()
     page.render_index_html(service_details)
diff --git a/services_only_template.html b/services_only_template.html
new file mode 100644
index 0000000..4ced4a1
--- /dev/null
+++ b/services_only_template.html
@@ -0,0 +1,7 @@
+        <div class="row">
+            <div class="col">
+                <div class="card text-white border-primary mb-3">
+                    {card_header}
+                </div>
+            </div>
+        </div>