umi/cogs/music.py

930 lines
29 KiB
Python

import discord
from discord.ext.commands import Cog, command, Context
from discord.ext.pages import Paginator
import asyncio # used to run async functions within regular functions
import subprocess # for running ffprobe and getting duration of files
from os import getenv, path, makedirs
from time import time # performance tracking
import random # for shuffling the queue
import math # for ceiling function in queue pages
from functools import partial
import logging
from youtube_search import YoutubeSearch
if not path.exists('.logs'):
makedirs('.logs')
logger = logging.getLogger(__name__)
logger.setLevel(logging.WARNING)
fh = logging.FileHandler('.logs/music.log')
formatter = logging.Formatter('%(asctime)s | %(name)s | [%(levelname)s] %(message)s', '%Y-%m-%d %H:%M:%S')
fh.setFormatter(formatter)
if not len(logger.handlers):
logger.addHandler(fh)
def setup(bot: discord.Bot):
bot.add_cog(Music(bot))
def format_time(d: int) -> str:
"""Convert seconds to timestamp"""
h = d // 3600
m = d % 3600 // 60
s = d % 60
if h:
return '{}:{:02}:{:02}'.format(h,m,s)
return '{}:{:02}'.format(m,s)
def format_date(d: str) -> str:
"""Convert YYYYMMDD to YYYY/MM/DD"""
return f"{d[:4]}/{d[4:6]}/{d[6:]}"
class Track:
def __init__(
self,
*,
source: str,
requester: discord.User,
title: str = None,
original_url: str = None,
duration = None,
author: str = None,
author_icon: str = None,
data: dict = None,
):
self.source = source
self.requester = requester
self.title = title
self.original_url = original_url
self.duration = duration
self.author = author
self.data = data
def __repr__(self):
return f"<Track {self.source=} {self.requester=} {self.title=} {self.duration=}"
def __str__(self):
title = f"**{self.title}**" if self.title else f"`{self.source}`"
duration = f" ({format_time(int(self.duration))})" if self.duration else " (?:??)"
return (
title + duration + f"\nRequested by {self.requester.display_name} ({self.requester})"
if self.requester
else
title + duration + f"\nRequested by ???"
)
class Player(discord.PCMVolumeTransformer):
def __init__(self,
source,
duration,
*,
data = None,
ffmpeg_options = {"options": "-vn"},
):
super().__init__(discord.FFmpegPCMAudio(source, **ffmpeg_options))
self.packets_read = 0
self.source = source
self.duration = duration
self.data = data
logger.info(f"Player created for {source}")
@classmethod
async def prepare_file(cls, track: Track, *, loop):
loop = loop or asyncio.get_event_loop()
logger.info(f"Preparing player from file: {track.source}")
return cls(track.source, track.duration, data = track.data, ffmpeg_options = {"options": "-vn"})
@classmethod
async def prepare_stream(cls, track: Track, *, loop):
loop = loop or asyncio.get_event_loop()
logger.info(f"Preparing player from stream: {track.source}")
to_run = partial(ytdl.extract_info, url = track.source, download = False)
data = await loop.run_in_executor(None, to_run)
logger.info(f"Stream URL: {data['url']}")
return cls(data['url'], track.duration, data = data, ffmpeg_options = {"options": "-vn", "before_options": "-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5"})
def __repr__(self):
return ''.join([f"{key=}\n" for key in self.__dict__ if key != "data"])
def __str__(self):
return (
f"{self.original}\n"
f"{self.progress}"
)
def read(self) -> bytes:
data = self.original.read()
if data:
self.packets_read += 1
return data
@property
def elapsed(self) -> float:
return self.packets_read * 0.02 # each packet is 20ms
@property
def progress(self) -> str:
elapsed = format_time(int(self.elapsed))
duration = format_time(int(self.duration)) if self.duration else "?:??"
return f"{elapsed} / {duration}"
@property
def length(self) -> str:
return format_time(int(self.duration))
class Music(Cog):
"""Play audio within a voice channel."""
REPEAT_NONE = 0
REPEAT_ONE = 1
REPEAT_ALL = 2
MAX_RESULTS = 5
PAGE_SIZE = 10
def __init__(self, bot: discord.Bot):
self.bot: discord.Bot = bot
self.q: list[Track] = [] # queue
self.track: Track | None = None
self.repeat_mode = Music.REPEAT_NONE
self.search_results: dict = None
self.i: int = -1 # initial value used for repeat mode
self.h: list[Track] = [] # history
print("Initialized Music cog")
@command(aliases=['start', 'summon', 'connect'])
async def join(self, ctx: Context, *, channel: discord.VoiceChannel = None):
"""Joins a voice channel"""
logger.info(f".join {channel}" if channel else ".join")
if not channel: # Upon a raw "join" command without a channel specified,
if not ctx.author.voice:
msg = await ctx.send(
"You must either be in a voice channel, "
"or specify a voice channel in order to use this command"
)
if msg:
logger.info(f"Message sent: no channel specified, and {ctx.author} is not in a voice channel")
return
channel = ctx.author.voice.channel # bind to your current vc channel.
if ctx.voice_client: # If the bot is in a different channel,
await ctx.voice_client.move_to(channel) # move to your channel.
logger.info(f"existing voice client moved to {channel}")
return
voice_client = await channel.connect() # Finally, join the chosen channel.
if voice_client:
logger.info("voice client created")
@command(aliases=['quit', 'dismiss', 'disconnect'])
async def leave(self, ctx: Context):
"""Stop+disconnect from voice"""
logger.info(".leave")
if ctx.voice_client:
await ctx.voice_client.disconnect()
logger.info("voice client disconnected")
def get_duration_from_file(self, filename: str):
cmd = subprocess.run(
[
'ffprobe', '-v', 'error', '-show_entries', 'format=duration', '-of',
'default=noprint_wrappers=1:nokey=1', filename
],
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT
)
return float(cmd.stdout)
async def get_tracks_from_query(self, ctx: Context, query: str):
logger.debug(f"get_tracks_from_query() called for query: {query}")
# Detect if the track should be downloaded
download = False
if query.endswith('!dl'):
download = True
query = query[:-3]
elif query.endswith('!download'):
download = True
query = query[:-9]
# Handle attachment playback
if query == "file":
logger.info(f"getting tracks from attachment")
return await self.get_tracks_from_attachments(ctx)
# Handle online playback
elif query.startswith('http'):
logger.info(f"getting tracks from url")
return await self.get_tracks_from_url(ctx, query, download=download)
# Handle local playback
elif tracks := await self.get_tracks_from_path(ctx, query):
logger.info(f"getting tracks from path to local file")
return tracks
# Do a youtube search if not found and no prior search
elif not self.search_results:
logger.info(f"performing a search result")
return await self.search_youtube(ctx, query=query)
# Handle prior search
try:
i = int(query) - 1
except ValueError:
logger.info(f"performing a search result")
return await self.search_youtube(ctx, query=query)
if i not in range(self.MAX_RESULTS + 1):
return await ctx.send(f"Please provide an integer between 1 and {self.MAX_RESULTS}")
url = f"https://youtube.com/watch?v={self.search_results[i]['id']}"
self.search_results = []
self.query = ""
logger.info(f"handling a prior search")
return await self.get_tracks_from_url(ctx, url)
async def get_tracks_from_url(self, ctx: Context, url: str, download: bool = False):
logger.debug(f"get_tracks_from_url() called for URL: {url}")
try:
data = ytdl.extract_info(url, download=download)
# logger.debug(f"{data=}")
# Detect tabs
if data['extractor'] == 'youtube:tab' and not "entries" in data:
logger.info("youtube:tab detected, no entries in data (so not a playlist)")
data = ytdl.extract_info(data['url'], download=download) # process the playlist url
logger.debug(f"{data=}")
except Exception as e:
logger.error("Exception thrown!")
logger.error(f"{e=}")
return e
# Detect playlists
entries = [data] # Assume that there is only one song.
if "entries" in data: # If we're wrong, just overwrite our singlet list.
entries = data["entries"] # yapf: disable
# Create Track objects
tracks = []
for entry in entries:
source = entry["url"]
original_url = entry.get("original_url")
title = entry.get("title", "(no title)")
duration = None
data = entry
if not "duration" in data and not "duration_string" in data:
logger.info("duration not found in entry's extracted data -- refetching")
logger.debug(f"{data=}")
start = time()
data = ytdl.extract_info(url, download=download)
logger.info(f"Refetching data took {time() - start} seconds")
if "duration" in data:
duration = data["duration"]
elif "duration_string" in data:
d = [int(x) for x in data["duration_string"].split(':')]
if len(d) == 2:
m,s = d
h = 0
elif len(d) == 3:
h,m,s = d
duration = s + 60*m + 3600*h
tracks.append(
Track(
source=source,
original_url=original_url,
requester=ctx.message.author,
title=title,
duration=duration,
data=data
)
)
logger.info(f"Got {len(tracks)} track(s) from URL")
logger.debug(f"{tracks=}")
return tracks
async def get_tracks_from_path(self, ctx: Context, query: str):
"""Attempt to load a local file from path"""
logger.debug(f"get_tracks_from_path() called for query: {query}")
if "/.." in query:
return None
filename = f"sounds/normalized/{query}"
try:
player = discord.FFmpegPCMAudio(filename)
except:
return None
if player.read():
logger.info("filename is readable from path")
return [
Track(
source=filename,
requester=ctx.message.author,
title=query,
duration=self.get_duration_from_file(filename)
)
]
return None
async def get_tracks_from_attachments(self, ctx: Context):
"""Fetch the attachment URL and convert it to a track"""
logger.debug(f"get_tracks_from_attachment() called")
attachments = ctx.message.attachments
tracks = []
for attachment in attachments:
try:
track = await self.get_tracks_from_url(ctx, attachment.url, download=False)
tracks += track
except Exception as e:
logger.error("Exception thrown!")
logger.error(f"{e=}")
msg = await ctx.send(
f"An error occurred while adding `{attachment.filename}`:\n"
f"```{e.exc_info[1]}```"
)
if msg:
logger.warning("Message sent: An error occurred while adding `{attachment.filename}`")
return e
logger.debug(f"{tracks=}")
return tracks
@command(name='search')
async def search_youtube(self, ctx: Context, *, query: str):
"""Do a YouTube search for the given query"""
logger.debug(f"search_youtube() called for query: {query}")
try:
self.query = query
self.search_results = YoutubeSearch(query, max_results=self.MAX_RESULTS).to_dict() # this returns a list, funnily enough
except Exception as e:
logger.error("Exception thrown!")
logger.error(f"{e=}")
msg = await ctx.send(
f"An error occurred while searching for `{query}`:\n"
f"```{e.exc_info[1]}```"
)
if msg:
logger.warning(f"Message sent: An error occurred while searching for `{query}`")
return e
await self.results(ctx)
@command()
async def results(self, ctx: Context):
"""Show results of a prior search"""
logger.debug(f"results() called")
if not self.search_results:
logger.info("No stored search results")
msg = await ctx.send("There are no stored search results right now.")
if msg:
logger.warning("Message sent: There are no stored search results right now.")
return
embeds = []
formatted_results = (
f"Performed a search for `{self.query}`.\n"
"Which track would you like to play?\n"
"Make your choice using the `play` command, e.g. `.play 1`.\n\n"
)
for i, result in enumerate(self.search_results):
result: dict
title = result.get('title', '<no title found>')
duration = result.get('duration', '?:??')
uploader = result.get('channel', '<no uploader found>')
views = result.get('views', '??? views')
date = result.get('publish_time', 'some time ago')
image = result.get('thumbnails', [None])[-1]
url = f"https://youtube.com/watch?v={result.get('id', '')}"
formatted_results += (
f"{i+1}: **{title}**\n"
f"{uploader} - {views} ({duration})\n"
)
embeds.append(
discord.Embed(
title = title,
description = f"Uploaded by {uploader} {date}\n{duration} - {views}",
url = url,
type = 'image',
colour = 0xff0000,
).set_thumbnail(
url = image,
)
)
msg = await ctx.send(formatted_results, embeds = embeds)
if msg:
logger.info("Message sent: formatted_results")
async def play_next(self, ctx: Context):
logger.debug("play_next() called")
if not ctx.voice_client:
return
if ctx.voice_client.is_playing():
return
logger.info(f"{self.track=}")
if self.repeat_mode == Music.REPEAT_NONE:
if not self.q:
return await ctx.send("Finished playing queue.")
self.track = self.q.pop(0)
logger.info("Repeat none -- popped track from queue")
elif self.repeat_mode == Music.REPEAT_ONE:
self.track = self.track
logger.info("Repeat one -- keeping track the same")
elif self.repeat_mode == Music.REPEAT_ALL:
self.i += 1
if self.i >= len(self.q):
self.i = 0
self.track = self.q[self.i]
logger.info("Repeat all -- advancing pointer without popping track")
logger.info(f"{self.track=}")
if self.track.source.startswith('http'):
# detect private or unplayable videos here
try:
data = ytdl.extract_info(self.track.source, download=False)
logger.debug(f"{data=}")
except Exception as e:
logger.error("Exception thrown!")
logger.error(f"{e=}")
await ctx.send(
f"`{self.track.source}` is unplayable -- skipping...\n"
f"```{e.exc_info[1]}```"
)
logger.warning(f"Skipping as unplayable: {self.track.source}")
return await self.play_next(ctx)
player = await Player.prepare_stream(self.track, loop = self.bot.loop)
else:
player = await Player.prepare_file(self.track, loop = self.bot.loop)
logger.info("playing Player on the voice client")
self.h += [self.track]
while not discord.opus.is_loaded():
for path in [
"/usr/lib/libopus.so",
"/usr/lib/libopus.so.0",
"/usr/lib/x86_64-linux-gnu/libopus.so.0" # debian WHY
]:
try:
discord.opus.load_opus(path)
except:
logger.warning(f"opus could not be loaded from {path} -- trying next?")
logger.error("opus could not be loaded!")
break
ctx.voice_client.play(
player,
after=lambda e: self.after(ctx)
)
def after(self, ctx: Context):
logger.debug("after() called")
if not ctx.voice_client:
logger.info("no voice client -- bot was disconnected from vc?")
self.track = None
self.q = []
asyncio.run_coroutine_threadsafe(
ctx.send(f"Clearing queue after bot left VC"),
self.bot.loop
).result()
logger.info("Cleared queue after bot left VC")
return
if not self.q and self.repeat_mode == Music.REPEAT_NONE:
logger.info("queue empty and not repeating")
self.track = None
asyncio.run_coroutine_threadsafe(
ctx.send(f"Finished playing queue."),
self.bot.loop
).result()
logger.info("Finished playing queue.")
return
if self.q and ctx.voice_client and not ctx.voice_client.is_playing():
logger.info("queue exists and voice client is not playing")
logger.debug(f"{self.q=}")
logger.info("playing next...")
asyncio.run_coroutine_threadsafe(
self.play_next(ctx),
self.bot.loop
).result()
return
def check_for_numbers(self, ctx: Context):
"""anti numbers action"""
logger.debug("check_for_numbers() called")
NUMBERS = 187024083471302656
PIZZA = 320294046935547905
RICKY = 949503750651936828
if ctx.author.id in [
NUMBERS,
PIZZA,
]:
return False
#if ctx.author.id != NUMBERS:
# return False
# if ctx.author.voice:
# members = ctx.author.voice.channel.members
# current_vc_members = {member.id for member in members}
# other_members = current_vc_members - {NUMBERS} - {RICKY}
# return (NUMBERS in current_vc_members) and not other_members
return True
async def add_to_queue(self,
ctx: Context,
query: str,
top: bool = False
):
logger.debug(f"add_to_queue({query}) called")
# Check for permission to add tracks
# allowed = self.check_for_numbers(ctx)
# if not allowed:
# logger.info(f"{ctx.author} is not allowed to add to queue")
# #return await ctx.send(
# #"You must be in a voice chat by yourself "
# #"in order to use this command."
# #)
# return await ctx.send("No 💜")
# Ensure we are connected to voice
if not ctx.voice_client:
logger.warning("no voice client")
if ctx.author.voice:
logger.info(f"moving voice client to {ctx.author.voice.channel}")
await ctx.author.voice.channel.connect()
else:
msg = await ctx.send(
"You are not connected to a voice channel. "
"Use the `join` command with a specified channel "
"or while connected to a channel before trying to "
"play any tracks."
)
if msg:
logger.info("Message sent: author not in voice, and no voice client exists")
return
# Guard against errors
tracks = await self.get_tracks_from_query(ctx, query)
if isinstance(tracks, Exception):
msg = await ctx.send(
f"An error occurred while trying to add `{query}` to the queue:\n"
f"```{tracks}```"
)
if msg:
logger.warning(f"Message sent: An error occurred while trying to add `{query}` to the queue")
return
if not tracks: # a search was performed instead
return
# Add track(s) to queue
if top:
self.q = tracks + self.q
else:
self.q = self.q + tracks
if ctx.voice_client.is_playing():
if top:
msg = await ctx.send(f"Added **{len(tracks)}** track(s) to top of queue.")
if msg:
logger.info(f"Message sent: Added **{len(tracks)}** track(s) to top of queue.")
return
else:
msg = await ctx.send(f"Added **{len(tracks)}** track(s) to queue.")
if msg:
logger.info(f"Message sent: Added **{len(tracks)}** track(s) to queue.")
return
# If not playing, start playing
if len(self.q) == 1:
msg = await ctx.send(f"Playing **{self.q[0].title}**")
if msg:
logger.info(f"Message sent: Playing **{self.q[0].title}**")
else:
msg = await ctx.send(f"Playing {len(tracks)} tracks.")
if msg:
logger.info(f"Message sent: Playing {len(tracks)} tracks.")
await self.play_next(ctx)
@command()
async def ytlogin(self, ctx: Context):
"""Print token and instructions for authorizing with YouTube"""
logger.info(f".ytlogin")
process = subprocess.Popen(
[
"yt-dlp",
"--netrc",
"--skip-download", # we don't care to actually download the video, we just wanna authorize
"https://www.youtube.com/watch?v=dQw4w9WgXcQ" # this can be anything, it just needs a youtube URL
],
stdout=subprocess.PIPE,
encoding="utf-8",
)
while True:
line = process.stdout.readline()
if not line:
logger.debug("Subprocess ran out of lines in stdout")
break
if "google.com" in line:
msg = await ctx.send(line)
if msg:
logger.info(f"Message sent: {line}")
break
timeout = time() + 30 # TODO: make this variable instead of hardcoded?
while (process.poll() is None) and (time() < timeout):
pass # this loop will break when the process exits with a return code, or when the timeout is exceeded
if process.poll() is None:
logger.debug("Process still hasn't exited yet, so we need to kill it and let the user know that the authorization process failed")
process.kill()
text = "Authorization process timed out."
msg = await ctx.send(text)
if msg:
logger.info(f"Message sent: {text}")
return
else:
logger.debug("Process has exited with a return code, so check if the return code indicates no errors.")
if process.returncode == 0:
text = "The process has finished with no errors, so you're probably logged in now."
msg = await ctx.send(text)
if msg:
logger.info(f"Message sent: {text}")
process.kill() # just in case
return
else:
process.kill() # just in case
out, err = process.communicate()
if err:
logger.error(f"An error occurred during the authorization process: {err}")
msg = await ctx.send(
f"An error occurred during the authorization process:\n",
f"```\n",
f"{err}",
f"```",
)
if msg:
logger.info(f"Message sent: An error occurred during the authorization process")
return
@command(aliases=['p', 'listen'])
async def play(self, ctx: Context, *, query: str = ""):
"""Add track(s) to queue"""
if not query:
msg = await ctx.send("No query detected")
if msg:
logger.info("Empty .play command was issued")
return
logger.info(f".play {query}")
return await self.add_to_queue(ctx, query, top=False)
@command(aliases=['ptop', 'top'])
async def playtop(self, ctx: Context, *, query: str = ""):
"""Add tracks to top of queue"""
if not query:
msg = await ctx.send("No query detected")
if msg:
logger.info("Empty .playtop command was issued")
return
logger.info(f".playtop {query}")
return await self.add_to_queue(ctx, query, top=True)
# TODO: repeat once, repeat all, repeat none (repeat/loop command)
# TODO: move positions of songs?
# TODO: cleanup command for clearing songs requested by users not in vc?
# TODO: remove duplicates?
# TODO: remove range of songs
# TODO: restart current song
# TODO: seek command? [no fuckin idea how]
# TODO: skip multiple songs?
# TODO: autoplay???? [???????????]
# TODO: filters? bass boost? nightcore? speed? [probs not]
@command(aliases=['q'])
async def queue(self, ctx: Context, p: int = 1):
"""Show tracks up next"""
logger.info(f".queue {p}" if p else ".queue")
# check that there is a queue and a current track
if not self.q and not self.track:
msg = await ctx.send("The queue is currently empty.")
if msg:
logger.info("Message sent: The queue is currently empty.")
return
# paginate the queue to just one page
full_queue = [self.track] + self.q
start = self.PAGE_SIZE * (p-1)
end = self.PAGE_SIZE * p
queue_page = full_queue[start:end]
# construct header
formatted_results = f"{len(self.q)} tracks on queue.\n"
total_pages = math.ceil(len(full_queue) / self.PAGE_SIZE)
formatted_results += f"Page {p} of {total_pages}:\n"
# construct page
for i, track in enumerate(queue_page):
if p == 1 and i == 0: # print nowplaying on first queue page
formatted_results += "=== Currently playing ===\n"
formatted_results += (
f"{start+i+1}: {track}\n"
)
if p == 1 and i == 0: # add separator on first page for actually queued tracks
formatted_results += "=== Up next ===\n"
# send text to channel
msg = await ctx.send(formatted_results)
if msg:
logger.info("Message sent: Sent queue page to channel")
@command(aliases=['h'])
async def history(self, ctx: Context, limit: int = 10):
"""Show recent actions"""
logger.info(f".history {limit}" if limit else ".history")
if not self.h:
msg = await ctx.send("No available history in this session.")
if msg:
logger.info("Message sent: No available history in this session.")
return
page = self.h[-limit:]
formatted_results = f"Last {len(page)} tracks played:\n"
for i, entry in enumerate(page):
formatted_results += (
f"{i - len(page)}: {entry}\n"
)
msg = await ctx.send(formatted_results)
if msg:
logger.info("Message sent: Sent history page to channel")
@command(aliases=['np'])
async def nowplaying(self, ctx: Context):
"""Show currently playing track"""
logger.info(".nowplaying")
if not self.track:
msg = await ctx.send("Nothing is currently playing")
if msg:
logger.info("Nothing is currently playing")
return
if not ctx.voice_client:
msg = await ctx.send("Bot is not currently connected to a voice channel")
if msg:
logger.info("Bot not connected to VC")
return
source: Player = ctx.voice_client.source
embed = discord.Embed(
title=f"{self.track.title}",
url=f"{self.track.original_url}" if self.track.original_url else None,
).add_field(
name="Progress",
value=f"{source.progress}",
).set_footer(
text=f"Requested by {self.track.requester.display_name} ({self.track.requester})",
icon_url=f"{self.track.requester.display_avatar.url}",
)
thumb = None
if self.track.data and "thumbnail" in self.track.data:
thumb = self.track.data['thumbnail']
elif self.track.data and "thumbnails" in self.track.data:
thumb = self.track.data['thumbnails'][0]['url']
if thumb:
embed.set_thumbnail(
url=thumb
)
msg = await ctx.send(
f"Now playing:\n{self.track}",
embed = embed
)
if msg:
logger.info(f"Message sent: Now playing: {self.track.title}")
@command()
async def skip(self, ctx: Context):
"""Start playing next track"""
logger.info(".skip")
if ctx.voice_client.is_playing():
if self.track:
msg = await ctx.send(f"Skipping: {self.track.title}")
if msg:
logger.info(f"Message sent: Skipping: {self.track.title}")
ctx.voice_client.stop()
@command()
async def remove(self, ctx: Context, i: int):
"""Remove track at given position"""
logger.info(f".remove {i}")
i -= 2 # convert to zero-indexing and also the np track is popped
logger.warning(f"trying to pop index {i}")
logger.warning(f"{self.q[i]=}")
track = self.q.pop(i)
msg = await ctx.send(f"Removed: {track.title}")
if msg:
logger.info(f"Message sent: Removed: {track.title}")
@command()
async def pause(self, ctx: Context):
"""Pause the currently playing track"""
logger.info(".pause")
if ctx.voice_client.is_playing():
ctx.voice_client.pause()
msg = await ctx.send("Playback is paused.")
if msg:
logger.info("Message sent: Playback is paused.")
@command()
async def resume(self, ctx: Context):
"""Resume playback of a paused track"""
logger.info(".resume")
if ctx.voice_client.is_paused():
ctx.voice_client.resume()
msg = await ctx.send("Playback is resumed.")
if msg:
logger.info("Message sent: Playback is resumed.")
@command()
async def shuffle(self, ctx: Context):
"""Randomizes the current queue"""
logger.info(".shuffle")
if not self.q:
return await ctx.send("There is no queue to shuffle")
logger.debug(f"{self.track=}")
logger.debug(f"{self.q=}")
random.shuffle(self.q)
logger.debug(f"{self.track=}")
logger.debug(f"{self.q=}")
msg = await ctx.send("Queue has been shuffled")
if msg:
logger.info("Message sent: Queue has been shuffled")
@command()
async def stop(self, ctx: Context):
"""Clear queue and stop playing"""
logger.info(".stop")
self.q = []
self.track = None
if ctx.voice_client:
if ctx.voice_client.is_connected():
ctx.voice_client.stop()
msg = await ctx.send("Stopped playing tracks and cleared queue.")
if msg:
logger.info("Message sent: Stopped playing tracks and cleared queue.")
@command()
async def clear(self, ctx: Context):
"""Clear queue, but keep playing"""
logger.info(".clear")
self.q = []
msg = await ctx.send("Queue has been cleared.")
if msg:
logger.info("Message sent: Queue has been cleared.")
@command()
async def refresh(self, ctx: Context):
"""Reset bot to fresh state"""
logger.info(".refresh")
self.q = []
self.track = []
self.h = []
msg = await ctx.send("Music bot has been refreshed")
if msg:
logger.info("Message sent: Bot has been refreshed")
@command(aliases=['v', 'vol'])
async def volume(self, ctx: Context, volume: int):
"""Changes the player's volume"""
logger.info(f".volume {volume}")
if ctx.voice_client is None:
return await ctx.send("Not connected to a voice channel.")
if volume not in range(101):
return await ctx.send(f"Please use an integer from 0 to 100")
ctx.voice_client.source.volume = volume / 100
await ctx.send(f"Changed volume to {volume}%")
@command(aliases=['list'])
async def catalogue(self, ctx: Context, subdirectory: str = ""):
"""Shows the available local files"""
logger.info(f".catalogue {subdirectory}" if subdirectory else ".catalogue")
if "../" in subdirectory:
return await ctx.send(f"Nice try, but that won't work.")
path = "."
if subdirectory:
path += f"/{subdirectory}"
cmd = subprocess.run(
f"cd sounds/normalized && find {path} -type f | sort",
shell=True,
stdout=subprocess.PIPE,
stderr=subprocess.STDOUT
)
results = cmd.stdout.decode('utf-8').split('\n')[:-1]
formatted_results = "```"
for result in results:
formatted_results += f"{result[2:]}\n"
formatted_results += "```"
await ctx.send(formatted_results)
"""
Initialize youtube-dl service.
"""
import yt_dlp as youtube_dl
youtube_dl.utils.bug_reports_message = lambda: ""
ytdl_format_options = {
"format": "bestaudio/best",
"outtmpl": ".cache/%(extractor)s-%(id)s-%(title)s.%(ext)s",
"restrictfilenames": True,
"noplaylist": True,
"nocheckcertificate": True,
"ignoreerrors": False,
"logtostderr": False,
"quiet": True,
"no_warnings": True,
"default_search": "auto",
# "source_address": "0.0.0.0", # Bind to ipv4 since ipv6 addresses cause issues
"extract_flat": True, # massive speedup for fetching metadata, at the cost of no upload date
#"cookiefile": "cookies.txt",
"cachedir": "cache",
"usenetrc": True
}
ytdl = youtube_dl.YoutubeDL(ytdl_format_options)