quick commit

src/guardden/cogs/admin.py

@@ -7,6 +7,7 @@ import discord
 from discord.ext import commands
 
 from guardden.bot import GuardDen
+from guardden.utils.ratelimit import RateLimitExceeded
 
 logger = logging.getLogger(__name__)
 
@@ -17,12 +18,32 @@ class Admin(commands.Cog):
     def __init__(self, bot: GuardDen) -> None:
         self.bot = bot
 
-    async def cog_check(self, ctx: commands.Context) -> bool:
+    def cog_check(self, ctx: commands.Context) -> bool:
         """Ensure only administrators can use these commands."""
         if not ctx.guild:
             return False
+        if not self.bot.is_owner_allowed(ctx.author.id):
+            return False
         return ctx.author.guild_permissions.administrator
 
+    async def cog_before_invoke(self, ctx: commands.Context) -> None:
+        if not ctx.command:
+            return
+        result = self.bot.rate_limiter.acquire_command(
+            ctx.command.qualified_name,
+            user_id=ctx.author.id,
+            guild_id=ctx.guild.id if ctx.guild else None,
+            channel_id=ctx.channel.id,
+        )
+        if result.is_limited:
+            raise RateLimitExceeded(result.reset_after)
+
+    async def cog_command_error(self, ctx: commands.Context, error: Exception) -> None:
+        if isinstance(error, RateLimitExceeded):
+            await ctx.send(
+                f"You're being rate limited. Try again in {error.retry_after:.1f} seconds."
+            )
+
     @commands.group(name="config", invoke_without_command=True)
     @commands.guild_only()
     async def config(self, ctx: commands.Context) -> None: