mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-24 03:54:20 +00:00
92 lines
3.8 KiB
Python
92 lines
3.8 KiB
Python
|
import typing
|
||
|
|
import discord
|
||
|
|
import asyncio
|
||
|
|
import datetime
|
||
|
|
from royalnet.commands import *
|
||
|
|
from royalnet.utils import asyncify
|
||
|
|
from royalnet.audio import YtdlDiscord
|
||
|
|
from royalnet.audio.playmodes import Playlist
|
||
|
|
from royalnet.bots import DiscordBot
|
||
|
|
|
||
|
|
|
||
|
|
class ZawarudoCommand(Command):
|
||
|
|
name: str = "zawarudo"
|
||
|
|
|
||
|
|
aliases = ["theworld", "world"]
|
||
|
|
|
||
|
|
description: str = "Ferma il tempo!"
|
||
|
|
|
||
|
|
syntax = "[ [guild] ] [1-9]"
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
async def _legacy_zawarudo_handler(bot: "DiscordBot", guild_name: typing.Optional[str], time: int):
|
||
|
|
# 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",
|
||
|
|
"outtmpl": f"./downloads/{datetime.datetime.now().timestamp()}_%(title)s.%(ext)s"
|
||
|
|
}
|
||
|
|
# Start downloading
|
||
|
|
zw_start: typing.List[YtdlDiscord] = await asyncify(YtdlDiscord.create_from_url,
|
||
|
|
"https://scaleway.steffo.eu/jojo/zawarudo_intro.mp3",
|
||
|
|
**ytdl_args)
|
||
|
|
zw_end: typing.List[YtdlDiscord] = await asyncify(YtdlDiscord.create_from_url,
|
||
|
|
"https://scaleway.steffo.eu/jojo/zawarudo_outro.mp3",
|
||
|
|
**ytdl_args)
|
||
|
|
old_playlist = bot.music_data[guild]
|
||
|
|
bot.music_data[guild].playmode = Playlist()
|
||
|
|
# Get voice client
|
||
|
|
vc: discord.VoiceClient = bot.client.find_voice_client_by_guild(guild)
|
||
|
|
channel: discord.VoiceChannel = vc.channel
|
||
|
|
affected: typing.List[typing.Union[discord.User, discord.Member]] = channel.members
|
||
|
|
await bot.add_to_music_data(zw_start, guild)
|
||
|
|
for member in affected:
|
||
|
|
if member.bot:
|
||
|
|
continue
|
||
|
|
await member.edit(mute=True)
|
||
|
|
await asyncio.sleep(time)
|
||
|
|
await bot.add_to_music_data(zw_end, guild)
|
||
|
|
for member in affected:
|
||
|
|
member: typing.Union[discord.User, discord.Member]
|
||
|
|
if member.bot:
|
||
|
|
continue
|
||
|
|
await member.edit(mute=False)
|
||
|
|
bot.music_data[guild] = old_playlist
|
||
|
|
await bot.advance_music_data(guild)
|
||
|
|
return {}
|
||
|
|
|
||
|
|
_event_name = "_legacy_zawarudo"
|
||
|
|
|
||
|
|
def __init__(self, interface: CommandInterface):
|
||
|
|
super().__init__(interface)
|
||
|
|
if interface.name == "discord":
|
||
|
|
interface.register_herald_action(self._event_name, self._legacy_zawarudo_handler)
|
||
|
|
|
||
|
|
async def run(self, args: CommandArgs, data: CommandData) -> None:
|
||
|
|
guild_name, time = args.match(r"(?:\[(.+)])?\s*(.+)?")
|
||
|
|
if time is None:
|
||
|
|
time = 5
|
||
|
|
else:
|
||
|
|
time = int(time)
|
||
|
|
if time < 1:
|
||
|
|
raise InvalidInputError("The World can't stop time for less than a second.")
|
||
|
|
if time > 10:
|
||
|
|
raise InvalidInputError("The World can stop time only for 10 seconds.")
|
||
|
|
await data.reply(f"🕒 ZA WARUDO! TOKI WO TOMARE!")
|
||
|
|
await self.interface.call_herald_action("discord", self._event_name, {
|
||
|
|
"guild_name": guild_name,
|
||
|
|
"time": time
|
||
|
|
})
|