(lib): Add ThreadedWebhook for threaded webhooks

plugin/lib.py

@@ -1,5 +1,8 @@
 from typing import Any
+import asyncio
+import discord
 from psycopg import sql
+from urllib.parse import urlparse, parse_qs
 
 from data import RawExpr, Expression
 
@@ -19,3 +22,44 @@ def asexpr(value: Any) -> RawExpr:
     Turn a value into an expression.
     """
     return RawExpr(sql.Placeholder(), (value,))
+
+async def fire_and_forget(awaitable, do_in=1, ignorable=(discord.HTTPException)):
+    await asyncio.sleep(do_in)
+    try:
+        await awaitable 
+    except ignorable:
+        pass
+    except Exception as e:
+        # TODO: Log unexpected exceptions
+        pass
+
+class ThreadedWebhook(discord.Webhook):
+    __slots__ = ("thread_id",)
+
+    def __init__(self, *args, thread_id=None, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.thread_id = thread_id
+
+    @classmethod
+    def from_url(cls, url: str, *args, **kwargs):
+        self = super().from_url(url, *args, **kwargs)
+        parse = urlparse(url)
+        if parse.query:
+            args = parse_qs(parse.query)
+            if "thread_id" in args:
+                self.thread_id = int(args["thread_id"][0])
+        return self
+
+    async def send(self, *args, **kwargs):
+        if self.thread_id is not None:
+            kwargs.setdefault("thread", discord.Object(self.thread_id))
+        return await super().send(*args, **kwargs)
+
+    async def test_webhook(self):
+        embed = discord.Embed(
+            title="Testing",
+            description="Testing logging webhook, feel free to delete."
+        )
+        result = await self.send(embed=embed, wait=True, silent=True)
+        asyncio.create_task(fire_and_forget(result.delete()))
+        await result