royalnet/royalpack/events/discord_queue.py

import discord
import pickle
import base64
from typing import *
import royalnet.commands as rc
from royalnet.serf.discord import *
from ..utils import RoyalQueue


class DiscordQueueEvent(rc.Event):
    name = "discord_queue"

    async def run(self,
                  guild_id: Optional[int] = None,
                  **kwargs) -> dict:
        if not isinstance(self.serf, DiscordSerf):
            raise rc.UnsupportedError()
        client: discord.Client = self.serf.client
        if len(self.serf.voice_players) == 1:
            voice_player: VoicePlayer = self.serf.voice_players[0]
        else:
            if guild_id is None:
                # TODO: trovare un modo per riprodurre canzoni su più server da Telegram
                raise rc.InvalidInputError("Non so in che Server riprodurre questo file...\n"
                                        "Invia il comando su Discord, per favore!")
            guild: discord.Guild = client.get_guild(guild_id)
            if guild is None:
                raise rc.InvalidInputError("Impossibile trovare il Server specificato.")
            candidate_players = self.serf.find_voice_players(guild)
            if len(candidate_players) == 0:
                raise rc.UserError("Il bot non è in nessun canale vocale.\n"
                                   "Evocalo prima con [c]summon[/c]!")
            elif len(candidate_players) == 1:
                voice_player = candidate_players[0]
            else:
                raise rc.CommandError("Non so di che Server visualizzare la coda...\n"
                                      "Invia il comando su Discord, per favore!")
        if isinstance(voice_player.playing, RoyalQueue):
            now_playing = voice_player.playing.now_playing
            return {
                "type": f"{voice_player.playing.__class__.__qualname__}",
                "now_playing": {
                    "title": now_playing.info.title,
                    "stringified_base64_pickled_discord_embed": str(base64.b64encode(pickle.dumps(now_playing.embed())),
                                                                    encoding="ascii")
                } if now_playing is not None else None,
                "next_up": [{
                    "title": ytd.info.title,
                    "stringified_base64_pickled_discord_embed": str(base64.b64encode(pickle.dumps(ytd.embed())),
                                                                    encoding="ascii")
                } for ytd in voice_player.playing.contents]
            }
        else:
            raise rc.CommandError(f"Non so come visualizzare il contenuto di un "
                               f"[c]{voice_player.playing.__class__.__qualname__}[/c].")