import asyncio import contextlib import datetime import heapq import json import logging import tarfile import math import random import re import os.path import time import traceback from collections import Counter, namedtuple from io import BytesIO from pathlib import Path from typing import List, Optional, Tuple, Union, cast, MutableMapping, Mapping import aiohttp import discord import lavalink from discord.embeds import EmptyEmbed from discord.utils import escape_markdown as escape from fuzzywuzzy import process from redbot.core import Config, bank, checks, commands from redbot.core.bot import Red from redbot.core.data_manager import cog_data_path from redbot.core.i18n import Translator, cog_i18n from redbot.core.utils.chat_formatting import bold, box, humanize_number, inline, pagify from redbot.core.utils.menus import ( DEFAULT_CONTROLS, close_menu, menu, next_page, prev_page, start_adding_reactions, ) from redbot.core.utils.predicates import MessagePredicate, ReactionPredicate from . import audio_dataclasses from .apis import MusicCache from .config import pass_config_to_dependencies from .converters import ComplexScopeParser, ScopeParser, get_lazy_converter, get_playlist_converter from .equalizer import Equalizer from .errors import ( DatabaseError, LavalinkDownloadFailed, MissingGuild, QueryUnauthorized, SpotifyFetchError, TooManyMatches, TrackEnqueueError, ) from .manager import ServerManager from .playlists import ( FakePlaylist, Playlist, create_playlist, delete_playlist, get_all_playlist, get_all_playlist_for_migration23, get_playlist, get_playlist_database, ) from .utils import * _ = Translator("Audio", __file__) __version__ = "1.1.1" __author__ = ["aikaterna", "Draper"] log = logging.getLogger("red.audio") _SCHEMA_VERSION = 3 LazyGreedyConverter = get_lazy_converter("--") PlaylistConverter = get_playlist_converter() @cog_i18n(_) class Audio(commands.Cog): """Play audio through voice channels.""" _default_lavalink_settings = { "host": "localhost", "rest_port": 2333, "ws_port": 2333, "password": "youshallnotpass", } def __init__(self, bot): super().__init__() self.bot: Red = bot self.config: Config = Config.get_conf(self, 2711759130, force_registration=True) self.skip_votes: MutableMapping[discord.Guild, List[discord.Member]] = {} self.play_lock: MutableMapping[int, bool] = {} self._daily_playlist_cache: MutableMapping[int, bool] = {} self._dj_status_cache: MutableMapping[int, Optional[bool]] = {} self._dj_role_cache: MutableMapping[int, Optional[int]] = {} self.session: aiohttp.ClientSession = aiohttp.ClientSession() self._connect_task: Optional[asyncio.Task] = None self._disconnect_task: Optional[asyncio.Task] = None self._cleaned_up: bool = False self._connection_aborted: bool = False self._manager: Optional[ServerManager] = None default_global: Mapping = dict( schema_version=1, cache_level=0, cache_age=365, status=False, use_external_lavalink=False, restrict=True, localpath=str(cog_data_path(raw_name="Audio")), url_keyword_blacklist=[], url_keyword_whitelist=[], **self._default_lavalink_settings, ) default_guild: Mapping = dict( auto_play=False, autoplaylist=dict(enabled=False, id=None, name=None, scope=None), disconnect=False, dj_enabled=False, dj_role=None, daily_playlists=False, emptydc_enabled=False, emptydc_timer=0, emptypause_enabled=False, emptypause_timer=0, jukebox=False, jukebox_price=0, maxlength=0, notify=False, repeat=False, shuffle=False, shuffle_bumped=True, thumbnail=False, volume=100, vote_enabled=False, vote_percent=0, room_lock=None, url_keyword_blacklist=[], url_keyword_whitelist=[], ) _playlist: Mapping = dict(id=None, author=None, name=None, playlist_url=None, tracks=[]) self.config.init_custom("EQUALIZER", 1) self.config.register_custom("EQUALIZER", eq_bands=[], eq_presets={}) self.config.init_custom(PlaylistScope.GLOBAL.value, 1) self.config.register_custom(PlaylistScope.GLOBAL.value, **_playlist) self.config.init_custom(PlaylistScope.GUILD.value, 2) self.config.register_custom(PlaylistScope.GUILD.value, **_playlist) self.config.init_custom(PlaylistScope.USER.value, 2) self.config.register_custom(PlaylistScope.USER.value, **_playlist) self.config.register_guild(**default_guild) self.config.register_global(**default_global) self.music_cache: Optional[MusicCache] = None self._error_counter: Counter = Counter() self._error_timer: MutableMapping[int, int] = {} self._disconnected_players: MutableMapping[int, bool] = {} # These has to be a task since this requires the bot to be ready # If it waits for ready in startup, we cause a deadlock during initial load # as initial load happens before the bot can ever be ready. self._init_task: asyncio.Task = self.bot.loop.create_task(self.initialize()) self._ready_event: asyncio.Event = asyncio.Event() async def cog_before_invoke(self, ctx: commands.Context): await self._ready_event.wait() # check for unsupported arch # Check on this needs refactoring at a later date # so that we have a better way to handle the tasks if self.llsetup in [ctx.command, ctx.command.root_parent]: pass elif self._connect_task and self._connect_task.cancelled(): await ctx.send( _( "You have attempted to run Audio's Lavalink server on an unsupported" " architecture. Only settings related commands will be available." ) ) raise RuntimeError( "Not running audio command due to invalid machine architecture for Lavalink." ) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) daily_cache = self._daily_playlist_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).daily_playlists() ) if dj_enabled: dj_role = self._dj_role_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_role() ) dj_role_obj = ctx.guild.get_role(dj_role) if not dj_role_obj: await self.config.guild(ctx.guild).dj_enabled.set(None) self._dj_status_cache[ctx.guild.id] = None await self.config.guild(ctx.guild).dj_role.set(None) self._dj_role_cache[ctx.guild.id] = None await self._embed_msg(ctx, title=_("No DJ role found. Disabling DJ mode.")) async def initialize(self) -> None: await self.bot.wait_until_ready() # Unlike most cases, we want the cache to exit before migration. try: pass_config_to_dependencies(self.config, self.bot, await self.config.localpath()) self.music_cache = MusicCache(self.bot, self.session) await self.music_cache.initialize(self.config) await self._migrate_config( from_version=await self.config.schema_version(), to_version=_SCHEMA_VERSION ) dat = get_playlist_database() if dat: dat.delete_scheduled() self._restart_connect() self._disconnect_task = self.bot.loop.create_task(self.disconnect_timer()) lavalink.register_event_listener(self.event_handler) except Exception as err: log.exception("Audio failed to start up, please report this issue.", exc_info=err) raise err self._ready_event.set() async def _migrate_config(self, from_version: int, to_version: int) -> None: database_entries = [] time_now = int(datetime.datetime.now(datetime.timezone.utc).timestamp()) if from_version == to_version: return if from_version < 2 <= to_version: all_guild_data = await self.config.all_guilds() all_playlist = {} for (guild_id, guild_data) in all_guild_data.items(): temp_guild_playlist = guild_data.pop("playlists", None) if temp_guild_playlist: guild_playlist = {} for (count, (name, data)) in enumerate(temp_guild_playlist.items(), 1): if not data or not name: continue playlist = {"id": count, "name": name, "guild": int(guild_id)} playlist.update(data) guild_playlist[str(count)] = playlist tracks_in_playlist = data.get("tracks", []) or [] for t in tracks_in_playlist: uri = t.get("info", {}).get("uri") if uri: t = {"loadType": "V2_COMPAT", "tracks": [t], "query": uri} data = json.dumps(t) if all( k in data for k in ["loadType", "playlistInfo", "isSeekable", "isStream"] ): database_entries.append( { "query": uri, "data": data, "last_updated": time_now, "last_fetched": time_now, } ) await asyncio.sleep(0) if guild_playlist: all_playlist[str(guild_id)] = guild_playlist await asyncio.sleep(0) await self.config.custom(PlaylistScope.GUILD.value).set(all_playlist) # new schema is now in place await self.config.schema_version.set(_SCHEMA_VERSION) # migration done, now let's delete all the old stuff for guild_id in all_guild_data: await self.config.guild( cast(discord.Guild, discord.Object(id=guild_id)) ).clear_raw("playlists") if from_version < 3 <= to_version: for scope in PlaylistScope.list(): scope_playlist = await get_all_playlist_for_migration23(scope) for p in scope_playlist: await p.save() await self.config.custom(scope).clear() await self.config.schema_version.set(_SCHEMA_VERSION) if database_entries: await self.music_cache.database.insert("lavalink", database_entries) def _restart_connect(self): if self._connect_task: self._connect_task.cancel() self._connect_task = self.bot.loop.create_task(self.attempt_connect()) async def attempt_connect(self, timeout: int = 50): self._connection_aborted = False max_retries = 5 retry_count = 0 while retry_count < max_retries: external = await self.config.use_external_lavalink() if external is False: settings = self._default_lavalink_settings host = settings["host"] password = settings["password"] rest_port = settings["rest_port"] ws_port = settings["ws_port"] if self._manager is not None: await self._manager.shutdown() self._manager = ServerManager() try: await self._manager.start() except LavalinkDownloadFailed as exc: await asyncio.sleep(1) if exc.should_retry: log.exception( "Exception whilst starting internal Lavalink server, retrying...", exc_info=exc, ) retry_count += 1 continue else: log.exception( "Fatal exception whilst starting internal Lavalink server, " "aborting...", exc_info=exc, ) self._connection_aborted = True raise except asyncio.CancelledError: log.exception("Invalid machine architecture, cannot run Lavalink.") raise except Exception as exc: log.exception( "Unhandled exception whilst starting internal Lavalink server, " "aborting...", exc_info=exc, ) self._connection_aborted = True raise else: break else: host = await self.config.host() password = await self.config.password() rest_port = await self.config.rest_port() ws_port = await self.config.ws_port() break else: log.critical( "Setting up the Lavalink server failed after multiple attempts. See above " "tracebacks for details." ) self._connection_aborted = True return retry_count = 0 while retry_count < max_retries: try: await lavalink.initialize( bot=self.bot, host=host, password=password, rest_port=rest_port, ws_port=ws_port, timeout=timeout, ) except asyncio.TimeoutError: log.error("Connecting to Lavalink server timed out, retrying...") if external is False and self._manager is not None: await self._manager.shutdown() retry_count += 1 await asyncio.sleep(1) # prevent busylooping except Exception as exc: log.exception( "Unhandled exception whilst connecting to Lavalink, aborting...", exc_info=exc ) self._connection_aborted = True raise else: break else: self._connection_aborted = True log.critical( "Connecting to the Lavalink server failed after multiple attempts. See above " "tracebacks for details." ) async def error_reset(self, player: lavalink.Player): guild = rgetattr(player, "channel.guild.id", None) if not guild: return now = time.time() seconds_allowed = 10 last_error = self._error_timer.setdefault(guild, now) if now - seconds_allowed > last_error: self._error_timer[guild] = 0 self._error_counter[guild] = 0 async def increase_error_counter(self, player: lavalink.Player) -> bool: guild = rgetattr(player, "channel.guild.id", None) if not guild: return False now = time.time() self._error_counter[guild] += 1 self._error_timer[guild] = now return self._error_counter[guild] >= 5 @staticmethod async def _players_check(): try: current = next( ( player.current for player in lavalink.active_players() if player.current is not None ), None, ) get_single_title = get_track_description_unformatted(current) playing_servers = len(lavalink.active_players()) except IndexError: get_single_title = None playing_servers = 0 return get_single_title, playing_servers async def _status_check(self, track, playing_servers): if playing_servers == 0: await self.bot.change_presence(activity=None) elif playing_servers == 1: await self.bot.change_presence( activity=discord.Activity(name=track, type=discord.ActivityType.listening) ) elif playing_servers > 1: await self.bot.change_presence( activity=discord.Activity( name=_("music in {} servers").format(playing_servers), type=discord.ActivityType.playing, ) ) async def event_handler( self, player: lavalink.Player, event_type: lavalink.LavalinkEvents, extra ): current_track = player.current current_channel = player.channel guild = rgetattr(current_channel, "guild", None) guild_id = rgetattr(guild, "id", None) current_requester = rgetattr(current_track, "requester", None) current_stream = rgetattr(current_track, "is_stream", None) current_length = rgetattr(current_track, "length", None) current_thumbnail = rgetattr(current_track, "thumbnail", None) current_extras = rgetattr(current_track, "extras", {}) guild_data = await self.config.guild(guild).all() repeat = guild_data["repeat"] notify = guild_data["notify"] disconnect = guild_data["disconnect"] autoplay = guild_data["auto_play"] description = get_track_description(current_track) status = await self.config.status() await self.error_reset(player) if event_type == lavalink.LavalinkEvents.TRACK_START: self.skip_votes[guild] = [] playing_song = player.fetch("playing_song") requester = player.fetch("requester") player.store("prev_song", playing_song) player.store("prev_requester", requester) player.store("playing_song", current_track) player.store("requester", current_requester) self.bot.dispatch("red_audio_track_start", guild, current_track, current_requester) if event_type == lavalink.LavalinkEvents.TRACK_END: prev_song = player.fetch("prev_song") prev_requester = player.fetch("prev_requester") self.bot.dispatch("red_audio_track_end", guild, prev_song, prev_requester) if event_type == lavalink.LavalinkEvents.QUEUE_END: prev_song = player.fetch("prev_song") prev_requester = player.fetch("prev_requester") self.bot.dispatch("red_audio_queue_end", guild, prev_song, prev_requester) if autoplay and not player.queue and player.fetch("playing_song") is not None: try: await self.music_cache.autoplay(player) except DatabaseError: notify_channel = player.fetch("channel") if notify_channel: notify_channel = self.bot.get_channel(notify_channel) await self._embed_msg( notify_channel, title=_("Couldn't get a valid track.") ) return if event_type == lavalink.LavalinkEvents.TRACK_START and notify: notify_channel = player.fetch("channel") prev_song = player.fetch("prev_song") if notify_channel: notify_channel = self.bot.get_channel(notify_channel) if player.fetch("notify_message") is not None: with contextlib.suppress(discord.HTTPException): await player.fetch("notify_message").delete() if ( autoplay and current_extras.get("autoplay") and ( prev_song is None or (hasattr(prev_song, "extras") and not prev_song.extras.get("autoplay")) ) ): await self._embed_msg(notify_channel, title=_("Auto Play Started.")) if not description: return if current_stream: dur = "LIVE" else: dur = lavalink.utils.format_time(current_length) thumb = None if await self.config.guild(guild).thumbnail() and current_thumbnail: thumb = current_thumbnail notify_message = await self._embed_msg( notify_channel, title=_("Now Playing"), description=description, footer=_("Track length: {length} | Requested by: {user}").format( length=dur, user=current_requester ), thumbnail=thumb, ) player.store("notify_message", notify_message) if event_type == lavalink.LavalinkEvents.TRACK_START and status: player_check = await self._players_check() await self._status_check(*player_check) if event_type == lavalink.LavalinkEvents.TRACK_END and status: await asyncio.sleep(1) if not player.is_playing: player_check = await self._players_check() await self._status_check(*player_check) if event_type == lavalink.LavalinkEvents.QUEUE_END: if not autoplay: notify_channel = player.fetch("channel") if notify_channel and notify: notify_channel = self.bot.get_channel(notify_channel) await self._embed_msg(notify_channel, title=_("Queue Ended.")) if disconnect: self.bot.dispatch("red_audio_audio_disconnect", guild) await player.disconnect() if status: player_check = await self._players_check() await self._status_check(*player_check) if event_type in [ lavalink.LavalinkEvents.TRACK_EXCEPTION, lavalink.LavalinkEvents.TRACK_STUCK, ]: message_channel = player.fetch("channel") while True: if current_track in player.queue: player.queue.remove(current_track) else: break if repeat: player.current = None if not guild_id: return self._error_counter.setdefault(guild_id, 0) if guild_id not in self._error_counter: self._error_counter[guild_id] = 0 early_exit = await self.increase_error_counter(player) if early_exit: self._disconnected_players[guild_id] = True self.play_lock[guild_id] = False eq = player.fetch("eq") player.queue = [] player.store("playing_song", None) if eq: await self.config.custom("EQUALIZER", guild_id).eq_bands.set(eq.bands) await player.stop() await player.disconnect() self.bot.dispatch("red_audio_audio_disconnect", guild) if message_channel: message_channel = self.bot.get_channel(message_channel) if early_exit: embed = discord.Embed( colour=(await self.bot.get_embed_color(message_channel)), title=_("Multiple errors detected"), description=_( "Closing the audio player " "due to multiple errors being detected. " "If this persists, please inform the bot owner " "as the Audio cog may be temporally unavailable." ), ) return await message_channel.send(embed=embed) else: description = description or "" if event_type == lavalink.LavalinkEvents.TRACK_STUCK: embed = discord.Embed( title=_("Track Stuck"), description="{}".format(description) ) else: embed = discord.Embed( title=_("Track Error"), description="{}\n{}".format(extra.replace("\n", ""), description), ) await message_channel.send(embed=embed) await player.skip() async def play_query( self, query: str, guild: discord.Guild, channel: discord.VoiceChannel, is_autoplay: bool = True, ): if not self._player_check(guild.me): try: if ( not channel.permissions_for(guild.me).connect or not channel.permissions_for(guild.me).move_members and userlimit(channel) ): log.error(f"I don't have permission to connect to {channel} in {guild}.") await lavalink.connect(channel) player = lavalink.get_player(guild.id) player.store("connect", datetime.datetime.utcnow()) except IndexError: log.debug( f"Connection to Lavalink has not yet been established" f" while trying to connect to to {channel} in {guild}." ) return query = audio_dataclasses.Query.process_input(query) restrict = await self.config.restrict() if restrict and match_url(query): valid_url = url_check(query) if not valid_url: raise QueryUnauthorized(f"{query} is not an allowed query.") elif not await is_allowed(guild, f"{query}", query_obj=query): raise QueryUnauthorized(f"{query} is not an allowed query.") player = lavalink.get_player(guild.id) player.store("channel", channel.id) player.store("guild", guild.id) await self._data_check(guild.me) ctx = namedtuple("Context", "message") (results, called_api) = await self.music_cache.lavalink_query(ctx(guild), player, query) if not results.tracks: log.debug(f"Query returned no tracks.") return track = results.tracks[0] if not await is_allowed( guild, f"{track.title} {track.author} {track.uri} {str(query._raw)}" ): log.debug(f"Query is not allowed in {guild} ({guild.id})") return track.extras["autoplay"] = is_autoplay player.add(player.channel.guild.me, track) self.bot.dispatch( "red_audio_track_auto_play", player.channel.guild, track, player.channel.guild.me ) if not player.current: await player.play() @commands.group() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def audioset(self, ctx: commands.Context): """Music configuration options.""" @audioset.command(name="dailyqueue") @checks.admin() async def _audioset_historical_queue(self, ctx: commands.Context): """Toggle daily queues. Daily queues creates a playlist for all tracks played today. """ daily_playlists = self._daily_playlist_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).daily_playlists() ) await self.config.guild(ctx.guild).daily_playlists.set(not daily_playlists) self._daily_playlist_cache[ctx.guild.id] = not daily_playlists await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Daily queues: {true_or_false}.").format( true_or_false=_("Enabled") if not daily_playlists else _("Disabled") ), ) @audioset.command() @checks.mod_or_permissions(manage_messages=True) async def dc(self, ctx: commands.Context): """Toggle the bot auto-disconnecting when done playing. This setting takes precedence over `[p]audioset emptydisconnect`. """ disconnect = await self.config.guild(ctx.guild).disconnect() autoplay = await self.config.guild(ctx.guild).auto_play() msg = "" msg += _("Auto-disconnection at queue end: {true_or_false}.").format( true_or_false=_("Enabled") if not disconnect else _("Disabled") ) if disconnect is not True and autoplay is True: msg += _("\nAuto-play has been disabled.") await self.config.guild(ctx.guild).auto_play.set(False) await self.config.guild(ctx.guild).disconnect.set(not disconnect) await self._embed_msg(ctx, title=_("Setting Changed"), description=msg) @audioset.group(name="restrictions") @checks.mod_or_permissions(manage_messages=True) async def _perms(self, ctx: commands.Context): """Manages the keyword whitelist and blacklist.""" @checks.is_owner() @_perms.group(name="global") async def _perms_global(self, ctx: commands.Context): """Manages the global keyword whitelist/blacklist.""" @_perms_global.group(name="whitelist") async def _perms_global_whitelist(self, ctx: commands.Context): """Manages the global keyword whitelist.""" @_perms_global.group(name="blacklist") async def _perms_global_blacklist(self, ctx: commands.Context): """Manages the global keyword blacklist.""" @_perms_global_blacklist.command(name="add") async def _perms_global_blacklist_add(self, ctx: commands.Context, *, keyword: str): """Adds a keyword to the blacklist.""" keyword = keyword.lower().strip() if not keyword: return await ctx.send_help() exists = False async with self.config.url_keyword_blacklist() as blacklist: if keyword in blacklist: exists = True else: blacklist.append(keyword) if exists: return await self._embed_msg(ctx, title=_("Keyword already in the blacklist.")) else: return await self._embed_msg( ctx, title=_("Blacklist Modified"), description=_("Added: `{blacklisted}` to the blacklist.").format( blacklisted=keyword ), ) @_perms_global_whitelist.command(name="add") async def _perms_global_whitelist_add(self, ctx: commands.Context, *, keyword: str): """Adds a keyword to the whitelist. If anything is added to whitelist, it will blacklist everything else. """ keyword = keyword.lower().strip() if not keyword: return await ctx.send_help() exists = False async with self.config.url_keyword_whitelist() as whitelist: if keyword in whitelist: exists = True else: whitelist.append(keyword) if exists: return await self._embed_msg(ctx, title=_("Keyword already in the whitelist.")) else: return await self._embed_msg( ctx, title=_("Whitelist Modified"), description=_("Added: `{whitelisted}` to the whitelist.").format( whitelisted=keyword ), ) @_perms_global_blacklist.command(name="delete", aliases=["del", "remove"]) async def _perms_global_blacklist_delete(self, ctx: commands.Context, *, keyword: str): """Removes a keyword from the blacklist.""" keyword = keyword.lower().strip() if not keyword: return await ctx.send_help() exists = True async with self.config.url_keyword_blacklist() as blacklist: if keyword not in blacklist: exists = False else: blacklist.remove(keyword) if not exists: return await self._embed_msg(ctx, title=_("Keyword is not in the blacklist.")) else: return await self._embed_msg( ctx, title=_("Blacklist Modified"), description=_("Removed: `{blacklisted}` from the blacklist.").format( blacklisted=keyword ), ) @_perms_global_whitelist.command(name="delete", aliases=["del", "remove"]) async def _perms_global_whitelist_delete(self, ctx: commands.Context, *, keyword: str): """Removes a keyword from the whitelist.""" keyword = keyword.lower().strip() if not keyword: return await ctx.send_help() exists = True async with self.config.url_keyword_whitelist() as whitelist: if keyword not in whitelist: exists = False else: whitelist.remove(keyword) if not exists: return await self._embed_msg(ctx, title=_("Keyword already in the whitelist.")) else: return await self._embed_msg( ctx, title=_("Whitelist Modified"), description=_("Removed: `{whitelisted}` from the whitelist.").format( whitelisted=keyword ), ) @_perms_global_whitelist.command(name="list") async def _perms_global_whitelist_list(self, ctx: commands.Context): """List all keywords added to the whitelist.""" whitelist = await self.config.url_keyword_whitelist() if not whitelist: return await self._embed_msg(ctx, title=_("Nothing in the whitelist.")) whitelist.sort() text = "" total = len(whitelist) pages = [] for i, entry in enumerate(whitelist, 1): text += f"{i}. [{entry}]" if i != total: text += "\n" if i % 10 == 0: pages.append(box(text, lang="ini")) text = "" else: pages.append(box(text, lang="ini")) embed_colour = await ctx.embed_colour() pages = list( discord.Embed(title="Global Whitelist", description=page, colour=embed_colour) for page in pages ) await menu(ctx, pages, DEFAULT_CONTROLS) @_perms_global_blacklist.command(name="list") async def _perms_global_blacklist_list(self, ctx: commands.Context): """List all keywords added to the blacklist.""" blacklist = await self.config.url_keyword_blacklist() if not blacklist: return await self._embed_msg(ctx, title=_("Nothing in the blacklist.")) blacklist.sort() text = "" total = len(blacklist) pages = [] for i, entry in enumerate(blacklist, 1): text += f"{i}. [{entry}]" if i != total: text += "\n" if i % 10 == 0: pages.append(box(text, lang="ini")) text = "" else: pages.append(box(text, lang="ini")) embed_colour = await ctx.embed_colour() pages = list( discord.Embed(title="Global Blacklist", description=page, colour=embed_colour) for page in pages ) await menu(ctx, pages, DEFAULT_CONTROLS) @_perms_global_whitelist.command(name="clear") async def _perms_global_whitelist_clear(self, ctx: commands.Context): """Clear all keywords from the whitelist.""" whitelist = await self.config.url_keyword_whitelist() if not whitelist: return await self._embed_msg(ctx, title=_("Nothing in the whitelist.")) await self.config.url_keyword_whitelist.clear() return await self._embed_msg( ctx, title=_("Whitelist Modified"), description=_("All entries have been removed from the whitelist."), ) @_perms_global_blacklist.command(name="clear") async def _perms_global_blacklist_clear(self, ctx: commands.Context): """Clear all keywords added to the blacklist.""" blacklist = await self.config.url_keyword_blacklist() if not blacklist: return await self._embed_msg(ctx, title=_("Nothing in the blacklist.")) await self.config.url_keyword_blacklist.clear() return await self._embed_msg( ctx, title=_("Blacklist Modified"), description=_("All entries have been removed from the blacklist."), ) @_perms.group(name="whitelist") async def _perms_whitelist(self, ctx: commands.Context): """Manages the keyword whitelist.""" @_perms.group(name="blacklist") async def _perms_blacklist(self, ctx: commands.Context): """Manages the keyword blacklist.""" @_perms_blacklist.command(name="add") async def _perms_blacklist_add(self, ctx: commands.Context, *, keyword: str): """Adds a keyword to the blacklist.""" keyword = keyword.lower().strip() if not keyword: return await ctx.send_help() exists = False async with self.config.guild(ctx.guild).url_keyword_blacklist() as blacklist: if keyword in blacklist: exists = True else: blacklist.append(keyword) if exists: return await self._embed_msg(ctx, title=_("Keyword already in the blacklist.")) else: return await self._embed_msg( ctx, title=_("Blacklist Modified"), description=_("Added: `{blacklisted}` to the blacklist.").format( blacklisted=keyword ), ) @_perms_whitelist.command(name="add") async def _perms_whitelist_add(self, ctx: commands.Context, *, keyword: str): """Adds a keyword to the whitelist. If anything is added to whitelist, it will blacklist everything else. """ keyword = keyword.lower().strip() if not keyword: return await ctx.send_help() exists = False async with self.config.guild(ctx.guild).url_keyword_whitelist() as whitelist: if keyword in whitelist: exists = True else: whitelist.append(keyword) if exists: return await self._embed_msg(ctx, title=_("Keyword already in the whitelist.")) else: return await self._embed_msg( ctx, title=_("Whitelist Modified"), description=_("Added: `{whitelisted}` to the whitelist.").format( whitelisted=keyword ), ) @_perms_blacklist.command(name="delete", aliases=["del", "remove"]) async def _perms_blacklist_delete(self, ctx: commands.Context, *, keyword: str): """Removes a keyword from the blacklist.""" keyword = keyword.lower().strip() if not keyword: return await ctx.send_help() exists = True async with self.config.guild(ctx.guild).url_keyword_blacklist() as blacklist: if keyword not in blacklist: exists = False else: blacklist.remove(keyword) if not exists: return await self._embed_msg(ctx, title=_("Keyword is not in the blacklist.")) else: return await self._embed_msg( ctx, title=_("Blacklist Modified"), description=_("Removed: `{blacklisted}` from the blacklist.").format( blacklisted=keyword ), ) @_perms_whitelist.command(name="delete", aliases=["del", "remove"]) async def _perms_whitelist_delete(self, ctx: commands.Context, *, keyword: str): """Removes a keyword from the whitelist.""" keyword = keyword.lower().strip() if not keyword: return await ctx.send_help() exists = True async with self.config.guild(ctx.guild).url_keyword_whitelist() as whitelist: if keyword not in whitelist: exists = False else: whitelist.remove(keyword) if not exists: return await self._embed_msg(ctx, title=_("Keyword already in the whitelist.")) else: return await self._embed_msg( ctx, title=_("Whitelist Modified"), description=_("Removed: `{whitelisted}` from the whitelist.").format( whitelisted=keyword ), ) @_perms_whitelist.command(name="list") async def _perms_whitelist_list(self, ctx: commands.Context): """List all keywords added to the whitelist.""" whitelist = await self.config.guild(ctx.guild).url_keyword_whitelist() if not whitelist: return await self._embed_msg(ctx, title=_("Nothing in the whitelist.")) whitelist.sort() text = "" total = len(whitelist) pages = [] for i, entry in enumerate(whitelist, 1): text += f"{i}. [{entry}]" if i != total: text += "\n" if i % 10 == 0: pages.append(box(text, lang="ini")) text = "" else: pages.append(box(text, lang="ini")) embed_colour = await ctx.embed_colour() pages = list( discord.Embed(title="Whitelist", description=page, colour=embed_colour) for page in pages ) await menu(ctx, pages, DEFAULT_CONTROLS) @_perms_blacklist.command(name="list") async def _perms_blacklist_list(self, ctx: commands.Context): """List all keywords added to the blacklist.""" blacklist = await self.config.guild(ctx.guild).url_keyword_blacklist() if not blacklist: return await self._embed_msg(ctx, title=_("Nothing in the blacklist.")) blacklist.sort() text = "" total = len(blacklist) pages = [] for i, entry in enumerate(blacklist, 1): text += f"{i}. [{entry}]" if i != total: text += "\n" if i % 10 == 0: pages.append(box(text, lang="ini")) text = "" else: pages.append(box(text, lang="ini")) embed_colour = await ctx.embed_colour() pages = list( discord.Embed(title="Blacklist", description=page, colour=embed_colour) for page in pages ) await menu(ctx, pages, DEFAULT_CONTROLS) @_perms_whitelist.command(name="clear") async def _perms_whitelist_clear(self, ctx: commands.Context): """Clear all keywords from the whitelist.""" whitelist = await self.config.guild(ctx.guild).url_keyword_whitelist() if not whitelist: return await self._embed_msg(ctx, title=_("Nothing in the whitelist.")) await self.config.guild(ctx.guild).url_keyword_whitelist.clear() return await self._embed_msg( ctx, title=_("Whitelist Modified"), description=_("All entries have been removed from the whitelist."), ) @_perms_blacklist.command(name="clear") async def _perms_blacklist_clear(self, ctx: commands.Context): """Clear all keywords added to the blacklist.""" blacklist = await self.config.guild(ctx.guild).url_keyword_blacklist() if not blacklist: return await self._embed_msg(ctx, title=_("Nothing in the blacklist.")) await self.config.guild(ctx.guild).url_keyword_blacklist.clear() return await self._embed_msg( ctx, title=_("Blacklist Modified"), description=_("All entries have been removed from the blacklist."), ) @audioset.group(name="autoplay") @checks.mod_or_permissions(manage_messages=True) async def _autoplay(self, ctx: commands.Context): """Change auto-play setting.""" @_autoplay.command(name="toggle") async def _autoplay_toggle(self, ctx: commands.Context): """Toggle auto-play when there no songs in queue.""" autoplay = await self.config.guild(ctx.guild).auto_play() repeat = await self.config.guild(ctx.guild).repeat() disconnect = await self.config.guild(ctx.guild).disconnect() msg = _("Auto-play when queue ends: {true_or_false}.").format( true_or_false=_("Enabled") if not autoplay else _("Disabled") ) await self.config.guild(ctx.guild).auto_play.set(not autoplay) if autoplay is not True and repeat is True: msg += _("\nRepeat has been disabled.") await self.config.guild(ctx.guild).repeat.set(False) if autoplay is not True and disconnect is True: msg += _("\nAuto-disconnecting at queue end has been disabled.") await self.config.guild(ctx.guild).disconnect.set(False) await self._embed_msg(ctx, title=_("Setting Changed"), description=msg) if self._player_check(ctx): await self._data_check(ctx) @_autoplay.command(name="playlist", usage=" [args]") async def _autoplay_playlist( self, ctx: commands.Context, playlist_matches: PlaylistConverter, *, scope_data: ScopeParser = None, ): """Set a playlist to auto-play songs from. **Usage**: ​ ​ ​ ​ `[p]audioset autoplay playlist_name_OR_id [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]audioset autoplay MyGuildPlaylist` ​ ​ ​ ​ `[p]audioset autoplay MyGlobalPlaylist --scope Global` ​ ​ ​ ​ `[p]audioset autoplay PersonalPlaylist --scope User --author Draper` """ if scope_data is None: scope_data = [None, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data try: playlist_id, playlist_arg, scope = await self._get_correct_playlist_id( ctx, playlist_matches, scope, author, guild, specified_user ) except TooManyMatches as e: return await self._embed_msg(ctx, title=str(e)) if playlist_id is None: return await self._embed_msg( ctx, title=_("No Playlist Found"), description=_("Could not match '{arg}' to a playlist").format(arg=playlist_arg), ) try: playlist = await get_playlist(playlist_id, scope, self.bot, guild, author) tracks = playlist.tracks if not tracks: return await self._embed_msg( ctx, title=_("No Tracks Found"), description=_("Playlist {name} has no tracks.").format(name=playlist.name), ) playlist_data = dict(enabled=True, id=playlist.id, name=playlist.name, scope=scope) await self.config.guild(ctx.guild).autoplaylist.set(playlist_data) except RuntimeError: return await self._embed_msg( ctx, title=_("No Playlist Found"), description=_("Playlist {id} does not exist in {scope} scope.").format( id=playlist_id, scope=humanize_scope(scope, the=True) ), ) except MissingGuild: return await self._embed_msg( ctx, title=_("Missing Arguments"), description=_("You need to specify the Guild ID for the guild to lookup."), ) else: return await self._embed_msg( ctx, title=_("Setting Changed"), description=_( "Playlist {name} (`{id}`) [**{scope}**] will be used for autoplay." ).format( name=playlist.name, id=playlist.id, scope=humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ), ), ) @_autoplay.command(name="reset") async def _autoplay_reset(self, ctx: commands.Context): """Resets auto-play to the default playlist.""" playlist_data = dict(enabled=False, id=None, name=None, scope=None) await self.config.guild(ctx.guild).autoplaylist.set(playlist_data) return await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Set auto-play playlist to default value."), ) @audioset.command() @checks.admin_or_permissions(manage_roles=True) async def dj(self, ctx: commands.Context): """Toggle DJ mode. DJ mode allows users with the DJ role to use audio commands. """ dj_role = self._dj_role_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_role() ) dj_role = ctx.guild.get_role(dj_role) if dj_role is None: await self._embed_msg( ctx, title=_("Missing DJ Role"), description=_( "Please set a role to use with DJ mode. Enter the role name or ID now." ), ) try: pred = MessagePredicate.valid_role(ctx) await ctx.bot.wait_for("message", timeout=15.0, check=pred) await ctx.invoke(self.role, role_name=pred.result) except asyncio.TimeoutError: return await self._embed_msg(ctx, title=_("Response timed out, try again later.")) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) await self.config.guild(ctx.guild).dj_enabled.set(not dj_enabled) self._dj_status_cache[ctx.guild.id] = not dj_enabled await self._embed_msg( ctx, title=_("Setting Changed"), description=_("DJ role: {true_or_false}.").format( true_or_false=_("Enabled") if not dj_enabled else _("Disabled") ), ) @audioset.command() @checks.mod_or_permissions(administrator=True) async def emptydisconnect(self, ctx: commands.Context, seconds: int): """Auto-disconnect from channel when bot is alone in it for x seconds, 0 to disable. `[p]audioset dc` takes precedence over this setting. """ if seconds < 0: return await self._embed_msg( ctx, title=_("Invalid Time"), description=_("Seconds can't be less than zero.") ) if 10 > seconds > 0: seconds = 10 if seconds == 0: enabled = False await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Empty disconnect disabled.") ) else: enabled = True await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Empty disconnect timer set to {num_seconds}.").format( num_seconds=dynamic_time(seconds) ), ) await self.config.guild(ctx.guild).emptydc_timer.set(seconds) await self.config.guild(ctx.guild).emptydc_enabled.set(enabled) @audioset.command() @checks.mod_or_permissions(administrator=True) async def emptypause(self, ctx: commands.Context, seconds: int): """Auto-pause after x seconds when room is empty, 0 to disable.""" if seconds < 0: return await self._embed_msg( ctx, title=_("Invalid Time"), description=_("Seconds can't be less than zero.") ) if 10 > seconds > 0: seconds = 10 if seconds == 0: enabled = False await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Empty pause disabled.") ) else: enabled = True await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Empty pause timer set to {num_seconds}.").format( num_seconds=dynamic_time(seconds) ), ) await self.config.guild(ctx.guild).emptypause_timer.set(seconds) await self.config.guild(ctx.guild).emptypause_enabled.set(enabled) @audioset.command() @checks.mod_or_permissions(administrator=True) async def jukebox(self, ctx: commands.Context, price: int): """Set a price for queueing tracks for non-mods, 0 to disable.""" if price < 0: return await self._embed_msg( ctx, title=_("Invalid Price"), description=_("Price can't be less than zero.") ) if price == 0: jukebox = False await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Jukebox mode disabled.") ) else: jukebox = True await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Track queueing command price set to {price} {currency}.").format( price=humanize_number(price), currency=await bank.get_currency_name(ctx.guild) ), ) await self.config.guild(ctx.guild).jukebox_price.set(price) await self.config.guild(ctx.guild).jukebox.set(jukebox) @audioset.command() @checks.is_owner() async def localpath(self, ctx: commands.Context, *, local_path=None): """Set the localtracks path if the Lavalink.jar is not run from the Audio data folder. Leave the path blank to reset the path to the default, the Audio data directory. """ if not local_path: await self.config.localpath.set(str(cog_data_path(raw_name="Audio"))) pass_config_to_dependencies( self.config, self.bot, str(cog_data_path(raw_name="Audio")) ) return await self._embed_msg( ctx, title=_("Setting Changed"), description=_( "The localtracks path location has been reset to {localpath}" ).format(localpath=str(cog_data_path(raw_name="Audio").absolute())), ) info_msg = _( "This setting is only for bot owners to set a localtracks folder location " "In the example below, the full path for 'ParentDirectory' " "must be passed to this command.\n" "The path must not contain spaces.\n" "```\n" "ParentDirectory\n" " |__ localtracks (folder)\n" " | |__ Awesome Album Name (folder)\n" " | |__01 Cool Song.mp3\n" " | |__02 Groovy Song.mp3\n" "```\n" "The folder path given to this command must contain the localtracks folder.\n" "**This folder and files need to be visible to the user where `" "Lavalink.jar` is being run from.**\n" "Use this command with no path given to reset it to the default, " "the Audio data directory for this bot.\n" "Do you want to continue to set the provided path for local tracks?" ) info = await ctx.maybe_send_embed(info_msg) start_adding_reactions(info, ReactionPredicate.YES_OR_NO_EMOJIS) pred = ReactionPredicate.yes_or_no(info, ctx.author) await ctx.bot.wait_for("reaction_add", check=pred) if not pred.result: with contextlib.suppress(discord.HTTPException): await info.delete() return temp = audio_dataclasses.LocalPath(local_path, forced=True) if not temp.exists() or not temp.is_dir(): return await self._embed_msg( ctx, title=_("Invalid Path"), description=_("{local_path} does not seem like a valid path.").format( local_path=local_path ), ) if not temp.localtrack_folder.exists(): warn_msg = _( "`{localtracks}` does not exist. " "The path will still be saved, but please check the path and " "create a localtracks folder in `{localfolder}` before attempting " "to play local tracks." ).format(localfolder=temp.absolute(), localtracks=temp.localtrack_folder.absolute()) await self._embed_msg(ctx, title=_("Invalid Environment"), description=warn_msg) local_path = str(temp.localtrack_folder.absolute()) await self.config.localpath.set(local_path) pass_config_to_dependencies(self.config, self.bot, local_path) return await self._embed_msg( ctx, title=_("Setting Changed"), description=_("The localtracks path location has been set to {localpath}").format( localpath=local_path ), ) @audioset.command() @checks.mod_or_permissions(administrator=True) async def maxlength(self, ctx: commands.Context, seconds: Union[int, str]): """Max length of a track to queue in seconds, 0 to disable. Accepts seconds or a value formatted like 00:00:00 (`hh:mm:ss`) or 00:00 (`mm:ss`). Invalid input will turn the max length setting off. """ if not isinstance(seconds, int): seconds = time_convert(seconds) if seconds < 0: return await self._embed_msg( ctx, title=_("Invalid length"), description=_("Length can't be less than zero.") ) if seconds == 0: await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Track max length disabled.") ) else: await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Track max length set to {seconds}.").format( seconds=dynamic_time(seconds) ), ) await self.config.guild(ctx.guild).maxlength.set(seconds) @audioset.command() @checks.mod_or_permissions(manage_messages=True) async def notify(self, ctx: commands.Context): """Toggle track announcement and other bot messages.""" notify = await self.config.guild(ctx.guild).notify() await self.config.guild(ctx.guild).notify.set(not notify) await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Notify mode: {true_or_false}.").format( true_or_false=_("Enabled") if not notify else _("Disabled") ), ) @audioset.command() @checks.is_owner() async def restrict(self, ctx: commands.Context): """Toggle the domain restriction on Audio. When toggled off, users will be able to play songs from non-commercial websites and links. When toggled on, users are restricted to YouTube, SoundCloud, Mixer, Vimeo, Twitch, and Bandcamp links. """ restrict = await self.config.restrict() await self.config.restrict.set(not restrict) await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Commercial links only: {true_or_false}.").format( true_or_false=_("Enabled") if not restrict else _("Disabled") ), ) @audioset.command() @checks.admin_or_permissions(manage_roles=True) async def role(self, ctx: commands.Context, *, role_name: discord.Role): """Set the role to use for DJ mode.""" await self.config.guild(ctx.guild).dj_role.set(role_name.id) self._dj_role_cache[ctx.guild.id] = role_name.id dj_role = self._dj_role_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_role() ) dj_role_obj = ctx.guild.get_role(dj_role) await self._embed_msg( ctx, title=_("Settings Changed"), description=_("DJ role set to: {role.name}.").format(role=dj_role_obj), ) @audioset.command() async def settings(self, ctx: commands.Context): """Show the current settings.""" is_owner = await ctx.bot.is_owner(ctx.author) global_data = await self.config.all() data = await self.config.guild(ctx.guild).all() dj_role_obj = ctx.guild.get_role(data["dj_role"]) dj_enabled = data["dj_enabled"] emptydc_enabled = data["emptydc_enabled"] emptydc_timer = data["emptydc_timer"] emptypause_enabled = data["emptypause_enabled"] emptypause_timer = data["emptypause_timer"] jukebox = data["jukebox"] jukebox_price = data["jukebox_price"] thumbnail = data["thumbnail"] dc = data["disconnect"] autoplay = data["auto_play"] maxlength = data["maxlength"] vote_percent = data["vote_percent"] current_level = CacheLevel(global_data["cache_level"]) song_repeat = _("Enabled") if data["repeat"] else _("Disabled") song_shuffle = _("Enabled") if data["shuffle"] else _("Disabled") bumpped_shuffle = _("Enabled") if data["shuffle_bumped"] else _("Disabled") song_notify = _("Enabled") if data["notify"] else _("Disabled") song_status = _("Enabled") if global_data["status"] else _("Disabled") spotify_cache = CacheLevel.set_spotify() youtube_cache = CacheLevel.set_youtube() lavalink_cache = CacheLevel.set_lavalink() has_spotify_cache = current_level.is_superset(spotify_cache) has_youtube_cache = current_level.is_superset(youtube_cache) has_lavalink_cache = current_level.is_superset(lavalink_cache) autoplaylist = data["autoplaylist"] vote_enabled = data["vote_enabled"] msg = "----" + _("Server Settings") + "---- \n" msg += _("Auto-disconnect: [{dc}]\n").format(dc=_("Enabled") if dc else _("Disabled")) msg += _("Auto-play: [{autoplay}]\n").format( autoplay=_("Enabled") if autoplay else _("Disabled") ) if emptydc_enabled: msg += _("Disconnect timer: [{num_seconds}]\n").format( num_seconds=dynamic_time(emptydc_timer) ) if emptypause_enabled: msg += _("Auto Pause timer: [{num_seconds}]\n").format( num_seconds=dynamic_time(emptypause_timer) ) if dj_enabled and dj_role_obj: msg += _("DJ Role: [{role.name}]\n").format(role=dj_role_obj) if jukebox: msg += _("Jukebox: [{jukebox_name}]\n").format(jukebox_name=jukebox) msg += _("Command price: [{jukebox_price}]\n").format( jukebox_price=humanize_number(jukebox_price) ) if maxlength > 0: msg += _("Max track length: [{tracklength}]\n").format( tracklength=dynamic_time(maxlength) ) msg += _( "Repeat: [{repeat}]\n" "Shuffle: [{shuffle}]\n" "Shuffle bumped: [{bumpped_shuffle}]\n" "Song notify msgs: [{notify}]\n" "Songs as status: [{status}]\n" ).format( repeat=song_repeat, shuffle=song_shuffle, notify=song_notify, status=song_status, bumpped_shuffle=bumpped_shuffle, ) if thumbnail: msg += _("Thumbnails: [{0}]\n").format( _("Enabled") if thumbnail else _("Disabled") ) if vote_percent > 0: msg += _( "Vote skip: [{vote_enabled}]\nSkip percentage: [{vote_percent}%]\n" ).format( vote_percent=vote_percent, vote_enabled=_("Enabled") if vote_enabled else _("Disabled"), ) if autoplay or autoplaylist["enabled"]: if autoplaylist["enabled"]: pname = autoplaylist["name"] pid = autoplaylist["id"] pscope = autoplaylist["scope"] if pscope == PlaylistScope.GUILD.value: pscope = f"Server" elif pscope == PlaylistScope.USER.value: pscope = f"User" else: pscope = "Global" else: pname = _("Cached") pid = _("Cached") pscope = _("Cached") msg += ( "\n---" + _("Auto-play Settings") + "--- \n" + _("Playlist name: [{pname}]\n") + _("Playlist ID: [{pid}]\n") + _("Playlist scope: [{pscope}]\n") ).format(pname=pname, pid=pid, pscope=pscope) if is_owner: msg += ( "\n---" + _("Cache Settings") + "--- \n" + _("Max age: [{max_age}]\n") + _("Spotify cache: [{spotify_status}]\n") + _("Youtube cache: [{youtube_status}]\n") + _("Lavalink cache: [{lavalink_status}]\n") ).format( max_age=str(await self.config.cache_age()) + " " + _("days"), spotify_status=_("Enabled") if has_spotify_cache else _("Disabled"), youtube_status=_("Enabled") if has_youtube_cache else _("Disabled"), lavalink_status=_("Enabled") if has_lavalink_cache else _("Disabled"), ) msg += _( "\n---" + _("Lavalink Settings") + "--- \n" "Cog version: [{version}]\n" "Red-Lavalink: [{redlava}]\n" "External server: [{use_external_lavalink}]\n" ).format( version=__version__, redlava=lavalink.__version__, use_external_lavalink=_("Enabled") if global_data["use_external_lavalink"] else _("Disabled"), ) if is_owner: msg += _("Localtracks path: [{localpath}]\n").format(**global_data) await self._embed_msg(ctx, description=box(msg, lang="ini")) @audioset.command() @checks.is_owner() async def spotifyapi(self, ctx: commands.Context): """Instructions to set the Spotify API tokens.""" message = _( "1. Go to Spotify developers and log in with your Spotify account.\n" "(https://developer.spotify.com/dashboard/applications)\n" '2. Click "Create An App".\n' "3. Fill out the form provided with your app name, etc.\n" '4. When asked if you\'re developing commercial integration select "No".\n' "5. Accept the terms and conditions.\n" "6. Copy your client ID and your client secret into:\n" "`{prefix}set api spotify client_id " "client_secret `" ).format(prefix=ctx.prefix) await ctx.maybe_send_embed(message) @checks.is_owner() @audioset.command() async def status(self, ctx: commands.Context): """Enable/disable tracks' titles as status.""" status = await self.config.status() await self.config.status.set(not status) await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Song titles as status: {true_or_false}.").format( true_or_false=_("Enabled") if not status else _("Disabled") ), ) @audioset.command() @checks.mod_or_permissions(administrator=True) async def thumbnail(self, ctx: commands.Context): """Toggle displaying a thumbnail on audio messages.""" thumbnail = await self.config.guild(ctx.guild).thumbnail() await self.config.guild(ctx.guild).thumbnail.set(not thumbnail) await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Thumbnail display: {true_or_false}.").format( true_or_false=_("Enabled") if not thumbnail else _("Disabled") ), ) @audioset.command() @checks.mod_or_permissions(administrator=True) async def vote(self, ctx: commands.Context, percent: int): """Percentage needed for non-mods to skip tracks, 0 to disable.""" if percent < 0: return await self._embed_msg( ctx, title=_("Invalid Time"), description=_("Seconds can't be less than zero.") ) elif percent > 100: percent = 100 if percent == 0: enabled = False await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Voting disabled. All users can use queue management commands."), ) else: enabled = True await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Vote percentage set to {percent}%.").format(percent=percent), ) await self.config.guild(ctx.guild).vote_percent.set(percent) await self.config.guild(ctx.guild).vote_enabled.set(enabled) @audioset.command() @checks.is_owner() async def youtubeapi(self, ctx: commands.Context): """Instructions to set the YouTube API key.""" message = _( f"1. Go to Google Developers Console and log in with your Google account.\n" "(https://console.developers.google.com/)\n" "2. You should be prompted to create a new project (name does not matter).\n" "3. Click on Enable APIs and Services at the top.\n" "4. In the list of APIs choose or search for YouTube Data API v3 and " "click on it. Choose Enable.\n" "5. Click on Credentials on the left navigation bar.\n" "6. Click on Create Credential at the top.\n" '7. At the top click the link for "API key".\n' "8. No application restrictions are needed. Click Create at the bottom.\n" "9. You now have a key to add to `{prefix}set api youtube api_key `" ).format(prefix=ctx.prefix) await ctx.maybe_send_embed(message) @audioset.command(name="cache", usage="level=[5, 3, 2, 1, 0, -1, -2, -3]") @checks.is_owner() async def _storage(self, ctx: commands.Context, *, level: int = None): """Sets the caching level. Level can be one of the following: 0: Disables all caching 1: Enables Spotify Cache 2: Enables YouTube Cache 3: Enables Lavalink Cache 5: Enables all Caches If you wish to disable a specific cache use a negative number. """ current_level = CacheLevel(await self.config.cache_level()) spotify_cache = CacheLevel.set_spotify() youtube_cache = CacheLevel.set_youtube() lavalink_cache = CacheLevel.set_lavalink() has_spotify_cache = current_level.is_superset(spotify_cache) has_youtube_cache = current_level.is_superset(youtube_cache) has_lavalink_cache = current_level.is_superset(lavalink_cache) if level is None: msg = ( _("Max age: [{max_age}]\n") + _("Spotify cache: [{spotify_status}]\n") + _("Youtube cache: [{youtube_status}]\n") + _("Lavalink cache: [{lavalink_status}]\n") ).format( max_age=str(await self.config.cache_age()) + " " + _("days"), spotify_status=_("Enabled") if has_spotify_cache else _("Disabled"), youtube_status=_("Enabled") if has_youtube_cache else _("Disabled"), lavalink_status=_("Enabled") if has_lavalink_cache else _("Disabled"), ) await self._embed_msg(ctx, title=_("Cache Settings"), description=box(msg, lang="ini")) return await ctx.send_help() if level not in [5, 3, 2, 1, 0, -1, -2, -3]: return await ctx.send_help() removing = level < 0 if level == 5: newcache = CacheLevel.all() elif level == 0: newcache = CacheLevel.none() elif level in [-3, 3]: if removing: newcache = current_level - lavalink_cache else: newcache = current_level + lavalink_cache elif level in [-2, 2]: if removing: newcache = current_level - youtube_cache else: newcache = current_level + youtube_cache elif level in [-1, 1]: if removing: newcache = current_level - spotify_cache else: newcache = current_level + spotify_cache else: return await ctx.send_help() has_spotify_cache = newcache.is_superset(spotify_cache) has_youtube_cache = newcache.is_superset(youtube_cache) has_lavalink_cache = newcache.is_superset(lavalink_cache) msg = ( _("Max age: [{max_age}]\n") + _("Spotify cache: [{spotify_status}]\n") + _("Youtube cache: [{youtube_status}]\n") + _("Lavalink cache: [{lavalink_status}]\n") ).format( max_age=str(await self.config.cache_age()) + " " + _("days"), spotify_status=_("Enabled") if has_spotify_cache else _("Disabled"), youtube_status=_("Enabled") if has_youtube_cache else _("Disabled"), lavalink_status=_("Enabled") if has_lavalink_cache else _("Disabled"), ) await self._embed_msg(ctx, title=_("Cache Settings"), description=box(msg, lang="ini")) await self.config.cache_level.set(newcache.value) @audioset.command(name="cacheage") @checks.is_owner() async def _cacheage(self, ctx: commands.Context, age: int): """Sets the cache max age. This commands allows you to set the max number of days before an entry in the cache becomes invalid. """ msg = "" if age < 7: msg = _( "Cache age cannot be less than 7 days. If you wish to disable it run " "{prefix}audioset cache.\n" ).format(prefix=ctx.prefix) age = 7 msg += _("I've set the cache age to {age} days").format(age=age) await self.config.cache_age.set(age) await self._embed_msg(ctx, title=_("Setting Changed"), description=msg) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True, add_reactions=True) async def audiostats(self, ctx: commands.Context): """Audio stats.""" server_num = len(lavalink.active_players()) total_num = len(lavalink.all_players()) localtracks = await self.config.localpath() msg = "" for p in lavalink.all_players(): connect_start = p.fetch("connect") connect_dur = dynamic_time( int((datetime.datetime.utcnow() - connect_start).total_seconds()) ) try: query = audio_dataclasses.Query.process_input(p.current.uri) if query.is_local: if p.current.title == "Unknown title": current_title = localtracks.LocalPath(p.current.uri).to_string_user() msg += "{} [`{}`]: **{}**\n".format( p.channel.guild.name, connect_dur, current_title ) else: current_title = p.current.title msg += "{} [`{}`]: **{} - {}**\n".format( p.channel.guild.name, connect_dur, p.current.author, current_title ) else: msg += "{} [`{}`]: **[{}]({})**\n".format( p.channel.guild.name, connect_dur, p.current.title, p.current.uri ) except AttributeError: msg += "{} [`{}`]: **{}**\n".format( p.channel.guild.name, connect_dur, _("Nothing playing.") ) if total_num == 0: return await self._embed_msg(ctx, title=_("Not connected anywhere.")) servers_embed = [] pages = 1 for page in pagify(msg, delims=["\n"], page_length=1500): em = discord.Embed( colour=await ctx.embed_colour(), title=_("Playing in {num}/{total} servers:").format( num=humanize_number(server_num), total=humanize_number(total_num) ), description=page, ) em.set_footer( text="Page {}/{}".format( humanize_number(pages), humanize_number((math.ceil(len(msg) / 1500))) ) ) pages += 1 servers_embed.append(em) await menu(ctx, servers_embed, DEFAULT_CONTROLS) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def bump(self, ctx: commands.Context, index: int): """Bump a track number to the top of the queue.""" dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) player = lavalink.get_player(ctx.guild.id) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Bump Track"), description=_("You must be in the voice channel to bump a track."), ) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Bump Track"), description=_("You need the DJ role to bump tracks."), ) if index > len(player.queue) or index < 1: return await self._embed_msg( ctx, title=_("Unable To Bump Track"), description=_("Song number must be greater than 1 and within the queue limit."), ) bump_index = index - 1 bump_song = player.queue[bump_index] bump_song.extras["bumped"] = True player.queue.insert(0, bump_song) removed = player.queue.pop(index) description = get_track_description(removed) await self._embed_msg( ctx, title=_("Moved track to the top of the queue."), description=description ) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def disconnect(self, ctx: commands.Context): """Disconnect from the voice channel.""" if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) else: dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) player = lavalink.get_player(ctx.guild.id) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable to disconnect"), description=_("You need the DJ role to disconnect."), ) if not await self._can_instaskip(ctx, ctx.author) and not await self._is_alone(ctx): return await self._embed_msg( ctx, title=_("There are other people listening to music.") ) else: await self._embed_msg(ctx, title=_("Disconnecting...")) self.bot.dispatch("red_audio_audio_disconnect", ctx.guild) self._play_lock(ctx, False) eq = player.fetch("eq") player.queue = [] player.store("playing_song", None) if eq: await self.config.custom("EQUALIZER", ctx.guild.id).eq_bands.set(eq.bands) await player.stop() await player.disconnect() @commands.group(invoke_without_command=True) @commands.guild_only() @commands.cooldown(1, 15, commands.BucketType.guild) @commands.bot_has_permissions(embed_links=True, add_reactions=True) async def eq(self, ctx: commands.Context): """Equalizer management.""" if not self._player_check(ctx): ctx.command.reset_cooldown(ctx) return await self._embed_msg(ctx, title=_("Nothing playing.")) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) player = lavalink.get_player(ctx.guild.id) eq = player.fetch("eq", Equalizer()) reactions = [ "\N{BLACK LEFT-POINTING TRIANGLE}", "\N{LEFTWARDS BLACK ARROW}", "\N{BLACK UP-POINTING DOUBLE TRIANGLE}", "\N{UP-POINTING SMALL RED TRIANGLE}", "\N{DOWN-POINTING SMALL RED TRIANGLE}", "\N{BLACK DOWN-POINTING DOUBLE TRIANGLE}", "\N{BLACK RIGHTWARDS ARROW}", "\N{BLACK RIGHT-POINTING TRIANGLE}", "\N{BLACK CIRCLE FOR RECORD}", "\N{INFORMATION SOURCE}", ] await self._eq_msg_clear(player.fetch("eq_message")) eq_message = await ctx.send(box(eq.visualise(), lang="ini")) if dj_enabled and not await self._can_instaskip(ctx, ctx.author): with contextlib.suppress(discord.HTTPException): await eq_message.add_reaction("\N{INFORMATION SOURCE}") else: start_adding_reactions(eq_message, reactions) eq_msg_with_reacts = await ctx.fetch_message(eq_message.id) player.store("eq_message", eq_msg_with_reacts) await self._eq_interact(ctx, player, eq, eq_msg_with_reacts, 0) @eq.command(name="delete", aliases=["del", "remove"]) async def _eq_delete(self, ctx: commands.Context, eq_preset: str): """Delete a saved eq preset.""" async with self.config.custom("EQUALIZER", ctx.guild.id).eq_presets() as eq_presets: eq_preset = eq_preset.lower() try: if eq_presets[eq_preset][ "author" ] != ctx.author.id and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Delete Preset"), description=_("You are not the author of that preset setting."), ) del eq_presets[eq_preset] except KeyError: return await self._embed_msg( ctx, title=_("Unable To Delete Preset"), description=_( "{eq_preset} is not in the eq preset list.".format( eq_preset=eq_preset.capitalize() ) ), ) except TypeError: if await self._can_instaskip(ctx, ctx.author): del eq_presets[eq_preset] else: return await self._embed_msg( ctx, title=_("Unable To Delete Preset"), description=_("You are not the author of that preset setting."), ) await self._embed_msg( ctx, title=_("The {preset_name} preset was deleted.".format(preset_name=eq_preset)) ) @eq.command(name="list") async def _eq_list(self, ctx: commands.Context): """List saved eq presets.""" eq_presets = await self.config.custom("EQUALIZER", ctx.guild.id).eq_presets() if not eq_presets.keys(): return await self._embed_msg(ctx, title=_("No saved equalizer presets.")) space = "\N{EN SPACE}" header_name = _("Preset Name") header_author = _("Author") header = box( "[{header_name}]{space}[{header_author}]\n".format( header_name=header_name, space=space * 9, header_author=header_author ), lang="ini", ) preset_list = "" for preset, bands in eq_presets.items(): try: author = self.bot.get_user(bands["author"]) except TypeError: author = "None" msg = f"{preset}{space * (22 - len(preset))}{author}\n" preset_list += msg page_list = [] colour = await ctx.embed_colour() for page in pagify(preset_list, delims=[", "], page_length=1000): formatted_page = box(page, lang="ini") embed = discord.Embed(colour=colour, description=f"{header}\n{formatted_page}") embed.set_footer( text=_("{num} preset(s)").format(num=humanize_number(len(list(eq_presets.keys())))) ) page_list.append(embed) await menu(ctx, page_list, DEFAULT_CONTROLS) @eq.command(name="load") async def _eq_load(self, ctx: commands.Context, eq_preset: str): """Load a saved eq preset.""" eq_preset = eq_preset.lower() eq_presets = await self.config.custom("EQUALIZER", ctx.guild.id).eq_presets() try: eq_values = eq_presets[eq_preset]["bands"] except KeyError: return await self._embed_msg( ctx, title=_("No Preset Found"), description=_( "Preset named {eq_preset} does not exist.".format(eq_preset=eq_preset) ), ) except TypeError: eq_values = eq_presets[eq_preset] if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) player = lavalink.get_player(ctx.guild.id) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Load Preset"), description=_("You need the DJ role to load equalizer presets."), ) await self.config.custom("EQUALIZER", ctx.guild.id).eq_bands.set(eq_values) await self._eq_check(ctx, player) eq = player.fetch("eq", Equalizer()) await self._eq_msg_clear(player.fetch("eq_message")) message = await ctx.send( content=box(eq.visualise(), lang="ini"), embed=discord.Embed( colour=await ctx.embed_colour(), title=_("The {eq_preset} preset was loaded.".format(eq_preset=eq_preset)), ), ) player.store("eq_message", message) @eq.command(name="reset") async def _eq_reset(self, ctx: commands.Context): """Reset the eq to 0 across all bands.""" if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Modify Preset"), description=_("You need the DJ role to reset the equalizer."), ) player = lavalink.get_player(ctx.guild.id) eq = player.fetch("eq", Equalizer()) for band in range(eq._band_count): eq.set_gain(band, 0.0) await self._apply_gains(ctx.guild.id, eq.bands) await self.config.custom("EQUALIZER", ctx.guild.id).eq_bands.set(eq.bands) player.store("eq", eq) await self._eq_msg_clear(player.fetch("eq_message")) message = await ctx.send( content=box(eq.visualise(), lang="ini"), embed=discord.Embed( colour=await ctx.embed_colour(), title=_("Equalizer values have been reset.") ), ) player.store("eq_message", message) @eq.command(name="save") @commands.cooldown(1, 15, commands.BucketType.guild) async def _eq_save(self, ctx: commands.Context, eq_preset: str = None): """Save the current eq settings to a preset.""" if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author): ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Save Preset"), description=_("You need the DJ role to save equalizer presets."), ) if not eq_preset: await self._embed_msg(ctx, title=_("Please enter a name for this equalizer preset.")) try: eq_name_msg = await ctx.bot.wait_for( "message", timeout=15.0, check=MessagePredicate.regex(fr"^(?!{re.escape(ctx.prefix)})", ctx), ) eq_preset = eq_name_msg.content.split(" ")[0].strip('"').lower() except asyncio.TimeoutError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Save Preset"), description=_( "No equalizer preset name entered, try the command again later." ), ) eq_exists_msg = None eq_preset = eq_preset.lower().lstrip(ctx.prefix) eq_presets = await self.config.custom("EQUALIZER", ctx.guild.id).eq_presets() eq_list = list(eq_presets.keys()) if len(eq_preset) > 20: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Save Preset"), description=_("Try the command again with a shorter name."), ) if eq_preset in eq_list: eq_exists_msg = await self._embed_msg( ctx, title=_("Preset name already exists, do you want to replace it?") ) start_adding_reactions(eq_exists_msg, ReactionPredicate.YES_OR_NO_EMOJIS) pred = ReactionPredicate.yes_or_no(eq_exists_msg, ctx.author) await ctx.bot.wait_for("reaction_add", check=pred) if not pred.result: await self._clear_react(eq_exists_msg) embed2 = discord.Embed( colour=await ctx.embed_colour(), title=_("Not saving preset.") ) ctx.command.reset_cooldown(ctx) return await eq_exists_msg.edit(embed=embed2) player = lavalink.get_player(ctx.guild.id) eq = player.fetch("eq", Equalizer()) to_append = {eq_preset: {"author": ctx.author.id, "bands": eq.bands}} new_eq_presets = {**eq_presets, **to_append} await self.config.custom("EQUALIZER", ctx.guild.id).eq_presets.set(new_eq_presets) embed3 = discord.Embed( colour=await ctx.embed_colour(), title=_( "Current equalizer saved to the {preset_name} preset.".format( preset_name=eq_preset ) ), ) if eq_exists_msg: await self._clear_react(eq_exists_msg) await eq_exists_msg.edit(embed=embed3) else: await self._embed_msg(ctx, embed=embed3) @eq.command(name="set") async def _eq_set(self, ctx: commands.Context, band_name_or_position, band_value: float): """Set an eq band with a band number or name and value. Band positions are 1-15 and values have a range of -0.25 to 1.0. Band names are 25, 40, 63, 100, 160, 250, 400, 630, 1k, 1.6k, 2.5k, 4k, 6.3k, 10k, and 16k Hz. Setting a band value to -0.25 nullifies it while +0.25 is double. """ if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Set Preset"), description=_("You need the DJ role to set equalizer presets."), ) player = lavalink.get_player(ctx.guild.id) band_names = [ "25", "40", "63", "100", "160", "250", "400", "630", "1k", "1.6k", "2.5k", "4k", "6.3k", "10k", "16k", ] eq = player.fetch("eq", Equalizer()) bands_num = eq._band_count if band_value > 1: band_value = 1 elif band_value <= -0.25: band_value = -0.25 else: band_value = round(band_value, 1) try: band_number = int(band_name_or_position) - 1 except ValueError: band_number = None if band_number not in range(0, bands_num) and band_name_or_position not in band_names: return await self._embed_msg( ctx, title=_("Invalid Band"), description=_( "Valid band numbers are 1-15 or the band names listed in " "the help for this command." ), ) if band_name_or_position in band_names: band_pos = band_names.index(band_name_or_position) band_int = False eq.set_gain(int(band_pos), band_value) await self._apply_gain(ctx.guild.id, int(band_pos), band_value) else: band_int = True eq.set_gain(band_number, band_value) await self._apply_gain(ctx.guild.id, band_number, band_value) await self._eq_msg_clear(player.fetch("eq_message")) await self.config.custom("EQUALIZER", ctx.guild.id).eq_bands.set(eq.bands) player.store("eq", eq) band_name = band_names[band_number] if band_int else band_name_or_position message = await ctx.send( content=box(eq.visualise(), lang="ini"), embed=discord.Embed( colour=await ctx.embed_colour(), title=_("Preset Modified"), description=_( "The {band_name}Hz band has been set to {band_value}.".format( band_name=band_name, band_value=band_value ) ), ), ) player.store("eq_message", message) @commands.group() @commands.guild_only() @commands.bot_has_permissions(embed_links=True, add_reactions=True) async def local(self, ctx: commands.Context): """Local playback commands.""" @local.command(name="folder", aliases=["start"]) async def local_folder( self, ctx: commands.Context, play_subfolders: Optional[bool] = True, *, folder: str = None ): """Play all songs in a localtracks folder.""" if not await self._localtracks_check(ctx): return if not folder: await ctx.invoke(self.local_play, play_subfolders=play_subfolders) else: folder = folder.strip() _dir = audio_dataclasses.LocalPath.joinpath(folder) if not _dir.exists(): return await self._embed_msg( ctx, title=_("Folder Not Found"), description=_("Localtracks folder named {name} does not exist.").format( name=folder ), ) query = audio_dataclasses.Query.process_input(_dir, search_subfolders=play_subfolders) await self._local_play_all(ctx, query, from_search=False if not folder else True) @local.command(name="play") async def local_play(self, ctx: commands.Context, play_subfolders: Optional[bool] = True): """Play a local track.""" if not await self._localtracks_check(ctx): return localtracks_folders = await self._localtracks_folders( ctx, search_subfolders=play_subfolders ) if not localtracks_folders: return await self._embed_msg(ctx, title=_("No album folders found.")) async with ctx.typing(): len_folder_pages = math.ceil(len(localtracks_folders) / 5) folder_page_list = [] for page_num in range(1, len_folder_pages + 1): embed = await self._build_search_page(ctx, localtracks_folders, page_num) folder_page_list.append(embed) async def _local_folder_menu( ctx: commands.Context, pages: list, controls: MutableMapping, message: discord.Message, page: int, timeout: float, emoji: str, ): if message: with contextlib.suppress(discord.HTTPException): await message.delete() await self._search_button_action(ctx, localtracks_folders, emoji, page) return None local_folder_controls = { "\N{DIGIT ONE}\N{COMBINING ENCLOSING KEYCAP}": _local_folder_menu, "\N{DIGIT TWO}\N{COMBINING ENCLOSING KEYCAP}": _local_folder_menu, "\N{DIGIT THREE}\N{COMBINING ENCLOSING KEYCAP}": _local_folder_menu, "\N{DIGIT FOUR}\N{COMBINING ENCLOSING KEYCAP}": _local_folder_menu, "\N{DIGIT FIVE}\N{COMBINING ENCLOSING KEYCAP}": _local_folder_menu, "\N{LEFTWARDS BLACK ARROW}": prev_page, "\N{CROSS MARK}": close_menu, "\N{BLACK RIGHTWARDS ARROW}": next_page, } dj_enabled = await self.config.guild(ctx.guild).dj_enabled() if dj_enabled and not await self._can_instaskip(ctx, ctx.author): return await menu(ctx, folder_page_list, DEFAULT_CONTROLS) else: await menu(ctx, folder_page_list, local_folder_controls) @local.command(name="search") async def local_search( self, ctx: commands.Context, search_subfolders: Optional[bool] = True, *, search_words ): """Search for songs across all localtracks folders.""" if not await self._localtracks_check(ctx): return all_tracks = await self._folder_list( ctx, ( audio_dataclasses.Query.process_input( audio_dataclasses.LocalPath( await self.config.localpath() ).localtrack_folder.absolute(), search_subfolders=search_subfolders, ) ), ) if not all_tracks: return await self._embed_msg(ctx, title=_("No album folders found.")) async with ctx.typing(): search_list = await self._build_local_search_list(all_tracks, search_words) if not search_list: return await self._embed_msg(ctx, title=_("No matches.")) return await ctx.invoke(self.search, query=search_list) async def _localtracks_folders( self, ctx: commands.Context, search_subfolders=False ) -> Optional[List[Union[Path, audio_dataclasses.LocalPath]]]: audio_data = audio_dataclasses.LocalPath( audio_dataclasses.LocalPath(None).localtrack_folder.absolute() ) if not await self._localtracks_check(ctx): return return ( await audio_data.subfolders_in_tree() if search_subfolders else await audio_data.subfolders() ) async def _folder_list( self, ctx: commands.Context, query: audio_dataclasses.Query ) -> Optional[List[audio_dataclasses.Query]]: if not await self._localtracks_check(ctx): return query = audio_dataclasses.Query.process_input(query) if not query.track.exists(): return return ( await query.track.tracks_in_tree() if query.search_subfolders else await query.track.tracks_in_folder() ) async def _folder_tracks( self, ctx, player: lavalink.player_manager.Player, query: audio_dataclasses.Query ) -> Optional[List[lavalink.rest_api.Track]]: if not await self._localtracks_check(ctx): return audio_data = audio_dataclasses.LocalPath(None) try: query.track.path.relative_to(audio_data.to_string()) except ValueError: return local_tracks = [] for local_file in await self._all_folder_tracks(ctx, query): trackdata, called_api = await self.music_cache.lavalink_query(ctx, player, local_file) with contextlib.suppress(IndexError): local_tracks.append(trackdata.tracks[0]) return local_tracks async def _local_play_all( self, ctx: commands.Context, query: audio_dataclasses.Query, from_search=False ) -> None: if not await self._localtracks_check(ctx): return if from_search: query = audio_dataclasses.Query.process_input( query.track.to_string(), invoked_from="local folder" ) await ctx.invoke(self.search, query=query) async def _all_folder_tracks( self, ctx: commands.Context, query: audio_dataclasses.Query ) -> Optional[List[audio_dataclasses.Query]]: if not await self._localtracks_check(ctx): return return ( await query.track.tracks_in_tree() if query.search_subfolders else await query.track.tracks_in_folder() ) async def _localtracks_check(self, ctx: commands.Context) -> bool: folder = audio_dataclasses.LocalPath(None) if folder.localtrack_folder.exists(): return True if ctx.invoked_with != "start": await self._embed_msg( ctx, title=_("Invalid Environment"), description=_("No localtracks folder.") ) return False @staticmethod async def _build_local_search_list(to_search, search_words): to_search_string = {i.track.name for i in to_search} search_results = process.extract(search_words, to_search_string, limit=50) search_list = [] for track_match, percent_match in search_results: if percent_match > 60: search_list.extend( [i.track.to_string_user() for i in to_search if i.track.name == track_match] ) await asyncio.sleep(0) return search_list @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True, add_reactions=True) async def now(self, ctx: commands.Context): """Now playing.""" if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) expected = ("⏮", "⏹", "⏯", "⏭") emoji = {"prev": "⏮", "stop": "⏹", "pause": "⏯", "next": "⏭"} player = lavalink.get_player(ctx.guild.id) if player.current: arrow = await draw_time(ctx) pos = lavalink.utils.format_time(player.position) if player.current.is_stream: dur = "LIVE" else: dur = lavalink.utils.format_time(player.current.length) song = get_track_description(player.current) song += _("\n Requested by: **{track.requester}**") song += "\n\n{arrow}`{pos}`/`{dur}`" song = song.format(track=player.current, arrow=arrow, pos=pos, dur=dur) else: song = _("Nothing.") if player.fetch("np_message") is not None: with contextlib.suppress(discord.HTTPException): await player.fetch("np_message").delete() embed = discord.Embed(title=_("Now Playing"), description=song) if await self.config.guild(ctx.guild).thumbnail() and player.current: if player.current.thumbnail: embed.set_thumbnail(url=player.current.thumbnail) shuffle = await self.config.guild(ctx.guild).shuffle() repeat = await self.config.guild(ctx.guild).repeat() autoplay = await self.config.guild(ctx.guild).auto_play() text = "" text += ( _("Auto-Play") + ": " + ("\N{WHITE HEAVY CHECK MARK}" if autoplay else "\N{CROSS MARK}") ) text += ( (" | " if text else "") + _("Shuffle") + ": " + ("\N{WHITE HEAVY CHECK MARK}" if shuffle else "\N{CROSS MARK}") ) text += ( (" | " if text else "") + _("Repeat") + ": " + ("\N{WHITE HEAVY CHECK MARK}" if repeat else "\N{CROSS MARK}") ) message = await self._embed_msg(ctx, embed=embed, footer=text) player.store("np_message", message) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) vote_enabled = await self.config.guild(ctx.guild).vote_enabled() if dj_enabled or vote_enabled: if not await self._can_instaskip(ctx, ctx.author) and not await self._is_alone(ctx): return if not player.queue: expected = ("⏹", "⏯") if player.current: task = start_adding_reactions(message, expected[:4]) else: task = None try: (r, u) = await self.bot.wait_for( "reaction_add", check=ReactionPredicate.with_emojis(expected, message, ctx.author), timeout=30.0, ) except asyncio.TimeoutError: return await self._clear_react(message, emoji) else: if task is not None: task.cancel() reacts = {v: k for k, v in emoji.items()} react = reacts[r.emoji] if react == "prev": await self._clear_react(message, emoji) await ctx.invoke(self.prev) elif react == "stop": await self._clear_react(message, emoji) await ctx.invoke(self.stop) elif react == "pause": await self._clear_react(message, emoji) await ctx.invoke(self.pause) elif react == "next": await self._clear_react(message, emoji) await ctx.invoke(self.skip) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def pause(self, ctx: commands.Context): """Pause or resume a playing track.""" dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) player = lavalink.get_player(ctx.guild.id) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Manage Tracks"), description=_("You must be in the voice channel to pause or resume."), ) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author) and not await self._is_alone(ctx): return await self._embed_msg( ctx, title=_("Unable To Manage Tracks"), description=_("You need the DJ role to pause or resume tracks."), ) if not player.current: return await self._embed_msg(ctx, title=_("Nothing playing.")) description = get_track_description(player.current) if player.current and not player.paused: await player.pause() return await self._embed_msg(ctx, title=_("Track Paused"), description=description) if player.current and player.paused: await player.pause(False) return await self._embed_msg(ctx, title=_("Track Resumed"), description=description) await self._embed_msg(ctx, title=_("Nothing playing.")) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def percent(self, ctx: commands.Context): """Queue percentage.""" if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) player = lavalink.get_player(ctx.guild.id) queue_tracks = player.queue requesters = {"total": 0, "users": {}} async def _usercount(req_username): if req_username in requesters["users"]: requesters["users"][req_username]["songcount"] += 1 requesters["total"] += 1 else: requesters["users"][req_username] = {} requesters["users"][req_username]["songcount"] = 1 requesters["total"] += 1 for track in queue_tracks: req_username = "{}#{}".format(track.requester.name, track.requester.discriminator) await _usercount(req_username) await asyncio.sleep(0) try: req_username = "{}#{}".format( player.current.requester.name, player.current.requester.discriminator ) await _usercount(req_username) except AttributeError: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) for req_username in requesters["users"]: percentage = float(requesters["users"][req_username]["songcount"]) / float( requesters["total"] ) requesters["users"][req_username]["percent"] = round(percentage * 100, 1) await asyncio.sleep(0) top_queue_users = heapq.nlargest( 20, [ (x, requesters["users"][x][y]) for x in requesters["users"] for y in requesters["users"][x] if y == "percent" ], key=lambda x: x[1], ) queue_user = ["{}: {:g}%".format(x[0], x[1]) for x in top_queue_users] queue_user_list = "\n".join(queue_user) await self._embed_msg( ctx, title=_("Queued and playing tracks:"), description=queue_user_list ) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def play(self, ctx: commands.Context, *, query: str): """Play a URL or search for a track.""" query = audio_dataclasses.Query.process_input(query) guild_data = await self.config.guild(ctx.guild).all() restrict = await self.config.restrict() if restrict and match_url(query): valid_url = url_check(query) if not valid_url: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("That URL is not allowed."), ) elif not await is_allowed(ctx.guild, f"{query}", query_obj=query): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("That track is not allowed.") ) if not self._player_check(ctx): if self._connection_aborted: msg = _("Connection to Lavalink has failed") desc = EmptyEmbed if await ctx.bot.is_owner(ctx.author): desc = _("Please check your console or logs for details.") return await self._embed_msg(ctx, title=msg, description=desc) try: if ( not ctx.author.voice.channel.permissions_for(ctx.me).connect or not ctx.author.voice.channel.permissions_for(ctx.me).move_members and userlimit(ctx.author.voice.channel) ): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("I don't have permission to connect to your channel."), ) await lavalink.connect(ctx.author.voice.channel) player = lavalink.get_player(ctx.guild.id) player.store("connect", datetime.datetime.utcnow()) except AttributeError: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Connect to a voice channel first."), ) except IndexError: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Connection to Lavalink has not yet been established."), ) if guild_data["dj_enabled"]: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("You need the DJ role to queue tracks."), ) player = lavalink.get_player(ctx.guild.id) player.store("channel", ctx.channel.id) player.store("guild", ctx.guild.id) await self._eq_check(ctx, player) await self._data_check(ctx) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("You must be in the voice channel to use the play command."), ) if not query.valid: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("No tracks found for `{query}`.").format( query=query.to_string_user() ), ) if len(player.queue) >= 10000: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Queue size limit reached.") ) if not await self._currency_check(ctx, guild_data["jukebox_price"]): return if query.is_spotify: return await self._get_spotify_tracks(ctx, query) try: await self._enqueue_tracks(ctx, query) except QueryUnauthorized as err: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=err.message ) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def bumpplay( self, ctx: commands.Context, play_now: Optional[bool] = False, *, query: str ): """Force play a URL or search for a track.""" query = audio_dataclasses.Query.process_input(query) if not query.single_track: return await self._embed_msg( ctx, title=_("Unable to bump track"), description=_("Only single tracks work with bump play."), ) guild_data = await self.config.guild(ctx.guild).all() restrict = await self.config.restrict() if restrict and match_url(query): valid_url = url_check(query) if not valid_url: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("That URL is not allowed."), ) elif not await is_allowed(ctx.guild, f"{query}", query_obj=query): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("That track is not allowed.") ) if not self._player_check(ctx): if self._connection_aborted: msg = _("Connection to Lavalink has failed") desc = EmptyEmbed if await ctx.bot.is_owner(ctx.author): desc = _("Please check your console or logs for details.") return await self._embed_msg(ctx, title=msg, description=desc) try: if ( not ctx.author.voice.channel.permissions_for(ctx.me).connect or not ctx.author.voice.channel.permissions_for(ctx.me).move_members and userlimit(ctx.author.voice.channel) ): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("I don't have permission to connect to your channel."), ) await lavalink.connect(ctx.author.voice.channel) player = lavalink.get_player(ctx.guild.id) player.store("connect", datetime.datetime.utcnow()) except AttributeError: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Connect to a voice channel first."), ) except IndexError: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Connection to Lavalink has not yet been established."), ) if guild_data["dj_enabled"]: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("You need the DJ role to queue tracks."), ) player = lavalink.get_player(ctx.guild.id) player.store("channel", ctx.channel.id) player.store("guild", ctx.guild.id) await self._eq_check(ctx, player) await self._data_check(ctx) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("You must be in the voice channel to use the play command."), ) if not query.valid: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("No tracks found for `{query}`.").format( query=query.to_string_user() ), ) if len(player.queue) >= 10000: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Queue size limit reached.") ) if not await self._currency_check(ctx, guild_data["jukebox_price"]): return try: if query.is_spotify: tracks = await self._get_spotify_tracks(ctx, query) else: tracks = await self._enqueue_tracks(ctx, query, enqueue=False) except QueryUnauthorized as err: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=err.message ) if isinstance(tracks, discord.Message): return elif not tracks: self._play_lock(ctx, False) title = _("Unable To Play Tracks") desc = _("No tracks found for `{query}`.").format(query=query.to_string_user()) embed = discord.Embed(title=title, description=desc) if await self.config.use_external_lavalink() and query.is_local: embed.description = _( "Local tracks will not work " "if the `Lavalink.jar` cannot see the track.\n" "This may be due to permissions or because Lavalink.jar is being run " "in a different machine than the local tracks." ) elif ( query.is_local and query.suffix in audio_dataclasses._PARTIALLY_SUPPORTED_MUSIC_EXT ): title = _("Track is not playable.") embed = discord.Embed(title=title) embed.description = _( "**{suffix}** is not a fully supported format and some " "tracks may not play." ).format(suffix=query.suffix) return await self._embed_msg(ctx, embed=embed) elif isinstance(tracks, discord.Message): return queue_dur = await track_remaining_duration(ctx) index = query.track_index seek = 0 if query.start_time: seek = query.start_time single_track = ( tracks if isinstance(tracks, lavalink.rest_api.Track) else tracks[index] if index else tracks[0] ) if seek and seek > 0: single_track.start_timestamp = seek * 1000 if not await is_allowed( ctx.guild, ( f"{single_track.title} {single_track.author} {single_track.uri} " f"{str(audio_dataclasses.Query.process_input(single_track))}" ), ): log.debug(f"Query is not allowed in {ctx.guild} ({ctx.guild.id})") self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("This track is not allowed in this server."), ) elif guild_data["maxlength"] > 0: if track_limit(single_track, guild_data["maxlength"]): single_track.requester = ctx.author player.queue.insert(0, single_track) player.maybe_shuffle() self.bot.dispatch( "red_audio_track_enqueue", player.channel.guild, single_track, ctx.author ) else: self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Track exceeds maximum length."), ) else: single_track.requester = ctx.author single_track.extras["bumped"] = True player.queue.insert(0, single_track) player.maybe_shuffle() self.bot.dispatch( "red_audio_track_enqueue", player.channel.guild, single_track, ctx.author ) description = get_track_description(single_track) footer = None if not play_now and not guild_data["shuffle"] and queue_dur > 0: footer = _("{time} until track playback: #1 in queue").format( time=lavalink.utils.format_time(queue_dur) ) await self._embed_msg( ctx, title=_("Track Enqueued"), description=description, footer=footer ) if not player.current: await player.play() elif play_now: await player.skip() self._play_lock(ctx, False) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def genre(self, ctx: commands.Context): """Pick a Spotify playlist from a list of categories to start playing.""" async def _category_search_menu( ctx: commands.Context, pages: list, controls: MutableMapping, message: discord.Message, page: int, timeout: float, emoji: str, ): if message: output = await self._genre_search_button_action(ctx, category_list, emoji, page) with contextlib.suppress(discord.HTTPException): await message.delete() return output async def _playlist_search_menu( ctx: commands.Context, pages: list, controls: MutableMapping, message: discord.Message, page: int, timeout: float, emoji: str, ): if message: output = await self._genre_search_button_action( ctx, playlists_list, emoji, page, playlist=True ) with contextlib.suppress(discord.HTTPException): await message.delete() return output category_search_controls = { "\N{DIGIT ONE}\N{COMBINING ENCLOSING KEYCAP}": _category_search_menu, "\N{DIGIT TWO}\N{COMBINING ENCLOSING KEYCAP}": _category_search_menu, "\N{DIGIT THREE}\N{COMBINING ENCLOSING KEYCAP}": _category_search_menu, "\N{DIGIT FOUR}\N{COMBINING ENCLOSING KEYCAP}": _category_search_menu, "\N{DIGIT FIVE}\N{COMBINING ENCLOSING KEYCAP}": _category_search_menu, "\N{LEFTWARDS BLACK ARROW}": prev_page, "\N{CROSS MARK}": close_menu, "\N{BLACK RIGHTWARDS ARROW}": next_page, } playlist_search_controls = { "\N{DIGIT ONE}\N{COMBINING ENCLOSING KEYCAP}": _playlist_search_menu, "\N{DIGIT TWO}\N{COMBINING ENCLOSING KEYCAP}": _playlist_search_menu, "\N{DIGIT THREE}\N{COMBINING ENCLOSING KEYCAP}": _playlist_search_menu, "\N{DIGIT FOUR}\N{COMBINING ENCLOSING KEYCAP}": _playlist_search_menu, "\N{DIGIT FIVE}\N{COMBINING ENCLOSING KEYCAP}": _playlist_search_menu, "\N{LEFTWARDS BLACK ARROW}": prev_page, "\N{CROSS MARK}": close_menu, "\N{BLACK RIGHTWARDS ARROW}": next_page, } api_data = await self._check_api_tokens() if any( [ not api_data["spotify_client_id"], not api_data["spotify_client_secret"], not api_data["youtube_api"], ] ): return await self._embed_msg( ctx, title=_("Invalid Environment"), description=_( "The owner needs to set the Spotify client ID, Spotify client secret, " "and YouTube API key before Spotify URLs or codes can be used. " "\nSee `{prefix}audioset youtubeapi` and `{prefix}audioset spotifyapi` " "for instructions." ).format(prefix=ctx.prefix), ) guild_data = await self.config.guild(ctx.guild).all() if not self._player_check(ctx): if self._connection_aborted: msg = _("Connection to Lavalink has failed") desc = EmptyEmbed if await ctx.bot.is_owner(ctx.author): desc = _("Please check your console or logs for details.") return await self._embed_msg(ctx, title=msg, description=desc) try: if ( not ctx.author.voice.channel.permissions_for(ctx.me).connect or not ctx.author.voice.channel.permissions_for(ctx.me).move_members and userlimit(ctx.author.voice.channel) ): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("I don't have permission to connect to your channel."), ) await lavalink.connect(ctx.author.voice.channel) player = lavalink.get_player(ctx.guild.id) player.store("connect", datetime.datetime.utcnow()) except AttributeError: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Connect to a voice channel first."), ) except IndexError: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Connection to Lavalink has not yet been established."), ) if guild_data["dj_enabled"]: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("You need the DJ role to queue tracks."), ) player = lavalink.get_player(ctx.guild.id) player.store("channel", ctx.channel.id) player.store("guild", ctx.guild.id) await self._eq_check(ctx, player) await self._data_check(ctx) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("You must be in the voice channel to use the genre command."), ) try: category_list = await self.music_cache.spotify_api.get_categories() except SpotifyFetchError as error: return await self._embed_msg( ctx, title=_("No categories found"), description=_(error.message).format(prefix=ctx.prefix), ) if not category_list: return await self._embed_msg(ctx, title=_("No categories found, try again later.")) len_folder_pages = math.ceil(len(category_list) / 5) category_search_page_list = [] for page_num in range(1, len_folder_pages + 1): embed = await self._build_genre_search_page( ctx, category_list, page_num, _("Categories") ) category_search_page_list.append(embed) await asyncio.sleep(0) cat_menu_output = await menu(ctx, category_search_page_list, category_search_controls) if not cat_menu_output: return await self._embed_msg(ctx, title=_("No categories selected, try again later.")) category_name, category_pick = cat_menu_output playlists_list = await self.music_cache.spotify_api.get_playlist_from_category( category_pick ) if not playlists_list: return await self._embed_msg(ctx, title=_("No categories found, try again later.")) len_folder_pages = math.ceil(len(playlists_list) / 5) playlists_search_page_list = [] for page_num in range(1, len_folder_pages + 1): embed = await self._build_genre_search_page( ctx, playlists_list, page_num, _("Playlists for {friendly_name}").format(friendly_name=category_name), playlist=True, ) playlists_search_page_list.append(embed) await asyncio.sleep(0) playlists_pick = await menu(ctx, playlists_search_page_list, playlist_search_controls) query = audio_dataclasses.Query.process_input(playlists_pick) if not query.valid: return await self._embed_msg(ctx, title=_("No tracks to play.")) if len(player.queue) >= 10000: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Queue size limit reached.") ) if not await self._currency_check(ctx, guild_data["jukebox_price"]): return if query.is_spotify: return await self._get_spotify_tracks(ctx, query) return await self._embed_msg( ctx, title=_("Couldn't find tracks for the selected playlist.") ) @staticmethod async def _genre_search_button_action( ctx: commands.Context, options, emoji, page, playlist=False ): try: if emoji == "\N{DIGIT ONE}\N{COMBINING ENCLOSING KEYCAP}": search_choice = options[0 + (page * 5)] elif emoji == "\N{DIGIT TWO}\N{COMBINING ENCLOSING KEYCAP}": search_choice = options[1 + (page * 5)] elif emoji == "\N{DIGIT THREE}\N{COMBINING ENCLOSING KEYCAP}": search_choice = options[2 + (page * 5)] elif emoji == "\N{DIGIT FOUR}\N{COMBINING ENCLOSING KEYCAP}": search_choice = options[3 + (page * 5)] elif emoji == "\N{DIGIT FIVE}\N{COMBINING ENCLOSING KEYCAP}": search_choice = options[4 + (page * 5)] else: search_choice = options[0 + (page * 5)] except IndexError: search_choice = options[-1] if not playlist: return list(search_choice.items())[0] else: return search_choice.get("uri") @staticmethod async def _build_genre_search_page( ctx: commands.Context, tracks, page_num, title, playlist=False ): search_num_pages = math.ceil(len(tracks) / 5) search_idx_start = (page_num - 1) * 5 search_idx_end = search_idx_start + 5 search_list = "" for i, entry in enumerate(tracks[search_idx_start:search_idx_end], start=search_idx_start): search_track_num = i + 1 if search_track_num > 5: search_track_num = search_track_num % 5 if search_track_num == 0: search_track_num = 5 if playlist: name = "**[{}]({})** - {}".format( entry.get("name"), entry.get("url"), str(entry.get("tracks")) + " " + _("tracks"), ) else: name = f"{list(entry.keys())[0]}" search_list += "`{}.` {}\n".format(search_track_num, name) await asyncio.sleep(0) embed = discord.Embed( colour=await ctx.embed_colour(), title=title, description=search_list ) embed.set_footer( text=_("Page {page_num}/{total_pages}").format( page_num=page_num, total_pages=search_num_pages ) ) return embed @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) @checks.mod_or_permissions(manage_messages=True) async def autoplay(self, ctx: commands.Context): """Starts auto play.""" if not self._player_check(ctx): if self._connection_aborted: msg = _("Connection to Lavalink has failed") desc = EmptyEmbed if await ctx.bot.is_owner(ctx.author): desc = _("Please check your console or logs for details.") return await self._embed_msg(ctx, title=msg, description=desc) try: if ( not ctx.author.voice.channel.permissions_for(ctx.me).connect or not ctx.author.voice.channel.permissions_for(ctx.me).move_members and userlimit(ctx.author.voice.channel) ): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("I don't have permission to connect to your channel."), ) await lavalink.connect(ctx.author.voice.channel) player = lavalink.get_player(ctx.guild.id) player.store("connect", datetime.datetime.utcnow()) except AttributeError: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Connect to a voice channel first."), ) except IndexError: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Connection to Lavalink has not yet been established."), ) guild_data = await self.config.guild(ctx.guild).all() if guild_data["dj_enabled"]: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("You need the DJ role to queue tracks."), ) player = lavalink.get_player(ctx.guild.id) player.store("channel", ctx.channel.id) player.store("guild", ctx.guild.id) await self._eq_check(ctx, player) await self._data_check(ctx) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("You must be in the voice channel to use the autoplay command."), ) if len(player.queue) >= 10000: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Queue size limit reached.") ) if not await self._currency_check(ctx, guild_data["jukebox_price"]): return try: await self.music_cache.autoplay(player) except DatabaseError: notify_channel = player.fetch("channel") if notify_channel: notify_channel = self.bot.get_channel(notify_channel) await self._embed_msg(notify_channel, title=_("Couldn't get a valid track.")) return if not guild_data["auto_play"]: await ctx.invoke(self._autoplay_toggle) if not guild_data["notify"] and ( (player.current and not player.current.extras.get("autoplay")) or not player.current ): await self._embed_msg(ctx, title=_("Auto play started.")) elif player.current: await self._embed_msg(ctx, title=_("Adding a track to queue.")) async def _get_spotify_tracks(self, ctx: commands.Context, query: audio_dataclasses.Query): if ctx.invoked_with in ["play", "genre"]: enqueue_tracks = True else: enqueue_tracks = False player = lavalink.get_player(ctx.guild.id) api_data = await self._check_api_tokens() if ( not api_data["spotify_client_id"] or not api_data["spotify_client_secret"] or not api_data["youtube_api"] ): return await self._embed_msg( ctx, title=_("Invalid Environment"), description=_( "The owner needs to set the Spotify client ID, Spotify client secret, " "and YouTube API key before Spotify URLs or codes can be used. " "\nSee `{prefix}audioset youtubeapi` and `{prefix}audioset spotifyapi` " "for instructions." ).format(prefix=ctx.prefix), ) try: if self.play_lock[ctx.message.guild.id]: return await self._embed_msg( ctx, title=_("Unable To Get Tracks"), description=_("Wait until the playlist has finished loading."), ) except KeyError: pass if query.single_track: try: res = await self.music_cache.spotify_query( ctx, "track", query.id, skip_youtube=True, notifier=None ) if not res: title = _("Nothing found.") embed = discord.Embed(title=title) if ( query.is_local and query.suffix in audio_dataclasses._PARTIALLY_SUPPORTED_MUSIC_EXT ): title = _("Track is not playable.") description = _( "**{suffix}** is not a fully supported " "format and some tracks may not play." ).format(suffix=query.suffix) embed = discord.Embed(title=title, description=description) return await self._embed_msg(ctx, embed=embed) except SpotifyFetchError as error: self._play_lock(ctx, False) return await self._embed_msg(ctx, title=_(error.message).format(prefix=ctx.prefix)) self._play_lock(ctx, False) try: if enqueue_tracks: new_query = audio_dataclasses.Query.process_input(res[0]) new_query.start_time = query.start_time return await self._enqueue_tracks(ctx, new_query) else: query = audio_dataclasses.Query.process_input(res[0]) try: result, called_api = await self.music_cache.lavalink_query( ctx, player, query ) except TrackEnqueueError: self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Unable to Get Track"), description=_( "I'm unable get a track from Lavalink at the moment, try again in a few minutes." ), ) tracks = result.tracks if not tracks: embed = discord.Embed(title=_("Nothing found.")) if ( query.is_local and query.suffix in audio_dataclasses._PARTIALLY_SUPPORTED_MUSIC_EXT ): embed = discord.Embed(title=_("Track is not playable.")) embed.description = _( "**{suffix}** is not a fully supported format and some " "tracks may not play." ).format(suffix=query.suffix) return await self._embed_msg(ctx, embed=embed) single_track = tracks[0] single_track.start_timestamp = query.start_time * 1000 single_track = [single_track] return single_track except KeyError: self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Invalid Environment"), description=_( "The Spotify API key or client secret has not been set properly. " "\nUse `{prefix}audioset spotifyapi` for instructions." ).format(prefix=ctx.prefix), ) elif query.is_album or query.is_playlist: self._play_lock(ctx, True) track_list = await self._spotify_playlist( ctx, "album" if query.is_album else "playlist", query, enqueue_tracks ) self._play_lock(ctx, False) return track_list else: return await self._embed_msg( ctx, title=_("Unable To Find Tracks"), description=_("This doesn't seem to be a supported Spotify URL or code."), ) async def _enqueue_tracks( self, ctx: commands.Context, query: Union[audio_dataclasses.Query, list], enqueue: bool = True, ): player = lavalink.get_player(ctx.guild.id) try: if self.play_lock[ctx.message.guild.id]: return await self._embed_msg( ctx, title=_("Unable To Get Tracks"), description=_("Wait until the playlist has finished loading."), ) except KeyError: self._play_lock(ctx, True) guild_data = await self.config.guild(ctx.guild).all() first_track_only = False single_track = None index = None playlist_data = None playlist_url = None seek = 0 if type(query) is not list: if not await is_allowed(ctx.guild, f"{query}", query_obj=query): raise QueryUnauthorized( _("{query} is not an allowed query.").format(query=query.to_string_user()) ) if query.single_track: first_track_only = True index = query.track_index if query.start_time: seek = query.start_time try: result, called_api = await self.music_cache.lavalink_query(ctx, player, query) except TrackEnqueueError: self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Unable to Get Track"), description=_( "I'm unable get a track from Lavalink at the moment, try again in a few minutes." ), ) tracks = result.tracks playlist_data = result.playlist_info if not enqueue: return tracks if not tracks: self._play_lock(ctx, False) title = _("Nothing found.") embed = discord.Embed(title=title) if result.exception_message: embed.set_footer(text=result.exception_message[:2000].replace("\n", "")) if await self.config.use_external_lavalink() and query.is_local: embed.description = _( "Local tracks will not work " "if the `Lavalink.jar` cannot see the track.\n" "This may be due to permissions or because Lavalink.jar is being run " "in a different machine than the local tracks." ) elif ( query.is_local and query.suffix in audio_dataclasses._PARTIALLY_SUPPORTED_MUSIC_EXT ): title = _("Track is not playable.") embed = discord.Embed(title=title) embed.description = _( "**{suffix}** is not a fully supported format and some " "tracks may not play." ).format(suffix=query.suffix) return await self._embed_msg(ctx, embed=embed) else: tracks = query queue_dur = await queue_duration(ctx) queue_total_duration = lavalink.utils.format_time(queue_dur) before_queue_length = len(player.queue) if not first_track_only and len(tracks) > 1: # a list of Tracks where all should be enqueued # this is a Spotify playlist already made into a list of Tracks or a # url where Lavalink handles providing all Track objects to use, like a # YouTube or Soundcloud playlist if len(player.queue) >= 10000: return await self._embed_msg(ctx, title=_("Queue size limit reached.")) track_len = 0 empty_queue = not player.queue for track in tracks: if len(player.queue) >= 10000: continue if not await is_allowed( ctx.guild, ( f"{track.title} {track.author} {track.uri} " f"{str(audio_dataclasses.Query.process_input(track))}" ), ): log.debug(f"Query is not allowed in {ctx.guild} ({ctx.guild.id})") continue elif guild_data["maxlength"] > 0: if track_limit(track, guild_data["maxlength"]): track_len += 1 player.add(ctx.author, track) self.bot.dispatch( "red_audio_track_enqueue", player.channel.guild, track, ctx.author ) else: track_len += 1 player.add(ctx.author, track) self.bot.dispatch( "red_audio_track_enqueue", player.channel.guild, track, ctx.author ) await asyncio.sleep(0) player.maybe_shuffle(0 if empty_queue else 1) if len(tracks) > track_len: maxlength_msg = " {bad_tracks} tracks cannot be queued.".format( bad_tracks=(len(tracks) - track_len) ) else: maxlength_msg = "" playlist_name = escape(playlist_data.name if playlist_data else _("No Title")) embed = discord.Embed( description=bold(f"[{playlist_name}]({playlist_url})") if playlist_url else playlist_name, title=_("Playlist Enqueued"), ) embed.set_footer( text=_("Added {num} tracks to the queue.{maxlength_msg}").format( num=track_len, maxlength_msg=maxlength_msg ) ) if not guild_data["shuffle"] and queue_dur > 0: embed.set_footer( text=_( "{time} until start of playlist playback: starts at #{position} in queue" ).format(time=queue_total_duration, position=before_queue_length + 1) ) if not player.current: await player.play() self._play_lock(ctx, False) message = await self._embed_msg(ctx, embed=embed) return tracks or message else: single_track = None # a ytsearch: prefixed item where we only need the first Track returned # this is in the case of [p]play , a single Spotify url/code # or this is a localtrack item try: if len(player.queue) >= 10000: return await self._embed_msg(ctx, title=_("Queue size limit reached.")) single_track = ( tracks if isinstance(tracks, lavalink.rest_api.Track) else tracks[index] if index else tracks[0] ) if seek and seek > 0: single_track.start_timestamp = seek * 1000 if not await is_allowed( ctx.guild, ( f"{single_track.title} {single_track.author} {single_track.uri} " f"{str(audio_dataclasses.Query.process_input(single_track))}" ), ): log.debug(f"Query is not allowed in {ctx.guild} ({ctx.guild.id})") self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("This track is not allowed in this server.") ) elif guild_data["maxlength"] > 0: if track_limit(single_track, guild_data["maxlength"]): player.add(ctx.author, single_track) player.maybe_shuffle() self.bot.dispatch( "red_audio_track_enqueue", player.channel.guild, single_track, ctx.author, ) else: self._play_lock(ctx, False) return await self._embed_msg(ctx, title=_("Track exceeds maximum length.")) else: player.add(ctx.author, single_track) player.maybe_shuffle() self.bot.dispatch( "red_audio_track_enqueue", player.channel.guild, single_track, ctx.author ) except IndexError: self._play_lock(ctx, False) title = _("Nothing found") desc = EmptyEmbed if await ctx.bot.is_owner(ctx.author): desc = _("Please check your console or logs for details.") return await self._embed_msg(ctx, title=title, description=desc) description = get_track_description(single_track) embed = discord.Embed(title=_("Track Enqueued"), description=description) if not guild_data["shuffle"] and queue_dur > 0: embed.set_footer( text=_("{time} until track playback: #{position} in queue").format( time=queue_total_duration, position=before_queue_length + 1 ) ) if not player.current: await player.play() self._play_lock(ctx, False) message = await self._embed_msg(ctx, embed=embed) return single_track or message async def _spotify_playlist( self, ctx: commands.Context, stype: str, query: audio_dataclasses.Query, enqueue: bool = False, ): player = lavalink.get_player(ctx.guild.id) try: embed1 = discord.Embed(title=_("Please wait, finding tracks...")) playlist_msg = await self._embed_msg(ctx, embed=embed1) notifier = Notifier( ctx, playlist_msg, { "spotify": _("Getting track {num}/{total}..."), "youtube": _("Matching track {num}/{total}..."), "lavalink": _("Loading track {num}/{total}..."), "lavalink_time": _("Approximate time remaining: {seconds}"), }, ) track_list = await self.music_cache.spotify_enqueue( ctx, stype, query.id, enqueue=enqueue, player=player, lock=self._play_lock, notifier=notifier, ) except SpotifyFetchError as error: self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Invalid Environment"), description=_(error.message).format(prefix=ctx.prefix), ) except (RuntimeError, aiohttp.ServerDisconnectedError): self._play_lock(ctx, False) error_embed = discord.Embed( title=_("The connection was reset while loading the playlist.") ) await self._embed_msg(ctx, embed=error_embed) return None except Exception as e: self._play_lock(ctx, False) raise e self._play_lock(ctx, False) return track_list async def can_manage_playlist( self, scope: str, playlist: Playlist, ctx: commands.Context, user, guild ): is_owner = await ctx.bot.is_owner(ctx.author) has_perms = False user_to_query = user guild_to_query = guild dj_enabled = None playlist_author = ( guild.get_member(playlist.author) if guild else self.bot.get_user(playlist.author) or user ) is_different_user = len({playlist.author, user_to_query.id, ctx.author.id}) != 1 is_different_guild = True if guild_to_query is None else ctx.guild.id != guild_to_query.id if is_owner: has_perms = True elif playlist.scope == PlaylistScope.USER.value: if not is_different_user: has_perms = True elif playlist.scope == PlaylistScope.GUILD.value: if not is_different_guild: dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if guild.owner_id == ctx.author.id: has_perms = True elif dj_enabled and await self._has_dj_role(ctx, ctx.author): has_perms = True elif await ctx.bot.is_mod(ctx.author): has_perms = True elif not dj_enabled and not is_different_user: has_perms = True if has_perms is False: if hasattr(playlist, "name"): msg = _( "You do not have the permissions to manage {name} (`{id}`) [**{scope}**]." ).format( user=playlist_author, name=playlist.name, id=playlist.id, scope=humanize_scope( playlist.scope, ctx=guild_to_query if playlist.scope == PlaylistScope.GUILD.value else playlist_author if playlist.scope == PlaylistScope.USER.value else None, ), ) elif playlist.scope == PlaylistScope.GUILD.value and ( is_different_guild or dj_enabled ): msg = _( "You do not have the permissions to manage that playlist in {guild}." ).format(guild=guild_to_query) elif ( playlist.scope in [PlaylistScope.GUILD.value, PlaylistScope.USER.value] and is_different_user ): msg = _( "You do not have the permissions to manage playlist owned by {user}." ).format(user=playlist_author) else: msg = _( "You do not have the permissions to manage " "playlists in {scope} scope.".format(scope=humanize_scope(scope, the=True)) ) await self._embed_msg(ctx, title=_("No access to playlist."), description=msg) return False return True async def _get_correct_playlist_id( self, context: commands.Context, matches: MutableMapping, scope: str, author: discord.User, guild: discord.Guild, specified_user: bool = False, ) -> Tuple[Optional[int], str, str]: """ Parameters ---------- context: commands.Context The context in which this is being called. matches: dict A dict of the matches found where key is scope and value is matches. scope:str The custom config scope. A value from :code:`PlaylistScope`. author: discord.User The user. guild: discord.Guild The guild. specified_user: bool Whether or not a user ID was specified via argparse. Returns ------- Tuple[Optional[int], str] Tuple of Playlist ID or None if none found and original user input. Raises ------ `TooManyMatches` When more than 10 matches are found or When multiple matches are found but none is selected. """ correct_scope_matches: List[Playlist] original_input = matches.get("arg") lazy_match = False if scope is None: correct_scope_matches_temp: MutableMapping = matches.get("all") lazy_match = True else: correct_scope_matches_temp: MutableMapping = matches.get(scope) guild_to_query = guild.id user_to_query = author.id correct_scope_matches_user = [] correct_scope_matches_guild = [] correct_scope_matches_global = [] if not correct_scope_matches_temp: return None, original_input, scope or PlaylistScope.GUILD.value if lazy_match or (scope == PlaylistScope.USER.value): correct_scope_matches_user = [ p for p in matches.get(PlaylistScope.USER.value) if user_to_query == p.scope_id ] if lazy_match or (scope == PlaylistScope.GUILD.value and not correct_scope_matches_user): if specified_user: correct_scope_matches_guild = [ p for p in matches.get(PlaylistScope.GUILD.value) if guild_to_query == p.scope_id and p.author == user_to_query ] else: correct_scope_matches_guild = [ p for p in matches.get(PlaylistScope.GUILD.value) if guild_to_query == p.scope_id ] if lazy_match or ( scope == PlaylistScope.GLOBAL.value and not correct_scope_matches_user and not correct_scope_matches_guild ): if specified_user: correct_scope_matches_global = [ p for p in matches.get(PlaylistScope.USGLOBALER.value) if p.author == user_to_query ] else: correct_scope_matches_global = [p for p in matches.get(PlaylistScope.GLOBAL.value)] correct_scope_matches = [ *correct_scope_matches_global, *correct_scope_matches_guild, *correct_scope_matches_user, ] match_count = len(correct_scope_matches) if match_count > 1: correct_scope_matches2 = [ p for p in correct_scope_matches if p.name == str(original_input).strip() ] if correct_scope_matches2: correct_scope_matches = correct_scope_matches2 elif original_input.isnumeric(): arg = int(original_input) correct_scope_matches3 = [p for p in correct_scope_matches if p.id == arg] if correct_scope_matches3: correct_scope_matches = correct_scope_matches3 match_count = len(correct_scope_matches) # We done all the trimming we can with the info available time to ask the user if match_count > 10: if original_input.isnumeric(): arg = int(original_input) correct_scope_matches = [p for p in correct_scope_matches if p.id == arg] if match_count > 10: raise TooManyMatches( _( "{match_count} playlists match {original_input}: " "Please try to be more specific, or use the playlist ID." ).format(match_count=match_count, original_input=original_input) ) elif match_count == 1: return correct_scope_matches[0].id, original_input, correct_scope_matches[0].scope elif match_count == 0: return None, original_input, scope # TODO : Convert this section to a new paged reaction menu when Toby Menus are Merged pos_len = 3 playlists = f"{'#':{pos_len}}\n" number = 0 correct_scope_matches = sorted(correct_scope_matches, key=lambda x: x.name.lower()) for number, playlist in enumerate(correct_scope_matches, 1): author = self.bot.get_user(playlist.author) or playlist.author or _("Unknown") line = _( "{number}." " <{playlist.name}>\n" " - Scope: < {scope} >\n" " - ID: < {playlist.id} >\n" " - Tracks: < {tracks} >\n" " - Author: < {author} >\n\n" ).format( number=number, playlist=playlist, scope=humanize_scope(playlist.scope), tracks=len(playlist.tracks), author=author, ) playlists += line embed = discord.Embed( title=_("{playlists} playlists found, which one would you like?").format( playlists=number ), description=box(playlists, lang="md"), colour=await context.embed_colour(), ) msg = await context.send(embed=embed) avaliable_emojis = ReactionPredicate.NUMBER_EMOJIS[1:] avaliable_emojis.append("🔟") emojis = avaliable_emojis[: len(correct_scope_matches)] emojis.append("\N{CROSS MARK}") start_adding_reactions(msg, emojis) pred = ReactionPredicate.with_emojis(emojis, msg, user=context.author) try: await context.bot.wait_for("reaction_add", check=pred, timeout=60) except asyncio.TimeoutError: with contextlib.suppress(discord.HTTPException): await msg.delete() raise TooManyMatches( _("Too many matches found and you did not select which one you wanted.") ) if emojis[pred.result] == "\N{CROSS MARK}": with contextlib.suppress(discord.HTTPException): await msg.delete() raise TooManyMatches( _("Too many matches found and you did not select which one you wanted.") ) with contextlib.suppress(discord.HTTPException): await msg.delete() return ( correct_scope_matches[pred.result].id, original_input, correct_scope_matches[pred.result].scope, ) @commands.group() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def playlist(self, ctx: commands.Context): """Playlist configuration options. Scope info: ​ ​ ​ ​ **Global**: ​ ​ ​ ​ ​ ​ ​ ​ Visible to all users of this bot. ​ ​ ​ ​ ​ ​ ​ ​ Only editable by bot owner. ​ ​ ​ ​ **Guild**: ​ ​ ​ ​ ​ ​ ​ ​ Visible to all users in this guild. ​ ​ ​ ​ ​ ​ ​ ​ Editable by bot owner, guild owner, guild admins, guild mods, DJ role and playlist creator. ​ ​ ​ ​ **User**: ​ ​ ​ ​ ​ ​ ​ ​ Visible to all bot users, if --author is passed. ​ ​ ​ ​ ​ ​ ​ ​ Editable by bot owner and creator. """ @playlist.command(name="append", usage=" [args]") async def _playlist_append( self, ctx: commands.Context, playlist_matches: PlaylistConverter, query: LazyGreedyConverter, *, scope_data: ScopeParser = None, ): """Add a track URL, playlist link, or quick search to a playlist. The track(s) will be appended to the end of the playlist. **Usage**: ​ ​ ​ ​ `[p]playlist append playlist_name_OR_id track_name_OR_url [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist append MyGuildPlaylist Hello by Adele` ​ ​ ​ ​ `[p]playlist append MyGlobalPlaylist Hello by Adele --scope Global` ​ ​ ​ ​ `[p]playlist append MyGlobalPlaylist Hello by Adele --scope Global --Author Draper#6666` """ if scope_data is None: scope_data = [None, ctx.author, ctx.guild, False] (scope, author, guild, specified_user) = scope_data if not await self._playlist_check(ctx): return try: (playlist_id, playlist_arg, scope) = await self._get_correct_playlist_id( ctx, playlist_matches, scope, author, guild, specified_user ) except TooManyMatches as e: return await self._embed_msg(ctx, title=str(e)) if playlist_id is None: return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Could not match '{arg}' to a playlist").format(arg=playlist_arg), ) try: playlist = await get_playlist(playlist_id, scope, self.bot, guild, author) except RuntimeError: return await self._embed_msg( ctx, title=_("Playlist {id} does not exist in {scope} scope.").format( id=playlist_id, scope=humanize_scope(scope, the=True) ), ) except MissingGuild: return await self._embed_msg( ctx, title=_("Missing Arguments"), description=_("You need to specify the Guild ID for the guild to lookup."), ) if not await self.can_manage_playlist(scope, playlist, ctx, author, guild): return player = lavalink.get_player(ctx.guild.id) to_append = await self._playlist_tracks( ctx, player, audio_dataclasses.Query.process_input(query) ) if isinstance(to_append, discord.Message): return None if not to_append: return await self._embed_msg( ctx, title=_("Could not find a track matching your query.") ) track_list = playlist.tracks current_count = len(track_list) to_append_count = len(to_append) tracks_obj_list = playlist.tracks_obj not_added = 0 if current_count + to_append_count > 10000: to_append = to_append[: 10000 - current_count] not_added = to_append_count - len(to_append) to_append_count = len(to_append) scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) appended = 0 if to_append and to_append_count == 1: to = lavalink.Track(to_append[0]) if to in tracks_obj_list: return await self._embed_msg( ctx, title=_("Skipping track"), description=_( "{track} is already in {playlist} (`{id}`) [**{scope}**]." ).format( track=to.title, playlist=playlist.name, id=playlist.id, scope=scope_name ), footer=_("Playlist limit reached: Could not add track.").format(not_added) if not_added > 0 else None, ) else: appended += 1 if to_append and to_append_count > 1: to_append_temp = [] for t in to_append: to = lavalink.Track(t) if to not in tracks_obj_list: appended += 1 to_append_temp.append(t) to_append = to_append_temp if appended > 0: track_list.extend(to_append) update = {"tracks": track_list, "url": None} await playlist.edit(update) if to_append_count == 1 and appended == 1: track_title = to_append[0]["info"]["title"] return await self._embed_msg( ctx, title=_("Track added"), description=_("{track} appended to {playlist} (`{id}`) [**{scope}**].").format( track=track_title, playlist=playlist.name, id=playlist.id, scope=scope_name ), ) desc = _("{num} tracks appended to {playlist} (`{id}`) [**{scope}**].").format( num=appended, playlist=playlist.name, id=playlist.id, scope=scope_name ) if to_append_count > appended: diff = to_append_count - appended desc += _("\n{existing} {plural} already in the playlist and were skipped.").format( existing=diff, plural=_("tracks are") if diff != 1 else _("track is") ) embed = discord.Embed(title=_("Playlist Modified"), description=desc) await self._embed_msg( ctx, embed=embed, footer=_("Playlist limit reached: Could not add track.").format(not_added) if not_added > 0 else None, ) @commands.cooldown(1, 150, commands.BucketType.member) @playlist.command(name="copy", usage=" [args]", cooldown_after_parsing=True) async def _playlist_copy( self, ctx: commands.Context, playlist_matches: PlaylistConverter, *, scope_data: ComplexScopeParser = None, ): """Copy a playlist from one scope to another. **Usage**: ​ ​ ​ ​ `[p]playlist copy playlist_name_OR_id [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --from-scope ​ ​ ​ ​ ​ ​ ​ ​ --from-author [user] ​ ​ ​ ​ ​ ​ ​ ​ --from-guild [guild] **Only the bot owner can use this** ​ ​ ​ ​ ​ ​ ​ ​ --to-scope ​ ​ ​ ​ ​ ​ ​ ​ --to-author [user] ​ ​ ​ ​ ​ ​ ​ ​ --to-guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist copy MyGuildPlaylist --from-scope Guild --to-scope Global` ​ ​ ​ ​ `[p]playlist copy MyGlobalPlaylist --from-scope Global --to-author Draper#6666 --to-scope User` ​ ​ ​ ​ `[p]playlist copy MyPersonalPlaylist --from-scope user --to-author Draper#6666 --to-scope Guild --to-guild Red - Discord Bot` """ if scope_data is None: scope_data = [ PlaylistScope.GUILD.value, ctx.author, ctx.guild, False, PlaylistScope.GUILD.value, ctx.author, ctx.guild, False, ] ( from_scope, from_author, from_guild, specified_from_user, to_scope, to_author, to_guild, specified_to_user, ) = scope_data try: playlist_id, playlist_arg, scope = await self._get_correct_playlist_id( ctx, playlist_matches, from_scope, from_author, from_guild, specified_from_user ) except TooManyMatches as e: ctx.command.reset_cooldown(ctx) return await self._embed_msg(ctx, title=str(e)) if playlist_id is None: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Could not match '{arg}' to a playlist.").format(arg=playlist_arg), ) temp_playlist = FakePlaylist(to_author.id, to_scope) if not await self.can_manage_playlist(to_scope, temp_playlist, ctx, to_author, to_guild): ctx.command.reset_cooldown(ctx) return try: from_playlist = await get_playlist( playlist_id, from_scope, self.bot, from_guild, from_author.id ) except RuntimeError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Playlist {id} does not exist in {scope} scope.").format( id=playlist_id, scope=humanize_scope(to_scope, the=True) ), ) except MissingGuild: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("You need to specify the Guild ID for the guild to lookup.") ) to_playlist = await create_playlist( ctx, to_scope, from_playlist.name, from_playlist.url, from_playlist.tracks, to_author, to_guild, ) if to_scope == PlaylistScope.GLOBAL.value: to_scope_name = "the Global" elif to_scope == PlaylistScope.USER.value: to_scope_name = to_author else: to_scope_name = to_guild if from_scope == PlaylistScope.GLOBAL.value: from_scope_name = "the Global" elif from_scope == PlaylistScope.USER.value: from_scope_name = from_author else: from_scope_name = from_guild return await self._embed_msg( ctx, title=_("Playlist Copied"), description=_( "Playlist {name} (`{from_id}`) copied from {from_scope} to {to_scope} (`{to_id}`)." ).format( name=from_playlist.name, from_id=from_playlist.id, from_scope=humanize_scope(from_scope, ctx=from_scope_name), to_scope=humanize_scope(to_scope, ctx=to_scope_name), to_id=to_playlist.id, ), ) @playlist.command(name="create", usage=" [args]") async def _playlist_create( self, ctx: commands.Context, playlist_name: str, *, scope_data: ScopeParser = None ): """Create an empty playlist. **Usage**: ​ ​ ​ ​ `[p]playlist create playlist_name [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist create MyGuildPlaylist` ​ ​ ​ ​ `[p]playlist create MyGlobalPlaylist --scope Global` ​ ​ ​ ​ `[p]playlist create MyPersonalPlaylist --scope User` """ if scope_data is None: scope_data = [PlaylistScope.GUILD.value, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data temp_playlist = FakePlaylist(author.id, scope) scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) if not await self.can_manage_playlist(scope, temp_playlist, ctx, author, guild): return playlist_name = playlist_name.split(" ")[0].strip('"')[:32] if playlist_name.isnumeric(): return await self._embed_msg( ctx, title=_("Invalid Playlist Name"), description=_( "Playlist names must be a single word (up to 32 " "characters) and not numbers only." ), ) playlist = await create_playlist(ctx, scope, playlist_name, None, None, author, guild) return await self._embed_msg( ctx, title=_("Playlist Created"), description=_("Empty playlist {name} (`{id}`) [**{scope}**] created.").format( name=playlist.name, id=playlist.id, scope=scope_name ), ) @playlist.command(name="delete", aliases=["del"], usage=" [args]") async def _playlist_delete( self, ctx: commands.Context, playlist_matches: PlaylistConverter, *, scope_data: ScopeParser = None, ): """Delete a saved playlist. **Usage**: ​ ​ ​ ​ `[p]playlist delete playlist_name_OR_id [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist delete MyGuildPlaylist` ​ ​ ​ ​ `[p]playlist delete MyGlobalPlaylist --scope Global` ​ ​ ​ ​ `[p]playlist delete MyPersonalPlaylist --scope User` """ if scope_data is None: scope_data = [None, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data try: playlist_id, playlist_arg, scope = await self._get_correct_playlist_id( ctx, playlist_matches, scope, author, guild, specified_user ) except TooManyMatches as e: return await self._embed_msg(ctx, title=str(e)) if playlist_id is None: return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Could not match '{arg}' to a playlist.").format(arg=playlist_arg), ) try: playlist = await get_playlist(playlist_id, scope, self.bot, guild, author) except RuntimeError: return await self._embed_msg( ctx, title=_("Playlist {id} does not exist in {scope} scope.").format( id=playlist_id, scope=humanize_scope(scope, the=True) ), ) except MissingGuild: return await self._embed_msg( ctx, title=_("You need to specify the Guild ID for the guild to lookup.") ) if not await self.can_manage_playlist(scope, playlist, ctx, author, guild): return scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) await delete_playlist(scope, playlist.id, guild or ctx.guild, author or ctx.author) await self._embed_msg( ctx, title=_("Playlist Deleted"), description=_("{name} (`{id}`) [**{scope}**] playlist deleted.").format( name=playlist.name, id=playlist.id, scope=scope_name ), ) @commands.cooldown(1, 30, commands.BucketType.member) @playlist.command( name="dedupe", usage=" [args]", cooldown_after_parsing=True ) async def _playlist_remdupe( self, ctx: commands.Context, playlist_matches: PlaylistConverter, *, scope_data: ScopeParser = None, ): """Remove duplicate tracks from a saved playlist. **Usage**: ​ ​ ​ ​ `[p]playlist dedupe playlist_name_OR_id [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist dedupe MyGuildPlaylist` ​ ​ ​ ​ `[p]playlist dedupe MyGlobalPlaylist --scope Global` ​ ​ ​ ​ `[p]playlist dedupe MyPersonalPlaylist --scope User` """ async with ctx.typing(): if scope_data is None: scope_data = [None, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data try: playlist_id, playlist_arg, scope = await self._get_correct_playlist_id( ctx, playlist_matches, scope, author, guild, specified_user ) except TooManyMatches as e: ctx.command.reset_cooldown(ctx) return await self._embed_msg(ctx, title=str(e)) scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) if playlist_id is None: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Could not match '{arg}' to a playlist.").format( arg=playlist_arg ), ) try: playlist = await get_playlist(playlist_id, scope, self.bot, guild, author) except RuntimeError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Playlist {id} does not exist in {scope} scope.").format( id=playlist_id, scope=humanize_scope(scope, the=True) ), ) except MissingGuild: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Missing Arguments"), description=_("You need to specify the Guild ID for the guild to lookup."), ) if not await self.can_manage_playlist(scope, playlist, ctx, author, guild): ctx.command.reset_cooldown(ctx) return track_objects = playlist.tracks_obj original_count = len(track_objects) unique_tracks = set() unique_tracks_add = unique_tracks.add track_objects = [ x for x in track_objects if not (x in unique_tracks or unique_tracks_add(x)) ] tracklist = [] for track in track_objects: track_keys = track._info.keys() track_values = track._info.values() track_id = track.track_identifier track_info = {} for k, v in zip(track_keys, track_values): track_info[k] = v keys = ["track", "info"] values = [track_id, track_info] track_obj = {} for key, value in zip(keys, values): track_obj[key] = value tracklist.append(track_obj) final_count = len(tracklist) if original_count - final_count != 0: await self._embed_msg( ctx, title=_("Playlist Modified"), description=_( "Removed {track_diff} duplicated " "tracks from {name} (`{id}`) [**{scope}**] playlist." ).format( name=playlist.name, id=playlist.id, track_diff=original_count - final_count, scope=scope_name, ), ) else: await self._embed_msg( ctx, title=_("Playlist Has Not Been Modified"), description=_( "{name} (`{id}`) [**{scope}**] playlist has no duplicate tracks." ).format(name=playlist.name, id=playlist.id, scope=scope_name), ) @checks.is_owner() @playlist.command( name="download", usage=" [v2=False] [args]", cooldown_after_parsing=True, ) @commands.bot_has_permissions(attach_files=True) @commands.cooldown(1, 30, commands.BucketType.guild) async def _playlist_download( self, ctx: commands.Context, playlist_matches: PlaylistConverter, v2: Optional[bool] = False, *, scope_data: ScopeParser = None, ): """Download a copy of a playlist. These files can be used with the `[p]playlist upload` command. Red v2-compatible playlists can be generated by passing True for the v2 variable. **Usage**: ​ ​ ​ ​ `[p]playlist download playlist_name_OR_id [v2=True_OR_False] [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist download MyGuildPlaylist True` ​ ​ ​ ​ `[p]playlist download MyGlobalPlaylist False --scope Global` ​ ​ ​ ​ `[p]playlist download MyPersonalPlaylist --scope User` """ if scope_data is None: scope_data = [None, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data try: playlist_id, playlist_arg, scope = await self._get_correct_playlist_id( ctx, playlist_matches, scope, author, guild, specified_user ) except TooManyMatches as e: ctx.command.reset_cooldown(ctx) return await self._embed_msg(ctx, title=str(e)) if playlist_id is None: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Could not match '{arg}' to a playlist.").format(arg=playlist_arg), ) try: playlist = await get_playlist(playlist_id, scope, self.bot, guild, author) except RuntimeError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Playlist {id} does not exist in {scope} scope.").format( id=playlist_id, scope=humanize_scope(scope, the=True) ), ) except MissingGuild: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Missing Arguments"), description=_("You need to specify the Guild ID for the guild to lookup."), ) schema = 2 version = "v3" if v2 is False else "v2" if not playlist.tracks: ctx.command.reset_cooldown(ctx) return await self._embed_msg(ctx, title=_("That playlist has no tracks.")) if version == "v2": v2_valid_urls = ["https://www.youtube.com/watch?v=", "https://soundcloud.com/"] song_list = [] for track in playlist.tracks: if track["info"]["uri"].startswith(tuple(v2_valid_urls)): song_list.append(track["info"]["uri"]) await asyncio.sleep(0) playlist_data = { "author": playlist.author, "link": playlist.url, "playlist": song_list, "name": playlist.name, } file_name = playlist.name else: # TODO: Keep new playlists backwards compatible, Remove me in a few releases playlist_data = playlist.to_json() playlist_songs_backwards_compatible = [ track["info"]["uri"] for track in playlist.tracks ] playlist_data["playlist"] = playlist_songs_backwards_compatible playlist_data["link"] = playlist.url file_name = playlist.id playlist_data.update({"schema": schema, "version": version}) playlist_data = json.dumps(playlist_data).encode("utf-8") to_write = BytesIO() to_write.write(playlist_data) to_write.seek(0) if to_write.getbuffer().nbytes > ctx.guild.filesize_limit - 10000: datapath = cog_data_path(raw_name="Audio") temp_file = datapath / f"{file_name}.txt" temp_tar = datapath / f"{file_name}.tar.gz" with temp_file.open("wb") as playlist_file: playlist_file.write(to_write.read()) with tarfile.open(str(temp_tar), "w:gz") as tar: tar.add( str(temp_file), arcname=str(temp_file.relative_to(datapath)), recursive=False ) try: if os.path.getsize(str(temp_tar)) > ctx.guild.filesize_limit - 10000: await ctx.send(_("This playlist is too large to be send in this server.")) else: await ctx.send( content=_("Playlist is too large, here is the compressed version."), file=discord.File(str(temp_tar)), ) except Exception: pass temp_file.unlink() temp_tar.unlink() else: await ctx.send(file=discord.File(to_write, filename=f"{file_name}.txt")) to_write.close() @commands.cooldown(1, 10, commands.BucketType.member) @playlist.command( name="info", usage=" [args]", cooldown_after_parsing=True ) async def _playlist_info( self, ctx: commands.Context, playlist_matches: PlaylistConverter, *, scope_data: ScopeParser = None, ): """Retrieve information from a saved playlist. **Usage**: ​ ​ ​ ​ `[p]playlist info playlist_name_OR_id [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist info MyGuildPlaylist` ​ ​ ​ ​ `[p]playlist info MyGlobalPlaylist --scope Global` ​ ​ ​ ​ `[p]playlist info MyPersonalPlaylist --scope User` """ if scope_data is None: scope_data = [None, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data try: playlist_id, playlist_arg, scope = await self._get_correct_playlist_id( ctx, playlist_matches, scope, author, guild, specified_user ) except TooManyMatches as e: ctx.command.reset_cooldown(ctx) return await self._embed_msg(ctx, title=str(e)) scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) if playlist_id is None: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Could not match '{arg}' to a playlist.").format(arg=playlist_arg), ) try: playlist = await get_playlist(playlist_id, scope, self.bot, guild, author) except RuntimeError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Playlist {id} does not exist in {scope} scope.").format( id=playlist_id, scope=humanize_scope(scope, the=True) ), ) except MissingGuild: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Missing Arguments"), description=_("You need to specify the Guild ID for the guild to lookup."), ) track_len = len(playlist.tracks) msg = "​" track_idx = 0 if track_len > 0: spaces = "\N{EN SPACE}" * (len(str(len(playlist.tracks))) + 2) for i, track in enumerate(playlist.tracks, start=1): if i % 500 == 0: # TODO: Improve when Toby menu's are merged await asyncio.sleep(0.1) track_idx = track_idx + 1 query = audio_dataclasses.Query.process_input(track["info"]["uri"]) if query.is_local: if track["info"]["title"] != "Unknown title": msg += "`{}.` **{} - {}**\n{}{}\n".format( track_idx, track["info"]["author"], track["info"]["title"], spaces, query.to_string_user(), ) else: msg += "`{}.` {}\n".format(track_idx, query.to_string_user()) else: msg += "`{}.` **[{}]({})**\n".format( track_idx, track["info"]["title"], track["info"]["uri"] ) await asyncio.sleep(0) else: msg = "No tracks." if not playlist.url: embed_title = _("Playlist info for {playlist_name} (`{id}`) [**{scope}**]:\n").format( playlist_name=playlist.name, id=playlist.id, scope=scope_name ) else: embed_title = _( "Playlist info for {playlist_name} (`{id}`) [**{scope}**]:\nURL: {url}" ).format( playlist_name=playlist.name, url=playlist.url, id=playlist.id, scope=scope_name ) page_list = [] pages = list(pagify(msg, delims=["\n"], page_length=2000)) total_pages = len(pages) for numb, page in enumerate(pages, start=1): embed = discord.Embed( colour=await ctx.embed_colour(), title=embed_title, description=page ) author_obj = self.bot.get_user(playlist.author) or playlist.author or _("Unknown") embed.set_footer( text=_("Page {page}/{pages} | Author: {author_name} | {num} track(s)").format( author_name=author_obj, num=track_len, pages=total_pages, page=numb ) ) page_list.append(embed) await menu(ctx, page_list, DEFAULT_CONTROLS) @commands.cooldown(1, 15, commands.BucketType.guild) @playlist.command(name="list", usage="[args]", cooldown_after_parsing=True) @commands.bot_has_permissions(add_reactions=True) async def _playlist_list(self, ctx: commands.Context, *, scope_data: ScopeParser = None): """List saved playlists. **Usage**: ​ ​ ​ ​ `[p]playlist list [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist list` ​ ​ ​ ​ `[p]playlist list --scope Global` ​ ​ ​ ​ `[p]playlist list --scope User` """ if scope_data is None: scope_data = [PlaylistScope.GUILD.value, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data try: playlists = await get_all_playlist(scope, self.bot, guild, author, specified_user) except MissingGuild: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Missing Arguments"), description=_("You need to specify the Guild ID for the guild to lookup."), ) if scope == PlaylistScope.GUILD.value: name = f"{guild.name}" elif scope == PlaylistScope.USER.value: name = f"{author}" else: name = "Global" if not playlists and specified_user: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("No saved playlists for {scope} created by {author}.").format( scope=name, author=author ), ) elif not playlists: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("No saved playlists for {scope}.").format(scope=name), ) playlist_list = [] space = "\N{EN SPACE}" for playlist in playlists: playlist_list.append( ("\n" + space * 4).join( ( bold(playlist.name), _("ID: {id}").format(id=playlist.id), _("Tracks: {num}").format(num=len(playlist.tracks)), _("Author: {name}\n").format( name=self.bot.get_user(playlist.author) or playlist.author or _("Unknown") ), ) ) ) await asyncio.sleep(0) abc_names = sorted(playlist_list, key=str.lower) len_playlist_list_pages = math.ceil(len(abc_names) / 5) playlist_embeds = [] for page_num in range(1, len_playlist_list_pages + 1): embed = await self._build_playlist_list_page(ctx, page_num, abc_names, name) playlist_embeds.append(embed) await asyncio.sleep(0) await menu(ctx, playlist_embeds, DEFAULT_CONTROLS) @staticmethod async def _build_playlist_list_page(ctx: commands.Context, page_num, abc_names, scope): plist_num_pages = math.ceil(len(abc_names) / 5) plist_idx_start = (page_num - 1) * 5 plist_idx_end = plist_idx_start + 5 plist = "" for i, playlist_info in enumerate( abc_names[plist_idx_start:plist_idx_end], start=plist_idx_start ): item_idx = i + 1 plist += "`{}.` {}".format(item_idx, playlist_info) await asyncio.sleep(0) embed = discord.Embed( colour=await ctx.embed_colour(), title=_("Playlists for {scope}:").format(scope=scope), description=plist, ) embed.set_footer( text=_("Page {page_num}/{total_pages} | {num} playlists.").format( page_num=page_num, total_pages=plist_num_pages, num=len(abc_names) ) ) return embed @playlist.command(name="queue", usage=" [args]", cooldown_after_parsing=True) @commands.cooldown(1, 300, commands.BucketType.member) async def _playlist_queue( self, ctx: commands.Context, playlist_name: str, *, scope_data: ScopeParser = None ): """Save the queue to a playlist. **Usage**: ​ ​ ​ ​ `[p]playlist queue playlist_name [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist queue MyGuildPlaylist` ​ ​ ​ ​ `[p]playlist queue MyGlobalPlaylist --scope Global` ​ ​ ​ ​ `[p]playlist queue MyPersonalPlaylist --scope User` """ async with ctx.typing(): if scope_data is None: scope_data = [PlaylistScope.GUILD.value, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) temp_playlist = FakePlaylist(author.id, scope) if not await self.can_manage_playlist(scope, temp_playlist, ctx, author, guild): ctx.command.reset_cooldown(ctx) return playlist_name = playlist_name.split(" ")[0].strip('"')[:32] if playlist_name.isnumeric(): ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Invalid Playlist Name"), description=_( "Playlist names must be a single word " "(up to 32 characters) and not numbers only." ), ) if not self._player_check(ctx): ctx.command.reset_cooldown(ctx) return await self._embed_msg(ctx, title=_("Nothing playing.")) player = lavalink.get_player(ctx.guild.id) if not player.queue: ctx.command.reset_cooldown(ctx) return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) tracklist = [] np_song = track_creator(player, "np") tracklist.append(np_song) queue_length = len(player.queue) to_add = player.queue not_added = 0 if queue_length > 10000: to_add = player.queue[:10000] not_added = queue_length - 10000 for i, track in enumerate(to_add, start=1): if i % 500 == 0: # TODO: Improve when Toby menu's are merged await asyncio.sleep(0.02) queue_idx = player.queue.index(track) track_obj = track_creator(player, queue_idx) tracklist.append(track_obj) playlist = await create_playlist( ctx, scope, playlist_name, None, tracklist, author, guild ) await asyncio.sleep(0) await self._embed_msg( ctx, title=_("Playlist Created"), description=_( "Playlist {name} (`{id}`) [**{scope}**] " "saved from current queue: {num} tracks added." ).format( name=playlist.name, num=len(playlist.tracks), id=playlist.id, scope=scope_name ), footer=_("Playlist limit reached: Could not add {} tracks.").format(not_added) if not_added > 0 else None, ) @playlist.command(name="remove", usage=" [args]") async def _playlist_remove( self, ctx: commands.Context, playlist_matches: PlaylistConverter, url: str, *, scope_data: ScopeParser = None, ): """Remove a track from a playlist by url. **Usage**: ​ ​ ​ ​ `[p]playlist remove playlist_name_OR_id url [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist remove MyGuildPlaylist https://www.youtube.com/watch?v=MN3x-kAbgFU` ​ ​ ​ ​ `[p]playlist remove MyGlobalPlaylist https://www.youtube.com/watch?v=MN3x-kAbgFU --scope Global` ​ ​ ​ ​ `[p]playlist remove MyPersonalPlaylist https://www.youtube.com/watch?v=MN3x-kAbgFU --scope User` """ if scope_data is None: scope_data = [None, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data try: playlist_id, playlist_arg, scope = await self._get_correct_playlist_id( ctx, playlist_matches, scope, author, guild, specified_user ) except TooManyMatches as e: return await self._embed_msg(ctx, title=str(e)) scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) if playlist_id is None: return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Could not match '{arg}' to a playlist.").format(arg=playlist_arg), ) try: playlist = await get_playlist(playlist_id, scope, self.bot, guild, author) except RuntimeError: return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Playlist {id} does not exist in {scope} scope.").format( id=playlist_id, scope=humanize_scope(scope, the=True) ), ) except MissingGuild: return await self._embed_msg( ctx, title=_("Missing Arguments"), description=_("You need to specify the Guild ID for the guild to lookup."), ) if not await self.can_manage_playlist(scope, playlist, ctx, author, guild): return track_list = playlist.tracks clean_list = [track for track in track_list if url != track["info"]["uri"]] if len(track_list) == len(clean_list): return await self._embed_msg(ctx, title=_("URL not in playlist.")) del_count = len(track_list) - len(clean_list) if not clean_list: await delete_playlist( scope=playlist.scope, playlist_id=playlist.id, guild=guild, author=playlist.author ) return await self._embed_msg(ctx, title=_("No tracks left, removing playlist.")) update = {"tracks": clean_list, "url": None} await playlist.edit(update) if del_count > 1: await self._embed_msg( ctx, title=_("Playlist Modified"), description=_( "{num} entries have been removed " "from the playlist {playlist_name} (`{id}`) [**{scope}**]." ).format( num=del_count, playlist_name=playlist.name, id=playlist.id, scope=scope_name ), ) else: await self._embed_msg( ctx, title=_("Playlist Modified"), description=_( "The track has been removed from the playlist: " "{playlist_name} (`{id}`) [**{scope}**]." ).format(playlist_name=playlist.name, id=playlist.id, scope=scope_name), ) @playlist.command(name="save", usage=" [args]", cooldown_after_parsing=True) @commands.cooldown(1, 60, commands.BucketType.member) async def _playlist_save( self, ctx: commands.Context, playlist_name: str, playlist_url: str, *, scope_data: ScopeParser = None, ): """Save a playlist from a url. **Usage**: ​ ​ ​ ​ `[p]playlist save name url [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist save MyGuildPlaylist https://www.youtube.com/playlist?list=PLx0sYbCqOb8Q_CLZC2BdBSKEEB59BOPUM` ​ ​ ​ ​ `[p]playlist save MyGlobalPlaylist https://www.youtube.com/playlist?list=PLx0sYbCqOb8Q_CLZC2BdBSKEEB59BOPUM --scope Global` ​ ​ ​ ​ `[p]playlist save MyPersonalPlaylist https://open.spotify.com/playlist/1RyeIbyFeIJVnNzlGr5KkR --scope User` """ if scope_data is None: scope_data = [PlaylistScope.GUILD.value, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) temp_playlist = FakePlaylist(author.id, scope) if not await self.can_manage_playlist(scope, temp_playlist, ctx, author, guild): return ctx.command.reset_cooldown(ctx) playlist_name = playlist_name.split(" ")[0].strip('"')[:32] if playlist_name.isnumeric(): ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Invalid Playlist Name"), description=_( "Playlist names must be a single word (up to 32 " "characters) and not numbers only." ), ) if not await self._playlist_check(ctx): ctx.command.reset_cooldown(ctx) return player = lavalink.get_player(ctx.guild.id) tracklist = await self._playlist_tracks( ctx, player, audio_dataclasses.Query.process_input(playlist_url) ) if isinstance(tracklist, discord.Message): return None if tracklist is not None: playlist_length = len(tracklist) not_added = 0 if playlist_length > 10000: tracklist = tracklist[:10000] not_added = playlist_length - 10000 playlist = await create_playlist( ctx, scope, playlist_name, playlist_url, tracklist, author, guild ) return await self._embed_msg( ctx, title=_("Playlist Created"), description=_( "Playlist {name} (`{id}`) [**{scope}**] saved: {num} tracks added." ).format(name=playlist.name, num=len(tracklist), id=playlist.id, scope=scope_name), footer=_("Playlist limit reached: Could not add {} tracks.").format(not_added) if not_added > 0 else None, ) @commands.cooldown(1, 30, commands.BucketType.member) @playlist.command( name="start", aliases=["play"], usage=" [args]", cooldown_after_parsing=True, ) async def _playlist_start( self, ctx: commands.Context, playlist_matches: PlaylistConverter, *, scope_data: ScopeParser = None, ): """Load a playlist into the queue. **Usage**: ​ ​ ​ ​` [p]playlist start playlist_name_OR_id [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist start MyGuildPlaylist` ​ ​ ​ ​ `[p]playlist start MyGlobalPlaylist --scope Global` ​ ​ ​ ​ `[p]playlist start MyPersonalPlaylist --scope User` """ if scope_data is None: scope_data = [None, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author): ctx.command.reset_cooldown(ctx) await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("You need the DJ role to start playing playlists."), ) return False try: playlist_id, playlist_arg, scope = await self._get_correct_playlist_id( ctx, playlist_matches, scope, author, guild, specified_user ) except TooManyMatches as e: ctx.command.reset_cooldown(ctx) return await self._embed_msg(ctx, title=str(e)) if playlist_id is None: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Could not match '{arg}' to a playlist").format(arg=playlist_arg), ) if not await self._playlist_check(ctx): ctx.command.reset_cooldown(ctx) return jukebox_price = await self.config.guild(ctx.guild).jukebox_price() if not await self._currency_check(ctx, jukebox_price): ctx.command.reset_cooldown(ctx) return maxlength = await self.config.guild(ctx.guild).maxlength() author_obj = self.bot.get_user(ctx.author.id) track_len = 0 playlist = None try: playlist = await get_playlist(playlist_id, scope, self.bot, guild, author) player = lavalink.get_player(ctx.guild.id) tracks = playlist.tracks_obj empty_queue = not player.queue for i, track in enumerate(tracks, start=1): if i % 500 == 0: # TODO: Improve when Toby menu's are merged await asyncio.sleep(0.02) if len(player.queue) >= 10000: continue if not await is_allowed( ctx.guild, ( f"{track.title} {track.author} {track.uri} " f"{str(audio_dataclasses.Query.process_input(track))}" ), ): log.debug(f"Query is not allowed in {ctx.guild} ({ctx.guild.id})") continue query = audio_dataclasses.Query.process_input(track.uri) if query.is_local: local_path = audio_dataclasses.LocalPath(track.uri) if not await self._localtracks_check(ctx): pass if not local_path.exists() and not local_path.is_file(): continue if maxlength > 0: if not track_limit(track.length, maxlength): continue player.add(author_obj, track) self.bot.dispatch( "red_audio_track_enqueue", player.channel.guild, track, ctx.author ) track_len += 1 await asyncio.sleep(0) player.maybe_shuffle(0 if empty_queue else 1) if len(tracks) > track_len: maxlength_msg = " {bad_tracks} tracks cannot be queued.".format( bad_tracks=(len(tracks) - track_len) ) else: maxlength_msg = "" if scope == PlaylistScope.GUILD.value: scope_name = f"{guild.name}" elif scope == PlaylistScope.USER.value: scope_name = f"{author}" else: scope_name = "Global" embed = discord.Embed( title=_("Playlist Enqueued"), description=_( "{name} - (`{id}`) [**{scope}**]\nAdded {num} " "tracks to the queue.{maxlength_msg}" ).format( num=track_len, maxlength_msg=maxlength_msg, name=playlist.name, id=playlist.id, scope=scope_name, ), ) await self._embed_msg(ctx, embed=embed) if not player.current: await player.play() return except RuntimeError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Playlist {id} does not exist in {scope} scope.").format( id=playlist_id, scope=humanize_scope(scope, the=True) ), ) except MissingGuild: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Missing Arguments"), description=_("You need to specify the Guild ID for the guild to lookup."), ) except TypeError: if playlist: return await ctx.invoke(self.play, query=playlist.url) @commands.cooldown(1, 60, commands.BucketType.member) @playlist.command( name="update", usage=" [args]", cooldown_after_parsing=True ) async def _playlist_update( self, ctx: commands.Context, playlist_matches: PlaylistConverter, *, scope_data: ScopeParser = None, ): """Updates all tracks in a playlist. **Usage**: ​ ​ ​ ​ `[p]playlist update playlist_name_OR_id [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist update MyGuildPlaylist` ​ ​ ​ ​ `[p]playlist update MyGlobalPlaylist --scope Global` ​ ​ ​ ​ `[p]playlist update MyPersonalPlaylist --scope User` """ if scope_data is None: scope_data = [None, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data try: playlist_id, playlist_arg, scope = await self._get_correct_playlist_id( ctx, playlist_matches, scope, author, guild, specified_user ) except TooManyMatches as e: ctx.command.reset_cooldown(ctx) return await self._embed_msg(ctx, title=str(e)) if playlist_id is None: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Could not match '{arg}' to a playlist.").format(arg=playlist_arg), ) if not await self._playlist_check(ctx): ctx.command.reset_cooldown(ctx) return try: playlist = await get_playlist(playlist_id, scope, self.bot, guild, author) if not await self.can_manage_playlist(scope, playlist, ctx, author, guild): return if playlist.url: player = lavalink.get_player(ctx.guild.id) added, removed, playlist = await self._maybe_update_playlist(ctx, player, playlist) else: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Invalid Playlist"), description=_("Custom playlists cannot be updated."), ) except RuntimeError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Playlist {id} does not exist in {scope} scope.").format( id=playlist_id, scope=humanize_scope(scope, the=True) ), ) except MissingGuild: return await self._embed_msg( ctx, title=_("Missing Arguments"), description=_("You need to specify the Guild ID for the guild to lookup."), ) else: scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) if added or removed: _colour = await ctx.embed_colour() removed_embeds = [] added_embeds = [] total_added = len(added) total_removed = len(removed) total_pages = math.ceil(total_removed / 10) + math.ceil(total_added / 10) page_count = 0 if removed: removed_text = "" for i, track in enumerate(removed, 1): if len(track.title) > 40: track_title = str(track.title).replace("[", "") track_title = "{}...".format((track_title[:40]).rstrip(" ")) else: track_title = track.title removed_text += f"`{i}.` **[{track_title}]({track.uri})**\n" if i % 10 == 0 or i == total_removed: page_count += 1 embed = discord.Embed( title=_("Tracks removed"), colour=_colour, description=removed_text ) text = _("Page {page_num}/{total_pages}").format( page_num=page_count, total_pages=total_pages ) embed.set_footer(text=text) removed_embeds.append(embed) removed_text = "" if added: added_text = "" for i, track in enumerate(added, 1): if len(track.title) > 40: track_title = str(track.title).replace("[", "") track_title = "{}...".format((track_title[:40]).rstrip(" ")) else: track_title = track.title added_text += f"`{i}.` **[{track_title}]({track.uri})**\n" if i % 10 == 0 or i == total_added: page_count += 1 embed = discord.Embed( title=_("Tracks added"), colour=_colour, description=added_text ) text = _("Page {page_num}/{total_pages}").format( page_num=page_count, total_pages=total_pages ) embed.set_footer(text=text) added_embeds.append(embed) added_text = "" embeds = removed_embeds + added_embeds await menu(ctx, embeds, DEFAULT_CONTROLS) else: return await self._embed_msg( ctx, title=_("Playlist Has Not Been Modified"), description=_("No changes for {name} (`{id}`) [**{scope}**].").format( id=playlist.id, name=playlist.name, scope=scope_name ), ) @checks.is_owner() @playlist.command(name="upload", usage="[args]") async def _playlist_upload(self, ctx: commands.Context, *, scope_data: ScopeParser = None): """Uploads a playlist file as a playlist for the bot. V2 and old V3 playlist will be slow. V3 Playlist made with `[p]playlist download` will load a lot faster. **Usage**: ​ ​ ​ ​ `[p]playlist upload [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist upload` ​ ​ ​ ​ `[p]playlist upload --scope Global` ​ ​ ​ ​ `[p]playlist upload --scope User` """ if scope_data is None: scope_data = [PlaylistScope.GUILD.value, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data temp_playlist = FakePlaylist(author.id, scope) if not await self.can_manage_playlist(scope, temp_playlist, ctx, author, guild): return if not await self._playlist_check(ctx): return player = lavalink.get_player(ctx.guild.id) await self._embed_msg( ctx, title=_( "Please upload the playlist file. Any other message will cancel this operation." ), ) try: file_message = await ctx.bot.wait_for( "message", timeout=30.0, check=MessagePredicate.same_context(ctx) ) except asyncio.TimeoutError: return await self._embed_msg(ctx, title=_("No file detected, try again later.")) try: file_url = file_message.attachments[0].url except IndexError: return await self._embed_msg(ctx, title=_("Upload cancelled.")) file_suffix = file_url.rsplit(".", 1)[1] if file_suffix != "txt": return await self._embed_msg(ctx, title=_("Only Red playlist files can be uploaded.")) try: async with self.session.request("GET", file_url) as r: uploaded_playlist = await r.json(content_type="text/plain", encoding="utf-8") except UnicodeDecodeError: return await self._embed_msg(ctx, title=_("Not a valid playlist file.")) new_schema = uploaded_playlist.get("schema", 1) >= 2 version = uploaded_playlist.get("version", "v2") if new_schema and version == "v3": uploaded_playlist_url = uploaded_playlist.get("playlist_url", None) track_list = uploaded_playlist.get("tracks", []) else: uploaded_playlist_url = uploaded_playlist.get("link", None) track_list = uploaded_playlist.get("playlist", []) if len(track_list) > 10000: return await self._embed_msg(ctx, title=_("This playlist is too large.")) uploaded_playlist_name = uploaded_playlist.get( "name", (file_url.split("/")[6]).split(".")[0] ) if ( not uploaded_playlist_url or not match_yt_playlist(uploaded_playlist_url) or not ( await self.music_cache.lavalink_query( ctx, player, audio_dataclasses.Query.process_input(uploaded_playlist_url) ) )[0].tracks ): if version == "v3": return await self._load_v3_playlist( ctx, scope, uploaded_playlist_name, uploaded_playlist_url, track_list, author, guild, ) return await self._load_v2_playlist( ctx, track_list, player, uploaded_playlist_url, uploaded_playlist_name, scope, author, guild, ) return await ctx.invoke( self._playlist_save, playlist_name=uploaded_playlist_name, playlist_url=uploaded_playlist_url, scope_data=(scope, author, guild, specified_user), ) @commands.cooldown(1, 60, commands.BucketType.member) @playlist.command( name="rename", usage=" [args]", cooldown_after_parsing=True ) async def _playlist_rename( self, ctx: commands.Context, playlist_matches: PlaylistConverter, new_name: str, *, scope_data: ScopeParser = None, ): """Rename an existing playlist. **Usage**: ​ ​ ​ ​ `[p]playlist rename playlist_name_OR_id new_name [args]` **Args**: ​ ​ ​ ​ The following are all optional: ​ ​ ​ ​ ​ ​ ​ ​ --scope ​ ​ ​ ​ ​ ​ ​ ​ --author [user] ​ ​ ​ ​ ​ ​ ​ ​ --guild [guild] **Only the bot owner can use this** **Scope** is one of the following: ​ ​ ​ ​ Global ​ ​ ​ ​ Guild ​ ​ ​ ​ User **Author** can be one of the following: ​ ​ ​ ​ User ID ​ ​ ​ ​ User Mention ​ ​ ​ ​ User Name#123 **Guild** can be one of the following: ​ ​ ​ ​ Guild ID ​ ​ ​ ​ Exact guild name Example use: ​ ​ ​ ​ `[p]playlist rename MyGuildPlaylist RenamedGuildPlaylist` ​ ​ ​ ​ `[p]playlist rename MyGlobalPlaylist RenamedGlobalPlaylist --scope Global` ​ ​ ​ ​ `[p]playlist rename MyPersonalPlaylist RenamedPersonalPlaylist --scope User` """ if scope_data is None: scope_data = [None, ctx.author, ctx.guild, False] scope, author, guild, specified_user = scope_data new_name = new_name.split(" ")[0].strip('"')[:32] if new_name.isnumeric(): ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Invalid Playlist Name"), description=_( "Playlist names must be a single word (up to 32 " "characters) and not numbers only." ), ) try: playlist_id, playlist_arg, scope = await self._get_correct_playlist_id( ctx, playlist_matches, scope, author, guild, specified_user ) except TooManyMatches as e: ctx.command.reset_cooldown(ctx) return await self._embed_msg(ctx, title=str(e)) if playlist_id is None: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Could not match '{arg}' to a playlist.").format(arg=playlist_arg), ) try: playlist = await get_playlist(playlist_id, scope, self.bot, guild, author) except RuntimeError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Playlist Not Found"), description=_("Playlist does not exist in {scope} scope.").format( scope=humanize_scope(scope, the=True) ), ) except MissingGuild: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Missing Arguments"), description=_("You need to specify the Guild ID for the guild to lookup."), ) if not await self.can_manage_playlist(scope, playlist, ctx, author, guild): ctx.command.reset_cooldown(ctx) return scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) old_name = playlist.name update = {"name": new_name} await playlist.edit(update) msg = _("'{old}' playlist has been renamed to '{new}' (`{id}`) [**{scope}**]").format( old=bold(old_name), new=bold(playlist.name), id=playlist.id, scope=scope_name ) await self._embed_msg(ctx, title=_("Playlist Modified"), description=msg) async def _load_v3_playlist( self, ctx: commands.Context, scope: str, uploaded_playlist_name: str, uploaded_playlist_url: str, track_list, author: Union[discord.User, discord.Member], guild: Union[discord.Guild], ): embed1 = discord.Embed(title=_("Please wait, adding tracks...")) playlist_msg = await self._embed_msg(ctx, embed=embed1) track_count = len(track_list) uploaded_track_count = len(track_list) await asyncio.sleep(1) embed2 = discord.Embed( colour=await ctx.embed_colour(), title=_("Loading track {num}/{total}...").format( num=track_count, total=uploaded_track_count ), ) await playlist_msg.edit(embed=embed2) playlist = await create_playlist( ctx, scope, uploaded_playlist_name, uploaded_playlist_url, track_list, author, guild ) scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) if not track_count: msg = _("Empty playlist {name} (`{id}`) [**{scope}**] created.").format( name=playlist.name, id=playlist.id, scope=scope_name ) elif uploaded_track_count != track_count: bad_tracks = uploaded_track_count - track_count msg = _( "Added {num} tracks from the {playlist_name} playlist. {num_bad} track(s) " "could not be loaded." ).format(num=track_count, playlist_name=playlist.name, num_bad=bad_tracks) else: msg = _("Added {num} tracks from the {playlist_name} playlist.").format( num=track_count, playlist_name=playlist.name ) embed3 = discord.Embed( colour=await ctx.embed_colour(), title=_("Playlist Saved"), description=msg ) await playlist_msg.edit(embed=embed3) database_entries = [] time_now = int(datetime.datetime.now(datetime.timezone.utc).timestamp()) for t in track_list: uri = t.get("info", {}).get("uri") if uri: t = {"loadType": "V2_COMPAT", "tracks": [t], "query": uri} data = json.dumps(t) if all(k in data for k in ["loadType", "playlistInfo", "isSeekable", "isStream"]): database_entries.append( { "query": uri, "data": data, "last_updated": time_now, "last_fetched": time_now, } ) if database_entries: await self.music_cache.database.insert("lavalink", database_entries) async def _load_v2_playlist( self, ctx: commands.Context, uploaded_track_list, player: lavalink.player_manager.Player, playlist_url: str, uploaded_playlist_name: str, scope: str, author: Union[discord.User, discord.Member], guild: Union[discord.Guild], ): track_list = [] track_count = 0 successful_count = 0 uploaded_track_count = len(uploaded_track_list) embed1 = discord.Embed(title=_("Please wait, adding tracks...")) playlist_msg = await self._embed_msg(ctx, embed=embed1) notifier = Notifier(ctx, playlist_msg, {"playlist": _("Loading track {num}/{total}...")}) for song_url in uploaded_track_list: track_count += 1 try: try: result, called_api = await self.music_cache.lavalink_query( ctx, player, audio_dataclasses.Query.process_input(song_url) ) except TrackEnqueueError: self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Unable to Get Track"), description=_( "I'm unable get a track from Lavalink at the moment, try again in a few " "minutes." ), ) track = result.tracks except Exception: continue try: track_obj = track_creator(player, other_track=track[0]) track_list.append(track_obj) successful_count += 1 except Exception: continue if (track_count % 2 == 0) or (track_count == len(uploaded_track_list)): await notifier.notify_user( current=track_count, total=len(uploaded_track_list), key="playlist" ) playlist = await create_playlist( ctx, scope, uploaded_playlist_name, playlist_url, track_list, author, guild ) scope_name = humanize_scope( scope, ctx=guild if scope == PlaylistScope.GUILD.value else author ) if not successful_count: msg = _("Empty playlist {name} (`{id}`) [**{scope}**] created.").format( name=playlist.name, id=playlist.id, scope=scope_name ) elif uploaded_track_count != successful_count: bad_tracks = uploaded_track_count - successful_count msg = _( "Added {num} tracks from the {playlist_name} playlist. {num_bad} track(s) " "could not be loaded." ).format(num=successful_count, playlist_name=playlist.name, num_bad=bad_tracks) else: msg = _("Added {num} tracks from the {playlist_name} playlist.").format( num=successful_count, playlist_name=playlist.name ) embed3 = discord.Embed( colour=await ctx.embed_colour(), title=_("Playlist Saved"), description=msg ) await playlist_msg.edit(embed=embed3) async def _maybe_update_playlist( self, ctx: commands.Context, player: lavalink.player_manager.Player, playlist: Playlist ) -> Tuple[List[lavalink.Track], List[lavalink.Track], Playlist]: if playlist.url is None: return [], [], playlist results = {} updated_tracks = await self._playlist_tracks( ctx, player, audio_dataclasses.Query.process_input(playlist.url) ) if isinstance(updated_tracks, discord.Message): return [], [], playlist if not updated_tracks: # No Tracks available on url Lets set it to none to avoid repeated calls here results["url"] = None if updated_tracks: # Tracks have been updated results["tracks"] = updated_tracks old_tracks = playlist.tracks_obj new_tracks = [lavalink.Track(data=track) for track in updated_tracks] removed = list(set(old_tracks) - set(new_tracks)) added = list(set(new_tracks) - set(old_tracks)) if removed or added: await playlist.edit(results) return added, removed, playlist async def _playlist_check(self, ctx: commands.Context): if not self._player_check(ctx): if self._connection_aborted: msg = _("Connection to Lavalink has failed") desc = EmptyEmbed if await ctx.bot.is_owner(ctx.author): desc = _("Please check your console or logs for details.") await self._embed_msg(ctx, title=msg, description=desc) return False try: if ( not ctx.author.voice.channel.permissions_for(ctx.me).connect or not ctx.author.voice.channel.permissions_for(ctx.me).move_members and userlimit(ctx.author.voice.channel) ): await self._embed_msg( ctx, title=_("Unable To Get Playlists"), description=_("I don't have permission to connect to your channel."), ) return False await lavalink.connect(ctx.author.voice.channel) player = lavalink.get_player(ctx.guild.id) player.store("connect", datetime.datetime.utcnow()) except IndexError: await self._embed_msg( ctx, title=_("Unable To Get Playlists"), description=_("Connection to Lavalink has not yet been established."), ) return False except AttributeError: await self._embed_msg( ctx, title=_("Unable To Get Playlists"), description=_("Connect to a voice channel first."), ) return False player = lavalink.get_player(ctx.guild.id) player.store("channel", ctx.channel.id) player.store("guild", ctx.guild.id) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): await self._embed_msg( ctx, title=_("Unable To Get Playlists"), description=_("You must be in the voice channel to use the playlist command."), ) return False await self._eq_check(ctx, player) await self._data_check(ctx) return True async def _playlist_tracks( self, ctx: commands.Context, player: lavalink.player_manager.Player, query: audio_dataclasses.Query, ): search = query.is_search tracklist = [] if query.is_spotify: try: if self.play_lock[ctx.message.guild.id]: return await self._embed_msg( ctx, title=_("Unable To Get Tracks"), description=_("Wait until the playlist has finished loading."), ) except KeyError: pass tracks = await self._get_spotify_tracks(ctx, query) if isinstance(tracks, discord.Message): return None if not tracks: embed = discord.Embed(title=_("Nothing found.")) if ( query.is_local and query.suffix in audio_dataclasses._PARTIALLY_SUPPORTED_MUSIC_EXT ): embed = discord.Embed(title=_("Track is not playable.")) embed.description = _( "**{suffix}** is not a fully supported format and some " "tracks may not play." ).format(suffix=query.suffix) return await self._embed_msg(ctx, embed=embed) for track in tracks: track_obj = track_creator(player, other_track=track) tracklist.append(track_obj) await asyncio.sleep(0) self._play_lock(ctx, False) elif query.is_search: try: result, called_api = await self.music_cache.lavalink_query(ctx, player, query) except TrackEnqueueError: self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Unable to Get Track"), description=_( "I'm unable get a track from Lavalink at the moment, try again in a few " "minutes." ), ) tracks = result.tracks if not tracks: embed = discord.Embed(title=_("Nothing found.")) if ( query.is_local and query.suffix in audio_dataclasses._PARTIALLY_SUPPORTED_MUSIC_EXT ): embed = discord.Embed(title=_("Track is not playable.")) embed.description = _( "**{suffix}** is not a fully supported format and some " "tracks may not play." ).format(suffix=query.suffix) return await self._embed_msg(ctx, embed=embed) else: try: result, called_api = await self.music_cache.lavalink_query(ctx, player, query) except TrackEnqueueError: self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Unable to Get Track"), description=_( "I'm unable get a track from Lavalink at the moment, try again in a few " "minutes." ), ) tracks = result.tracks if not search and len(tracklist) == 0: for track in tracks: track_obj = track_creator(player, other_track=track) tracklist.append(track_obj) await asyncio.sleep(0) elif len(tracklist) == 0: track_obj = track_creator(player, other_track=tracks[0]) tracklist.append(track_obj) return tracklist @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def prev(self, ctx: commands.Context): """Skip to the start of the previously played track.""" if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) vote_enabled = await self.config.guild(ctx.guild).vote_enabled() is_alone = await self._is_alone(ctx) is_requester = await self.is_requester(ctx, ctx.author) can_skip = await self._can_instaskip(ctx, ctx.author) player = lavalink.get_player(ctx.guild.id) if (not ctx.author.voice or ctx.author.voice.channel != player.channel) and not can_skip: return await self._embed_msg( ctx, title=_("Unable To Skip Tracks"), description=_("You must be in the voice channel to skip the track."), ) if vote_enabled or vote_enabled and dj_enabled: if not await self._can_instaskip(ctx, ctx.author) and not await self._is_alone(ctx): return await self._embed_msg( ctx, title=_("Unable To Skip Tracks"), description=_("There are other people listening - vote to skip instead."), ) if dj_enabled and not vote_enabled: if not (can_skip or is_requester) and not is_alone: return await self._embed_msg( ctx, title=_("Unable To Skip Tracks"), description=_( "You need the DJ role or be the track requester " "to enqueue the previous song tracks." ), ) if player.fetch("prev_song") is None: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("No previous track.") ) else: track = player.fetch("prev_song") player.add(player.fetch("prev_requester"), track) self.bot.dispatch("red_audio_track_enqueue", player.channel.guild, track, ctx.author) queue_len = len(player.queue) bump_song = player.queue[-1] player.queue.insert(0, bump_song) player.queue.pop(queue_len) await player.skip() description = get_track_description(player.current) embed = discord.Embed(title=_("Replaying Track"), description=description) await self._embed_msg(ctx, embed=embed) @commands.group(invoke_without_command=True) @commands.guild_only() @commands.bot_has_permissions(embed_links=True, add_reactions=True) async def queue(self, ctx: commands.Context, *, page: int = 1): """List the songs in the queue.""" async def _queue_menu( ctx: commands.Context, pages: list, controls: MutableMapping, message: discord.Message, page: int, timeout: float, emoji: str, ): if message: await ctx.send_help(self.queue) with contextlib.suppress(discord.HTTPException): await message.delete() return None queue_controls = { "\N{LEFTWARDS BLACK ARROW}": prev_page, "\N{CROSS MARK}": close_menu, "\N{BLACK RIGHTWARDS ARROW}": next_page, "\N{INFORMATION SOURCE}": _queue_menu, } if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) player = lavalink.get_player(ctx.guild.id) if player.current and not player.queue: arrow = await draw_time(ctx) pos = lavalink.utils.format_time(player.position) if player.current.is_stream: dur = "LIVE" else: dur = lavalink.utils.format_time(player.current.length) song = get_track_description(player.current) song += _("\n Requested by: **{track.requester}**") song += "\n\n{arrow}`{pos}`/`{dur}`" song = song.format(track=player.current, arrow=arrow, pos=pos, dur=dur) embed = discord.Embed(title=_("Now Playing"), description=song) if await self.config.guild(ctx.guild).thumbnail() and player.current: if player.current.thumbnail: embed.set_thumbnail(url=player.current.thumbnail) shuffle = await self.config.guild(ctx.guild).shuffle() repeat = await self.config.guild(ctx.guild).repeat() autoplay = await self.config.guild(ctx.guild).auto_play() text = "" text += ( _("Auto-Play") + ": " + ("\N{WHITE HEAVY CHECK MARK}" if autoplay else "\N{CROSS MARK}") ) text += ( (" | " if text else "") + _("Shuffle") + ": " + ("\N{WHITE HEAVY CHECK MARK}" if shuffle else "\N{CROSS MARK}") ) text += ( (" | " if text else "") + _("Repeat") + ": " + ("\N{WHITE HEAVY CHECK MARK}" if repeat else "\N{CROSS MARK}") ) embed.set_footer(text=text) message = await self._embed_msg(ctx, embed=embed) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) vote_enabled = await self.config.guild(ctx.guild).vote_enabled() if dj_enabled or vote_enabled: if not await self._can_instaskip(ctx, ctx.author) and not await self._is_alone( ctx ): return expected = ("⏹", "⏯") emoji = {"stop": "⏹", "pause": "⏯"} if player.current: task = start_adding_reactions(message, expected[:4]) else: task = None try: (r, u) = await self.bot.wait_for( "reaction_add", check=ReactionPredicate.with_emojis(expected, message, ctx.author), timeout=30.0, ) except asyncio.TimeoutError: return await self._clear_react(message, emoji) else: if task is not None: task.cancel() reacts = {v: k for k, v in emoji.items()} react = reacts[r.emoji] if react == "stop": await self._clear_react(message, emoji) return await ctx.invoke(self.stop) elif react == "pause": await self._clear_react(message, emoji) return await ctx.invoke(self.pause) return elif not player.current and not player.queue: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) async with ctx.typing(): limited_queue = player.queue[:500] # TODO: Improve when Toby menu's are merged len_queue_pages = math.ceil(len(limited_queue) / 10) queue_page_list = [] for page_num in range(1, len_queue_pages + 1): embed = await self._build_queue_page(ctx, limited_queue, player, page_num) queue_page_list.append(embed) await asyncio.sleep(0) if page > len_queue_pages: page = len_queue_pages return await menu(ctx, queue_page_list, queue_controls, page=(page - 1)) async def _build_queue_page( self, ctx: commands.Context, queue: list, player: lavalink.player_manager.Player, page_num ): shuffle = await self.config.guild(ctx.guild).shuffle() repeat = await self.config.guild(ctx.guild).repeat() autoplay = await self.config.guild(ctx.guild).auto_play() queue_num_pages = math.ceil(len(queue) / 10) queue_idx_start = (page_num - 1) * 10 queue_idx_end = queue_idx_start + 10 if len(player.queue) > 500: queue_list = "__Too many songs in the queue, only showing the first 500__.\n\n" else: queue_list = "" try: arrow = await draw_time(ctx) except AttributeError: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) pos = lavalink.utils.format_time(player.position) if player.current.is_stream: dur = "LIVE" else: dur = lavalink.utils.format_time(player.current.length) query = audio_dataclasses.Query.process_input(player.current) if query.is_stream: queue_list += _("**Currently livestreaming:**\n") queue_list += "**[{current.title}]({current.uri})**\n".format(current=player.current) queue_list += _("Requested by: **{user}**").format(user=player.current.requester) queue_list += f"\n\n{arrow}`{pos}`/`{dur}`\n\n" elif query.is_local: if player.current.title != "Unknown title": queue_list += "\n".join( ( _("Playing: ") + "**{current.author} - {current.title}**".format(current=player.current), audio_dataclasses.LocalPath(player.current.uri).to_string_user(), _("Requested by: **{user}**\n").format(user=player.current.requester), f"{arrow}`{pos}`/`{dur}`\n\n", ) ) else: queue_list += "\n".join( ( _("Playing: ") + audio_dataclasses.LocalPath(player.current.uri).to_string_user(), _("Requested by: **{user}**\n").format(user=player.current.requester), f"{arrow}`{pos}`/`{dur}`\n\n", ) ) else: queue_list += _("Playing: ") queue_list += "**[{current.title}]({current.uri})**\n".format(current=player.current) queue_list += _("Requested by: **{user}**").format(user=player.current.requester) queue_list += f"\n\n{arrow}`{pos}`/`{dur}`\n\n" for i, track in enumerate(queue[queue_idx_start:queue_idx_end], start=queue_idx_start): if i % 100 == 0: # TODO: Improve when Toby menu's are merged await asyncio.sleep(0.1) if len(track.title) > 40: track_title = str(track.title).replace("[", "") track_title = "{}...".format((track_title[:40]).rstrip(" ")) else: track_title = track.title req_user = track.requester track_idx = i + 1 query = audio_dataclasses.Query.process_input(track) if query.is_local: if track.title == "Unknown title": queue_list += f"`{track_idx}.` " + ", ".join( ( bold(audio_dataclasses.LocalPath(track.uri).to_string_user()), _("requested by **{user}**\n").format(user=req_user), ) ) else: queue_list += f"`{track_idx}.` **{track.author} - {track_title}**, " + _( "requested by **{user}**\n" ).format(user=req_user) else: queue_list += f"`{track_idx}.` **[{track_title}]({track.uri})**, " queue_list += _("requested by **{user}**\n").format(user=req_user) await asyncio.sleep(0) embed = discord.Embed( colour=await ctx.embed_colour(), title="Queue for __{guild.name}__".format(guild=ctx.guild), description=queue_list, ) if await self.config.guild(ctx.guild).thumbnail() and player.current.thumbnail: embed.set_thumbnail(url=player.current.thumbnail) queue_dur = await queue_duration(ctx) queue_total_duration = lavalink.utils.format_time(queue_dur) text = _( "Page {page_num}/{total_pages} | {num_tracks} tracks, {num_remaining} remaining\n" ).format( page_num=humanize_number(page_num), total_pages=humanize_number(queue_num_pages), num_tracks=len(player.queue), num_remaining=queue_total_duration, ) text += ( _("Auto-Play") + ": " + ("\N{WHITE HEAVY CHECK MARK}" if autoplay else "\N{CROSS MARK}") ) text += ( (" | " if text else "") + _("Shuffle") + ": " + ("\N{WHITE HEAVY CHECK MARK}" if shuffle else "\N{CROSS MARK}") ) text += ( (" | " if text else "") + _("Repeat") + ": " + ("\N{WHITE HEAVY CHECK MARK}" if repeat else "\N{CROSS MARK}") ) embed.set_footer(text=text) return embed @staticmethod async def _build_queue_search_list(queue_list, search_words): track_list = [] queue_idx = 0 for i, track in enumerate(queue_list, start=1): if i % 100 == 0: # TODO: Improve when Toby menu's are merged await asyncio.sleep(0.1) queue_idx = queue_idx + 1 if not match_url(track.uri): query = audio_dataclasses.Query.process_input(track) if track.title == "Unknown title": track_title = query.track.to_string_user() else: track_title = "{} - {}".format(track.author, track.title) else: track_title = track.title song_info = {str(queue_idx): track_title} track_list.append(song_info) await asyncio.sleep(0) search_results = process.extract(search_words, track_list, limit=50) search_list = [] for search, percent_match in search_results: for queue_position, title in search.items(): if percent_match > 89: search_list.append([queue_position, title]) return search_list @staticmethod async def _build_queue_search_page(ctx: commands.Context, page_num, search_list): search_num_pages = math.ceil(len(search_list) / 10) search_idx_start = (page_num - 1) * 10 search_idx_end = search_idx_start + 10 track_match = "" for i, track in enumerate( search_list[search_idx_start:search_idx_end], start=search_idx_start ): if i % 100 == 0: # TODO: Improve when Toby menu's are merged await asyncio.sleep(0.1) track_idx = i + 1 if type(track) is str: track_location = audio_dataclasses.LocalPath(track).to_string_user() track_match += "`{}.` **{}**\n".format(track_idx, track_location) else: track_match += "`{}.` **{}**\n".format(track[0], track[1]) await asyncio.sleep(0) embed = discord.Embed( colour=await ctx.embed_colour(), title=_("Matching Tracks:"), description=track_match ) embed.set_footer( text=(_("Page {page_num}/{total_pages}") + " | {num_tracks} tracks").format( page_num=humanize_number(page_num), total_pages=humanize_number(search_num_pages), num_tracks=len(search_list), ) ) return embed @queue.command(name="clear") @commands.guild_only() async def _queue_clear(self, ctx: commands.Context): """Clears the queue.""" try: player = lavalink.get_player(ctx.guild.id) except KeyError: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if not self._player_check(ctx) or not player.queue: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author) and not await self._is_alone(ctx): return await self._embed_msg( ctx, title=_("Unable To Clear Queue"), description=_("You need the DJ role to clear the queue."), ) player.queue.clear() await self._embed_msg( ctx, title=_("Queue Modified"), description=_("The queue has been cleared.") ) @queue.command(name="clean") @commands.guild_only() async def _queue_clean(self, ctx: commands.Context): """Removes songs from the queue if the requester is not in the voice channel.""" try: player = lavalink.get_player(ctx.guild.id) except KeyError: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if not self._player_check(ctx) or not player.queue: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author) and not await self._is_alone(ctx): return await self._embed_msg( ctx, title=_("Unable To Clean Queue"), description=_("You need the DJ role to clean the queue."), ) clean_tracks = [] removed_tracks = 0 listeners = player.channel.members for track in player.queue: if track.requester in listeners: clean_tracks.append(track) else: removed_tracks += 1 player.queue = clean_tracks if removed_tracks == 0: await self._embed_msg(ctx, title=_("Removed 0 tracks.")) else: await self._embed_msg( ctx, title=_("Removed racks from the queue"), description=_( "Removed {removed_tracks} tracks queued by members " "outside of the voice channel." ).format(removed_tracks=removed_tracks), ) @queue.command(name="cleanself") @commands.guild_only() async def _queue_cleanself(self, ctx: commands.Context): """Removes all tracks you requested from the queue.""" try: player = lavalink.get_player(ctx.guild.id) except KeyError: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) if not self._player_check(ctx) or not player.queue: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) clean_tracks = [] removed_tracks = 0 for track in player.queue: if track.requester != ctx.author: clean_tracks.append(track) else: removed_tracks += 1 player.queue = clean_tracks if removed_tracks == 0: await self._embed_msg(ctx, title=_("Removed 0 tracks.")) else: await self._embed_msg( ctx, title=_("Removed tracks from the queue"), description=_( "Removed {removed_tracks} tracks queued by {member.display_name}." ).format(removed_tracks=removed_tracks, member=ctx.author), ) @queue.command(name="search") @commands.guild_only() async def _queue_search(self, ctx: commands.Context, *, search_words: str): """Search the queue.""" try: player = lavalink.get_player(ctx.guild.id) except KeyError: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) if not self._player_check(ctx) or not player.queue: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) search_list = await self._build_queue_search_list(player.queue, search_words) if not search_list: return await self._embed_msg(ctx, title=_("No matches.")) len_search_pages = math.ceil(len(search_list) / 10) search_page_list = [] for page_num in range(1, len_search_pages + 1): embed = await self._build_queue_search_page(ctx, page_num, search_list) search_page_list.append(embed) await menu(ctx, search_page_list, DEFAULT_CONTROLS) @queue.command(name="shuffle") @commands.guild_only() @commands.cooldown(1, 30, commands.BucketType.guild) async def _queue_shuffle(self, ctx: commands.Context): """Shuffles the queue.""" dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author) and not await self._is_alone(ctx): ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Shuffle Queue"), description=_("You need the DJ role to shuffle the queue."), ) if not self._player_check(ctx): ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Shuffle Queue"), description=_("There's nothing in the queue."), ) try: if ( not ctx.author.voice.channel.permissions_for(ctx.me).connect or not ctx.author.voice.channel.permissions_for(ctx.me).move_members and userlimit(ctx.author.voice.channel) ): ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Shuffle Queue"), description=_("I don't have permission to connect to your channel."), ) await lavalink.connect(ctx.author.voice.channel) player = lavalink.get_player(ctx.guild.id) player.store("connect", datetime.datetime.utcnow()) except AttributeError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Shuffle Queue"), description=_("Connect to a voice channel first."), ) except IndexError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Shuffle Queue"), description=_("Connection to Lavalink has not yet been established."), ) except KeyError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Shuffle Queue"), description=_("There's nothing in the queue."), ) if not self._player_check(ctx) or not player.queue: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Shuffle Queue"), description=_("There's nothing in the queue."), ) player.force_shuffle(0) return await self._embed_msg(ctx, title=_("Queue has been shuffled.")) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def repeat(self, ctx: commands.Context): """Toggle repeat.""" dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author) and not await self._has_dj_role( ctx, ctx.author ): return await self._embed_msg( ctx, title=_("Unable To Toggle Repeat"), description=_("You need the DJ role to toggle repeat."), ) if self._player_check(ctx): await self._data_check(ctx) player = lavalink.get_player(ctx.guild.id) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Toggle Repeat"), description=_("You must be in the voice channel to toggle repeat."), ) autoplay = await self.config.guild(ctx.guild).auto_play() repeat = await self.config.guild(ctx.guild).repeat() msg = "" msg += _("Repeat tracks: {true_or_false}.").format( true_or_false=_("Enabled") if not repeat else _("Disabled") ) await self.config.guild(ctx.guild).repeat.set(not repeat) if repeat is not True and autoplay is True: msg += _("\nAuto-play has been disabled.") await self.config.guild(ctx.guild).auto_play.set(False) embed = discord.Embed(title=_("Setting Changed"), description=msg) await self._embed_msg(ctx, embed=embed) if self._player_check(ctx): await self._data_check(ctx) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def remove(self, ctx: commands.Context, index: int): """Remove a specific track number from the queue.""" dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) player = lavalink.get_player(ctx.guild.id) if not player.queue: return await self._embed_msg(ctx, title=_("Nothing queued.")) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Modify Queue"), description=_("You need the DJ role to remove tracks."), ) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Modify Queue"), description=_("You must be in the voice channel to manage the queue."), ) if index > len(player.queue) or index < 1: return await self._embed_msg( ctx, title=_("Unable To Modify Queue"), description=_("Song number must be greater than 1 and within the queue limit."), ) index -= 1 removed = player.queue.pop(index) removed_title = get_track_description(removed) await self._embed_msg( ctx, title=_("Removed track from queue"), description=_("Removed {track} from the queue.").format(track=removed_title), ) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True, add_reactions=True) async def search(self, ctx: commands.Context, *, query: str): """Pick a track with a search. Use `[p]search list ` to queue all tracks found on YouTube. `[p]search sc` will search SoundCloud instead of YouTube. """ async def _search_menu( ctx: commands.Context, pages: list, controls: MutableMapping, message: discord.Message, page: int, timeout: float, emoji: str, ): if message: await self._search_button_action(ctx, tracks, emoji, page) with contextlib.suppress(discord.HTTPException): await message.delete() return None search_controls = { "\N{DIGIT ONE}\N{COMBINING ENCLOSING KEYCAP}": _search_menu, "\N{DIGIT TWO}\N{COMBINING ENCLOSING KEYCAP}": _search_menu, "\N{DIGIT THREE}\N{COMBINING ENCLOSING KEYCAP}": _search_menu, "\N{DIGIT FOUR}\N{COMBINING ENCLOSING KEYCAP}": _search_menu, "\N{DIGIT FIVE}\N{COMBINING ENCLOSING KEYCAP}": _search_menu, "\N{LEFTWARDS BLACK ARROW}": prev_page, "\N{CROSS MARK}": close_menu, "\N{BLACK RIGHTWARDS ARROW}": next_page, } if not self._player_check(ctx): if self._connection_aborted: msg = _("Connection to Lavalink has failed") desc = EmptyEmbed if await ctx.bot.is_owner(ctx.author): desc = _("Please check your console or logs for details.") return await self._embed_msg(ctx, title=msg, description=desc) try: if ( not ctx.author.voice.channel.permissions_for(ctx.me).connect or not ctx.author.voice.channel.permissions_for(ctx.me).move_members and userlimit(ctx.author.voice.channel) ): return await self._embed_msg( ctx, title=_("Unable To Search For Tracks"), description=_("I don't have permission to connect to your channel."), ) await lavalink.connect(ctx.author.voice.channel) player = lavalink.get_player(ctx.guild.id) player.store("connect", datetime.datetime.utcnow()) except AttributeError: return await self._embed_msg( ctx, title=_("Unable To Search For Tracks"), description=_("Connect to a voice channel first."), ) except IndexError: return await self._embed_msg( ctx, title=_("Unable To Search For Tracks"), description=_("Connection to Lavalink has not yet been established."), ) player = lavalink.get_player(ctx.guild.id) guild_data = await self.config.guild(ctx.guild).all() player.store("channel", ctx.channel.id) player.store("guild", ctx.guild.id) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Search For Tracks"), description=_("You must be in the voice channel to enqueue tracks."), ) await self._eq_check(ctx, player) await self._data_check(ctx) before_queue_length = len(player.queue) if not isinstance(query, list): query = audio_dataclasses.Query.process_input(query) restrict = await self.config.restrict() if restrict and match_url(query): valid_url = url_check(query) if not valid_url: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("That URL is not allowed."), ) if not await is_allowed(ctx.guild, f"{query}", query_obj=query): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("That track is not allowed."), ) if query.invoked_from == "search list" or query.invoked_from == "local folder": if query.invoked_from == "search list" and not query.is_local: try: result, called_api = await self.music_cache.lavalink_query( ctx, player, query ) except TrackEnqueueError: self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Unable to Get Track"), description=_( "I'm unable get a track from Lavalink at the moment, try again in a " "few " "minutes." ), ) tracks = result.tracks else: try: tracks = await self._folder_tracks(ctx, player, query) except TrackEnqueueError: self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Unable to Get Track"), description=_( "I'm unable get a track from Lavalink at the moment, try again in a " "few " "minutes." ), ) if not tracks: embed = discord.Embed(title=_("Nothing found.")) if await self.config.use_external_lavalink() and query.is_local: embed.description = _( "Local tracks will not work " "if the `Lavalink.jar` cannot see the track.\n" "This may be due to permissions or because Lavalink.jar is being run " "in a different machine than the local tracks." ) elif ( query.is_local and query.suffix in audio_dataclasses._PARTIALLY_SUPPORTED_MUSIC_EXT ): embed = discord.Embed(title=_("Track is not playable.")) embed.description = _( "**{suffix}** is not a fully supported format and some " "tracks may not play." ).format(suffix=query.suffix) return await self._embed_msg(ctx, embed=embed) queue_dur = await queue_duration(ctx) queue_total_duration = lavalink.utils.format_time(queue_dur) if guild_data["dj_enabled"]: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("You need the DJ role to queue tracks."), ) track_len = 0 empty_queue = not player.queue for track in tracks: if len(player.queue) >= 10000: continue if not await is_allowed( ctx.guild, ( f"{track.title} {track.author} {track.uri} " f"{str(audio_dataclasses.Query.process_input(track))}" ), ): log.debug(f"Query is not allowed in {ctx.guild} ({ctx.guild.id})") continue elif guild_data["maxlength"] > 0: if track_limit(track, guild_data["maxlength"]): track_len += 1 player.add(ctx.author, track) self.bot.dispatch( "red_audio_track_enqueue", player.channel.guild, track, ctx.author ) else: track_len += 1 player.add(ctx.author, track) self.bot.dispatch( "red_audio_track_enqueue", player.channel.guild, track, ctx.author ) if not player.current: await player.play() await asyncio.sleep(0) player.maybe_shuffle(0 if empty_queue else 1) if len(tracks) > track_len: maxlength_msg = " {bad_tracks} tracks cannot be queued.".format( bad_tracks=(len(tracks) - track_len) ) else: maxlength_msg = "" songembed = discord.Embed( title=_("Queued {num} track(s).{maxlength_msg}").format( num=track_len, maxlength_msg=maxlength_msg ) ) if not guild_data["shuffle"] and queue_dur > 0: songembed.set_footer( text=_( "{time} until start of search playback: starts at #{position} in queue" ).format(time=queue_total_duration, position=before_queue_length + 1) ) return await self._embed_msg(ctx, embed=songembed) elif query.is_local and query.single_track: tracks = await self._folder_list(ctx, query) elif query.is_local and query.is_album: if ctx.invoked_with == "folder": return await self._local_play_all(ctx, query, from_search=True) else: tracks = await self._folder_list(ctx, query) else: try: result, called_api = await self.music_cache.lavalink_query(ctx, player, query) except TrackEnqueueError: self._play_lock(ctx, False) return await self._embed_msg( ctx, title=_("Unable to Get Track"), description=_( "I'm unable get a track from Lavalink at the moment," "try again in a few minutes." ), ) tracks = result.tracks if not tracks: embed = discord.Embed(title=_("Nothing found.")) if await self.config.use_external_lavalink() and query.is_local: embed.description = _( "Local tracks will not work " "if the `Lavalink.jar` cannot see the track.\n" "This may be due to permissions or because Lavalink.jar is being run " "in a different machine than the local tracks." ) elif ( query.is_local and query.suffix in audio_dataclasses._PARTIALLY_SUPPORTED_MUSIC_EXT ): embed = discord.Embed(title=_("Track is not playable.")) embed.description = _( "**{suffix}** is not a fully supported format and some " "tracks may not play." ).format(suffix=query.suffix) return await self._embed_msg(ctx, embed=embed) else: tracks = query dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) len_search_pages = math.ceil(len(tracks) / 5) search_page_list = [] for page_num in range(1, len_search_pages + 1): embed = await self._build_search_page(ctx, tracks, page_num) search_page_list.append(embed) await asyncio.sleep(0) if dj_enabled and not await self._can_instaskip(ctx, ctx.author): return await menu(ctx, search_page_list, DEFAULT_CONTROLS) await menu(ctx, search_page_list, search_controls) async def _search_button_action(self, ctx: commands.Context, tracks, emoji, page): if not self._player_check(ctx): if self._connection_aborted: msg = _("Connection to Lavalink has failed.") description = EmptyEmbed if await ctx.bot.is_owner(ctx.author): description = _("Please check your console or logs for details.") return await self._embed_msg(ctx, title=msg, description=description) try: await lavalink.connect(ctx.author.voice.channel) player = lavalink.get_player(ctx.guild.id) player.store("connect", datetime.datetime.utcnow()) except AttributeError: return await self._embed_msg(ctx, title=_("Connect to a voice channel first.")) except IndexError: return await self._embed_msg( ctx, title=_("Connection to Lavalink has not yet been established.") ) player = lavalink.get_player(ctx.guild.id) guild_data = await self.config.guild(ctx.guild).all() if len(player.queue) >= 10000: return await self._embed_msg( ctx, title=_("Unable To Play Tracks"), description=_("Queue size limit reached.") ) if not await self._currency_check(ctx, guild_data["jukebox_price"]): return try: if emoji == "\N{DIGIT ONE}\N{COMBINING ENCLOSING KEYCAP}": search_choice = tracks[0 + (page * 5)] elif emoji == "\N{DIGIT TWO}\N{COMBINING ENCLOSING KEYCAP}": search_choice = tracks[1 + (page * 5)] elif emoji == "\N{DIGIT THREE}\N{COMBINING ENCLOSING KEYCAP}": search_choice = tracks[2 + (page * 5)] elif emoji == "\N{DIGIT FOUR}\N{COMBINING ENCLOSING KEYCAP}": search_choice = tracks[3 + (page * 5)] elif emoji == "\N{DIGIT FIVE}\N{COMBINING ENCLOSING KEYCAP}": search_choice = tracks[4 + (page * 5)] else: search_choice = tracks[0 + (page * 5)] except IndexError: search_choice = tracks[-1] if getattr(search_choice, "uri", None): description = get_track_description(search_choice) else: search_choice = audio_dataclasses.Query.process_input(search_choice) if search_choice.track.exists() and search_choice.track.is_dir(): return await ctx.invoke(self.search, query=search_choice) elif search_choice.track.exists() and search_choice.track.is_file(): search_choice.invoked_from = "localtrack" return await ctx.invoke(self.play, query=search_choice) songembed = discord.Embed(title=_("Track Enqueued"), description=description) queue_dur = await queue_duration(ctx) queue_total_duration = lavalink.utils.format_time(queue_dur) before_queue_length = len(player.queue) if not await is_allowed( ctx.guild, ( f"{search_choice.title} {search_choice.author} {search_choice.uri} " f"{str(audio_dataclasses.Query.process_input(search_choice))}" ), ): log.debug(f"Query is not allowed in {ctx.guild} ({ctx.guild.id})") self._play_lock(ctx, False) return await self._embed_msg(ctx, title=_("This track is not allowed in this server.")) elif guild_data["maxlength"] > 0: if track_limit(search_choice.length, guild_data["maxlength"]): player.add(ctx.author, search_choice) player.maybe_shuffle() self.bot.dispatch( "red_audio_track_enqueue", player.channel.guild, search_choice, ctx.author ) else: return await self._embed_msg(ctx, title=_("Track exceeds maximum length.")) else: player.add(ctx.author, search_choice) player.maybe_shuffle() self.bot.dispatch( "red_audio_track_enqueue", player.channel.guild, search_choice, ctx.author ) if not guild_data["shuffle"] and queue_dur > 0: songembed.set_footer( text=_("{time} until track playback: #{position} in queue").format( time=queue_total_duration, position=before_queue_length + 1 ) ) if not player.current: await player.play() return await self._embed_msg(ctx, embed=songembed) @staticmethod def _format_search_options(search_choice): query = audio_dataclasses.Query.process_input(search_choice) description = get_track_description(search_choice) return description, query @staticmethod async def _build_search_page(ctx: commands.Context, tracks, page_num): search_num_pages = math.ceil(len(tracks) / 5) search_idx_start = (page_num - 1) * 5 search_idx_end = search_idx_start + 5 search_list = "" command = ctx.invoked_with folder = False for i, track in enumerate(tracks[search_idx_start:search_idx_end], start=search_idx_start): search_track_num = i + 1 if search_track_num > 5: search_track_num = search_track_num % 5 if search_track_num == 0: search_track_num = 5 try: query = audio_dataclasses.Query.process_input(track.uri) if query.is_local: search_list += "`{0}.` **{1}**\n[{2}]\n".format( search_track_num, track.title, audio_dataclasses.LocalPath(track.uri).to_string_user(), ) else: search_list += "`{0}.` **[{1}]({2})**\n".format( search_track_num, track.title, track.uri ) except AttributeError: track = audio_dataclasses.Query.process_input(track) if track.is_local and command != "search": search_list += "`{}.` **{}**\n".format( search_track_num, track.to_string_user() ) if track.is_album: folder = True elif command == "search": search_list += "`{}.` **{}**\n".format( search_track_num, track.to_string_user() ) else: search_list += "`{}.` **{}**\n".format( search_track_num, track.to_string_user() ) await asyncio.sleep(0) if hasattr(tracks[0], "uri") and hasattr(tracks[0], "track_identifier"): title = _("Tracks Found:") footer = _("search results") elif folder: title = _("Folders Found:") footer = _("local folders") else: title = _("Files Found:") footer = _("local tracks") embed = discord.Embed( colour=await ctx.embed_colour(), title=title, description=search_list ) embed.set_footer( text=(_("Page {page_num}/{total_pages}") + " | {num_results} {footer}").format( page_num=page_num, total_pages=search_num_pages, num_results=len(tracks), footer=footer, ) ) return embed @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def seek(self, ctx: commands.Context, seconds: Union[int, str]): """Seek ahead or behind on a track by seconds or a to a specific time. Accepts seconds or a value formatted like 00:00:00 (`hh:mm:ss`) or 00:00 (`mm:ss`). """ dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) vote_enabled = await self.config.guild(ctx.guild).vote_enabled() is_alone = await self._is_alone(ctx) is_requester = await self.is_requester(ctx, ctx.author) can_skip = await self._can_instaskip(ctx, ctx.author) if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) player = lavalink.get_player(ctx.guild.id) if (not ctx.author.voice or ctx.author.voice.channel != player.channel) and not can_skip: return await self._embed_msg( ctx, title=_("Unable To Seek Tracks"), description=_("You must be in the voice channel to use seek."), ) if vote_enabled and not can_skip and not is_alone: return await self._embed_msg( ctx, title=_("Unable To Seek Tracks"), description=_("There are other people listening - vote to skip instead."), ) if dj_enabled and not (can_skip or is_requester) and not is_alone: return await self._embed_msg( ctx, title=_("Unable To Seek Tracks"), description=_("You need the DJ role or be the track requester to use seek."), ) if player.current: if player.current.is_stream: return await self._embed_msg( ctx, title=_("Unable To Seek Tracks"), description=_("Can't seek on a stream.") ) else: try: int(seconds) abs_position = False except ValueError: abs_position = True seconds = time_convert(seconds) if seconds == 0: return await self._embed_msg( ctx, title=_("Unable To Seek Tracks"), description=_("Invalid input for the time to seek."), ) if not abs_position: time_sec = int(seconds) * 1000 seek = player.position + time_sec if seek <= 0: await self._embed_msg( ctx, title=_("Moved {num_seconds}s to 00:00:00").format( num_seconds=seconds ), ) else: await self._embed_msg( ctx, title=_("Moved {num_seconds}s to {time}").format( num_seconds=seconds, time=lavalink.utils.format_time(seek) ), ) await player.seek(seek) else: await self._embed_msg( ctx, title=_("Moved to {time}").format( time=lavalink.utils.format_time(seconds * 1000) ), ) await player.seek(seconds * 1000) else: await self._embed_msg(ctx, title=_("Nothing playing.")) @commands.group(autohelp=False) @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def shuffle(self, ctx: commands.Context): """Toggle shuffle.""" if ctx.invoked_subcommand is None: dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Toggle Shuffle"), description=_("You need the DJ role to toggle shuffle."), ) if self._player_check(ctx): await self._data_check(ctx) player = lavalink.get_player(ctx.guild.id) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Toggle Shuffle"), description=_("You must be in the voice channel to toggle shuffle."), ) shuffle = await self.config.guild(ctx.guild).shuffle() await self.config.guild(ctx.guild).shuffle.set(not shuffle) await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Shuffle tracks: {true_or_false}.").format( true_or_false=_("Enabled") if not shuffle else _("Disabled") ), ) if self._player_check(ctx): await self._data_check(ctx) @shuffle.command(name="bumped") @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def _shuffle_bumpped(self, ctx: commands.Context): """Toggle bumped track shuffle. Set this to disabled if you wish to avoid bumped songs being shuffled. This takes priority over `[p]shuffle`. """ dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Toggle Shuffle"), description=_("You need the DJ role to toggle shuffle."), ) if self._player_check(ctx): await self._data_check(ctx) player = lavalink.get_player(ctx.guild.id) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Toggle Shuffle"), description=_("You must be in the voice channel to toggle shuffle."), ) bumped = await self.config.guild(ctx.guild).shuffle_bumped() await self.config.guild(ctx.guild).shuffle_bumped.set(not bumped) await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Shuffle bumped tracks: {true_or_false}.").format( true_or_false=_("Enabled") if not bumped else _("Disabled") ), ) if self._player_check(ctx): await self._data_check(ctx) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def sing(self, ctx: commands.Context): """Make Red sing one of her songs.""" ids = ( "zGTkAVsrfg8", "cGMWL8cOeAU", "vFrjMq4aL-g", "WROI5WYBU_A", "41tIUr_ex3g", "f9O2Rjn1azc", ) url = f"https://www.youtube.com/watch?v={random.choice(ids)}" await ctx.invoke(self.play, query=url) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def skip(self, ctx: commands.Context, skip_to_track: int = None): """Skip to the next track, or to a given track number.""" if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) player = lavalink.get_player(ctx.guild.id) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Skip Tracks"), description=_("You must be in the voice channel to skip the music."), ) if not player.current: return await self._embed_msg(ctx, title=_("Nothing playing.")) dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) vote_enabled = await self.config.guild(ctx.guild).vote_enabled() is_alone = await self._is_alone(ctx) is_requester = await self.is_requester(ctx, ctx.author) can_skip = await self._can_instaskip(ctx, ctx.author) if dj_enabled and not vote_enabled: if not (can_skip or is_requester) and not is_alone: return await self._embed_msg( ctx, title=_("Unable To Skip Tracks"), description=_( "You need the DJ role or be the track requester to skip tracks." ), ) if ( is_requester and not can_skip and isinstance(skip_to_track, int) and skip_to_track > 1 ): return await self._embed_msg( ctx, title=_("Unable To Skip Tracks"), description=_("You can only skip the current track."), ) if vote_enabled: if not can_skip: if skip_to_track is not None: return await self._embed_msg( ctx, title=_("Unable To Skip Tracks"), description=_( "Can't skip to a specific track in vote mode without the DJ role." ), ) if ctx.author.id in self.skip_votes[ctx.message.guild]: self.skip_votes[ctx.message.guild].remove(ctx.author.id) reply = _("I removed your vote to skip.") else: self.skip_votes[ctx.message.guild].append(ctx.author.id) reply = _("You voted to skip.") num_votes = len(self.skip_votes[ctx.message.guild]) vote_mods = [] for member in player.channel.members: can_skip = await self._can_instaskip(ctx, member) if can_skip: vote_mods.append(member) num_members = len(player.channel.members) - len(vote_mods) vote = int(100 * num_votes / num_members) percent = await self.config.guild(ctx.guild).vote_percent() if vote >= percent: self.skip_votes[ctx.message.guild] = [] await self._embed_msg(ctx, title=_("Vote threshold met.")) return await self._skip_action(ctx) else: reply += _( " Votes: {num_votes}/{num_members}" " ({cur_percent}% out of {required_percent}% needed)" ).format( num_votes=humanize_number(num_votes), num_members=humanize_number(num_members), cur_percent=vote, required_percent=percent, ) return await self._embed_msg(ctx, title=reply) else: return await self._skip_action(ctx, skip_to_track) else: return await self._skip_action(ctx, skip_to_track) async def _can_instaskip(self, ctx: commands.Context, member: discord.Member): dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if member.bot: return True if member.id == ctx.guild.owner_id: return True if dj_enabled: if await self._has_dj_role(ctx, member): return True if await ctx.bot.is_owner(member): return True if await ctx.bot.is_mod(member): return True if await self._channel_check(ctx): return True return False @staticmethod async def _is_alone(ctx: commands.Context): channel_members = rgetattr(ctx, "guild.me.voice.channel.members", []) nonbots = sum(m.id != ctx.author.id for m in channel_members if not m.bot) return nonbots < 1 async def _has_dj_role(self, ctx: commands.Context, member: discord.Member): dj_role = self._dj_role_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_role() ) dj_role_obj = ctx.guild.get_role(dj_role) return dj_role_obj in ctx.guild.get_member(member.id).roles @staticmethod async def is_requester(ctx: commands.Context, member: discord.Member): try: player = lavalink.get_player(ctx.guild.id) log.debug(f"Current requester is {player.current}") return player.current.requester.id == member.id except Exception as e: log.error(e) return False async def _skip_action(self, ctx: commands.Context, skip_to_track: int = None): player = lavalink.get_player(ctx.guild.id) autoplay = await self.config.guild(player.channel.guild).auto_play() if not player.current or (not player.queue and not autoplay): try: pos, dur = player.position, player.current.length except AttributeError: return await self._embed_msg(ctx, title=_("There's nothing in the queue.")) time_remain = lavalink.utils.format_time(dur - pos) if player.current.is_stream: embed = discord.Embed(title=_("There's nothing in the queue.")) embed.set_footer( text=_("Currently livestreaming {track}").format(track=player.current.title) ) else: embed = discord.Embed(title=_("There's nothing in the queue.")) embed.set_footer( text=_("{time} left on {track}").format( time=time_remain, track=player.current.title ) ) return await self._embed_msg(ctx, embed=embed) elif autoplay and not player.queue: embed = discord.Embed( title=_("Track Skipped"), description=get_track_description(player.current) ) await self._embed_msg(ctx, embed=embed) return await player.skip() queue_to_append = [] if skip_to_track is not None and skip_to_track != 1: if skip_to_track < 1: return await self._embed_msg( ctx, title=_("Track number must be equal to or greater than 1.") ) elif skip_to_track > len(player.queue): return await self._embed_msg( ctx, title=_( "There are only {queuelen} songs currently queued.".format( queuelen=len(player.queue) ) ), ) embed = discord.Embed( title=_("{skip_to_track} Tracks Skipped".format(skip_to_track=skip_to_track)) ) await self._embed_msg(ctx, embed=embed) if player.repeat: queue_to_append = player.queue[0 : min(skip_to_track - 1, len(player.queue) - 1)] player.queue = player.queue[ min(skip_to_track - 1, len(player.queue) - 1) : len(player.queue) ] else: embed = discord.Embed( title=_("Track Skipped"), description=get_track_description(player.current) ) await self._embed_msg(ctx, embed=embed) self.bot.dispatch("red_audio_skip_track", player.channel.guild, player.current, ctx.author) await player.play() player.queue += queue_to_append @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def stop(self, ctx: commands.Context): """Stop playback and clear the queue.""" dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) vote_enabled = await self.config.guild(ctx.guild).vote_enabled() if not self._player_check(ctx): return await self._embed_msg(ctx, title=_("Nothing playing.")) player = lavalink.get_player(ctx.guild.id) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Stop Player"), description=_("You must be in the voice channel to stop the music."), ) if vote_enabled or vote_enabled and dj_enabled: if not await self._can_instaskip(ctx, ctx.author) and not await self._is_alone(ctx): return await self._embed_msg( ctx, title=_("Unable To Stop Player"), description=_("There are other people listening - vote to skip instead."), ) if dj_enabled and not vote_enabled: if not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Stop Player"), description=_("You need the DJ role to stop the music."), ) if ( player.is_playing or (not player.is_playing and player.paused) or player.queue or getattr(player.current, "extras", {}).get("autoplay") ): eq = player.fetch("eq") if eq: await self.config.custom("EQUALIZER", ctx.guild.id).eq_bands.set(eq.bands) player.queue = [] player.store("playing_song", None) player.store("prev_requester", None) player.store("prev_song", None) player.store("requester", None) await player.stop() await self._embed_msg(ctx, title=_("Stopping...")) @commands.command() @commands.guild_only() @commands.cooldown(1, 15, commands.BucketType.guild) @commands.bot_has_permissions(embed_links=True) async def summon(self, ctx: commands.Context): """Summon the bot to a voice channel.""" dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) vote_enabled = await self.config.guild(ctx.guild).vote_enabled() is_alone = await self._is_alone(ctx) is_requester = await self.is_requester(ctx, ctx.author) can_skip = await self._can_instaskip(ctx, ctx.author) if vote_enabled or (vote_enabled and dj_enabled): if not can_skip and not is_alone: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Join Voice Channel"), description=_("There are other people listening."), ) if dj_enabled and not vote_enabled: if not (can_skip or is_requester) and not is_alone: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Join Voice Channel"), description=_("You need the DJ role to summon the bot."), ) try: if ( not ctx.author.voice.channel.permissions_for(ctx.me).connect or not ctx.author.voice.channel.permissions_for(ctx.me).move_members and userlimit(ctx.author.voice.channel) ): ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Join Voice Channel"), description=_("I don't have permission to connect to your channel."), ) if not self._player_check(ctx): await lavalink.connect(ctx.author.voice.channel) player = lavalink.get_player(ctx.guild.id) player.store("connect", datetime.datetime.utcnow()) else: player = lavalink.get_player(ctx.guild.id) if ctx.author.voice.channel == player.channel: ctx.command.reset_cooldown(ctx) return await player.move_to(ctx.author.voice.channel) except AttributeError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Join Voice Channel"), description=_("Connect to a voice channel first."), ) except IndexError: ctx.command.reset_cooldown(ctx) return await self._embed_msg( ctx, title=_("Unable To Join Voice Channel"), description=_("Connection to Lavalink has not yet been established."), ) @commands.command() @commands.guild_only() @commands.bot_has_permissions(embed_links=True) async def volume(self, ctx: commands.Context, vol: int = None): """Set the volume, 1% - 150%.""" dj_enabled = self._dj_status_cache.setdefault( ctx.guild.id, await self.config.guild(ctx.guild).dj_enabled() ) if not vol: vol = await self.config.guild(ctx.guild).volume() embed = discord.Embed(title=_("Current Volume:"), description=str(vol) + "%") if not self._player_check(ctx): embed.set_footer(text=_("Nothing playing.")) return await self._embed_msg(ctx, embed=embed) if self._player_check(ctx): player = lavalink.get_player(ctx.guild.id) if ( not ctx.author.voice or ctx.author.voice.channel != player.channel ) and not await self._can_instaskip(ctx, ctx.author): return await self._embed_msg( ctx, title=_("Unable To Change Volume"), description=_("You must be in the voice channel to change the volume."), ) if dj_enabled: if not await self._can_instaskip(ctx, ctx.author) and not await self._has_dj_role( ctx, ctx.author ): return await self._embed_msg( ctx, title=_("Unable To Change Volume"), description=_("You need the DJ role to change the volume."), ) if vol < 0: vol = 0 if vol > 150: vol = 150 await self.config.guild(ctx.guild).volume.set(vol) if self._player_check(ctx): await lavalink.get_player(ctx.guild.id).set_volume(vol) else: await self.config.guild(ctx.guild).volume.set(vol) if self._player_check(ctx): await lavalink.get_player(ctx.guild.id).set_volume(vol) embed = discord.Embed(title=_("Volume:"), description=str(vol) + "%") if not self._player_check(ctx): embed.set_footer(text=_("Nothing playing.")) await self._embed_msg(ctx, embed=embed) @commands.group(aliases=["llset"]) @commands.guild_only() @commands.bot_has_permissions(embed_links=True) @checks.is_owner() async def llsetup(self, ctx: commands.Context): """Lavalink server configuration options.""" @llsetup.command() async def external(self, ctx: commands.Context): """Toggle using external lavalink servers.""" external = await self.config.use_external_lavalink() await self.config.use_external_lavalink.set(not external) if external: embed = discord.Embed( title=_("Setting Changed"), description=_("External lavalink server: {true_or_false}.").format( true_or_false=_("Enabled") if not external else _("Disabled") ), ) await self._embed_msg(ctx, embed=embed) else: if self._manager is not None: await self._manager.shutdown() await self._embed_msg( ctx, title=_("Setting Changed"), description=_("External lavalink server: {true_or_false}.").format( true_or_false=_("Enabled") if not external else _("Disabled") ), ) self._restart_connect() @llsetup.command() async def host(self, ctx: commands.Context, host: str): """Set the lavalink server host.""" await self.config.host.set(host) footer = None if await self._check_external(): footer = _("External lavalink server set to True.") await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Host set to {host}.").format(host=host), footer=footer, ) self._restart_connect() @llsetup.command() async def password(self, ctx: commands.Context, password: str): """Set the lavalink server password.""" await self.config.password.set(str(password)) footer = None if await self._check_external(): footer = _("External lavalink server set to True.") await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Server password set to {password}.").format(password=password), footer=footer, ) self._restart_connect() @llsetup.command() async def restport(self, ctx: commands.Context, rest_port: int): """Set the lavalink REST server port.""" await self.config.rest_port.set(rest_port) footer = None if await self._check_external(): footer = _("External lavalink server set to True.") await self._embed_msg( ctx, title=_("Setting Changed"), description=_("REST port set to {port}.").format(port=rest_port), footer=footer, ) self._restart_connect() @llsetup.command() async def wsport(self, ctx: commands.Context, ws_port: int): """Set the lavalink websocket server port.""" await self.config.ws_port.set(ws_port) footer = None if await self._check_external(): footer = _("External lavalink server set to True.") await self._embed_msg( ctx, title=_("Setting Changed"), description=_("Websocket port set to {port}.").format(port=ws_port), footer=footer, ) self._restart_connect() @staticmethod async def _apply_gain(guild_id: int, band, gain): const = { "op": "equalizer", "guildId": str(guild_id), "bands": [{"band": band, "gain": gain}], } try: await lavalink.get_player(guild_id).node.send({**const}) except (KeyError, IndexError): pass @staticmethod async def _apply_gains(guild_id: int, gains): const = { "op": "equalizer", "guildId": str(guild_id), "bands": [{"band": x, "gain": y} for x, y in enumerate(gains)], } try: await lavalink.get_player(guild_id).node.send({**const}) except (KeyError, IndexError): pass async def _channel_check(self, ctx: commands.Context): try: player = lavalink.get_player(ctx.guild.id) except KeyError: return False try: in_channel = sum( not m.bot for m in ctx.guild.get_member(self.bot.user.id).voice.channel.members ) except AttributeError: return False if not ctx.author.voice: user_channel = None else: user_channel = ctx.author.voice.channel if in_channel == 0 and user_channel: if ( (player.channel != user_channel) and not player.current and player.position == 0 and len(player.queue) == 0 ): await player.move_to(user_channel) return True else: return False async def _check_api_tokens(self): spotify = await self.bot.get_shared_api_tokens("spotify") youtube = await self.bot.get_shared_api_tokens("youtube") return { "spotify_client_id": spotify.get("client_id", ""), "spotify_client_secret": spotify.get("client_secret", ""), "youtube_api": youtube.get("api_key", ""), } async def _check_external(self): external = await self.config.use_external_lavalink() if not external: if self._manager is not None: await self._manager.shutdown() await self.config.use_external_lavalink.set(True) return True else: return False async def _clear_react(self, message: discord.Message, emoji: MutableMapping = None): """Non blocking version of clear_react.""" return self.bot.loop.create_task(clear_react(self.bot, message, emoji)) async def _currency_check(self, ctx: commands.Context, jukebox_price: int): jukebox = await self.config.guild(ctx.guild).jukebox() if jukebox and not await self._can_instaskip(ctx, ctx.author): can_spend = await bank.can_spend(ctx.author, jukebox_price) if can_spend: await bank.withdraw_credits(ctx.author, jukebox_price) else: credits_name = await bank.get_currency_name(ctx.guild) bal = await bank.get_balance(ctx.author) await self._embed_msg( ctx, title=_("Not enough {currency}").format(currency=credits_name), description=_( "{required_credits} {currency} required, but you have {bal}." ).format( currency=credits_name, required_credits=humanize_number(jukebox_price), bal=humanize_number(bal), ), ) return can_spend else: return True async def _data_check(self, ctx: commands.Context): player = lavalink.get_player(ctx.guild.id) shuffle = await self.config.guild(ctx.guild).shuffle() repeat = await self.config.guild(ctx.guild).repeat() volume = await self.config.guild(ctx.guild).volume() shuffle_bumped = await self.config.guild(ctx.guild).shuffle_bumped() player.repeat = repeat player.shuffle = shuffle player.shuffle_bumped = shuffle_bumped if player.volume != volume: await player.set_volume(volume) async def disconnect_timer(self): stop_times = {} pause_times = {} while True: for p in lavalink.all_players(): server = p.channel.guild if [self.bot.user] == p.channel.members: stop_times.setdefault(server.id, time.time()) pause_times.setdefault(server.id, time.time()) else: stop_times.pop(server.id, None) if p.paused and server.id in pause_times: try: await p.pause(False) except Exception: log.error( "Exception raised in Audio's emptypause_timer.", exc_info=True ) pause_times.pop(server.id, None) servers = stop_times.copy() servers.update(pause_times) for sid in servers: server_obj = self.bot.get_guild(sid) if sid in stop_times and await self.config.guild(server_obj).emptydc_enabled(): emptydc_timer = await self.config.guild(server_obj).emptydc_timer() if (time.time() - stop_times[sid]) >= emptydc_timer: stop_times.pop(sid) try: player = lavalink.get_player(sid) await player.stop() await player.disconnect() except Exception as err: log.error("Exception raised in Audio's emptydc_timer.", exc_info=True) if "No such player for that guild" in str(err): stop_times.pop(sid, None) elif ( sid in pause_times and await self.config.guild(server_obj).emptypause_enabled() ): emptypause_timer = await self.config.guild(server_obj).emptypause_timer() if (time.time() - pause_times.get(sid)) >= emptypause_timer: try: await lavalink.get_player(sid).pause() except Exception as err: if "No such player for that guild" in str(err): pause_times.pop(sid, None) log.error( "Exception raised in Audio's emptypause_timer.", exc_info=True ) await asyncio.sleep(5) async def _embed_msg(self, ctx: commands.Context, **kwargs): colour = kwargs.get("colour") or kwargs.get("color") or await self.bot.get_embed_color(ctx) error = kwargs.get("error", False) success = kwargs.get("success", False) title = kwargs.get("title", EmptyEmbed) or EmptyEmbed _type = kwargs.get("type", "rich") or "rich" url = kwargs.get("url", EmptyEmbed) or EmptyEmbed description = kwargs.get("description", EmptyEmbed) or EmptyEmbed timestamp = kwargs.get("timestamp") footer = kwargs.get("footer") thumbnail = kwargs.get("thumbnail") contents = dict(title=title, type=_type, url=url, description=description) embed = kwargs.get("embed").to_dict() if hasattr(kwargs.get("embed"), "to_dict") else {} colour = embed.get("color") if embed.get("color") else colour contents.update(embed) if timestamp and isinstance(timestamp, datetime.datetime): contents["timestamp"] = timestamp embed = discord.Embed.from_dict(contents) embed.color = colour if footer: embed.set_footer(text=footer) if thumbnail: embed.set_thumbnail(url=thumbnail) return await ctx.send(embed=embed) async def _eq_check(self, ctx: commands.Context, player: lavalink.Player): eq = player.fetch("eq", Equalizer()) config_bands = await self.config.custom("EQUALIZER", ctx.guild.id).eq_bands() if not config_bands: config_bands = eq.bands await self.config.custom("EQUALIZER", ctx.guild.id).eq_bands.set(eq.bands) if eq.bands != config_bands: band_num = list(range(0, eq._band_count)) band_value = config_bands eq_dict = {} for k, v in zip(band_num, band_value): eq_dict[k] = v for band, value in eq_dict.items(): eq.set_gain(band, value) player.store("eq", eq) await self._apply_gains(ctx.guild.id, config_bands) async def _eq_interact( self, ctx: commands.Context, player: lavalink.Player, eq, message, selected ): player.store("eq", eq) emoji = { "far_left": "\N{BLACK LEFT-POINTING TRIANGLE}", "one_left": "\N{LEFTWARDS BLACK ARROW}", "max_output": "\N{BLACK UP-POINTING DOUBLE TRIANGLE}", "output_up": "\N{UP-POINTING SMALL RED TRIANGLE}", "output_down": "\N{DOWN-POINTING SMALL RED TRIANGLE}", "min_output": "\N{BLACK DOWN-POINTING DOUBLE TRIANGLE}", "one_right": "\N{BLACK RIGHTWARDS ARROW}", "far_right": "\N{BLACK RIGHT-POINTING TRIANGLE}", "reset": "\N{BLACK CIRCLE FOR RECORD}", "info": "\N{INFORMATION SOURCE}", } selector = f'{" " * 8}{" " * selected}^^' try: await message.edit(content=box(f"{eq.visualise()}\n{selector}", lang="ini")) except discord.errors.NotFound: return try: (react_emoji, react_user) = await self._get_eq_reaction(ctx, message, emoji) except TypeError: return if not react_emoji: await self.config.custom("EQUALIZER", ctx.guild.id).eq_bands.set(eq.bands) await self._clear_react(message, emoji) if react_emoji == "\N{LEFTWARDS BLACK ARROW}": await remove_react(message, react_emoji, react_user) await self._eq_interact(ctx, player, eq, message, max(selected - 1, 0)) if react_emoji == "\N{BLACK RIGHTWARDS ARROW}": await remove_react(message, react_emoji, react_user) await self._eq_interact(ctx, player, eq, message, min(selected + 1, 14)) if react_emoji == "\N{UP-POINTING SMALL RED TRIANGLE}": await remove_react(message, react_emoji, react_user) _max = "{:.2f}".format(min(eq.get_gain(selected) + 0.1, 1.0)) eq.set_gain(selected, float(_max)) await self._apply_gain(ctx.guild.id, selected, _max) await self._eq_interact(ctx, player, eq, message, selected) if react_emoji == "\N{DOWN-POINTING SMALL RED TRIANGLE}": await remove_react(message, react_emoji, react_user) _min = "{:.2f}".format(max(eq.get_gain(selected) - 0.1, -0.25)) eq.set_gain(selected, float(_min)) await self._apply_gain(ctx.guild.id, selected, _min) await self._eq_interact(ctx, player, eq, message, selected) if react_emoji == "\N{BLACK UP-POINTING DOUBLE TRIANGLE}": await remove_react(message, react_emoji, react_user) _max = 1.0 eq.set_gain(selected, _max) await self._apply_gain(ctx.guild.id, selected, _max) await self._eq_interact(ctx, player, eq, message, selected) if react_emoji == "\N{BLACK DOWN-POINTING DOUBLE TRIANGLE}": await remove_react(message, react_emoji, react_user) _min = -0.25 eq.set_gain(selected, _min) await self._apply_gain(ctx.guild.id, selected, _min) await self._eq_interact(ctx, player, eq, message, selected) if react_emoji == "\N{BLACK LEFT-POINTING TRIANGLE}": await remove_react(message, react_emoji, react_user) selected = 0 await self._eq_interact(ctx, player, eq, message, selected) if react_emoji == "\N{BLACK RIGHT-POINTING TRIANGLE}": await remove_react(message, react_emoji, react_user) selected = 14 await self._eq_interact(ctx, player, eq, message, selected) if react_emoji == "\N{BLACK CIRCLE FOR RECORD}": await remove_react(message, react_emoji, react_user) for band in range(eq._band_count): eq.set_gain(band, 0.0) await self._apply_gains(ctx.guild.id, eq.bands) await self._eq_interact(ctx, player, eq, message, selected) if react_emoji == "\N{INFORMATION SOURCE}": await remove_react(message, react_emoji, react_user) await ctx.send_help(self.eq) await self._eq_interact(ctx, player, eq, message, selected) @staticmethod async def _eq_msg_clear(eq_message: discord.Message): if eq_message is not None: with contextlib.suppress(discord.HTTPException): await eq_message.delete() async def _get_eq_reaction(self, ctx: commands.Context, message: discord.Message, emoji): try: reaction, user = await self.bot.wait_for( "reaction_add", check=lambda r, u: r.message.id == message.id and u.id == ctx.author.id and r.emoji in emoji.values(), timeout=30, ) except asyncio.TimeoutError: await self._clear_react(message, emoji) return None else: return reaction.emoji, user def _play_lock(self, ctx: commands.Context, tf): if tf: self.play_lock[ctx.message.guild.id] = True else: self.play_lock[ctx.message.guild.id] = False def _player_check(self, ctx: commands.Context): if self._connection_aborted: return False try: lavalink.get_player(ctx.guild.id) return True except IndexError: return False except KeyError: return False @commands.Cog.listener() async def on_red_audio_track_start( self, guild: discord.Guild, track: lavalink.Track, requester: discord.Member ): daily_cache = self._daily_playlist_cache.setdefault( guild.id, await self.config.guild(guild).daily_playlists() ) scope = PlaylistScope.GUILD.value today = datetime.date.today() midnight = datetime.datetime.combine(today, datetime.datetime.min.time()) if daily_cache: name = f"Daily playlist - {today}" today_id = int(time.mktime(today.timetuple())) track_identifier = track.track_identifier track = track_to_json(track) try: playlist = await get_playlist( playlist_number=today_id, scope=PlaylistScope.GUILD.value, bot=self.bot, guild=guild, author=self.bot.user, ) except RuntimeError: playlist = None if playlist: tracks = playlist.tracks tracks.append(track) await playlist.edit({"tracks": tracks}) else: playlist = Playlist( bot=self.bot, scope=scope, author=self.bot.user.id, playlist_id=today_id, name=name, playlist_url=None, tracks=[track], guild=guild, ) await playlist.save() with contextlib.suppress(Exception): too_old = midnight - datetime.timedelta(days=8) too_old_id = int(time.mktime(too_old.timetuple())) await delete_playlist( scope=scope, playlist_id=too_old_id, guild=guild, author=self.bot.user ) @commands.Cog.listener() async def on_voice_state_update( self, member: discord.Member, before: discord.VoiceState, after: discord.VoiceState ): await self._ready_event.wait() if after.channel != before.channel: try: self.skip_votes[before.channel.guild].remove(member.id) except (ValueError, KeyError, AttributeError): pass @commands.Cog.listener() async def on_red_audio_queue_end( self, guild: discord.Guild, track: lavalink.Track, requester: discord.Member ): await self.music_cache.database.clean_up_old_entries() await asyncio.sleep(5) dat = get_playlist_database() if dat: dat.delete_scheduled() await asyncio.sleep(5) def cog_unload(self): if not self._cleaned_up: self.bot.dispatch("red_audio_unload", self) self.session.detach() self.bot.loop.create_task(self._close_database()) if self._disconnect_task: self._disconnect_task.cancel() if self._connect_task: self._connect_task.cancel() if self._init_task: self._init_task.cancel() lavalink.unregister_event_listener(self.event_handler) self.bot.loop.create_task(lavalink.close()) if self._manager is not None: self.bot.loop.create_task(self._manager.shutdown()) self._cleaned_up = True @bump.error @disconnect.error @genre.error @local_folder.error @local_play.error @local_search.error @play.error @prev.error @search.error @_playlist_append.error @_playlist_save.error @_playlist_update.error @_playlist_upload.error async def _clear_lock_on_error(self, ctx: commands.Context, error): # TODO: Change this in a future PR # FIXME: This seems to be consuming tracebacks and not adding them to last traceback # which is handled by on_command_error # Make it so that this can be used to show user friendly errors if not isinstance( getattr(error, "original", error), ( commands.CheckFailure, commands.UserInputError, commands.DisabledCommand, commands.CommandOnCooldown, ), ): self._play_lock(ctx, False) await self.music_cache.run_tasks(ctx) message = "Error in command '{}'. Check your console or logs for details.".format( ctx.command.qualified_name ) await ctx.send(inline(message)) exception_log = "Exception in command '{}'\n" "".format(ctx.command.qualified_name) exception_log += "".join( traceback.format_exception(type(error), error, error.__traceback__) ) self.bot._last_exception = exception_log await ctx.bot.on_command_error( ctx, getattr(error, "original", error), unhandled_by_cog=True ) async def cog_after_invoke(self, ctx: commands.Context): await self._process_db(ctx) async def _process_db(self, ctx: commands.Context): await self.music_cache.run_tasks(ctx) async def _close_database(self): await self.music_cache.run_all_pending_tasks() self.music_cache.database.close() __del__ = cog_unload