From eeb7a40d495bd5f595d7ba0f8a350b23411e9b3f Mon Sep 17 00:00:00 2001
From: Joinemm <joonas@rautiola.co>
Date: Sun, 31 Dec 2023 02:44:20 +0200
Subject: [PATCH] Add webhook endpoint for receiving sponsor events

---
 cogs/webserver.py   | 58 +++++++++++++++++++++++++++++++++++++++++++++
 modules/keychain.py |  1 +
 2 files changed, 59 insertions(+)

diff --git a/cogs/webserver.py b/cogs/webserver.py
index 14b72cf..f773c29 100644
--- a/cogs/webserver.py
+++ b/cogs/webserver.py
@@ -33,6 +33,7 @@ def __init__(self, bot):
         self.app.router.add_get("/documentation", self.command_list)
         self.app.router.add_get("/donators", self.donator_list)
         self.app.router.add_get("/metrics", aio.web.server_stats)
+        self.app.router.add_post("/webhook", self.webhook)
 
     async def cog_load(self):
         self.cache_stats.start()
@@ -112,6 +113,63 @@ async def ping_handler(self, request):
     async def website_statistics(self, request):
         return web.json_response(self.cached)
 
+    async def webhook(self, request):
+        x_github_event: str | None = request.headers.get("x-github-event")
+        if not x_github_event:
+            return web.HTTPBadRequest()
+        if x_github_event == "ping":
+            return web.json_response({"status": "pong"})
+        if x_github_event != "sponsorship":
+            return web.HTTPMethodNotAllowed(x_github_event, ["sponsorship", "ping"])
+
+        data = await request.json()
+        action = data["action"]
+        data = data["sponsorship"]
+        tier = data["tier"]
+
+        if data["privacy_level"] == "private":
+            sponsor = {
+                "login": "Anonymous",
+                "html_url": "",
+                "avatar_url": "https://gutegymnasiet.se/wp-content/uploads/2023/12/anonymous-icon-0.png",
+            }
+        else:
+            sponsor = data["sponsor"]
+
+        color = "333333"
+
+        if action == "created":
+            description = (
+                f"Just sponsored with **${tier['monthly_price_in_dollars']}**"
+                f" {'one time' if tier['is_one_time'] else 'per month'}! :heart:"
+            )
+            color = "6cc644"
+        elif action == "cancelled":
+            description = "Just cancelled their sponsorship :("
+            color = "bd2c00"
+        else:
+            description = action
+            print("unknown webhook data:", data)
+
+        embed_data = {
+            "embeds": [
+                {
+                    "color": int(color, 16),
+                    "author": {
+                        "name": sponsor["login"],
+                        "url": sponsor["html_url"],
+                        "icon_url": sponsor["avatar_url"],
+                    },
+                    "description": description,
+                }
+            ]
+        }
+        async with self.bot.session.post(
+            self.bot.keychain.SPONSORS_WEBHOOK_URL,
+            json=embed_data,
+        ) as response:
+            return web.json_response({"status": response.status})
+
     async def command_list(self, request):
         return web.json_response(self.cached_command_list)
 
diff --git a/modules/keychain.py b/modules/keychain.py
index 6c95557..11cce34 100644
--- a/modules/keychain.py
+++ b/modules/keychain.py
@@ -39,6 +39,7 @@ def __init__(self):
         self.GIPHY_API_KEY: str = ""
         self.LASTFM_USERNAME: str = ""
         self.LASTFM_PASSWORD: str = ""
+        self.SPONSORS_WEBHOOK_URL: str = ""
 
         for name in self.__dict__:
             value = os.environ.get(name)