Add purge command

extensions/__init__.py

@@ -3,3 +3,4 @@ from bot_bde import bot
 bot.load_extension("extensions.help")
 bot.load_extension("extensions.speak")
 bot.load_extension("extensions.extension")
+bot.load_extension("extensions.purge")

extensions/purge.py

@@ -0,0 +1,98 @@
+from asyncio import sleep
+
+from discord.ext import commands
+from discord import Embed, RawReactionActionEvent
+from discord.ext.commands import CommandNotFound
+
+from bot_bde.logger import logger
+
+
+extension_name = "purge"
+logger = logger.getChild(extension_name)
+
+
+class Purge(commands.Cog):
+    def __init__(self, bot: commands.Bot):
+        self.bot = bot
+        self.purges = {}
+
+    @commands.group("purge", pass_context=True)
+    @commands.guild_only()
+    async def purge(self, ctx: commands.Context):
+        if ctx.invoked_subcommand is None:
+            if ctx.message.author.guild_permissions.manage_messages:
+                self.purges[ctx.message.author.id] = ctx.message
+            await ctx.message.add_reaction("\U0001f44d")
+
+            await sleep(2*60)
+            try:
+                if self.purges[ctx.message.author.id] == ctx.message:
+                    await ctx.message.clear_reactions()
+                    del self.purges[ctx.message.author.id]
+            except:
+                pass
+
+        else:
+            await ctx.message.add_reaction("\u274C")
+
+    @purge.group("help", pass_context=True)
+    @commands.guild_only()
+    async def purge_help(self, ctx: commands.Context):
+        embed = Embed(title="Speak help")
+        embed.add_field(name="speak", value="Join the waiting list", inline=False)
+        embed.add_field(name="speak remove [@pepole, @...]",
+                        value="Remove yourself or mentioned person from the waiting list", inline=False)
+        embed.add_field(name="speak list", value="Show the waiting list", inline=False)
+        embed.add_field(name="Speak setup [strict]",
+                        value="Set your current voice channel as the speak channel, you cant add the argument `strict` "
+                              "to mute everyone except you and the current speaker", inline=False)
+        embed.add_field(name="speak next",
+                        value="Give the turn to the next waiter, if strict mode is enabled the last person get muted "
+                              "and the next unmuted", inline=False)
+        embed.add_field(name="speak mute", value="Mute everyone on the speak channel except you", inline=False)
+        embed.add_field(name="speak unmute", value="Unmute everyone on the speak channel except you", inline=False)
+        embed.add_field(name="speak clear", value="Clear the speak by unmute everyone and forget the list & channel",
+                        inline=False)
+        await ctx.send(embed=embed)
+
+    @commands.Cog.listener()
+    async def on_raw_reaction_add(self, payload: RawReactionActionEvent):
+        user = self.bot.get_user(payload.user_id)
+        message = await self.bot.get_guild(payload.guild_id).get_channel(payload.channel_id)\
+            .fetch_message(payload.message_id)
+        if user.id in self.purges:
+            if message.channel == self.purges[user.id].channel:
+                async with message.channel.typing():
+                    await message.channel.purge(before=self.purges[user.id], after=message,
+                                                         limit=None)
+                    await self.purges[user.id].delete()
+                    await message.delete()
+                    del self.purges[user.id]
+
+    @commands.Cog.listener()
+    async def on_command_error(self, ctx: commands.Context, error):
+        if isinstance(error, CommandNotFound):
+            await ctx.message.add_reaction("\u2753")
+        else:
+            await ctx.send("An error occurred !")
+            raise error
+
+
+def setup(bot):
+    logger.info(f"Loading...")
+    try:
+        bot.add_cog(Purge(bot))
+    except Exception as e:
+        logger.error(f"Error loading: {e}")
+    else:
+        logger.info(f"Load successful")
+
+
+def teardown(bot):
+    logger.info(f"Unloading...")
+    try:
+        bot.remove_cog("Speak")
+    except Exception as e:
+        logger.error(f"Error unloading: {e}")
+    else:
+        logger.info(f"Unload successful")