From 09fe26633f39472c1b87e709868776f20d636c61 Mon Sep 17 00:00:00 2001
From: Alex Kelly <kellya@arachnitech.com>
Date: Wed, 3 Nov 2021 00:19:52 -0400
Subject: [PATCH] add basic functionality for pingdom formatter

---
 matrix_webhook/formatters.py | 61 ++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/matrix_webhook/formatters.py b/matrix_webhook/formatters.py
index c59249e..813b071 100644
--- a/matrix_webhook/formatters.py
+++ b/matrix_webhook/formatters.py
@@ -1,4 +1,5 @@
 """Formatters for matrix webhook."""
+from datetime import datetime
 
 
 def grafana(data, headers):
@@ -30,3 +31,63 @@ def github(data, headers):
         data["body"] = "notification from github"
     data["digest"] = headers["X-Hub-Signature-256"].replace("sha256=", "")
     return data
+
+
+def pingdom(data, headers):
+    """Pretty-print a pingdom notification."""
+    #JSON data formatting was obtained from https://www.pingdom.com/resources/webhooks/
+    check_id = data["check_id"]
+    check_name = data["check_name"]
+    current_state = data["current_state"]
+    local_time = datetime.fromtimestamp(data["state_changed_timestamp"])
+
+    if data["check_type"].lower() == "http":
+        try:
+            check_url = data["check_params"]["full_url"]
+            data["body"] = (
+                f"###{check_name} is {current_state}\n\n{check_url}"
+                f" marked {current_state} at {local_time} ⚬ "
+                f"[view details](https://my.pingdom.com/reports/responsetime#check={check_id})"
+            )
+        except Exception as error:
+            data[
+                "body"
+                ] = (
+                    f"Error: An attempt to post from pingdom was malformed "
+                    "(or I don't know how to handle what was sent).\n\n"
+                    f"{repr(error)}"
+                )
+    elif data["check_type"].lower() == "dns":
+        # There are a bunch of values that are blanke when you do a test
+        # so ignore them if value is unset
+        try
+            first_ip = data["first_probe"]["ip"]
+        except KeyError:
+            first_ip = "unknown"
+        try:
+            second_ip = data["second_probe"]["ip"]
+        except KeyError:
+            second_ip = "unknown"
+        try:
+            first_location  = data["first_probe"]["location"]
+        except:
+            first_location = "unknown"
+        try:
+            second_location = data["second_probe"]["location"]
+        except:
+            second_location = "unknown"
+        try:
+            expected_ip = data["check_params"]["expected_ip"]
+            data["body"] = (
+                f"###{check_name} is {current_state}\n\n"
+                f"expected {expected_ip} but got:\n\n"
+                f"    {first_ip} ({first_location})\n\n"
+                f"    {second_ip} ({second_location})\n\n"
+                f" marked {current_state} at {local_time} ⚬ "
+                f"[view details](https://my.pingdom.com/reports/responsetime#check={check_id})"
+            )
+
+        except Exception as error:
+            print(error)
+
+    return data