mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 13:34:28 +00:00
93 lines
4 KiB
Python
93 lines
4 KiB
Python
import typing
|
||
import pickle
|
||
import discord
|
||
from royalnet.commands import *
|
||
from royalnet.utils import numberemojiformat
|
||
from royalnet.bots import DiscordBot
|
||
|
||
|
||
class QueueCommand(Command):
|
||
name: str = "queue"
|
||
|
||
aliases = ["q"]
|
||
|
||
description: str = "Visualizza la coda di riproduzione attuale."
|
||
|
||
syntax = "[ [guild] ]"
|
||
|
||
@staticmethod
|
||
async def _legacy_queue_handler(bot: "DiscordBot", guild_name: typing.Optional[str]):
|
||
# Find the matching guild
|
||
if guild_name:
|
||
guilds: typing.List[discord.Guild] = bot.client.find_guild_by_name(guild_name)
|
||
else:
|
||
guilds = bot.client.guilds
|
||
if len(guilds) == 0:
|
||
raise CommandError("No guilds with the specified name found.")
|
||
if len(guilds) > 1:
|
||
raise CommandError("Multiple guilds with the specified name found.")
|
||
guild = list(bot.client.guilds)[0]
|
||
# Check if the guild has a PlayMode
|
||
playmode = bot.music_data.get(guild)
|
||
if not playmode:
|
||
return {
|
||
"type": None
|
||
}
|
||
try:
|
||
queue = playmode.queue_preview()
|
||
except NotImplementedError:
|
||
return {
|
||
"type": playmode.__class__.__name__
|
||
}
|
||
return {
|
||
"type": playmode.__class__.__name__,
|
||
"queue":
|
||
{
|
||
"strings": [str(dfile.info) for dfile in queue],
|
||
"pickled_embeds": str(pickle.dumps([dfile.info.to_discord_embed() for dfile in queue]))
|
||
}
|
||
}
|
||
|
||
_event_name = "_legacy_queue"
|
||
|
||
def __init__(self, interface: CommandInterface):
|
||
super().__init__(interface)
|
||
if interface.name == "discord":
|
||
interface.register_herald_action(self._event_name, self._legacy_queue_handler)
|
||
|
||
async def run(self, args: CommandArgs, data: CommandData) -> None:
|
||
guild_name, = args.match(r"(?:\[(.+)])?")
|
||
response = await self.interface.call_herald_action("discord", self._event_name, {"guild_name": guild_name})
|
||
if response["type"] is None:
|
||
await data.reply("ℹ️ Non c'è nessuna coda di riproduzione attiva al momento.")
|
||
return
|
||
elif "queue" not in response:
|
||
await data.reply(f"ℹ️ La coda di riproduzione attuale ([c]{response['type']}[/c]) non permette l'anteprima.")
|
||
return
|
||
if response["type"] == "Playlist":
|
||
if len(response["queue"]["strings"]) == 0:
|
||
message = f"ℹ️ Questa [c]Playlist[/c] è vuota."
|
||
else:
|
||
message = f"ℹ️ Questa [c]Playlist[/c] contiene {len(response['queue']['strings'])} elementi, e i prossimi saranno:\n"
|
||
elif response["type"] == "Pool":
|
||
if len(response["queue"]["strings"]) == 0:
|
||
message = f"ℹ️ Questo [c]Pool[/c] è vuoto."
|
||
else:
|
||
message = f"ℹ️ Questo [c]Pool[/c] contiene {len(response['queue']['strings'])} elementi, tra cui:\n"
|
||
elif response["type"] == "Layers":
|
||
if len(response["queue"]["strings"]) == 0:
|
||
message = f"ℹ️ Nessun elemento è attualmente in riproduzione, pertanto non ci sono [c]Layers[/c]:"
|
||
else:
|
||
message = f"ℹ️ I [c]Layers[/c] dell'elemento attualmente in riproduzione sono {len(response['queue']['strings'])}, tra cui:\n"
|
||
else:
|
||
if len(response["queue"]["strings"]) == 0:
|
||
message = f"ℹ️ Il PlayMode attuale, [c]{response['type']}[/c], è vuoto.\n"
|
||
else:
|
||
message = f"ℹ️ Il PlayMode attuale, [c]{response['type']}[/c], contiene {len(response['queue']['strings'])} elementi:\n"
|
||
if self.interface.name == "discord":
|
||
await data.reply(message)
|
||
for embed in pickle.loads(eval(response["queue"]["pickled_embeds"]))[:5]:
|
||
await data.message.channel.send(embed=embed)
|
||
else:
|
||
message += numberemojiformat(response["queue"]["strings"][:10])
|
||
await data.reply(message)
|