FTW-Bot/Music.py

import asyncio
import discord
from discord.ext import commands
from ctypes.util import find_library
import youtube_dl
if not discord.opus.is_loaded():
    # the 'opus' library here is opus.dll on windows
    # or libopus.so on linux in the current directory
    # you should replace this with the location the
    # opus library is located in and with the proper filename.
    # note that on windows this DLL is automatically provided for you
    discord.opus.load_opus(find_library("opus"))

def __init__(self, bot):
        self.bot = bot

class VoiceEntry:
    def __init__(self, message, player):
        self.requester = message.author
        self.channel = message.channel
        self.player = player

    def __str__(self):
        fmt = ' {0.title} uploaded by {0.uploader} and requested by {1.display_name}'
        duration = self.player.duration
        if duration:
            fmt = fmt + ' [length: {0[0]}m {0[1]}s]'.format(divmod(duration, 60))
        return fmt.format(self.player, self.requester)

class VoiceState:
    def __init__(self, bot):
        self.current = None
        self.voice = None
        self.bot = bot
        self.play_next_song = asyncio.Event()
        self.songs = asyncio.Queue()
        self.skip_votes = set() # a set of user_ids that voted
        self.audio_player = self.bot.loop.create_task(self.audio_player_task())

    def is_playing(self):
        if self.voice is None or self.current is None:
            return False

        player = self.current.player
        return not player.is_done()

    @property
    def player(self):
        return self.current.player

    def skip(self):
        self.skip_votes.clear()
        if self.is_playing():
            self.player.stop()

    def toggle_next(self):
        self.bot.loop.call_soon_threadsafe(self.play_next_song.set)

    async def audio_player_task(self):
        while True:
            self.play_next_song.clear()
            self.current = await self.songs.get()
            print("Musi en cours lancer: "+str(self.current))
            embed=discord.Embed(title="Music", description="En cours", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="play", value="Joue maintenant \n" + str(self.current), inline=True)
            await self.bot.send_message(self.current.channel, embed=embed)
            self.current.player.start()
            await self.play_next_song.wait()

class Music:
    """Commandes de musique.
    """
    def __init__(self, bot):
        self.bot = bot
        self.voice_states = {}

    def get_voice_state(self, server):
        state = self.voice_states.get(server.id)
        if state is None:
            state = VoiceState(self.bot)
            self.voice_states[server.id] = state

        return state

    async def create_voice_client(self, channel):
        voice = await self.bot.join_voice_channel(channel)
        state = self.get_voice_state(channel.server)
        state.voice = voice

    def __unload(self):
        for state in self.voice_states.values():
            try:
                state.audio_player.cancel()
                if state.voice:
                    self.bot.loop.create_task(state.voice.disconnect())
            except:
                pass

    def is_listening(self, user_channel):
        for bot_channel in self.bot.voice_clients:
            if bot_channel.channel == user_channel:
                return True
            return False

    #Definition du group music
    @commands.group(pass_context=True)
    async def music(self, ctx):
        if ctx.invoked_subcommand is None:
            await ctx.invoke(self.help)

    #Un simple help
    @music.command(pass_context=True, no_pm=True)
    async def help(self, ctx):
        embed=discord.Embed(title="Music", description="Aide", color=0x80ff00)
        embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
        embed.add_field(name="play", value="Lancer une musique", inline=True)
        embed.add_field(name="playing", value="Voir se qui passe", inline=True)
        embed.add_field(name="pause", value="MEt en pause la musique, ne marche pas", inline=True)
        embed.add_field(name="resume", value="Continue la musique", inline=True)
        embed.add_field(name="skip", value="Passer la musique", inline=True)
        embed.add_field(name="stop", value="Arreter la musique", inline=True)
        embed.add_field(name="summon", value="Faire apparaitre le bot", inline=True)
        embed.add_field(name="volume", value="DEfinir le volume de la musique", inline=True)
        await self.bot.say(embed=embed)

    #Permmet de faire connecter vocalement le bot
    @music.command(pass_context=True, no_pm=True)
    async def summon(self, ctx):
        """Invoque le bot dans le channel vocal.
        Ne fonctionne que si l'utilisateur est déja dans un channel."""
        summoned_channel = ctx.message.author.voice_channel
        if summoned_channel is None:
            print("Commande music summon lancer par: "+str(ctx.message.author)+" refuser car il n'est pas dans un channel vocal")
            embed=discord.Embed(title="Music", description="Erreur", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="summon", value="Vous n'étes pas dans un channel vocal !", inline=True)
            await self.bot.say(embed=embed)
            return False

        state = self.get_voice_state(ctx.message.server)
        if state.voice is None:
            print("Commande music summon lancer par: "+str(ctx.message.author))
            state.voice = await self.bot.join_voice_channel(summoned_channel)
            embed=discord.Embed(title="Music", description="", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="summon", value="Il se materialise a vous !", inline=True)
            await self.bot.say(embed=embed)
        else:
            print("Commande music summon lancer par: "+str(ctx.message.author))
            await state.voice.move_to(summoned_channel)
            embed=discord.Embed(title="Music", description="", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="summon", value="Il se téléporte a vous !", inline=True)
            await self.bot.say(embed=embed)

        return True

    #Permet de mettre a jouer une musique
    @music.command(pass_context=True, no_pm=True)
    async def play(self, ctx, *, song : str):
        """Joue une musique.
        S'il y a une musique qui joue deja, alors elle est mise dans
        la queue jusqu'a la derniere musique de la queue.
        Le bot recherche automatiquement sur youtube.
        La liste des sites supportés est trouvée ici:
        https://rg3.github.io/youtube-dl/supportedsites.html
        """
        state = self.get_voice_state(ctx.message.server)
        opts = {
            "format":"bestaudio/worstvideo",
            'default_search': 'auto',
            'quiet': True,
        }

        if state.voice is None:
            success = await ctx.invoke(self.summon)
            print("Commande music play lancer par: "+str(ctx.message.author))
            #embed=discord.Embed(title="Music", description="", color=0x80ff00)
            #embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            #embed.add_field(name="play", value="Chargement de la musique...", inline=True)
            await self.bot.say("Chargement de la musique...")
            if not success:
                return

        try:
            player = await state.voice.create_ytdl_player(song, ytdl_options=opts, after=state.toggle_next)
        except Exception as e:
            fmt = 'Une erreur est arrivé lors du traitement de la requete: ```py\n{}: {}\n```'
            print("Commande music play lancer par: "+str(ctx.message.author)+" erreur: "+fmt.format(type(e).__name__, e))
            embed=discord.Embed(title="Music", description="Erreur", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="play", value=fmt.format(type(e).__name__, e), inline=True)
            await self.bot.send_message(ctx.message.channel, embed=embed)
        else:
            player.volume = 0.6
            entry = VoiceEntry(ctx.message, player)
            print("Commande music play lancer par: "+str(ctx.message.author)+" mise en queue")
            embed=discord.Embed(title="Music", description="", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="play", value="La musique \n" + str(entry)+" \na été mise en queue", inline=True)
            await self.bot.say(embed=embed)
            await state.songs.put(entry)


    #Permmet de definit le volume
    @music.command(pass_context=True, no_pm=True)
    async def volume(self, ctx, value : int):
        """Définie le volume du bot."""
        voice_channel_id = ctx.message.author.voice_channel
        if self.is_listening(voice_channel_id) == True:
            state = self.get_voice_state(ctx.message.server)
            if state.is_playing():
                player = state.player
                player.volume = value / 100
                print("Commande music volume lancer par: "+str(ctx.message.author))
                embed=discord.Embed(title="Music", description="", color=0x80ff00)
                embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
                embed.add_field(name="volume", value='Volume mit à {:.0%}'.format(player.volume), inline=True)
                await self.bot.say(embed=embed)

        elif self.is_listening(voice_channel_id) == False:
            print("Commande music volume lancer par: "+str(ctx.message.author)+" refuser car il n'est pas dans le channel vocal")
            embed=discord.Embed(title="Music", description="Erreur", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="volume", value="Vous n'étes pas dans le channel vocal !", inline=True)
            await self.bot.say(embed=embed)

    #Reprend la music
    @music.command(pass_context=True, no_pm=True)
    async def resume(self, ctx):
        voice_channel_id = ctx.message.author.voice_channel
        if self.is_listening(voice_channel_id) == True:
            state = self.get_voice_state(ctx.message.server)
            if state.is_playing():
                player = state.player
                player.resume()
                print("Commande music resume lancer par: "+str(ctx.message.author))
                embed=discord.Embed(title="Music", description="", color=0x80ff00)
                embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
                embed.add_field(name="resume", value="Reprise de la musique", inline=True)
                await self.bot.say(embed=embed)

        elif self.is_listening(voice_channel_id) == False:
            print("Commande music resume lancer par: "+str(ctx.message.author)+" refuser car il n'est pas dans le channel vocal")
            embed=discord.Embed(title="Music", description="Erreur", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="resume", value="Vous n'étes pas dans le channel vocal !", inline=True)
            await self.bot.say(embed=embed)

    #Arret la musique et fait quitter le bot
    @music.command(pass_context=True, no_pm=True)
    async def stop(self, ctx):
        """Arrete la musique jouée et quitte le channel.
        Cela vide aussi la queue.
        """
        voice_channel_id = ctx.message.author.voice_channel
        if self.is_listening(voice_channel_id) == True:
            server = ctx.message.server
            state = self.get_voice_state(server)

            if state.is_playing():
                player = state.player
                player.stop()

            try:
                state.audio_player.cancel()
                del self.voice_states[server.id]
                await state.voice.disconnect()
                print("Commande music stop lancer par: "+str(ctx.message.author))
                embed=discord.Embed(title="Music", description="", color=0x80ff00)
                embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
                embed.add_field(name="stop", value="Queue vidée et channel quitté.", inline=True)
                await self.bot.say(embed=embed)
            except:
                pass

        elif self.is_listening(voice_channel_id) == False:
            print("Commande music stop lancer par: "+str(ctx.message.author)+" refuser car il n'est pas dans le channel vocal")
            embed=discord.Embed(title="Music", description="Erreur", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="stop", value="Vous n'étes pas dans le channel vocal !", inline=True)
            await self.bot.say(embed=embed)

    #Permet de passer la musique en cours
    @music.command(pass_context=True, no_pm=True)
    async def skip(self, ctx):
        """Vote pour passer la chanson en cours.
        Il faut trois votes pour passer la chanson.
        """
        voice_channel_id = ctx.message.author.voice_channel
        if self.is_listening(voice_channel_id) == True:
            state = self.get_voice_state(ctx.message.server)
            if not state.is_playing():
                print("Commande music skip lancer par: "+str(ctx.message.author)+" refuser car rien est jouer")
                embed=discord.Embed(title="Music", description="Erreur", color=0x80ff00)
                embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
                embed.add_field(name="skip", value="Aucune chanson n'est jouée actuellement...", inline=True)
                await self.bot.say(embed=embed)
                return

            voter = ctx.message.author
            if voter == state.current.requester:
                print("Commande music skip lancer par: "+str(ctx.message.author)+" requete envoyer")
                embed=discord.Embed(title="Music", description="", color=0x80ff00)
                embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
                embed.add_field(name="skip", value="Une requete pour passer la chanson a été faite.", inline=True)
                await self.bot.say(embed=embed)
                state.skip()
            elif voter.id not in state.skip_votes:
                state.skip_votes.add(voter.id)
                total_votes = len(state.skip_votes)
                if total_votes >= 3:
                    print("Commande music skip lancer")
                    embed=discord.Embed(title="Music", description="", color=0x80ff00)
                    embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
                    embed.add_field(name="skip", value="Vote pour passer effectué, chanson passée...", inline=True)
                    await self.bot.say(embed=embed)
                    state.skip()
                else:
                    embed=discord.Embed(title="Music", description="", color=0x80ff00)
                    embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
                    embed.add_field(name="skip", value="Vote pour passer effectué, actuellement à [{}/3]".format(total_votes), inline=True)
                    await self.bot.say(embed=embed)
            else:
                embed=discord.Embed(title="Music", description="", color=0x80ff00)
                embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
                embed.add_field(name="skip", value="Vous avez deja voté pour passer cette chanson.", inline=True)
                await self.bot.say(embed=embed)

        elif self.is_listening(voice_channel_id) == False:
            print("Commande music skip lancer par: "+str(ctx.message.author)+" refuser car il n'est pas dans le channel vocal")
            embed=discord.Embed(title="Music", description="Erreur", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="skip", value="Vous n'étes pas dans le channel vocal !", inline=True)
            await self.bot.say(embed=embed)

    #Affiche la musique actuellement lancer
    @music.command(pass_context=True, no_pm=True)
    async def playing(self, ctx):

        state = self.get_voice_state(ctx.message.server)
        if state.current is None:
            print("Commande music playing lancer par: "+str(ctx.message.author)+" refuser car rien est jouer")
            embed=discord.Embed(title="Music", description="Erreur", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="playing", value="Aucune chanson n'est jouée actuellement...", inline=True)
            await self.bot.say(embed=embed)
        else:
            skip_count = len(state.skip_votes)
            print("Commande music playing lancer par: "+str(ctx.message.author))
            embed=discord.Embed(title="Music", description="", color=0x80ff00)
            embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
            embed.add_field(name="playing", value="Joue actuellement {} [skips: {}/3]".format(state.current, skip_count), inline=True)
            await self.bot.say(embed=embed)

def setup(bot):
    bot.add_cog(Music(bot))
    print('Music charger')