Administrator-py/extensions/speak.py

from discord.ext import commands
from discord import Member, VoiceState, Embed
from discord.ext.commands import CommandNotFound

from bot_bde.logger import logger


extension_name = "speak"
logger = logger.getChild(extension_name)


class Speak(commands.Cog):
    def __init__(self, bot: commands.Bot):
        self.bot = bot
        self.strict = False
        self.voice_chan = None
        self.waiting = []
        self.lastSpeaker = None

    @commands.group("speak", pass_context=True)
    async def speak(self, ctx: commands.Context):
        if ctx.invoked_subcommand is None:
            if ctx.author.voice is None or ctx.author.voice.channel is None:
                await ctx.send("Your not in a voice channel !")
            elif self.voice_chan is None:
                await ctx.send("Voice channel not set !")
            elif ctx.author.voice.channel.id != self.voice_chan:
                await ctx.send("Your not in the good voice channel !")
            elif ctx.author.id in self.waiting:
                await ctx.message.add_reaction("\u274C")
            else:
                self.waiting.append(ctx.author.id)
                await ctx.message.add_reaction("\U0001f44d")

    @speak.group("remove", pass_context=True)
    async def speak_remove(self, ctx: commands.Context):
        if len(ctx.message.mentions) != 0:
            if self.voice_chan and ctx.guild.get_channel(self.voice_chan).permissions_for(ctx.author).mute_members:
                for speaker in ctx.message.mentions:
                    self.waiting.remove(speaker.id)
                await ctx.message.add_reaction("\U0001f44d")
            else:
                await ctx.message.add_reaction("\u274C")
        elif ctx.author.id in self.waiting:
            self.waiting.remove(ctx.author.id)
            await ctx.message.add_reaction("\U0001f44d")
        else:
            await ctx.message.add_reaction("\u274C")

    @speak.group("list", pass_context=True)
    async def speak_list(self, ctx: commands.Context):
        if ctx.author.voice.channel is None or not self.voice_chan:
            await ctx.message.add_reaction("\u274C")
        else:
            embed = Embed(title="Waiting list")
            for i, speaker in enumerate(self.waiting):
                embed.add_field(name=f"N°{i+1}", value=ctx.guild.get_member(speaker).display_name, inline=True)
            await ctx.send(embed=embed)

    @speak.group("next", pass_context=True)
    async def speak_next(self, ctx: commands.Context):
        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:
                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])
                self.lastSpeaker = self.waiting[0]
                await ctx.send(f"It's {user.mention} turn")
                if self.strict:
                    await user.edit(mute=False)
            else:
                self.lastSpeaker = None
                await ctx.send("Nobody left !")

    @speak.group("help", pass_context=True)
    async def speak_help(self, ctx: commands.Context):
        await ctx.send("help")

    @speak.group("setup", pass_context=True)
    async def speak_setup(self, ctx: commands.Context, *args):
        if not ctx.author.voice.channel.permissions_for(ctx.author).mute_members:
            await ctx.message.add_reaction("\u274C")
        else:
            if len(args) != 0 and args[0] == "strict":
                self.strict = True
                for client in ctx.author.voice.channel.members:
                    if client != ctx.author and not client.bot:
                        await client.edit(mute=True)
            self.voice_chan = ctx.author.voice.channel.id
            await ctx.message.add_reaction("\U0001f44d")

    @speak.group("clear", pass_context=True)
    async def speak_clear(self, ctx: commands.Context):
        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:
            self.waiting = []
            self.lastSpeaker = None
            await self.speak_unmute(ctx)
            self.strict = False
            self.voice_chan = None
            await ctx.message.add_reaction("\U0001f44d")

    @speak.group("mute", pass_context=True)
    async def speak_mute(self, ctx: commands.Context):
        if not self.voice_chan or not ctx.guild.get_channel(self.voice_chan).permissions_for(ctx.author).mute_members or not self.voice_chan:
            await ctx.message.add_reaction("\u274C")
        else:
            for client in ctx.author.voice.channel.members:
                if client != ctx.author and not client.bot:
                    await client.edit(mute=True)
            await ctx.message.add_reaction("\U0001f44d")

    @speak.group("unmute", pass_context=True)
    async def speak_unmute(self, ctx: commands.Context):
        if not self.voice_chan or not ctx.guild.get_channel(self.voice_chan).permissions_for(ctx.author).mute_members or not self.voice_chan:
            await ctx.message.add_reaction("\u274C")
        else:
            for client in ctx.author.voice.channel.members:
                if client != ctx.author and not client.bot:
                    await client.edit(mute=False)
            await ctx.message.add_reaction("\U0001f44d")

    @commands.Cog.listener()
    async def on_voice_state_update(self, member: Member, before: VoiceState, after: VoiceState):
        if (before is None or before.channel is None or before.channel.id != self.voice_chan) and\
                (after is not None and after.channel is not None and after.channel.id == self.voice_chan and self.strict):
            await member.edit(mute=True)
        elif (before is not None or before.channel is not None and before.channel.id == self.voice_chan) and\
                (after is None or after.channel is None or after.channel.id != self.voice_chan):
            await member.edit(mute=False)

    @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(Speak(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")