Added moderation extension from the other repo

leek/moderation/__init__.py

@@ -0,0 +1,3 @@
+# flake8: noqa: F401
+
+from .moderation import Moderation

leek/moderation/moderation.en-US.json

@@ -0,0 +1,10 @@
+{
+    "MODERATION_COMMAND_CLEAR_NAME": "clear",
+    "MODERATION_COMMAND_CLEAR_HELP": "Clears all of the messages in a channel.",
+    "MODERATION_COMMAND_CLEAR_INVALID": "{0} is not a valid id!",
+    "MODERATION_COMMAND_CLEAR_NO_PERMS": "The bot does not has permissions to delete messages!",
+    "MODERATION_COMMAND_CLEAR_HTTP_ERROR": "Unable to delete messages: Code {0}",
+    "MODERATION_COMMAND_CLEAR_LIMIT_LOCAL": "Rate limited for {0} seconds, please wait...",
+    "MODERATION_COMMAND_CLEAR_LIMIT_GLOBAL": "Rate limited **globally** for {0} seconds.\nStopping.",
+    "MODERATION_COMMAND_CLEAR_DONE": "Done!"
+}

leek/moderation/moderation.py

@@ -0,0 +1,84 @@
+"""
+Moderation tools for the Leek bot.
+"""
+
+import asyncio
+import logging
+from typing import Optional
+
+from discord import ApplicationContext, Cog, Forbidden, HTTPException, Message, NotFound, Permissions, slash_command
+from leek import LeekBot, get_default, get_localizations, localize
+
+LOGGER = logging.getLogger("leek_moderation")
+PERMISSIONS = Permissions(manage_messages=True)
+
+
+class Moderation(Cog):
+    """
+    Set of tools for the Moderation of Discord servers.
+    """
+    def __init__(self, bot: LeekBot):
+        """
+        Creates a new moderation cog.
+        :param bot: The bot instance to use.
+        """
+        self.bot: LeekBot = bot
+
+    async def _safely_delete(self, ctx: ApplicationContext, message: Message) -> bool:
+        try:
+            await message.delete(reason=f"Clear by {ctx.user} ({ctx.user})")
+        except Forbidden:
+            await ctx.send(localize("MODERATION_COMMAND_CLEAR_NO_PERMS", ctx.locale))
+            return False
+        except NotFound:
+            return True
+        except HTTPException as e:
+            if e.code != 429:
+                await ctx.send(localize("MODERATION_COMMAND_CLEAR_HTTP_ERROR", ctx.locale, e.code))
+                return False
+
+            response = await e.response.json()
+            retry = response["retry_after"]
+
+            if response["global"]:
+                await ctx.send(localize("MODERATION_COMMAND_CLEAR_LIMIT_GLOBAL", ctx.locale, retry))
+                return False
+
+            await ctx.send(localize("MODERATION_COMMAND_CLEAR_LIMIT_LOCAL", ctx.locale, retry),
+                           delete_after=10)
+            await asyncio.sleep(retry + 1)
+            return await self._safely_delete(ctx, message)
+        else:
+            return True
+
+    @slash_command(name=get_default("MODERATION_COMMAND_CLEAR_NAME"),
+                   name_localizations=get_localizations("MODERATION_COMMAND_CLEAR_NAME"),
+                   description=get_default("MODERATION_COMMAND_CLEAR_HELP"),
+                   description_localizations=get_localizations("MODERATION_COMMAND_CLEAR_HELP"),
+                   default_member_permissions=PERMISSIONS)
+    async def clear(self, ctx: ApplicationContext, keep: Optional[str]) -> None:
+        """
+        Clears the messages of a channel.
+        :param ctx: The context of the application.
+        :param keep: The id of the message that should stay.
+        """
+        if keep is not None:
+            try:
+                keep_id = int(keep)
+            except ValueError:
+                await ctx.respond(localize("MODERATION_COMMAND_CLEAR_INVALID", ctx.locale, keep), ephemeral=True)
+                return
+        else:
+            keep_id = 0
+
+        await ctx.defer(ephemeral=True)
+
+        def should_keep(msg: Message):  # noqa: ANN202
+            return msg.id == keep_id
+
+        async for message in ctx.channel.history(limit=100):
+            if should_keep(message):
+                continue
+            await self._safely_delete(ctx, message)
+
+        await ctx.followup.send(localize("MODERATION_COMMAND_CLEAR_DONE", ctx.locale), ephemeral=True)