From bf907ac0e1dc32d495a4fdcce5769611b7c56bf9 Mon Sep 17 00:00:00 2001
From: cswimr <seaswimmerthefsh@gmail.com>
Date: Sat, 25 Jan 2025 22:46:59 +0000
Subject: [PATCH] feat(hotreload): init

---
 hotreload/__init__.py  |  5 +++
 hotreload/hotreload.py | 70 ++++++++++++++++++++++++++++++++++++++++++
 hotreload/info.json    | 17 ++++++++++
 pyproject.toml         |  1 +
 uv.lock                |  2 ++
 5 files changed, 95 insertions(+)
 create mode 100644 hotreload/__init__.py
 create mode 100644 hotreload/hotreload.py
 create mode 100644 hotreload/info.json

diff --git a/hotreload/__init__.py b/hotreload/__init__.py
new file mode 100644
index 0000000..7140931
--- /dev/null
+++ b/hotreload/__init__.py
@@ -0,0 +1,5 @@
+from .hotreload import HotReload
+
+
+async def setup(bot):
+    await bot.add_cog(HotReload(bot))
diff --git a/hotreload/hotreload.py b/hotreload/hotreload.py
new file mode 100644
index 0000000..73513f5
--- /dev/null
+++ b/hotreload/hotreload.py
@@ -0,0 +1,70 @@
+from pathlib import Path
+
+from red_commons.logging import RedTraceLogger, getLogger
+from redbot.core import commands
+from redbot.core.bot import Red
+from redbot.core.core_commands import CoreLogic
+from redbot.core.utils.chat_formatting import bold, humanize_list
+from watchdog.events import FileSystemEvent, FileSystemEventHandler
+from watchdog.observers import Observer
+
+
+class HotReload(commands.Cog):
+    """Automatically reload cogs in local cog paths on file change."""
+
+    __author__ = ["[cswimr](https://www.coastalcommits.com/cswimr)"]
+    __git__ = "https://www.coastalcommits.com/cswimr/SeaCogs"
+    __version__ = "1.0.0-rc1"
+    __documentation__ = "https://seacogs.coastalcommits.com/hotreload/"
+
+    def __init__(self, bot: Red) -> None:
+        super().__init__()
+        self.bot: Red = bot
+        self.logger: RedTraceLogger = getLogger(name="red.SeaCogs.HotReload")
+
+    def format_help_for_context(self, ctx: commands.Context) -> str:
+        pre_processed = super().format_help_for_context(ctx) or ""
+        n = "\n" if "\n\n" not in pre_processed else ""
+        text = [
+            f"{pre_processed}{n}",
+            f"{bold('Cog Version:')} [{self.__version__}]({self.__git__})",
+            f"{bold('Author:')} {humanize_list(self.__author__)}",
+            f"{bold('Documentation:')} {self.__documentation__}",
+        ]
+        return "\n".join(text)
+
+    async def get_paths(self) -> tuple[Path]:
+        """Retrieve user defined paths."""
+        cog_manager = self.bot._cog_mgr
+        cog_paths = await cog_manager.user_defined_paths()
+        return (Path(path) for path in cog_paths)
+
+    async def start_observer(self) -> None:
+        """Start the observer to watch for file changes."""
+        observer = Observer()
+        for path in await self.get_paths():
+            observer.schedule(HotReloadHandler(self.bot, path), path, recursive=True)
+        observer.start()
+        self.logger.info("Started observer.")
+
+
+class HotReloadHandler(FileSystemEventHandler):
+    """Handler for file changes."""
+
+    def __init__(self, bot: Red, path: Path) -> None:
+        self.bot: Red = bot
+        self.path: Path = path
+        self.logger: RedTraceLogger = getLogger(name="red.SeaCogs.HotReloadHandler")
+
+    def on_modified(self, event: FileSystemEvent) -> None:
+        """Handle file modification events."""
+        if event.is_directory:
+            return
+        self.logger.info(f"File {event.src_path} has been modified.")
+        # self.bot.loop.create_task(self.reload_cog())
+
+    async def reload_cog(self, cog_name: str) -> None:
+        """Reload modified cog."""
+        core_logic = CoreLogic(bot=self.bot)
+        core_logic._reload(pkg_names=(cog_name,))
+        self.logger.info(f"Reloaded {cog_name} cog.")
diff --git a/hotreload/info.json b/hotreload/info.json
new file mode 100644
index 0000000..7d47a1e
--- /dev/null
+++ b/hotreload/info.json
@@ -0,0 +1,17 @@
+{
+  "author" : ["cswimr"],
+  "install_msg" : "Thank you for installing HotReload!",
+  "name" : "HotReload",
+  "short" : "Automatically reload cogs in local cog paths on file change.",
+  "description" : "Automatically reload cogs in local cog paths on file change.",
+  "end_user_data_statement" : "This cog does not store end user data.",
+  "hidden": false,
+  "disabled": false,
+  "min_bot_version": "3.5.0",
+  "min_python_version": [3, 10, 0],
+  "requirements": ["watchdog"],
+  "tags": [
+    "utility",
+    "development"
+  ]
+}
diff --git a/pyproject.toml b/pyproject.toml
index 2c40dee..ccf98da 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -18,6 +18,7 @@ dependencies = [
     "py-dactyl",
     "pydantic>=2.9.2",
     "red-discordbot>=3.5.14",
+    "watchdog>=5.0.3",
     "websockets>=13.1",
 ]
 
diff --git a/uv.lock b/uv.lock
index 9056ced..a029a28 100644
--- a/uv.lock
+++ b/uv.lock
@@ -1667,6 +1667,7 @@ dependencies = [
     { name = "py-dactyl" },
     { name = "pydantic" },
     { name = "red-discordbot" },
+    { name = "watchdog" },
     { name = "websockets" },
 ]
 
@@ -1706,6 +1707,7 @@ requires-dist = [
     { name = "py-dactyl", git = "https://github.com/cswimr/pydactyl" },
     { name = "pydantic", specifier = ">=2.9.2" },
     { name = "red-discordbot", specifier = ">=3.5.14" },
+    { name = "watchdog", specifier = ">=5.0.3" },
     { name = "websockets", specifier = ">=13.1" },
 ]