import asyncio import discord from discord.ext import commands class VoiceEntry: def __init__(self, message, player): self.requester = message.author self.channel = message.channel self.player = player def __str__(self): fmt = '*{0.title}* verzocht door {1.display_name}' duration = self.player.duration if duration: fmt = fmt + ' [{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) @asyncio.coroutine def audio_player_task(self): while True: self.play_next_song.clear() self.current = yield from self.songs.get() yield from self.bot.send_message(self.current.channel, 'U luistert naar: ' + str(self.current)) self.current.player.start() yield from self.play_next_song.wait() 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('opus') class Music: """Play that funky music white boy. """ def __init__(self, bot): self.bot = bot self.voice_states = {} def __check(self, ctx): return True 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 @asyncio.coroutine def create_voice_client(self, channel): voice = yield from 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 @commands.command(pass_context=True, no_pm=True) @asyncio.coroutine def join(self, ctx, *, channel : discord.Channel): """Verplaats stroopie naar een bepaald kanaal.""" try: yield from self.create_voice_client(channel) except discord.ClientException: yield from self.bot.say('Already in a voice channel...') except discord.InvalidArgument: yield from self.bot.say('This is not a voice channel...') else: yield from self.bot.say('Ready to play audio in ' + channel.name) @commands.command(pass_context=True, no_pm=True) @asyncio.coroutine def summon(self, ctx): """Roep stroopie naar jouw kanaal toe..""" summoned_channel = ctx.message.author.voice_channel if summoned_channel is None: yield from self.bot.say('You are not in a voice channel.') return False state = self.get_voice_state(ctx.message.server) if state.voice is None: state.voice = yield from self.bot.join_voice_channel(summoned_channel) else: yield from state.voice.move_to(summoned_channel) return True @commands.command(pass_context=True, no_pm=True) @asyncio.coroutine def play(self, ctx, *, song : str): """Zet een liedje in de queue """ state = self.get_voice_state(ctx.message.server) opts = { 'default_search': 'auto', 'quiet': True, 'no-check-certificate': True, } if state.voice is None: success = yield from ctx.invoke(self.summon) if not success: return try: player = yield from state.voice.create_ytdl_player(song, ytdl_options=opts, after=state.toggle_next) except Exception as e: fmt = 'Error: ```py\n{}: {}\n```' yield from self.bot.send_message(ctx.message.channel, fmt.format(type(e).__name__, e)) else: player.volume = 0.6 entry = VoiceEntry(ctx.message, player) yield from self.bot.say('In de queue: ' + str(entry)) yield from state.songs.put(entry) @commands.command(pass_context=True, no_pm=True) @asyncio.coroutine def volume(self, ctx, value : int): """zet het volume. 1-100""" state = self.get_voice_state(ctx.message.server) if state.is_playing(): player = state.player player.volume = value / 100 yield from self.bot.say('Zet het volume op {:.0%}'.format(player.volume)) @commands.command(pass_context=True, no_pm=True) @asyncio.coroutine def pause(self, ctx): """Pauzeer het huidige liedje""" state = self.get_voice_state(ctx.message.server) if state.is_playing(): player = state.player player.pause() @commands.command(pass_context=True, no_pm=True) @asyncio.coroutine def resume(self, ctx): """onpauzeer.""" state = self.get_voice_state(ctx.message.server) if state.is_playing(): player = state.player player.resume() @commands.command(pass_context=True, no_pm=True) @asyncio.coroutine def stop(self, ctx): """Kappen nou. Stoppen met spelen, gaat uit het kanaal en leegt de queue. """ 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] yield from state.voice.disconnect() except: pass @commands.command(pass_context=True, no_pm=True) @asyncio.coroutine def skip(self, ctx): """skipperdepip """ state = self.get_voice_state(ctx.message.server) if not state.is_playing(): yield from self.bot.say('Not playing any music right now...') return voter = ctx.message.author if voter == state.current.requester: yield from self.bot.say('Requester requested skipping song...') 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: yield from self.bot.say('Skip vote passed, skipping song...') state.skip() else: yield from self.bot.say('Skip vote added, currently at [{}/3]'.format(total_votes)) else: yield from self.bot.say('You have already voted to skip this song.') @commands.command(pass_context=True, no_pm=True) @asyncio.coroutine def playing(self, ctx): """Info over het huidige liedje.""" state = self.get_voice_state(ctx.message.server) if state.current is None: yield from self.bot.say('Not playing anything.') else: skip_count = len(state.skip_votes) yield from self.bot.say('Now playing {} [skips: {}/3]'.format(state.current, skip_count))