Initial commit.

fjerkroa_bot/__init__.py

@@ -0,0 +1 @@
+from .discord_bot import FjerkroaBot, main

fjerkroa_bot/discord_bot.py

@@ -0,0 +1,74 @@
+import sys
+import argparse
+import json
+import discord
+from discord import Message
+from discord.ext import commands
+from watchdog.observers import Observer
+from watchdog.events import FileSystemEventHandler
+
+
+class ConfigFileHandler(FileSystemEventHandler):
+    def __init__(self, on_modified):
+        self._on_modified = on_modified
+
+    def on_modified(self, event):
+        self._on_modified(event)
+
+
+class FjerkroaBot(commands.Bot):
+    def __init__(self, config_file: str):
+        assert not hasattr(self, 'config_file')
+        assert not hasattr(self, 'config')
+        assert not hasattr(self, 'observer')
+        assert not hasattr(self, 'file_handler')
+
+        self.config_file = config_file
+        self.config = self.load_config(self.config_file)
+        intents = discord.Intents.default()
+        intents.message_content = True
+        intents.members = True
+
+        self.observer = Observer()
+        self.file_handler = ConfigFileHandler(self.on_config_file_modified)
+        self.observer.schedule(self.file_handler, path=".", recursive=False)
+        self.observer.start()
+
+        super().__init__(command_prefix="!", case_insensitive=True, intents=intents)
+
+    @classmethod
+    def load_config(self, config_file: str = "config.json"):
+        with open(config_file, "r") as file:
+            return json.load(file)
+
+    def on_config_file_modified(self, event):
+        if event.src_path == self.config_file:
+            self.config = self.load_config(self.config_file)
+
+    async def on_ready(self):
+        print(f"We have logged in as {self.user}")
+
+    async def on_message(self, message: Message) -> None:
+        await message.channel.send("Hello!")
+
+    async def close(self):
+        self.observer.stop()
+        await super().close()
+
+
+def main() -> int:
+    from .logging import setup_logging
+    setup_logging()
+    parser = argparse.ArgumentParser(description='Fjerkroa AI bot')
+    parser.add_argument('--config', type=str, default='config.json', help='Config file.')
+    args = parser.parse_args()
+
+    config = FjerkroaBot.load_config(args.config)
+    bot = FjerkroaBot(args.config)
+    bot.run(config["discord_token"])
+
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())

fjerkroa_bot/logging.py

@@ -0,0 +1,11 @@
+import sys
+import logging
+
+
+def setup_logging():
+    logging_handler = logging.StreamHandler(stream=sys.stdout)
+    logging.basicConfig(
+        level=logging.INFO,
+        format="%(asctime)s [%(levelname)s] %(message)s",
+        handlers=[logging_handler],
+    )