aboutsummaryrefslogtreecommitdiff
path: root/code/cogs/play.py
diff options
context:
space:
mode:
Diffstat (limited to 'code/cogs/play.py')
-rw-r--r--code/cogs/play.py171
1 files changed, 132 insertions, 39 deletions
diff --git a/code/cogs/play.py b/code/cogs/play.py
index 895f5ab..f25b3d8 100644
--- a/code/cogs/play.py
+++ b/code/cogs/play.py
@@ -6,8 +6,10 @@ from cogs.music import Music
from lavalink import LoadType
import re
from cogs.music import LavalinkVoiceClient
+import requests
from global_variables import BOT_COLOR
+from custom_source import CustomSource
url_rx = re.compile(r"https?://(?:www\.)?.+")
@@ -42,57 +44,148 @@ class Play(commands.Cog):
)
return await interaction.response.send_message(embed=embed)
- if not url_rx.match(query):
- ytsearch = f"scsearch:{query}"
- results = await player.node.get_tracks(ytsearch)
+ # If a Spotify link is found, act accordingly
+ # We use a custom source for this (I tried the LavaSrc plugin, but Spotify
+ # links would just result in random shit being played whenever ytsearch was removed)
+ if "open.spotify.com" in query:
+ embed = discord.Embed(color=BOT_COLOR)
+
+ if "open.spotify.com/playlist" in query:
+ playlist_id = query.split("playlist/")[1].split("?si=")[0]
+ playlist_url = f"https://api.spotify.com/v1/playlists/{playlist_id}"
+ response = requests.get(playlist_url, headers=self.bot.spotify_headers)
+ if response.status_code == 200:
+ playlist = response.json()
+
+ embed.title = "Playlist Queued"
+ embed.description = f"**{playlist['name']}** from **{playlist['owner']['display_name']}**\n` {len(playlist['tracks']['items'])} ` tracks\n\nQueued by: {interaction.user.mention}"
+ embed.set_thumbnail(url=playlist["images"][0]["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)
+
+ tracks = await CustomSource.load_playlist(
+ self, interaction.user, playlist
+ )
+ for track in tracks["tracks"]:
+ player.add(requester=interaction.user, track=track)
+
+ if "open.spotify.com/album" in query:
+ album_id = query.split("album/")[1]
+ album_url = f"https://api.spotify.com/v1/albums/{album_id}"
+ response = requests.get(album_url, headers=self.bot.spotify_headers)
+ if response.status_code == 200:
+ album = response.json()
+
+ embed.title = "Album Queued"
+ embed.description = f"**{album['name']}** by **{album['artists'][0]['name']}**\n` {len(album['tracks']['items'])} ` tracks\n\nQueued by: {interaction.user.mention}"
+ embed.set_thumbnail(url=album["images"][0]["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)
+
+ tracks = await CustomSource.load_album(
+ self, interaction.user, album
+ )
+ for track in tracks["tracks"]:
+ player.add(requester=interaction.user, track=track)
+
+ if "open.spotify.com/track" in query:
+ track_id = query.split("track/")[1]
+ track_url = f"https://api.spotify.com/v1/tracks/{track_id}"
+ response = requests.get(track_url, headers=self.bot.spotify_headers)
+ if response.status_code == 200:
+ track = response.json()
+
+ embed.title = "Track Queued"
+ embed.description = f"**{track['name']}** by **{track['artists'][0]['name']}**\n\nQueued by: {interaction.user.mention}"
+ embed.set_thumbnail(url=track["album"]["images"][0]["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)
+
+ results = await CustomSource.load_item(
+ self, interaction.user, track
+ )
+ player.add(requester=interaction.user, track=results.tracks[0])
+
+ if "open.spotify.com/artists" in query:
+ embed.title = "Artists Cannot Be Played"
+ embed.description = "I cannot play just artists, you must provide a song/album/playlist. Please try again."
+ return await interaction.response.send_message(
+ embed=embed, ephemeral=True
+ )
+
+ # For anything else, do default searches and attempt to find track and play
+
+ else:
+ if not url_rx.match(query):
+ ytsearch = f"scsearch:{query}"
+ results = await player.node.get_tracks(ytsearch)
+
+ if not results.tracks or results.load_type in (
+ LoadType.EMPTY,
+ LoadType.ERROR,
+ ):
+ scsearch = f"dzsearch:{query}"
+ results = await player.node.get_tracks(scsearch)
+ 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,
):
- scsearch = f"dzsearch:{query}"
- results = await player.node.get_tracks(scsearch)
- else:
- results = await player.node.get_tracks(query)
+ 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
+ )
- embed = discord.Embed(color=BOT_COLOR)
+ elif results.load_type == LoadType.PLAYLIST:
+ tracks = results.tracks
- 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)
+ for track in tracks:
+ player.add(requester=interaction.user, track=track)
- elif results.load_type == LoadType.PLAYLIST:
- tracks = results.tracks
+ 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)
- for track in tracks:
+ else:
+ track = results.tracks[0]
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"
+ 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"
)
- + " 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)
+ 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:
generated by cgit v1.2.3-70-g09d2 (git 2.46.0) at 2025-07-18 19:17:13 +0000