1 files changed, 414 insertions, 0 deletions

diff --git a/systems/monitoring-1/status/app.py b/systems/monitoring-1/status/app.py
new file mode 100755
index 0000000..ff92891
--- /dev/null
+++ b/systems/monitoring-1/status/app.py
@@ -0,0 +1,414 @@
+from flask import Flask, request, render_template_string, jsonify, make_response
+from flask_login import LoginManager, UserMixin, login_required
+import socket
+import json
+import time
+import os
+
+login_manager = LoginManager()
+app = Flask(__name__)
+login_manager.init_app(app)
+
+STATUS = [
+        "ok",
+        "warning",
+        "error",
+        "unknown"
+        ]
+
+HOST_STATUS = [
+        "up",
+        "down",
+        "unreachable",
+        ]
+
+#### Push
+AUTHORIZED_KEYS = os.environ.get("TOKENS", "").split()
+COMMAND_FILE = "/var/run/naemon/naemon.cmd"
+
+ERROR_NO_REQUEST_HANDLER="NO REQUEST HANDLER"
+ERROR_NO_TOKEN_SUPPLIED="NO TOKEN"
+ERROR_BAD_TOKEN_SUPPLIED="BAD TOKEN"
+
+ERROR_BAD_COMMAND_FILE="BAD COMMAND FILE"
+ERROR_COMMAND_FILE_OPEN_WRITE="COMMAND FILE UNWRITEABLE"
+ERROR_COMMAND_FILE_OPEN="CANNOT OPEN COMMAND FILE"
+ERROR_BAD_WRITE="WRITE ERROR"
+
+ERROR_BAD_DATA="BAD DATA"
+ERROR_BAD_JSON="BAD JSON"
+
+ERROR_NO_CORRECT_STATUS="NO STATUS WAS CORRECT"
+#### /Push
+
+def get_lq(request):
+    # https://mathias-kettner.de/checkmk_livestatus.html
+    socket_path="/var/run/naemon/live"
+    s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+    s.connect(socket_path)
+    s.send(request.encode())
+    s.shutdown(socket.SHUT_WR)
+    chunks = []
+    while len(chunks) == 0 or len(chunks[-1]) > 0:
+        chunks.append(s.recv(4096))
+    s.close()
+    return b"".join(chunks).decode()
+
+class Host:
+    def __init__(self, name, alias, status, webname, vhost):
+        self.name = name
+        self.alias = alias
+        self.webname = webname or alias
+        self.vhost = vhost
+        self.status = status
+        self.services = []
+
+    @classmethod
+    def parse_hosts(cls, payload, vhost):
+        parsed = filter(lambda x: x.vhost == vhost, [cls.parse(p) for p in json.loads(payload)])
+        return {p.name: p for p in parsed}
+
+    @classmethod
+    def parse(cls, payload):
+        return cls(payload[0], payload[1], HOST_STATUS[payload[2]], payload[3].get("WEBSTATUS_NAME"), payload[3].get("WEBSTATUS_VHOST"))
+
+    def __repr__(self):
+        return "Host {}: {} ({})".format(self.name, self.alias, self.webname)
+
+    @classmethod
+    def query(cls, vhost):
+        answer = get_lq("""GET hosts
+Filter: groups >= webstatus-hosts
+Columns: name alias state custom_variables
+OutputFormat: json
+""")
+        return cls.parse_hosts(answer, vhost)
+
+    def fill_services(self, services):
+        self.services = [service for service in services if service.host == self.name]
+
+class ServiceGroup:
+    def __init__(self, name, alias):
+        self.name = name
+        self.alias = alias
+        self.services = []
+
+    @classmethod
+    def parse_groups(cls, payload):
+        parsed = [cls.parse(p) for p in json.loads(payload)]
+        return {p.name: p for p in parsed}
+
+    @classmethod
+    def parse(cls, payload):
+        return cls(payload[0], payload[1])
+
+    @classmethod
+    def query(cls):
+        answer = get_lq("""GET servicegroups
+Filter: name ~ ^webstatus-
+Columns: name alias custom_variables
+OutputFormat: json
+""")
+        return cls.parse_groups(answer)
+
+    def fill_services(self, services, hosts):
+        self.services = [service for service in services if any([group == self.name for group in service.groups]) and service.host in hosts]
+
+    def __repr__(self):
+        return "ServiceGroup {}: {}".format(self.name, self.alias)
+
+class Service:
+    def __init__(self, name, host, groups, status, webname, url, description, infos):
+        self.name = name
+        self.host = host
+        self.groups = groups
+        self.status = status
+        self.webname = webname
+        self.url = url
+        self.description = description
+        self.infos = infos
+
+    @classmethod
+    def parse_services(cls, payload):
+        parsed = json.loads(payload)
+        return [cls.parse(p) for p in parsed if cls.valid(p[2])]
+
+    @staticmethod
+    def valid(groups):
+        return any([b.startswith("webstatus-") for b in groups])
+
+    @classmethod
+    def parse(cls, payload):
+        return cls(payload[0],
+                payload[1],
+                payload[2],
+                STATUS[payload[3]],
+                payload[4].get("WEBSTATUS_NAME"),
+                payload[4].get("WEBSTATUS_URL"),
+                payload[5],
+                payload[6])
+
+    @classmethod
+    def query(cls):
+        answer = get_lq("""GET services
+Columns: display_name host_name groups state custom_variables description plugin_output
+OutputFormat: json
+""")
+        return cls.parse_services(answer)
+
+    def __repr__(self):
+        return "Service {}: {}".format(self.name, self.webname)
+
+def get_infos(vhost):
+    hosts = Host.query(vhost)
+    servicegroups = ServiceGroup.query()
+    services = Service.query()
+
+    for host in hosts:
+        hosts[host].fill_services(services)
+    for group in servicegroups:
+        servicegroups[group].fill_services(services, hosts)
+    return (hosts, servicegroups, services)
+
+TEMPLATE='''<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.1//EN" "http://www.w3.org/TR/xhtml11/DTD/xhtml11.dtd">
+<html lang="en" xmlns="http://www.w3.org/1999/xhtml">
+    <head>
+        <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
+        <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+        <title>Status</title>
+        <meta name="referrer" content="no-referrer" />
+        <style type="text/css">
+        ul {
+            list-style: none;
+            margin: 0px;
+        }
+        ul li:nth-child(2n) {
+            background-color: rgb(240, 240, 240);
+        }
+        li.resource, li.service {
+            margin: 1px 0px;
+        }
+        span.status {
+            display: inline-block;
+            width: 150px;
+            text-align: center;
+            margin-right: 5px;
+            font-variant: small-caps;
+            font-size: 1.2em;
+        }
+        .status_ok,.status_up {
+            background-color: rgba(0, 255, 0, 0.5);;
+        }
+        .status_warning {
+            background-color: rgba(255, 255, 0, 0.5);;
+        }
+        .status_error,.status_down {
+            background-color: rgba(255, 0, 0, 0.5);;
+        }
+        .status_unknown,.status_unreachable {
+            background-color: rgba(0, 0, 255, 0.5);;
+        }
+        .infos {
+            margin-left: 40px;
+            color: rgb(100, 100, 100);
+        }
+        div#services {
+            column-count: auto;
+            column-width: 36em;
+        }
+        div.servicegroup {
+            -webkit-column-break-inside: avoid;
+            break-inside: avoid;
+        }
+        h3.servicegroup_title, h3.host_title {
+            margin: 1px 0px;
+        }
+        span.service_host, span.infos {
+            float: right;
+            display: inline-block;
+            color: rgb(100, 100, 100);
+        }
+        </style>
+    </head>
+    <body>
+        <h2>Hosts</h2>
+        {%- for host in hosts.values() %}
+            <h3 class="host_title">
+                <span class="status status_{{ host.status }}">{{ host.status }}</span>
+                <span class="host">{{ host.webname }}</span>
+            </h3>
+            {%- for service in servicegroups["webstatus-resources"].services if service.host == host.name -%}
+                {%- if loop.first %}
+                <ul class="resources">
+                {% endif %}
+
+                <li class="resource">
+                    <span class="status status_{{ service.status }}">{{ service.status }}</span>
+                    <span class="description">{{ service.description }}</span>
+                    <span class="infos">{{ service.infos }}</span>
+                </li>
+
+                {%- if loop.last %}
+                </ul>
+                {% endif %}
+            {% endfor %}
+        {%- endfor %}
+
+        {%- for group in servicegroups.values() if group.services and group.name != "webstatus-resources" %}
+        {%- if loop.first %}
+        <h2>Services</h2>
+        <div id="services">
+        {%- endif %}
+            <div class="servicegroup">
+            <h3 class="servicegroup_title">{{ group.alias }}</h3>
+            {%- for service in group.services if service.host in hosts -%}
+                {%- if loop.first %}
+                <ul class="services">
+                {% endif %}
+
+                <li class="service" title="{{ service.infos }}">
+                    <span class="status status_{{ service.status }}">{{ service.status }}</span>
+                    <span class="description">
+                        {% if service.url and service.url.startswith("https://") %}
+                        <a href="{{ service.url }}">{{ service.webname or service.description }}</a>
+                        {% else %}
+                        {{ service.webname or service.description }}
+                        {% endif %}
+                    </span>
+                    <span class="service_host">{{ hosts[service.host].webname }}</span>
+                </li>
+
+                {%- if loop.last %}
+                </ul>
+                {% endif %}
+            {%- endfor -%}
+            </div>
+        {%- if loop.last %}
+        </div>
+        {% endif %}
+        {%- endfor %}
+    </body>
+</html>
+'''
+
+@login_manager.request_loader
+def load_user_from_request(request):
+    api_key = request.headers.get('Token')
+    if api_key in AUTHORIZED_KEYS:
+        return UserMixin()
+    content = request.get_json(force=True, silent=True)
+    if content is not None and content.get("token") in AUTHORIZED_KEYS:
+        return UserMixin()
+
+@app.route("/live", methods=["POST"])
+@login_required
+def live():
+    query = request.get_data()
+    result = get_lq(query.decode() + "\n")
+    resp = make_response(result)
+    resp.content_type = "text/plain"
+    return resp
+
+@app.route("/", methods=["GET"])
+def get():
+    (hosts, servicegroups, services) = get_infos(request.host)
+    resp = make_response(render_template_string(TEMPLATE, hosts=hosts, servicegroups=servicegroups))
+    resp.content_type = "text/html"
+    return resp
+
+@app.route("/", methods=["POST"])
+@login_required
+def push():
+    content = request.get_json(force=True, silent=True)
+    if content is None:
+        return ERROR_BAD_JSON
+    if content.get("cmd") != "submitcheck":
+        return render_error(ERROR_NO_REQUEST_HANDLER)
+    if "checkresult" not in content or not isinstance(content["checkresult"], list):
+        return render_error(ERROR_BAD_DATA)
+
+    checks = 0
+    errors = 0
+    for check in map(lambda x: CheckResult.from_json(x), content["checkresult"]):
+        if check is None:
+            errors += 1
+            continue
+        try:
+            write_check_output(check)
+        except Exception as e:
+            return render_error(str(e))
+        checks += 1
+    return render_response(checks, errors)
+
+def write_check_output(check):
+    if check.type== "service":
+        command = "[{time}] PROCESS_SERVICE_CHECK_RESULT;{hostname};{servicename};{state};{output}";
+    else:
+        command = "[{time}] PROCESS_HOST_CHECK_RESULT;{hostname};{state};{output}";
+    formatted = command.format(
+            time=int(time.time()),
+            hostname=check.hostname,
+            state=check.state,
+            output=check.output,
+            servicename=check.servicename,
+        )
+
+    if not os.path.exists(COMMAND_FILE):
+        raise Exception(ERROR_BAD_COMMAND_FILE)
+    if not os.access(COMMAND_FILE, os.W_OK):
+        raise Exception(ERROR_COMMAND_FILE_OPEN_WRITE)
+    if not os.access(COMMAND_FILE, os.W_OK):
+        raise Exception(ERROR_COMMAND_FILE_OPEN_WRITE)
+    try:
+        with open(COMMAND_FILE, "w") as c:
+            c.write(formatted + "\n")
+    except Exception as e:
+        raise Exception(ERROR_BAD_WRITE)
+
+def render_error(error):
+    return jsonify({
+            "status": "error",
+            "message": error,
+            })
+
+def render_response(checks, errors):
+    if checks > 0:
+        return jsonify({
+            "status": "ok",
+            "result": {
+                "checks": checks,
+                "errors": errors,
+                }
+            })
+    else:
+        return jsonify({
+            "status": "error",
+            "message": ERROR_NO_CORRECT_STATUS,
+            })
+
+class CheckResult:
+    def __init__(self, hostname, state, output, servicename, checktype):
+        self.hostname = hostname
+        self.state = state
+        self.output = output
+        self.servicename = servicename
+        self.type = checktype
+
+    @classmethod
+    def from_json(klass, j):
+        if not isinstance(j, dict):
+            return None
+        for key in ["hostname", "state", "output"]:
+            if key not in j or not isinstance(j[key], str):
+                return None
+        for key in ["servicename", "type"]:
+            if key in j and not isinstance(j[key], str):
+                return None
+        return klass(
+                j["hostname"],
+                j["state"],
+                j["output"],
+                j.get("servicename", ""),
+                j.get("type", "host"))
+