From 53020ae0f2386f5bfeaec644f3e9fc364c136e1c Mon Sep 17 00:00:00 2001
From: agatha <agatha@juggalol.com>
Date: Thu, 6 Jun 2024 13:55:29 -0400
Subject: [PATCH] initial commit

---
 .gitignore       |  5 ++++
 main.py          | 65 ++++++++++++++++++++++++++++++++++++++++++++++++
 requirements.txt |  2 ++
 3 files changed, 72 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 main.py
 create mode 100644 requirements.txt

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..49895dc
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,5 @@
+.idea/
+venv/
+__pycache__/
+*.py[cod]
+config.json
diff --git a/main.py b/main.py
new file mode 100644
index 0000000..2134328
--- /dev/null
+++ b/main.py
@@ -0,0 +1,65 @@
+"""Matrix Bot Framework"""
+import asyncio
+import json
+from loguru import logger
+from nio import AsyncClient, LoginResponse
+
+
+class MatrixBot:
+    def __init__(self, config: dict):
+        # TODO: Test configuration for required settings
+        self.config = config
+
+        self.client = AsyncClient(
+            homeserver=self.config['homeserver'],
+            user=self.config['username']
+        )
+
+    async def send_message(self, message: str):
+        if not self.client.access_token:
+            logged_in = await self.login()
+            if not logged_in:
+                return
+
+        await self.client.room_send(
+            room_id=self.config['room_id'],
+            message_type="m.room.message",
+            content={
+                "msgtype": "m.text",
+                "body": message
+            }
+        )
+        logger.info("Message sent")
+
+    async def login(self):
+        response = await self.client.login(
+            password=self.config['password']
+        )
+        if isinstance(response, LoginResponse):
+            logger.info(f"Logged in as {self.config['username']}")
+        else:
+            logger.error(f"Failed to login as {self.config['username']}: {response}")
+            return False
+
+        return True
+
+    async def logout(self):
+        await self.client.logout()
+        await self.client.close()
+        logger.info(f"Logged out from {self.config['homeserver']}")
+
+
+def load_config(path: str) -> dict:
+    with open(path, 'r', encoding='utf-8') as f:
+        return json.loads(f.read())
+
+
+async def main():
+    config = load_config("config.json")
+    bot = MatrixBot(config['matrix'])
+    await bot.send_message("beep boop, i'm a bot")
+    await bot.logout()
+
+
+if __name__ == '__main__':
+    asyncio.get_event_loop().run_until_complete(main())
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..855170c
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,2 @@
+matrix-nio
+loguru
\ No newline at end of file