From be1cd2e0b0da4f79110f35498b8f1ec5624be3ab Mon Sep 17 00:00:00 2001
From: flifloo <flifloo@gmail.com>
Date: Wed, 8 Apr 2020 19:28:05 +0200
Subject: [PATCH] Add of speak reaction command

---
 extensions/speak.py | 33 ++++++++++++++++++++++++++++++---
 1 file changed, 30 insertions(+), 3 deletions(-)

diff --git a/extensions/speak.py b/extensions/speak.py
index 36c1385..29f5443 100644
--- a/extensions/speak.py
+++ b/extensions/speak.py
@@ -16,6 +16,8 @@ class Speak(commands.Cog):
         self.voice_chan = None
         self.waiting = []
         self.lastSpeaker = None
+        self.reaction = []
+        self.lastReaction = None
 
     @commands.group("speak", pass_context=True)
     @commands.guild_only()
@@ -33,6 +35,17 @@ class Speak(commands.Cog):
                 self.waiting.append(ctx.author.id)
                 await ctx.message.add_reaction("\U0001f44d")
 
+    @speak.group("react", pass_context=True)
+    @commands.guild_only()
+    async def speak_react(self, ctx: commands.Context):
+        if ctx.author.voice is None or ctx.author.voice.channel is None or self.voice_chan is None or \
+                ctx.author.voice.channel.id != self.voice_chan or ctx.author.id in self.reaction or \
+                self.lastSpeaker is None or self.lastSpeaker == ctx.author.id:
+            await ctx.message.add_reaction("\u274C")
+        else:
+            self.reaction.append(ctx.author.id)
+            await ctx.message.add_reaction("\U0001f44d")
+
     @speak.group("remove", pass_context=True)
     @commands.guild_only()
     async def speak_remove(self, ctx: commands.Context):
@@ -66,18 +79,30 @@ class Speak(commands.Cog):
         if not self.voice_chan or not ctx.guild.get_channel(self.voice_chan).permissions_for(ctx.author).mute_members:
             await ctx.message.add_reaction("\u274C")
         else:
-            if self.lastSpeaker:
+            if self.lastReaction:
+                self.reaction.remove(self.lastReaction)
+                if self.strict:
+                    await ctx.guild.get_member(self.lastReaction).edit(mute=True)
+            if self.lastSpeaker and len(self.reaction) == 0:
                 self.waiting.remove(self.lastSpeaker)
                 if self.strict:
                     await ctx.guild.get_member(self.lastSpeaker).edit(mute=True)
-            if len(self.waiting) != 0:
-                user : Member = ctx.guild.get_member(self.waiting[0])
+            if len(self.reaction) != 0 and self.lastSpeaker is not None:
+                user: Member = ctx.guild.get_member(self.reaction[0])
+                self.lastReaction = self.reaction[0]
+                await ctx.send(f"{user.mention} react on {ctx.guild.get_member(self.lastSpeaker).mention} speak !")
+                if self.strict:
+                    await user.edit(mute=False)
+            elif len(self.waiting) != 0:
+                user: Member = ctx.guild.get_member(self.waiting[0])
                 self.lastSpeaker = self.waiting[0]
+                self.lastReaction = None
                 await ctx.send(f"It's {user.mention} turn")
                 if self.strict:
                     await user.edit(mute=False)
             else:
                 self.lastSpeaker = None
+                self.lastReaction = None
                 await ctx.send("Nobody left !")
 
     @speak.group("help", pass_context=True)
@@ -123,6 +148,8 @@ class Speak(commands.Cog):
         else:
             self.waiting = []
             self.lastSpeaker = None
+            self.reaction = []
+            self.lastReaction = None
             for client in speak_channel.members:
                 if client != ctx.author and not client.bot:
                     await client.edit(mute=False)