From afb18c5234628c9f052f44d94ad26c0d94ed3f12 Mon Sep 17 00:00:00 2001
From: flifloo <flifloo@gmail.com>
Date: Wed, 8 Apr 2020 18:18:04 +0200
Subject: [PATCH] Add purge command

---
 extensions/__init__.py |  1 +
 extensions/purge.py    | 98 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 99 insertions(+)
 create mode 100644 extensions/purge.py

diff --git a/extensions/__init__.py b/extensions/__init__.py
index b8eff8f..9953568 100644
--- a/extensions/__init__.py
+++ b/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")
diff --git a/extensions/purge.py b/extensions/purge.py
new file mode 100644
index 0000000..5ce6e40
--- /dev/null
+++ b/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")