From 5dccc2d4bb834d4364989eaff80b0cb3fa4a9bf2 Mon Sep 17 00:00:00 2001
From: polyfloyd <floyd@polyfloyd.net>
Date: Sun, 1 Jun 2025 16:41:04 +0200
Subject: [PATCH] Add a IRC wrapper for Discord command implementations

---
 .gitignore       |   1 +
 ircbot.py        | 115 +++++++++++++++++++++++++++++++++++++++++++++++
 pyproject.toml   |   4 ++
 requirements.txt |  11 +++--
 uv.lock          |  18 ++++++++
 5 files changed, 145 insertions(+), 4 deletions(-)
 create mode 100755 ircbot.py

diff --git a/.gitignore b/.gitignore
index 1bc83ee..ba92e59 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,3 @@
 /.ruff_cache
 /.venv
+__pycache__
diff --git a/ircbot.py b/ircbot.py
new file mode 100755
index 0000000..38baf1f
--- /dev/null
+++ b/ircbot.py
@@ -0,0 +1,115 @@
+import asyncio
+import os
+import sys
+
+import pydle
+from discord_webhook import DiscordEmbed
+
+import commands as botcommands
+
+
+class DiscordAuthor:
+    def __init__(self, nick):
+        self.nick = nick
+        self.global_name = nick
+
+
+class DiscordContext:
+    def __init__(self, bot, target, source, message):
+        self._bot = bot
+        self._target = target
+        self._source = source
+        self._message = message
+        self.author = DiscordAuthor(source)
+
+    def typing(self):
+        class NilTyping:
+            async def __aenter__(self):
+                return self
+
+            async def __aexit__(self, *exc):
+                return None
+
+        return NilTyping()
+
+    async def reply(self, m):
+        lines = m.strip().split("\n")
+        lines = [f"{self._source}: {line}" for line in lines]
+        await self._bot.message(self._target, "\n".join(lines))
+
+
+class DiscordImplBot(pydle.Client):
+    def __init__(self, channel, nickname, *, push_messages=None, prefix="!"):
+        super().__init__(nickname, realname=nickname)
+        self._channel = channel
+        self._cmd_prefix = prefix
+        self._cmds = {}
+        self._push_messages = push_messages
+        self._push_messages_task = None
+
+    async def on_connect(self):
+        await self.join(self._channel)
+        if self._push_messages is not None:
+            self._push_messages_task = asyncio.create_task(self._handle_push_messages())
+
+    async def on_disconnect(self):
+        if self._push_messages_task is not None:
+            self._push_messages_task.cancel()
+            self._push_messages_task = None
+
+    async def _handle_push_messages(self):
+        async for message in self._push_messages():
+            if isinstance(message, DiscordEmbed):
+                await self.message(self._channel, f"{message.title} {message.url}")
+            else:
+                await self.message(self._channel, message)
+
+    async def on_message(self, target, source, message):
+        # Don't respond to our own messages, as this leads to a positive feedback loop.
+        if source == self.nickname:
+            return
+        if not message.startswith(self._cmd_prefix):
+            return
+
+        cmd_fn = self._cmds.get(message.removeprefix(self._cmd_prefix))
+        if not cmd_fn:
+            return
+
+        ctx = DiscordContext(self, target, source, message)
+        await cmd_fn(ctx)
+
+    # Discord API: Register a new command
+    def command(self, *, name=None, description=None):
+        def _reg_cmd(handler):
+            nonlocal name
+            name = name or handler.__name__
+            self._cmds[name] = handler
+
+        return _reg_cmd
+
+
+def main(*, server, channel, nick, mqtt_host):
+    def push_messages():
+        return botcommands.run_events(mqtt_host)
+
+    bot = DiscordImplBot(
+        channel,
+        nick,
+        push_messages=push_messages,
+    )
+    botcommands.setup(bot, mqtt_host)
+    bot.run(server, tls=True)
+
+
+if __name__ == "__main__":
+    mqtt_host = os.getenv("MQTT_HOST")
+    if not mqtt_host:
+        print("MQTT_HOST unset")
+        sys.exit(1)
+
+    main(
+        server="irc.libera.chat",
+        channel="#bitlair-bot-test",
+        nick="Bitlair",
+        mqtt_host=mqtt_host,
+    )
diff --git a/pyproject.toml b/pyproject.toml
index cc655ca..3f1316c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -8,5 +8,9 @@ dependencies = [
     "aiomqtt>=2.4.0",
     "discord-py>=2.5.2",
     "discord-webhook>=1.4.1",
+    "pydle[sasl]",
     "pytz>=2025.2",
 ]
+
+[tool.uv.sources]
+pydle = { git = "https://github.com/TheHolyRoger/pydle", branch = "fix-py10" }
diff --git a/requirements.txt b/requirements.txt
index d7615c5..f336bad 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 # This file was autogenerated by uv via the following command:
-#    uv export --format requirements-txt
+#    uv export --format=requirements.txt
 aiohappyeyeballs==2.6.1 \
     --hash=sha256:c3f9d0113123803ccadfdf3f0faa505bc78e6a72d1cc4806cbd719826e943558 \
     --hash=sha256:f349ba8f4b75cb25c99c5c2d84e997e485204d2902a9597802b0371f09331fb8
@@ -38,9 +38,7 @@ aiohttp==3.11.18 \
     --hash=sha256:ea4cf2488156e0f281f93cc2fd365025efcba3e2d217cbe3df2840f8c73db261 \
     --hash=sha256:f8af2ef3b4b652ff109f98087242e2ab974b2b2b496304063585e3d78de0b000 \
     --hash=sha256:f9e6710ebebfce2ba21cee6d91e7452d1125100f41b906fb5af3da8c78b764c1
-    # via
-    #   discord-bot
-    #   discord-py
+    # via discord-py
 aiomqtt==2.4.0 \
     --hash=sha256:721296e2b79df5f6c7c4dfc91700ae0166953a4127735c92637859619dbd84e4 \
     --hash=sha256:ab0f18fc5b7ffaa57451c407417d674db837b00a9c7d953cccd02be64f046c17
@@ -308,6 +306,11 @@ propcache==0.3.1 \
     # via
     #   aiohttp
     #   yarl
+pure-sasl==0.6.2 \
+    --hash=sha256:53c1355f5da95e2b85b2cc9a6af435518edc20c81193faa0eea65fdc835138f4
+    # via pydle
+pydle @ git+https://github.com/TheHolyRoger/pydle@befd10fb340a1934fbdbc9c8ccaca68620cba0f6
+    # via discord-bot
 pytz==2025.2 \
     --hash=sha256:360b9e3dbb49a209c21ad61809c7fb453643e048b38924c765813546746e81c3 \
     --hash=sha256:5ddf76296dd8c44c26eb8f4b6f35488f3ccbf6fbbd7adee0b7262d43f0ec2f00
diff --git a/uv.lock b/uv.lock
index 173362c..2c06120 100644
--- a/uv.lock
+++ b/uv.lock
@@ -189,6 +189,7 @@ dependencies = [
     { name = "aiomqtt" },
     { name = "discord-py" },
     { name = "discord-webhook" },
+    { name = "pydle", extra = ["sasl"] },
     { name = "pytz" },
 ]
 
@@ -197,6 +198,7 @@ requires-dist = [
     { name = "aiomqtt", specifier = ">=2.4.0" },
     { name = "discord-py", specifier = ">=2.5.2" },
     { name = "discord-webhook", specifier = ">=1.4.1" },
+    { name = "pydle", extras = ["sasl"], git = "https://github.com/TheHolyRoger/pydle?branch=fix-py10" },
     { name = "pytz", specifier = ">=2025.2" },
 ]
 
@@ -420,6 +422,22 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/b8/d3/c3cb8f1d6ae3b37f83e1de806713a9b3642c5895f0215a62e1a4bd6e5e34/propcache-0.3.1-py3-none-any.whl", hash = "sha256:9a8ecf38de50a7f518c21568c80f985e776397b902f1ce0b01f799aba1608b40", size = 12376, upload-time = "2025-03-26T03:06:10.5Z" },
 ]
 
+[[package]]
+name = "pure-sasl"
+version = "0.6.2"
+source = { registry = "https://pypi.org/simple" }
+sdist = { url = "https://files.pythonhosted.org/packages/83/b7/a0d688f86c869073cc28c0640899394a1cf68a6d87ee78a09565e9037da6/pure-sasl-0.6.2.tar.gz", hash = "sha256:53c1355f5da95e2b85b2cc9a6af435518edc20c81193faa0eea65fdc835138f4", size = 11617, upload-time = "2019-10-14T21:43:57.13Z" }
+
+[[package]]
+name = "pydle"
+version = "1.0.1"
+source = { git = "https://github.com/TheHolyRoger/pydle?branch=fix-py10#befd10fb340a1934fbdbc9c8ccaca68620cba0f6" }
+
+[package.optional-dependencies]
+sasl = [
+    { name = "pure-sasl" },
+]
+
 [[package]]
 name = "pytz"
 version = "2025.2"