From ad08f69df1bc0ebd8688d2eda66ce437c3a32793 Mon Sep 17 00:00:00 2001
From: Yannik Schmidt <sheppy@atlantishq.de>
Date: Mon, 1 Jan 2024 20:41:29 +0100
Subject: [PATCH] wip:

---
 icingatools.py                    | 12 ++++++++++--
 server.py                         | 30 +++++++++++++++++++++++++++---
 templates/add_modify_service.html |  3 +++
 templates/service_info.html       |  5 +++++
 4 files changed, 45 insertions(+), 5 deletions(-)

diff --git a/icingatools.py b/icingatools.py
index d48c767..bb5a295 100644
--- a/icingatools.py
+++ b/icingatools.py
@@ -83,7 +83,15 @@ def build_icinga_link_for_service(user, service_name, static_configured, app):
         url_fmt = "{base}/icingaweb2/monitoring/list/services?service={service}&modifyFilter=1"
         name = service_name
 
-    return url_fmt.format(base=app.config["ICINGA_WEB_URL"],
-                            host=app.config["ASYNC_ICINGA_DUMMY_HOST"],
+    icinga_web_url = app.config.get("ICINGA_WEB_URL")
+    if not icinga_web_url:
+        icinga_web_url = "ICINGA_WEB_URL_NOT_SET:"
+
+    dummy_host=app.config.get("ASYNC_ICINGA_DUMMY_HOST")
+    if not dummy_host:
+        dummy_host = "ASYNC_ICINGA_DUMMY_HOST_NOT_SET:"
+
+    return url_fmt.format(base=icinga_web_url,
+                            host=dummy_host,
                             service=name)
 
diff --git a/server.py b/server.py
index 85bf7ea..0da59fa 100755
--- a/server.py
+++ b/server.py
@@ -12,7 +12,7 @@ import secrets
 
 import flask_wtf
 from flask_wtf import FlaskForm
-from wtforms import StringField, SubmitField, BooleanField, DecimalField, HiddenField
+from wtforms import StringField, SubmitField, BooleanField, DecimalField, HiddenField, SelectField
 from wtforms.validators import DataRequired, Length
 
 from sqlalchemy import Column, Integer, String, Boolean, or_, and_
@@ -41,6 +41,7 @@ class Service(db.Model):
     token   = Column(String)
     timeout = Column(Integer)
     owner   = Column(String)
+    special_type = Column(String)
 
     staticly_configured = Column(Boolean)
 
@@ -57,6 +58,19 @@ class Status(db.Model):
         dt = datetime.datetime.fromtimestamp(self.timestamp)
         return dt.strftime("%d. %B %Y at %H:%M")
 
+class SMARTStatus(db.Model):
+
+    __tablename__ = "smart"
+
+    service     = Column(String, primary_key=True)
+    timestamp   = Column(Integer, primary_key=True)
+    power_cycles = Column(Integer)
+    temperatur   = Column(Integer)
+    avail_spare  = Column(Integer)
+    unsafe_shutdowns  = Column(Integer)
+    critical_warning = Column(Integer)
+    model_number = Column(String)
+
 def buildReponseDict(status, service=None):
 
     if not status:
@@ -108,6 +122,7 @@ class EntryForm(FlaskForm):
 
     service = StringField("Service Name")
     service_hidden = HiddenField("service_hidden")
+    special_type = SelectField("Type", choices=["Default", "SMART"])
     timeout = DecimalField("Timeout in days", default=30)
 
 def create_entry(form, user):
@@ -117,8 +132,13 @@ def create_entry(form, user):
     service_name = form.service.data or form.service_hidden.data
 
     day_delta = datetime.timedelta(days=int(form.timeout.data))
+
+    special_type = form.special_type.data
+    if form.special_type == "Default":
+        special_type = None
+
     service = Service(service=service_name, timeout=day_delta.total_seconds(),
-                        owner=user, token=token)
+                        owner=user, token=token, special_type=special_type)
 
     # service.data set = create, service_hidden.data = modify #
     if form.service.data:
@@ -148,8 +168,11 @@ def service_details():
     icinga_link = icingatools.build_icinga_link_for_service(user, service.service,
                         service.staticly_configured, app)
 
+    smart_entry_list = db.session.query(SMARTStatus).filter(SMARTStatus.service==service.service)
+    smart_entry = smart_entry_list.order_by(SMARTStatus.timestamp.desc()).first()
+
     return flask.render_template("service_info.html", service=service, flask=flask,
-                                    user=user, status_list=status_list, icinga_link=icinga_link)
+                    user=user, status_list=status_list, icinga_link=icinga_link, smart_entry=smart_entry)
 
 
 @app.route("/entry-form", methods=["GET", "POST", "DELETE"])
@@ -182,6 +205,7 @@ def create_interface():
         service = db.session.query(Service).filter(Service.service == modify_service_name).first()
         if service and service.owner == user:
             form.service.default = service.service
+            form.special_type.default = service.special_type
             form.timeout.default = datetime.timedelta(seconds=service.timeout).days
             form.service_hidden.default = service.service
             form.process()
diff --git a/templates/add_modify_service.html b/templates/add_modify_service.html
index 6c7ab9a..4176c3b 100644
--- a/templates/add_modify_service.html
+++ b/templates/add_modify_service.html
@@ -37,6 +37,9 @@
                 </br>
                 {{ form.timeout.label }}    {{ form.timeout() }}        </br>
 
+                </br>
+                {{ form.special_type.label }}    {{ form.special_type() }}        </br>
+
                 {% if is_modification %}
                 <input class="form-button mt-4" type="submit" value="Send Modification">
                 {% else %}
diff --git a/templates/service_info.html b/templates/service_info.html
index a03aac4..cfbb6a5 100644
--- a/templates/service_info.html
+++ b/templates/service_info.html
@@ -33,6 +33,11 @@
             <h2 class="service-name">Service: {{ service.service }}</h2>
             {% if service.staticly_configured %}
             {% endif %}
+            {% if service.special_type == "SMART" %}
+            <div class="service-info-button mt-3" style="background-color: orange;">
+                Smart Monitor {% if smart_entry %} for: {{ smart.model_number }} {% endif %}
+            </div>
+            {% endif %}
 
             {% if not service.staticly_configured %}
             <a class="service-info-button mt-3" style="background-color: orange;"