320 lines
15 KiB
Python
320 lines
15 KiB
Python
import discord
|
|
import datetime
|
|
from discord import app_commands
|
|
from discord.ext import commands
|
|
from lavalink import LoadType
|
|
import re
|
|
import requests
|
|
|
|
from cogs.music import Music, LavalinkVoiceClient
|
|
from utils.config import BOT_COLOR
|
|
from utils.custom_sources import SpotifySource, AppleSource
|
|
|
|
|
|
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)
|
|
|
|
# Notify users that YouTube links are not allowed
|
|
|
|
if "youtube.com" in query or "youtu.be" in query:
|
|
embed = discord.Embed(
|
|
title="YouTube Not Supported",
|
|
description="Unfortunately, YouTube does not allow bots to stream from their platform. Try sending a link for a different platform, or simply type the name of the song and I will automatically find it on a supported platform.",
|
|
color=BOT_COLOR,
|
|
)
|
|
return await interaction.response.send_message(embed=embed, ephemeral=True)
|
|
|
|
###
|
|
### APPLE MUSIC links, perform API requests and load all tracks from the playlist/album/track
|
|
###
|
|
|
|
if "music.apple.com" in query:
|
|
if not self.bot.apple_headers:
|
|
embed = discord.Embed(
|
|
title="Apple Music Error",
|
|
description="Apple Music support seems to be broken at the moment. Please try again and fill out a bug report with </bug:1224840889906499626> if this continues to happen.",
|
|
color=BOT_COLOR,
|
|
)
|
|
return await interaction.response.send_message(embed=embed, ephemeral=True)
|
|
|
|
embed = discord.Embed(color=BOT_COLOR)
|
|
|
|
if "/playlist/" in query and "?i=" not in query:
|
|
playlist_id = query.split("/playlist/")[1].split("/")[1]
|
|
# Get all of the tracks in the playlist (limit at 250)
|
|
playlist_url = f"https://api.music.apple.com/v1/catalog/us/playlists/{playlist_id}/tracks?limit=100"
|
|
response = requests.get(playlist_url, headers=self.bot.apple_headers)
|
|
if response.status_code == 200:
|
|
playlist = response.json()
|
|
# Get the general playlist info (name, artwork)
|
|
playlist_info_url = f"https://api.music.apple.com/v1/catalog/us/playlists/{playlist_id}"
|
|
playlist_info = requests.get(playlist_info_url, headers=self.bot.apple_headers)
|
|
playlist_info = playlist_info.json()
|
|
try:
|
|
artwork_url = playlist_info["data"][0]["attributes"]["artwork"]["url"].replace(
|
|
"{w}x{h}", "300x300"
|
|
)
|
|
except KeyError:
|
|
artwork_url = None
|
|
|
|
embed.title = "Playlist Queued"
|
|
embed.description = f"**{playlist_info['data'][0]['attributes']['name']}**\n` {len(playlist['data'])} ` tracks\n\nQueued by: {interaction.user.mention}"
|
|
embed.set_thumbnail(url=artwork_url)
|
|
embed.set_footer(
|
|
text=datetime.datetime.now(datetime.timezone.utc).strftime(
|
|
"%Y-%m-%d %H:%M:%S"
|
|
)
|
|
+ " UTC"
|
|
)
|
|
# Add small alert if the playlist is the max size
|
|
if len(playlist["data"]) == 100:
|
|
embed.description += "\n\n*This playlist is longer than the 100 song maximum. Only the first 100 songs will be queued.*"
|
|
|
|
await interaction.response.send_message(embed=embed)
|
|
|
|
tracks = await AppleSource.load_playlist(
|
|
self, interaction.user, playlist
|
|
)
|
|
for track in tracks["tracks"]:
|
|
player.add(requester=interaction.user, track=track)
|
|
|
|
# If there is an album, not a specific song within the album
|
|
if "/album/" in query and "?i=" not in query:
|
|
album_id = query.split("/album/")[1].split("/")[1]
|
|
album_url = f"https://api.music.apple.com/v1/catalog/us/albums/{album_id}"
|
|
response = requests.get(album_url, headers=self.bot.apple_headers)
|
|
if response.status_code == 200:
|
|
album = response.json()
|
|
|
|
embed.title = "Album Queued"
|
|
embed.description = f"**{album['data'][0]['attributes']['name']}** by **{album['data'][0]['attributes']['artistName']}**\n` {len(album['data'][0]['relationships']['tracks']['data'])} ` tracks\n\nQueued by: {interaction.user.mention}"
|
|
embed.set_thumbnail(
|
|
url=album["data"][0]["attributes"]["artwork"]["url"].replace(
|
|
"{w}x{h}", "300x300"
|
|
)
|
|
)
|
|
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 AppleSource.load_album(self, interaction.user, album)
|
|
for track in tracks["tracks"]:
|
|
player.add(requester=interaction.user, track=track)
|
|
|
|
# If there is a specific song
|
|
if "/album/" in query and "?i=" in query:
|
|
song_id = query.split("/album/")[1].split("?i=")[1]
|
|
song_url = f"https://api.music.apple.com/v1/catalog/us/songs/{song_id}"
|
|
response = requests.get(song_url, headers=self.bot.apple_headers)
|
|
if response.status_code == 200:
|
|
song = response.json()
|
|
|
|
embed.title = "Song Queued"
|
|
embed.description = f"**{song['data'][0]['attributes']['name']}** by **{song['data'][0]['attributes']['artistName']}**\n\nQueued by: {interaction.user.mention}"
|
|
embed.set_thumbnail(
|
|
url=song["data"][0]["attributes"]["artwork"]["url"].replace(
|
|
"{w}x{h}", "300x300"
|
|
)
|
|
)
|
|
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 AppleSource.load_item(self, interaction.user, song)
|
|
player.add(requester=interaction.user, track=results.tracks[0])
|
|
|
|
###
|
|
### SPOTIFY links, perform API requests and load all tracks from the playlist/album/track
|
|
###
|
|
|
|
elif "open.spotify.com" in query:
|
|
if not self.bot.spotify_headers:
|
|
embed = discord.Embed(
|
|
title="Spotify Error",
|
|
description="Spotify support seems to be broken at the moment. Please try again and fill out a bug report with </bug:1224840889906499626> if this continues to happen.",
|
|
color=BOT_COLOR,
|
|
)
|
|
return await interaction.response.send_message(embed=embed, ephemeral=True)
|
|
|
|
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 SpotifySource.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 SpotifySource.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 SpotifySource.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, use default Lavalink providers to search the query
|
|
###
|
|
|
|
else:
|
|
if not url_rx.match(query):
|
|
dzsearch = f"dzsearch:{query}"
|
|
results = await player.node.get_tracks(dzsearch)
|
|
|
|
if not results.tracks or results.load_type in (
|
|
LoadType.EMPTY,
|
|
LoadType.ERROR,
|
|
):
|
|
ytmsearch = f"ytmsearch:{query}"
|
|
results = await player.node.get_tracks(ytmsearch)
|
|
|
|
if not results.tracks or results.load_type in (
|
|
LoadType.EMPTY,
|
|
LoadType.ERROR,
|
|
):
|
|
ytsearch = f"ytsearch:{query}"
|
|
results = await player.node.get_tracks(ytsearch)
|
|
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:1224840889906499626> 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]
|
|
player.add(requester=interaction.user, track=track)
|
|
|
|
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"
|
|
)
|
|
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))
|