mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 13:34:28 +00:00
76 lines
3.4 KiB
Python
76 lines
3.4 KiB
Python
import typing
|
|
import pickle
|
|
import datetime
|
|
import discord
|
|
from royalnet.commands import *
|
|
from royalnet.utils import asyncify
|
|
from royalnet.audio import YtdlDiscord
|
|
from royalnet.bots import DiscordBot
|
|
|
|
|
|
class YoutubeCommand(Command):
|
|
name: str = "youtube"
|
|
|
|
aliases = ["yt"]
|
|
|
|
description: str = "Cerca un video su YouTube e lo aggiunge alla coda della chat vocale."
|
|
|
|
syntax = "[ [guild] ] {url}"
|
|
|
|
@classmethod
|
|
async def _legacy_youtube_handler(cls, bot: "DiscordBot", guild_name: typing.Optional[str], search: 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("Server non trovato.")
|
|
if len(guilds) > 1:
|
|
raise CommandError("Il nome del server è ambiguo.")
|
|
guild = list(bot.client.guilds)[0]
|
|
# Ensure the guild has a PlayMode before adding the file to it
|
|
if not bot.music_data.get(guild):
|
|
raise CommandError("Il bot non è in nessun canale vocale.")
|
|
# Create url
|
|
ytdl_args = {
|
|
"format": "bestaudio/best",
|
|
"outtmpl": f"./downloads/{datetime.datetime.now().timestamp()}_%(title)s.%(ext)s"
|
|
}
|
|
# Start downloading
|
|
dfiles: typing. List[YtdlDiscord] = await asyncify(YtdlDiscord.create_from_url, f'ytsearch:{search}', **ytdl_args)
|
|
await bot.add_to_music_data(dfiles, guild)
|
|
# Create response dictionary
|
|
return {
|
|
"videos": [{
|
|
"title": dfile.info.title,
|
|
"discord_embed_pickle": str(pickle.dumps(dfile.info.to_discord_embed()))
|
|
} for dfile in dfiles]
|
|
}
|
|
|
|
_event_name = "_legacy_youtube"
|
|
|
|
def __init__(self, interface: CommandInterface):
|
|
super().__init__(interface)
|
|
if interface.name == "discord":
|
|
interface.register_herald_action(self._event_name, self._legacy_youtube_handler)
|
|
|
|
async def run(self, args: CommandArgs, data: CommandData) -> None:
|
|
guild_name, search = args.match(r"(?:\[(.+)])?\s*<?(.+)>?")
|
|
if search.startswith("http://") or search.startswith("https://"):
|
|
raise CommandError(f"Il comando [c]{self.interface.prefix}youtube[/c] funziona solo per cercare video su"
|
|
f" YouTube con un dato nome.\n"
|
|
f"Se vuoi riprodurre una canzone da un URL, usa [c]{self.interface.prefix}play[/c]!")
|
|
response = await self.interface.call_herald_action("discord", self._event_name, {
|
|
"guild_name": guild_name,
|
|
"search": search
|
|
})
|
|
if len(response["videos"]) == 0:
|
|
raise CommandError(f"Il video non può essere scaricato a causa di un blocco imposto da YouTube.")
|
|
for video in response["videos"]:
|
|
if self.interface.name == "discord":
|
|
# This is one of the unsafest things ever
|
|
embed = pickle.loads(eval(video["discord_embed_pickle"]))
|
|
await data.message.channel.send(content="▶️ Aggiunto alla coda:", embed=embed)
|
|
else:
|
|
await data.reply(f"▶️ Aggiunto alla coda: [i]{video['title']}[/i]")
|