From b27aa39ea8caf61720e90c08ce635fcb51fd3d8a Mon Sep 17 00:00:00 2001
From: Parker <contact@pkrm.dev>
Date: Fri, 12 Jul 2024 03:48:02 -0500
Subject: [PATCH] Overhaul cog command

---
 code/cogs/owner/cog.py        | 75 +++++++++++++++++++++++++++++++++++
 code/cogs/owner/load_cog.py   | 22 ----------
 code/cogs/owner/reload_cog.py | 22 ----------
 3 files changed, 75 insertions(+), 44 deletions(-)
 create mode 100644 code/cogs/owner/cog.py
 delete mode 100644 code/cogs/owner/load_cog.py
 delete mode 100644 code/cogs/owner/reload_cog.py

diff --git a/code/cogs/owner/cog.py b/code/cogs/owner/cog.py
new file mode 100644
index 0000000..78dc246
--- /dev/null
+++ b/code/cogs/owner/cog.py
@@ -0,0 +1,75 @@
+from discord.ext import commands
+
+
+class CogCommands(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.group(invoke_without_command=True)
+    @commands.dm_only()
+    @commands.is_owner()
+    async def cog(self, ctx):
+        await ctx.author.send(f"This is a group command. Use `{self.bot.command_prefix}cog load/unload/reload` followed by the name of the cog.")
+
+    @cog.command()
+    @commands.dm_only()
+    @commands.is_owner()
+    async def load(self, ctx: commands.Context, *, cog: str = None):
+        if not cog:
+            return await ctx.send("No cog provided.")
+
+        cog = cog.lower()
+        await self.bot.load_extension(f"cogs.{cog}")
+
+        await ctx.send(f"Cog `{cog}` has been loaded.")
+
+    @load.error
+    async def cog_load_error(self, ctx, error):
+        if isinstance(error.original, commands.ExtensionAlreadyLoaded):
+            return await ctx.send(f"Cog is already loaded.")
+        if isinstance(error.original, commands.ExtensionNotFound):
+            return await ctx.send("Cog does not exist.")
+        else:
+            return await ctx.send("An unknown error occurred.")
+
+    @cog.command()
+    @commands.dm_only()
+    @commands.is_owner()
+    async def unload(self, ctx: commands.Context, *, cog: str = None):
+        if not cog:
+            return await ctx.send("No cog provided.")
+
+        cog = cog.lower()
+        await self.bot.unload_extension(f"cogs.{cog}")
+
+        await ctx.send(f"Cog `{cog}` has been unloaded.")
+
+    @unload.error
+    async def cog_unload_error(self, ctx, error):
+        if isinstance(error.original, commands.ExtensionNotLoaded):
+            return await ctx.send("Cog not loaded. It might be that the cog does not exist.")
+        else:
+            return await ctx.send("An unknown error occurred.")
+
+    @cog.command()
+    @commands.dm_only()
+    @commands.is_owner()
+    async def reload(self, ctx: commands.Context, *, cog: str = None):
+        if not cog:
+            return await ctx.send("No cog provided.")
+
+        cog = cog.lower()
+        await self.bot.reload_extension(f"cogs.{cog}")
+
+        await ctx.send(f"Cog `{cog}` has been reloaded.")
+
+    @reload.error
+    async def cog_reload_error(self, ctx, error):
+        if isinstance(error.original, commands.ExtensionNotLoaded):
+            return await ctx.send("Cog not loaded. It might be that the cog does not exist.")
+        else:
+            return await ctx.send("An unknown error occurred.")
+
+
+async def setup(bot):
+    await bot.add_cog(CogCommands(bot))
diff --git a/code/cogs/owner/load_cog.py b/code/cogs/owner/load_cog.py
deleted file mode 100644
index b59de85..0000000
--- a/code/cogs/owner/load_cog.py
+++ /dev/null
@@ -1,22 +0,0 @@
-from discord.ext import commands
-
-
-class LoadCog(commands.Cog):
-    def __init__(self, bot):
-        self.bot = bot
-
-    @commands.command()
-    @commands.dm_only()
-    @commands.is_owner()
-    async def loadcog(self, ctx: commands.Context, cog: str = None):
-        if not cog:
-            return await ctx.send("No cog provided.")
-
-        cog = cog.lower()
-        await self.bot.load_extension(f"cogs.{cog}")
-
-        await ctx.send(f"Cog {cog} has been loaded")
-
-
-async def setup(bot):
-    await bot.add_cog(LoadCog(bot))
diff --git a/code/cogs/owner/reload_cog.py b/code/cogs/owner/reload_cog.py
deleted file mode 100644
index 5cf6d9d..0000000
--- a/code/cogs/owner/reload_cog.py
+++ /dev/null
@@ -1,22 +0,0 @@
-from discord.ext import commands
-
-
-class ReloadCog(commands.Cog):
-    def __init__(self, bot):
-        self.bot = bot
-
-    @commands.command()
-    @commands.dm_only()
-    @commands.is_owner()
-    async def reloadcog(self, ctx: commands.Context, cog: str = None):
-        if not cog:
-            return await ctx.send("No cog provided.")
-
-        cog = cog.lower()
-        await self.bot.reload_extension(f"cogs.{cog}")
-
-        await ctx.send(f"Cog {cog} has been reloaded")
-
-
-async def setup(bot):
-    await bot.add_cog(ReloadCog(bot))