Merge branch '16-skip-button' into 'develop'
Resolve "Add skip button" Closes #16 See merge request ekzyis/musicube!14
This commit is contained in:
commit
9f584a078b
30
src/bot.py
30
src/bot.py
|
@ -29,6 +29,7 @@ class Music(commands.Cog):
|
|||
self._bot = bot
|
||||
self._queue = asyncio.Queue()
|
||||
self._queue_lock = asyncio.Lock()
|
||||
self._current_skip_message = None
|
||||
self._ytdl_params = {
|
||||
'cookiefile': os.environ.get('YOUTUBE_COOKIES'),
|
||||
'format': 'bestaudio/best',
|
||||
|
@ -57,6 +58,7 @@ class Music(commands.Cog):
|
|||
def _next(self):
|
||||
"""Trigger playback of next song."""
|
||||
self._queue.task_done()
|
||||
self._current_skip_message = None
|
||||
if self._queue_lock.locked():
|
||||
self._queue_lock.release()
|
||||
|
||||
|
@ -80,7 +82,8 @@ class Music(commands.Cog):
|
|||
self._next()
|
||||
ctx.voice_client.play(audio, after=after)
|
||||
embed = NowPlayingMessage(title=song.title, url=song.webpage_url)
|
||||
await ctx.send(embed=embed)
|
||||
msg = await ctx.send(embed=embed)
|
||||
await self._add_skip_button(msg)
|
||||
# pylint: disable=broad-except
|
||||
except Exception as err:
|
||||
print(f"Error during playback: {err}")
|
||||
|
@ -89,6 +92,19 @@ class Music(commands.Cog):
|
|||
await ctx.send(embed=embed)
|
||||
self._next()
|
||||
|
||||
async def _add_skip_button(self, msg):
|
||||
await msg.add_reaction("⏭️")
|
||||
self._current_skip_message = msg
|
||||
|
||||
@commands.Cog.listener()
|
||||
async def on_reaction_add(self, reaction, user):
|
||||
if not user.bot \
|
||||
and self._current_skip_message \
|
||||
and reaction.message.id == self._current_skip_message.id \
|
||||
and reaction.emoji == "⏭️":
|
||||
voice_client = reaction.message.guild.voice_client
|
||||
self._skip(voice_client)
|
||||
|
||||
@_handle_playback.before_loop
|
||||
async def before_handle_playback(self):
|
||||
await self._bot.wait_until_ready()
|
||||
|
@ -109,12 +125,18 @@ class Music(commands.Cog):
|
|||
embed = QueuedMessage(title=song.title, url=song.webpage_url)
|
||||
await ctx.send(embed=embed)
|
||||
|
||||
def _skip(self, voice_client):
|
||||
"""Skip to next song."""
|
||||
if voice_client is None or not voice_client.is_playing():
|
||||
raise commands.CommandError("No song playing")
|
||||
# This skips to next song because the bot does not differentiate between
|
||||
# a song stopping because it is finished or because it was manually stopped.
|
||||
voice_client.stop()
|
||||
|
||||
@commands.command()
|
||||
async def skip(self, ctx):
|
||||
async with ctx.typing():
|
||||
if ctx.voice_client is None or not ctx.voice_client.is_playing():
|
||||
raise commands.CommandError("No song playing")
|
||||
ctx.voice_client.stop()
|
||||
self._skip(ctx.voice_client)
|
||||
|
||||
@commands.command()
|
||||
async def stop(self, ctx):
|
||||
|
|
Loading…
Reference in New Issue