From d82a0ba9bdfbc7d59a3dc77db39efc1973c02aa4 Mon Sep 17 00:00:00 2001
From: Alex Kelly <kellya@arachnitech.com>
Date: Wed, 3 Nov 2021 16:12:52 -0400
Subject: [PATCH] first attempt at this disaster

---
 .../__pycache__/github.cpython-310.pyc        | Bin 0 -> 939 bytes
 matrix_webhook/formatters/github.py           |  19 ++++++++++
 matrix_webhook/formatters/grafana.py          |  16 ++++++++
 .../{formatters.py => formatters/pingdom.py}  |  35 +-----------------
 matrix_webhook/handler.py                     |   2 +
 5 files changed, 39 insertions(+), 33 deletions(-)
 create mode 100644 matrix_webhook/formatters/__pycache__/github.cpython-310.pyc
 create mode 100644 matrix_webhook/formatters/github.py
 create mode 100644 matrix_webhook/formatters/grafana.py
 rename matrix_webhook/{formatters.py => formatters/pingdom.py} (65%)

diff --git a/matrix_webhook/formatters/__pycache__/github.cpython-310.pyc b/matrix_webhook/formatters/__pycache__/github.cpython-310.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..ac481eee08159fabc91e44dfa5cece6e09fc400a
GIT binary patch
literal 939
zcmYk4PjAyO7{(tv|GLt_U>mG(iG+lfLsx{ffu=HX7=qJ;G!9e8gxuI|!kW4`F0d%Q
zzy!OpYc&au`wn~wt~tVm&%g!bbc6X|`}O<gm)Fl@IXhbgfa729*>?*3G{MC|0Xzbb
z)&OvDa)>rK-9})ACU-t601U}8wch~bgp)cr-KT2n6S{<3+&-*uXA>XD3ujEn*4Q39
zW0$)*<z5{V@XEj4%!xCmIh}x4b87;ADR(B|wcMV7&*ts~{PIcl2>eP5N8odHY+|oI
z_RNWcI{4Mx<JU^&Z0_;-a<_2e+=qo0UO(WA({FRauh;Pf{03k8N{_(nhPRF2R(AWD
z5?UWLhAJLtpZUF5N288EkUH+hA=9xOtYnvW8c$;Vbku1)ej^6Du!o~G3dy-9Kq;RM
zVBW~G0HB8)d+wMVAt!ZwrmyKXPN9sH0WIjt-Gc0&lEUUpv-0%!^JXLy(cBmP{sC(a
zm3$>aoi>?bVKj)f2%~1gw2I#~-il5n<^Eat@;vNHl`yS^N}JPLtPBr|YO5coI+V$9
z!!f<N09;gv`zO13R@LgukoD7zmH(xIQ@G}iLKT)0-L}KJTBx?$5nZW7+cP|-L`HWX
z750E7B3s;Bj<g=8>&^3@hcapYCP8+|FG<{&13$CY{oRpi#Q*W}*w5(Ra%KnX)oNjP
zBtIy@NvzY%YWTZ_mxwfFJ<+yCs-MxInk`-Q#qTPaoEetQ?=*}*+i`Ee^hk-uz16kM
zOCxq~b?sq+w#(z5NcC1g3Qvh)pM@f*D5FN<MS^jmQZ)lWna5oaR*+6X!nPYq(@^P<
zGJPx0<lU)s<^`nJ0o1H2p=9pQP3sP+k!4b$3&bXB5x`b3U8e}+|8y!jl~yA8s4q53
Il1bL^7j=H|NdN!<

literal 0
HcmV?d00001

diff --git a/matrix_webhook/formatters/github.py b/matrix_webhook/formatters/github.py
new file mode 100644
index 0000000..9f6226e
--- /dev/null
+++ b/matrix_webhook/formatters/github.py
@@ -0,0 +1,19 @@
+import plugins
+
+
+@plugins.register
+def github(data, headers):
+    """Pretty-print a github notification."""
+    # TODO: Write nice useful formatters. This is only an example.
+    if headers["X-GitHub-Event"] == "push":
+        pusher, ref, a, b, c = [
+            data[k] for k in ["pusher", "ref", "after", "before", "compare"]
+        ]
+        pusher = f"[@{pusher['name']}](https://github.com/{pusher['name']})"
+        data["body"] = f"{pusher} pushed on {ref}: [{b} → {a}]({c}):\n\n"
+        for commit in data["commits"]:
+            data["body"] += f"- [{commit['message']}]({commit['url']})\n"
+    else:
+        data["body"] = "notification from github"
+    data["digest"] = headers["X-Hub-Signature-256"].replace("sha256=", "")
+    return data
diff --git a/matrix_webhook/formatters/grafana.py b/matrix_webhook/formatters/grafana.py
new file mode 100644
index 0000000..e78344f
--- /dev/null
+++ b/matrix_webhook/formatters/grafana.py
@@ -0,0 +1,16 @@
+import plugins
+
+
+@plugins.register
+def grafana(data, headers):
+    """Pretty-print a grafana notification."""
+    text = ""
+    if "title" in data:
+        text = "#### " + data["title"] + "\n"
+    if "message" in data:
+        text = text + data["message"] + "\n\n"
+    if "evalMatches" in data:
+        for match in data["evalMatches"]:
+            text = text + "* " + match["metric"] + ": " + str(match["value"]) + "\n"
+    data["body"] = text
+    return data
diff --git a/matrix_webhook/formatters.py b/matrix_webhook/formatters/pingdom.py
similarity index 65%
rename from matrix_webhook/formatters.py
rename to matrix_webhook/formatters/pingdom.py
index 48deaee..1bffb2f 100644
--- a/matrix_webhook/formatters.py
+++ b/matrix_webhook/formatters/pingdom.py
@@ -1,38 +1,7 @@
-"""Formatters for matrix webhook."""
-from datetime import datetime
-
-
-def grafana(data, headers):
-    """Pretty-print a grafana notification."""
-    text = ""
-    if "title" in data:
-        text = "#### " + data["title"] + "\n"
-    if "message" in data:
-        text = text + data["message"] + "\n\n"
-    if "evalMatches" in data:
-        for match in data["evalMatches"]:
-            text = text + "* " + match["metric"] + ": " + str(match["value"]) + "\n"
-    data["body"] = text
-    return data
-
-
-def github(data, headers):
-    """Pretty-print a github notification."""
-    # TODO: Write nice useful formatters. This is only an example.
-    if headers["X-GitHub-Event"] == "push":
-        pusher, ref, a, b, c = [
-            data[k] for k in ["pusher", "ref", "after", "before", "compare"]
-        ]
-        pusher = f"[@{pusher['name']}](https://github.com/{pusher['name']})"
-        data["body"] = f"{pusher} pushed on {ref}: [{b} → {a}]({c}):\n\n"
-        for commit in data["commits"]:
-            data["body"] += f"- [{commit['message']}]({commit['url']})\n"
-    else:
-        data["body"] = "notification from github"
-    data["digest"] = headers["X-Hub-Signature-256"].replace("sha256=", "")
-    return data
+import plugins
 
 
+@plugin.register
 def pingdom(data, headers):
     """Pretty-print a pingdom notification."""
     #JSON data formatting was obtained from https://www.pingdom.com/resources/webhooks/
diff --git a/matrix_webhook/handler.py b/matrix_webhook/handler.py
index 3460810..b3c81df 100644
--- a/matrix_webhook/handler.py
+++ b/matrix_webhook/handler.py
@@ -4,6 +4,7 @@ import json
 import logging
 from http import HTTPStatus
 from hmac import HMAC
+import plugins
 
 from markdown import markdown
 
@@ -18,6 +19,7 @@ async def matrix_webhook(request):
 
     This one handles a POST, checks its content, and forwards it to the matrix room.
     """
+    formatters = plugins.names_factory(__package__)
     LOGGER.debug(f"Handling {request=}")
     data_b = await request.read()