407 lines
20 KiB
Python
407 lines
20 KiB
Python
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("Musique en cours lancée: "+str(self.current))
|
|
#await self.bot.change_presence(game=discord.Game(name='faire de la musique: '+str(self.current)))
|
|
embed=discord.Embed(title="Musique", description="En cours", color=0x80ff00)
|
|
embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
|
|
embed.add_field(name="play", value="Jouée acutellement: \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 Musique:
|
|
"""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 set_pause(self, server):
|
|
state = self.get_voice_state(server)
|
|
player = state.player
|
|
player.pause()
|
|
|
|
def set_resume(self,server):
|
|
state = self.get_voice_state(server)
|
|
player = state.player
|
|
player.resume()
|
|
|
|
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="Musique", 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 quelle musique est lancée", inline=True)
|
|
embed.add_field(name="pause", value="Met en pause la musique", inline=True)
|
|
embed.add_field(name="resume", value="Résume la musique", inline=True)
|
|
embed.add_field(name="skip", value="Passer la musique", inline=True)
|
|
embed.add_field(name="stop", value="Arrêter la musique", inline=True)
|
|
embed.add_field(name="summon", value="Faire apparaître le bot", inline=True)
|
|
embed.add_field(name="volume", value="Définir 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 musique summon lancée par: "+str(ctx.message.author)+" refuser car il n'est pas dans un channel vocal")
|
|
embed=discord.Embed(title="Musique", 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 musique summon lancée par: "+str(ctx.message.author))
|
|
state.voice = await self.bot.join_voice_channel(summoned_channel)
|
|
embed=discord.Embed(title="Musique", description="", color=0x80ff00)
|
|
embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
|
|
embed.add_field(name="summon", value="Il se matérialise devant vous !", inline=True)
|
|
await self.bot.say(embed=embed)
|
|
else:
|
|
print("Commande music summon lancée par: "+str(ctx.message.author))
|
|
await state.voice.move_to(summoned_channel)
|
|
embed=discord.Embed(title="Musique", 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 déjà, 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 musique play lancée par: "+str(ctx.message.author))
|
|
#embed=discord.Embed(title="Musique", 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)
|
|
message = 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)
|
|
#await self.bot.delete_message(message)
|
|
|
|
|
|
except Exception as e:
|
|
fmt = 'Une erreur est arrivé lors du traitement de la requete: ```py\n{}: {}\n```'
|
|
print("Commande musique play lancée par: "+str(ctx.message.author)+" erreur: "+fmt.format(type(e).__name__, e))
|
|
embed=discord.Embed(title="Musique", 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 musique play lancée par: "+str(ctx.message.author)+" mise en queue")
|
|
embed=discord.Embed(title="Musique", 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 lancée par: "+str(ctx.message.author))
|
|
embed=discord.Embed(title="Musique", 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 lancée par: "+str(ctx.message.author)+" refuser car il n'est pas dans le channel vocal")
|
|
embed=discord.Embed(title="Musique", 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)
|
|
|
|
#MEt en pause la musique
|
|
@music.command(pass_context=True,no_pm=True)
|
|
async def pause(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():
|
|
print("Commande musique pause lancée par: "+str(ctx.message.author))
|
|
embed=discord.Embed(title="Musique", description="", color=0x80ff00)
|
|
embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
|
|
embed.add_field(name="pause", value="Mise en pause de la musique", inline=True)
|
|
await self.bot.say(embed=embed)
|
|
await self.set_pause(ctx.message.author.server)
|
|
|
|
elif self.is_listening(voice_channel_id) == False:
|
|
print("Commande musique resume lancée par: "+str(ctx.message.author)+" refuser car il n'est pas dans le channel vocal")
|
|
embed=discord.Embed(title="Musique", 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)
|
|
|
|
|
|
#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():
|
|
print("Commande musique resume lancée par: "+str(ctx.message.author))
|
|
embed=discord.Embed(title="Musique", 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)
|
|
await self.set_resume(ctx.message.author.server)
|
|
|
|
elif self.is_listening(voice_channel_id) == False:
|
|
print("Commande musique resume lancée par: "+str(ctx.message.author)+" refuser car il n'est pas dans le channel vocal")
|
|
embed=discord.Embed(title="Musique", 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)
|
|
|
|
async def on_voice_state_update(self,before,after):
|
|
state = self.get_voice_state(after.server)
|
|
if type(state.voice) != type(None):
|
|
if type(before.voice_channel) != type(None):
|
|
if state.voice.channel == before.voice_channel:
|
|
if len(before.voice_channel.voice_members)<2:
|
|
try:await self.get_pause(state.voice.channel.server)
|
|
except:pass
|
|
if type(after.voice_channel) != type(None):
|
|
if len(after.voice_channel.voice_members)>=2:
|
|
try:await self.get_resume(state.voice.channel.server)
|
|
except:pass
|
|
|
|
#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 musique stop lancée par: "+str(ctx.message.author))
|
|
embed=discord.Embed(title="Musique", 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 musique stop lancée par: "+str(ctx.message.author)+" refuser car il n'est pas dans le channel vocal")
|
|
embed=discord.Embed(title="Musique", 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 musique skip lancée par: "+str(ctx.message.author)+" refuser car rien est jouer")
|
|
embed=discord.Embed(title="Musique", 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 musique skip lancée par: "+str(ctx.message.author)+" requete envoyer")
|
|
embed=discord.Embed(title="Musique", 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 musique skip lancée")
|
|
embed=discord.Embed(title="Musique", 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="Musique", 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="Musique", description="", color=0x80ff00)
|
|
embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
|
|
embed.add_field(name="skip", value="Vous avez déjà voté pour passer cette chanson.", inline=True)
|
|
await self.bot.say(embed=embed)
|
|
|
|
elif self.is_listening(voice_channel_id) == False:
|
|
print("Commande musique skip lancée par: "+str(ctx.message.author)+" refuser car il n'est pas dans le channel vocal")
|
|
embed=discord.Embed(title="Musique", 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 lancée
|
|
@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 musique playing lancée par: "+str(ctx.message.author)+" refuser car rien est jouer")
|
|
embed=discord.Embed(title="Musique", 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 musique playing lancée par: "+str(ctx.message.author))
|
|
embed=discord.Embed(title="Musique", description="", color=0x80ff00)
|
|
embed.set_thumbnail(url="http://www.icone-png.com/png/16/15638.png")
|
|
embed.add_field(name="playing", value="Jouée actuellement {} [skips: {}/3]".format(state.current, skip_count), inline=True)
|
|
await self.bot.say(embed=embed)
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Musique(bot))
|
|
print('Musique chargée')
|