diff options
| -rw-r--r-- | code/cogs/play.py | 100 |
1 files changed, 100 insertions, 0 deletions
diff --git a/code/cogs/play.py b/code/cogs/play.py new file mode 100644 index 0000000..454881b --- /dev/null +++ b/code/cogs/play.py @@ -0,0 +1,100 @@ +import discord +import datetime +from discord import app_commands +from discord.ext import commands +from cogs.music import Music +from lavalink import LoadType +import re +from cogs.music import LavalinkVoiceClient + +from global_variables import BOT_COLOR + + +url_rx = re.compile(r"https?://(?:www\.)?.+") + + +class Play(commands.Cog): + def __init__(self, bot): + self.bot = bot + + + @app_commands.command() + @app_commands.check(Music.create_player) + @app_commands.describe(query="Name or link of song") + async def play(self, interaction: discord.Interaction, query: str): + "Play a song from your favorite music provider" + player = self.bot.lavalink.player_manager.get(interaction.guild.id) + + if not url_rx.match(query): + ytsearch = f"ytsearch:{query}" + results = await player.node.get_tracks(ytsearch) + + if not results.tracks or results.load_type in ( + LoadType.EMPTY, + LoadType.ERROR, + ): + scsearch = f"scsearch:{query}" + results = await player.node.get_tracks(scsearch) + + if not results.tracks or results.load_type in ( + LoadType.EMPTY, + LoadType.ERROR, + ): + dzsearch = f"dzsearch:{query}" + results = await player.node.get_tracks(dzsearch) + else: + results = await player.node.get_tracks(query) + + embed = discord.Embed(color=BOT_COLOR) + + if not results.tracks or results.load_type in (LoadType.EMPTY, LoadType.ERROR): + embed.title = "Nothing Found" + embed.description = "Nothing for that query could be found. If this continues happening for other songs, please run `/bug` to let the developer know." + return await interaction.response.send_message(embed=embed, ephemeral=True) + + elif results.load_type == LoadType.PLAYLIST: + tracks = results.tracks + + for track in tracks: + player.add(requester=interaction.user, track=track) + + embed.title = "Songs Queued!" + embed.description = f"**{results.playlist_info.name}**\n` {len(tracks)} ` tracks\n\nQueued by: {interaction.user.mention}" + 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) + + else: + track = results.tracks[0] + + embed.title = "Track Queued" + embed.description = f"**{track.title}** by **{track.author}**\n\nQueued by: {interaction.user.mention}" + embed.set_thumbnail(url=track.artwork_url) + embed.set_footer( + text=datetime.datetime.now(datetime.timezone.utc).strftime( + "%Y-%m-%d %H:%M:%S" + ) + + " UTC" + ) + + player.add(requester=interaction.user, track=track) + await interaction.response.send_message(embed=embed) + + # Only join the voice channel now, so that the bot doesn't join if nothing is found + if not interaction.guild.voice_client: + await interaction.user.voice.channel.connect( + cls=LavalinkVoiceClient, self_deaf=True + ) + + # We don't want to call .play() if the player is playing as that will + # effectively skip the current track + if not player.is_playing: + await player.play() + + +async def setup(bot): + await bot.add_cog(Play(bot)) |