Compare commits
No commits in common. "186eda49344742682909302775474630f937b0d8" and "f08b9f076633b57ec66b445c5bb7a873f7766326" have entirely different histories.
186eda4934
...
f08b9f0766
@ -11,7 +11,6 @@ class Command(enum.Enum):
|
|||||||
LEAVE = "leave"
|
LEAVE = "leave"
|
||||||
PAUSE = "pause"
|
PAUSE = "pause"
|
||||||
PLAY = "play"
|
PLAY = "play"
|
||||||
PLAYING = "playing"
|
|
||||||
PURGE = "purge"
|
PURGE = "purge"
|
||||||
QUEUE = "queue"
|
QUEUE = "queue"
|
||||||
RELOAD = "reload"
|
RELOAD = "reload"
|
||||||
@ -25,14 +24,6 @@ def match_token(token: str) -> list[Command]:
|
|||||||
if token.lower() == "r":
|
if token.lower() == "r":
|
||||||
return [Command.RELOAD]
|
return [Command.RELOAD]
|
||||||
|
|
||||||
if exact_match := list(
|
|
||||||
filter(
|
|
||||||
lambda command: command.value == token.lower(),
|
|
||||||
Command.__members__.values(),
|
|
||||||
)
|
|
||||||
):
|
|
||||||
return exact_match
|
|
||||||
|
|
||||||
return list(
|
return list(
|
||||||
filter(
|
filter(
|
||||||
lambda command: command.value.startswith(token.lower()),
|
lambda command: command.value.startswith(token.lower()),
|
||||||
|
@ -1,11 +1,7 @@
|
|||||||
import itertools
|
import math
|
||||||
|
|
||||||
import disnake
|
|
||||||
import disnake_paginator
|
|
||||||
|
|
||||||
import arguments
|
import arguments
|
||||||
import commands
|
import commands
|
||||||
import constants
|
|
||||||
import utils
|
import utils
|
||||||
import youtubedl
|
import youtubedl
|
||||||
from state import client, players
|
from state import client, players
|
||||||
@ -54,11 +50,6 @@ async def queue_or_play(message):
|
|||||||
action="store_true",
|
action="store_true",
|
||||||
help="play the specified song immediately",
|
help="play the specified song immediately",
|
||||||
)
|
)
|
||||||
parser.add_argument(
|
|
||||||
"--next",
|
|
||||||
action="store_true",
|
|
||||||
help="play the specified song next",
|
|
||||||
)
|
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"-t",
|
"-t",
|
||||||
"--remove-title",
|
"--remove-title",
|
||||||
@ -70,10 +61,37 @@ async def queue_or_play(message):
|
|||||||
type=int,
|
type=int,
|
||||||
help="remove queued songs by queuer",
|
help="remove queued songs by queuer",
|
||||||
)
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"-d",
|
||||||
|
"--duration",
|
||||||
|
action="store_true",
|
||||||
|
help="print duration of queued songs",
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
"-p",
|
||||||
|
"--page",
|
||||||
|
type=int,
|
||||||
|
default=1,
|
||||||
|
help="print the specified page of the queue",
|
||||||
|
)
|
||||||
if not (args := await parser.parse_args(message, tokens)):
|
if not (args := await parser.parse_args(message, tokens)):
|
||||||
return
|
return
|
||||||
|
|
||||||
if args.clear:
|
if args.duration:
|
||||||
|
queued_songs = players[message.guild.id].queue
|
||||||
|
formatted_duration = utils.format_duration(
|
||||||
|
sum(
|
||||||
|
[
|
||||||
|
queued.player.duration if queued.player.duration else 0
|
||||||
|
for queued in queued_songs
|
||||||
|
]
|
||||||
|
)
|
||||||
|
)
|
||||||
|
await utils.reply(
|
||||||
|
message,
|
||||||
|
f"queue is **{formatted_duration or '0 seconds'}** long (**{len(queued_songs)}** queued)",
|
||||||
|
)
|
||||||
|
elif args.clear:
|
||||||
players[message.guild.id].queue.clear()
|
players[message.guild.id].queue.clear()
|
||||||
await utils.add_check_reaction(message)
|
await utils.add_check_reaction(message)
|
||||||
return
|
return
|
||||||
@ -93,7 +111,7 @@ async def queue_or_play(message):
|
|||||||
targets.append(queued)
|
targets.append(queued)
|
||||||
continue
|
continue
|
||||||
if q := args.remove_queuer:
|
if q := args.remove_queuer:
|
||||||
if q == queued.trigger_message.author.id:
|
if q == queued.queuer:
|
||||||
targets.append(queued)
|
targets.append(queued)
|
||||||
if not args.remove_multiple:
|
if not args.remove_multiple:
|
||||||
targets = targets[:1]
|
targets = targets[:1]
|
||||||
@ -105,25 +123,6 @@ async def queue_or_play(message):
|
|||||||
f"removed **{len(targets)}** queued {'song' if len(targets) == 1 else 'songs'}",
|
f"removed **{len(targets)}** queued {'song' if len(targets) == 1 else 'songs'}",
|
||||||
)
|
)
|
||||||
elif query := args.query:
|
elif query := args.query:
|
||||||
if (
|
|
||||||
not message.channel.permissions_for(message.author).manage_channels
|
|
||||||
and len(
|
|
||||||
list(
|
|
||||||
filter(
|
|
||||||
lambda queued: queued.trigger_message.author.id
|
|
||||||
== message.author.id,
|
|
||||||
players[message.guild.id].queue,
|
|
||||||
)
|
|
||||||
)
|
|
||||||
)
|
|
||||||
>= 5
|
|
||||||
):
|
|
||||||
await utils.reply(
|
|
||||||
message,
|
|
||||||
"you can only queue **5 songs** without the manage channels permission!",
|
|
||||||
)
|
|
||||||
return
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
async with message.channel.typing():
|
async with message.channel.typing():
|
||||||
player = await youtubedl.YTDLSource.from_url(
|
player = await youtubedl.YTDLSource.from_url(
|
||||||
@ -132,13 +131,14 @@ async def queue_or_play(message):
|
|||||||
player.volume = float(args.volume) / 100.0
|
player.volume = float(args.volume) / 100.0
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
await utils.reply(
|
await utils.reply(
|
||||||
message, f"**failed to queue:** `{e}`", suppress_embeds=True
|
message,
|
||||||
|
f"**unable to queue {query}:** {e}",
|
||||||
)
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
queued = youtubedl.QueuedSong(player, message)
|
queued = youtubedl.QueuedSong(player, message.author.id)
|
||||||
|
|
||||||
if args.now or args.next:
|
if args.now:
|
||||||
players[message.guild.id].queue_add_front(queued)
|
players[message.guild.id].queue_add_front(queued)
|
||||||
else:
|
else:
|
||||||
players[message.guild.id].queue_add(queued)
|
players[message.guild.id].queue_add(queued)
|
||||||
@ -157,71 +157,48 @@ async def queue_or_play(message):
|
|||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
if tokens[0].lower() == "play":
|
if tokens[0].lower() == "play":
|
||||||
await resume(message)
|
message.guild.voice_client.resume()
|
||||||
|
await utils.reply(
|
||||||
|
message,
|
||||||
|
"resumed!",
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
if players[message.guild.id].queue:
|
args.page = max(
|
||||||
formatted_duration = utils.format_duration(
|
min(args.page, math.ceil(len(players[message.guild.id].queue) / 10)), 1
|
||||||
sum(
|
)
|
||||||
[
|
queue_list = lambda: "\n".join(
|
||||||
queued.player.duration if queued.player.duration else 0
|
[
|
||||||
for queued in players[message.guild.id].queue
|
f"**{i + 1}.** {queued.format(with_queuer=True, hide_preview=True)}"
|
||||||
]
|
for i, queued in list(enumerate(players[message.guild.id].queue))[
|
||||||
)
|
(args.page - 1) * 10 : args.page * 10
|
||||||
|
]
|
||||||
|
]
|
||||||
|
)
|
||||||
|
currently_playing = (
|
||||||
|
lambda: f"**0.** {'(paused) ' if message.guild.voice_client.is_paused() else ''} {players[message.guild.id].current.format(with_queuer=True)}"
|
||||||
|
)
|
||||||
|
if (
|
||||||
|
not players[message.guild.id].queue
|
||||||
|
and not message.guild.voice_client.source
|
||||||
|
):
|
||||||
|
await utils.reply(
|
||||||
|
message,
|
||||||
|
"nothing is playing or queued!",
|
||||||
|
)
|
||||||
|
elif not players[message.guild.id].queue:
|
||||||
|
await utils.reply(message, currently_playing())
|
||||||
|
elif not message.guild.voice_client.source:
|
||||||
|
await utils.reply(
|
||||||
|
message,
|
||||||
|
queue_list(),
|
||||||
)
|
)
|
||||||
|
|
||||||
def embed(description):
|
|
||||||
e = disnake.Embed(
|
|
||||||
title="Queued",
|
|
||||||
description=description,
|
|
||||||
color=constants.EMBED_COLOR,
|
|
||||||
)
|
|
||||||
if formatted_duration:
|
|
||||||
e.set_footer(text=f"{formatted_duration} long")
|
|
||||||
return e
|
|
||||||
|
|
||||||
await disnake_paginator.ButtonPaginator(
|
|
||||||
invalid_user_function=utils.invalid_user_handler,
|
|
||||||
color=constants.EMBED_COLOR,
|
|
||||||
segments=list(
|
|
||||||
map(
|
|
||||||
embed,
|
|
||||||
[
|
|
||||||
"\n\n".join(
|
|
||||||
[
|
|
||||||
f"**{i + 1}.** {queued.format(show_queuer=True, hide_preview=True, multiline=True)}"
|
|
||||||
for i, queued in batch
|
|
||||||
]
|
|
||||||
)
|
|
||||||
for batch in itertools.batched(
|
|
||||||
enumerate(players[message.guild.id].queue), 10
|
|
||||||
)
|
|
||||||
],
|
|
||||||
)
|
|
||||||
),
|
|
||||||
).start(disnake_paginator.wrappers.MessageInteractionWrapper(message))
|
|
||||||
else:
|
else:
|
||||||
await utils.reply(
|
await utils.reply(
|
||||||
message,
|
message,
|
||||||
"nothing is queued!",
|
currently_playing() + "\n" + queue_list(),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
async def playing(message):
|
|
||||||
if not command_allowed(message):
|
|
||||||
return
|
|
||||||
|
|
||||||
if message.guild.voice_client.source:
|
|
||||||
await utils.reply(
|
|
||||||
message,
|
|
||||||
f"{'(paused) ' if message.guild.voice_client.is_paused() else ''} {players[message.guild.id].current.format(show_queuer=True)}",
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
await utils.reply(
|
|
||||||
message,
|
|
||||||
"nothing is playing!",
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
async def skip(message):
|
async def skip(message):
|
||||||
if not command_allowed(message):
|
if not command_allowed(message):
|
||||||
return
|
return
|
||||||
@ -339,9 +316,7 @@ def play_next(message, once=False):
|
|||||||
message.channel.send(f"error while trying to play: `{e}`")
|
message.channel.send(f"error while trying to play: `{e}`")
|
||||||
)
|
)
|
||||||
return
|
return
|
||||||
client.loop.create_task(
|
client.loop.create_task(message.channel.send(f"**0.** {queued.format()}"))
|
||||||
message.channel.send(f"**0.** {queued.format(show_queuer=True)}")
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
async def ensure_joined(message):
|
async def ensure_joined(message):
|
||||||
|
@ -19,7 +19,6 @@ RELOADABLE_MODULES = [
|
|||||||
]
|
]
|
||||||
|
|
||||||
YTDL_OPTIONS = {
|
YTDL_OPTIONS = {
|
||||||
"color": "never",
|
|
||||||
"default_search": "auto",
|
"default_search": "auto",
|
||||||
"format": "bestaudio/best",
|
"format": "bestaudio/best",
|
||||||
"ignoreerrors": False,
|
"ignoreerrors": False,
|
||||||
|
17
core.py
17
core.py
@ -12,7 +12,7 @@ import commands
|
|||||||
import constants
|
import constants
|
||||||
import core
|
import core
|
||||||
import utils
|
import utils
|
||||||
from state import client, command_locks, executed_messages
|
from state import client, command_locks
|
||||||
|
|
||||||
|
|
||||||
async def on_message(message):
|
async def on_message(message):
|
||||||
@ -80,23 +80,20 @@ async def on_message(message):
|
|||||||
|
|
||||||
if len(output) > 2000:
|
if len(output) > 2000:
|
||||||
output = output.replace("`", "\\`")
|
output = output.replace("`", "\\`")
|
||||||
await disnake_paginator.ButtonPaginator(
|
pager = disnake_paginator.ButtonPaginator(
|
||||||
prefix=f"```\n",
|
prefix=f"```\n",
|
||||||
suffix="```",
|
suffix="```",
|
||||||
invalid_user_function=utils.invalid_user_handler,
|
|
||||||
color=constants.EMBED_COLOR,
|
color=constants.EMBED_COLOR,
|
||||||
segments=disnake_paginator.split(output),
|
segments=disnake_paginator.split(output),
|
||||||
).start(
|
invalid_user_function=utils.invalid_user_handler,
|
||||||
|
)
|
||||||
|
await pager.start(
|
||||||
disnake_paginator.wrappers.MessageInteractionWrapper(message)
|
disnake_paginator.wrappers.MessageInteractionWrapper(message)
|
||||||
)
|
)
|
||||||
elif len(output.strip()) == 0:
|
elif len(output.strip()) == 0:
|
||||||
await utils.add_check_reaction(message)
|
await utils.add_check_reaction(message)
|
||||||
else:
|
else:
|
||||||
if message.id in executed_messages:
|
await message.channel.send(output)
|
||||||
await executed_messages[message.id].edit(output)
|
|
||||||
else:
|
|
||||||
response = await message.channel.send(output)
|
|
||||||
executed_messages[message.id] = response
|
|
||||||
case C.CLEAR | C.PURGE if message.author.id in constants.OWNERS:
|
case C.CLEAR | C.PURGE if message.author.id in constants.OWNERS:
|
||||||
await commands.tools.clear(message)
|
await commands.tools.clear(message)
|
||||||
case C.JOIN:
|
case C.JOIN:
|
||||||
@ -119,8 +116,6 @@ async def on_message(message):
|
|||||||
await commands.bot.help(message)
|
await commands.bot.help(message)
|
||||||
case C.UPTIME:
|
case C.UPTIME:
|
||||||
await commands.bot.uptime(message)
|
await commands.bot.uptime(message)
|
||||||
case C.PLAYING:
|
|
||||||
await commands.voice.playing(message)
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
await utils.reply(
|
await utils.reply(
|
||||||
message,
|
message,
|
||||||
|
26
events.py
26
events.py
@ -16,32 +16,24 @@ async def trigger_dynamic_handlers(event_type: str, *data):
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@client.event
|
||||||
|
async def on_message_edit(before, after):
|
||||||
|
await events.trigger_dynamic_handlers("on_message_edit", before, after)
|
||||||
|
|
||||||
|
await core.on_message(after)
|
||||||
|
|
||||||
|
|
||||||
|
@client.event
|
||||||
async def on_message(message):
|
async def on_message(message):
|
||||||
await events.trigger_dynamic_handlers("on_message", message)
|
await events.trigger_dynamic_handlers("on_message", message)
|
||||||
|
|
||||||
await core.on_message(message)
|
await core.on_message(message)
|
||||||
|
|
||||||
|
|
||||||
async def on_message_edit(before, after):
|
@client.event
|
||||||
await events.trigger_dynamic_handlers("on_message_edit", before, after)
|
|
||||||
|
|
||||||
if before.content == after.content:
|
|
||||||
return
|
|
||||||
|
|
||||||
await core.on_message(after)
|
|
||||||
|
|
||||||
|
|
||||||
async def on_voice_state_update(member, before, after):
|
async def on_voice_state_update(member, before, after):
|
||||||
await events.trigger_dynamic_handlers(
|
await events.trigger_dynamic_handlers(
|
||||||
"on_voice_state_update", member, before, after
|
"on_voice_state_update", member, before, after
|
||||||
)
|
)
|
||||||
|
|
||||||
await core.on_voice_state_update(member, before, after)
|
await core.on_voice_state_update(member, before, after)
|
||||||
|
|
||||||
|
|
||||||
for k, v in client.get_listeners().items():
|
|
||||||
for f in v:
|
|
||||||
client.remove_listener(f, k)
|
|
||||||
client.add_listener(on_message, "on_message")
|
|
||||||
client.add_listener(on_message_edit, "on_message_edit")
|
|
||||||
client.add_listener(on_voice_state_update, "on_voice_state_update")
|
|
||||||
|
3
state.py
3
state.py
@ -2,9 +2,8 @@ import time
|
|||||||
|
|
||||||
import disnake
|
import disnake
|
||||||
|
|
||||||
command_locks = {}
|
|
||||||
executed_messages = {}
|
|
||||||
players = {}
|
players = {}
|
||||||
|
command_locks = {}
|
||||||
|
|
||||||
intents = disnake.Intents.default()
|
intents = disnake.Intents.default()
|
||||||
intents.message_content = True
|
intents.message_content = True
|
||||||
|
6
utils.py
6
utils.py
@ -33,10 +33,8 @@ async def add_check_reaction(message):
|
|||||||
await message.add_reaction("✅")
|
await message.add_reaction("✅")
|
||||||
|
|
||||||
|
|
||||||
async def reply(message, *args, **kwargs):
|
async def reply(message, *args):
|
||||||
await message.reply(
|
await message.reply(*args, allowed_mentions=disnake.AllowedMentions.none())
|
||||||
*args, **kwargs, allowed_mentions=disnake.AllowedMentions.none()
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
async def invalid_user_handler(interaction):
|
async def invalid_user_handler(interaction):
|
||||||
|
69
youtubedl.py
69
youtubedl.py
@ -1,6 +1,5 @@
|
|||||||
import asyncio
|
import asyncio
|
||||||
import collections
|
import collections
|
||||||
from dataclasses import dataclass
|
|
||||||
from typing import Any, Optional
|
from typing import Any, Optional
|
||||||
|
|
||||||
import disnake
|
import disnake
|
||||||
@ -51,43 +50,10 @@ class YTDLSource(disnake.PCMVolumeTransformer):
|
|||||||
return self.__repr__()
|
return self.__repr__()
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class QueuedSong:
|
|
||||||
player: YTDLSource
|
|
||||||
trigger_message: disnake.Message
|
|
||||||
|
|
||||||
def format(self, show_queuer=False, hide_preview=False, multiline=False) -> str:
|
|
||||||
if multiline:
|
|
||||||
return (
|
|
||||||
f"[`{self.player.title}`]({'<' if hide_preview else ''}{self.player.original_url}{'>' if hide_preview else ''})\n**duration:** {self.format_duration(self.player.duration) if self.player.duration else '[live]'}"
|
|
||||||
+ (
|
|
||||||
f", **queuer:** <@{self.trigger_message.author.id}>"
|
|
||||||
if show_queuer
|
|
||||||
else ""
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
return (
|
|
||||||
f"[`{self.player.title}`]({'<' if hide_preview else ''}{self.player.original_url}{'>' if hide_preview else ''}) [{self.format_duration(self.player.duration) if self.player.duration else 'live'}]"
|
|
||||||
+ (f" (<@{self.trigger_message.author.id}>)" if show_queuer else "")
|
|
||||||
)
|
|
||||||
|
|
||||||
def format_duration(self, duration: int) -> str:
|
|
||||||
hours, duration = divmod(duration, 3600)
|
|
||||||
minutes, duration = divmod(duration, 60)
|
|
||||||
segments = [hours, minutes, duration]
|
|
||||||
if len(segments) == 3 and segments[0] == 0:
|
|
||||||
del segments[0]
|
|
||||||
return f"{':'.join(f'{s:0>2}' for s in segments)}"
|
|
||||||
|
|
||||||
def __str__(self):
|
|
||||||
return self.__repr__()
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
|
||||||
class QueuedPlayer:
|
class QueuedPlayer:
|
||||||
queue = collections.deque()
|
def __init__(self):
|
||||||
current: Optional[QueuedSong] = None
|
self.queue = collections.deque()
|
||||||
|
self.current = None
|
||||||
|
|
||||||
def queue_pop(self):
|
def queue_pop(self):
|
||||||
popped = self.queue.popleft()
|
popped = self.queue.popleft()
|
||||||
@ -100,6 +66,35 @@ class QueuedPlayer:
|
|||||||
def queue_add_front(self, item):
|
def queue_add_front(self, item):
|
||||||
self.queue.appendleft(item)
|
self.queue.appendleft(item)
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return f"<QueuedPlayer current={self.current} queue={self.queue}>"
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
return self.__repr__()
|
||||||
|
|
||||||
|
|
||||||
|
class QueuedSong:
|
||||||
|
def __init__(self, player, queuer):
|
||||||
|
self.player = player
|
||||||
|
self.queuer = queuer
|
||||||
|
|
||||||
|
def format(self, with_queuer=False, hide_preview=False) -> str:
|
||||||
|
return (
|
||||||
|
f"[`{self.player.title}`]({'<' if hide_preview else ''}{self.player.original_url}{'>' if hide_preview else ''}) [{self.format_duration(self.player.duration) if self.player.duration else 'live'}]"
|
||||||
|
+ (f" (<@{self.queuer}>)" if with_queuer else "")
|
||||||
|
)
|
||||||
|
|
||||||
|
def format_duration(self, duration: int) -> str:
|
||||||
|
hours, duration = divmod(duration, 3600)
|
||||||
|
minutes, duration = divmod(duration, 60)
|
||||||
|
segments = [hours, minutes, duration]
|
||||||
|
if len(segments) == 3 and segments[0] == 0:
|
||||||
|
del segments[0]
|
||||||
|
return f"{':'.join(f'{s:0>2}' for s in segments)}"
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return f"<QueuedSong player={self.player} queuer={self.queuer}>"
|
||||||
|
|
||||||
def __str__(self):
|
def __str__(self):
|
||||||
return self.__repr__()
|
return self.__repr__()
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user