diff options
| -rw-r--r-- | code/cogs/skip.py | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/code/cogs/skip.py b/code/cogs/skip.py new file mode 100644 index 0000000..51d466e --- /dev/null +++ b/code/cogs/skip.py @@ -0,0 +1,63 @@ +import discord +import datetime +from discord import app_commands +from discord.ext import commands +from cogs.music import Music +import asyncio + +from global_variables import BOT_COLOR + + +class Skip(commands.Cog): + def __init__(self, bot): + self.bot = bot + + @app_commands.command() + @app_commands.check(Music.create_player) + @app_commands.describe(number="Optional: Number of songs to skip, default is 1") + async def skip(self, interaction: discord.Interaction, number: int = 1): + "Skips the song that is currently playing" + player = self.bot.lavalink.player_manager.get(interaction.guild.id) + + embed = discord.Embed(color=BOT_COLOR) + + if number != 1: + if number < 1: + embed.title = "Invalid Number" + embed.description = "The number option cannot be less than 1" + return await interaction.response.send_message( + embed=embed, ephemeral=True + ) + + elif number > len(player.queue): + embed.title = "Number too Large" + embed.description = "The number you entered is larger than the number of songs in queue. If you want to stop playing music entirely, try the `/stop` command." + return await interaction.response.send_message( + embed=embed, ephemeral=True + ) + else: + for i in range(number - 2, -1, -1): + player.queue.pop(i) + + await player.skip() + + # It takes a sec for the new track to be grabbed and played + # So just wait a sec before sending the message + await asyncio.sleep(0.5) + embed = discord.Embed( + title="Track Skipped", + description=f"**Now Playing: [{player.current.title}]({player.current.uri})** by {player.current.author}\n\nQueued by: {player.current.requester.mention}", + color=BOT_COLOR, + ) + embed.set_thumbnail(url=player.current.artwork_url) + embed.set_footer( + text=datetime.datetime.now(datetime.timezone.utc).strftime( + "%Y-%m-%d %H:%M:%S" + ) + + " UTC" + ) + await interaction.response.send_message(embed=embed) + + +async def setup(bot): + await bot.add_cog(Skip(bot)) |