royalnet/discordbot.py

import random
import re
# discord.py has a different name
# noinspection PyPackageRequirements
import discord
# noinspection PyPackageRequirements
import discord.opus
# noinspection PyPackageRequirements
import discord.voice_client
import functools
import sys
import db
import youtube_dl
import concurrent.futures
import typing
import os
import asyncio
import configparser
import async_timeout
import raven
import logging
import datetime
import sqlalchemy.exc
import coloredlogs
import errors
import math
import enum

logging.getLogger().disabled = True
logger = logging.getLogger(__name__)
os.environ["COLOREDLOGS_LOG_FORMAT"] = "%(asctime)s %(levelname)s %(name)s %(message)s"
coloredlogs.install(level="DEBUG", logger=logger)

# Number emojis from one to ten
number_emojis = [":one:",
                 ":two:",
                 ":three:",
                 ":four:",
                 ":five:",
                 ":six:",
                 ":seven:",
                 ":eight:",
                 ":nine:",
                 ":keycap_ten:"]

# Init the event loop
loop = asyncio.get_event_loop()

# FFmpeg settings
ffmpeg_settings = {}

# Init the executor
executor = concurrent.futures.ThreadPoolExecutor(max_workers=3)


class Succ:
    """All calls to this class return itself."""

    def __bool__(self):
        return False

    def __getattr__(self, attr):
        return Succ()

    def __call__(self, *args, **kwargs):
        return Succ()

    def __str__(self):
        return "succ"

    def __repr__(self):
        return "<Succ>"


class Video:
    def __init__(self, enqueuer: typing.Optional[discord.Member] = None):
        self.is_ready = False
        self.name = None
        self.enqueuer = enqueuer
        self.audio_source = None

    def __str__(self):
        if self.name is None:
            return "_Untitled_"
        return self.name

    def plain_text(self):
        """Title without formatting to be printed on terminals."""
        if self.name is None:
            return "Untitled"
        return self.name

    def database_text(self):
        """The text to be stored in the database for the stats. Usually the same as plain_text()."""
        if self.name is None:
            raise errors.VideoHasNoName()
        return self.name

    def __repr__(self):
        return f"<Video {self.name} ({'' if self.is_ready else 'not '}ready) added by {self.enqueuer}>"

    def ready_up(self):
        """Prepare the video for playback in some way. For example, download it."""
        raise NotImplementedError()

    def make_audio_source(self):
        """Create an AudioSource to be played through Discord, and store and return it."""
        raise NotImplementedError()

    def get_suggestion(self):
        """Get the next suggested video, to be used when the queue is in LoopMode.FOLLOW_SUGGESTION"""
        raise NotImplementedError()


# this is not an abstract class pycharm wtf
# noinspection PyAbstractClass
class YoutubeDLVideo(Video):
    """A file sourcing from YoutubeDL."""

    def __init__(self, url, enqueuer: typing.Optional[discord.Member] = None):
        super().__init__(enqueuer)
        self.url = url
        self.info = None

    def get_info(self):
        """Get info about the video."""
        if self.info:
            return
        with youtube_dl.YoutubeDL({"quiet": True,
                                   "ignoreerrors": True,
                                   "simulate": True}) as ytdl:
            data = ytdl.extract_info(url=self.url, download=False)
        if data is None:
            raise errors.VideoInfoExtractionFailed()
        if "entries" in data:
            raise errors.VideoIsPlaylist()
        self.info = data
        self.name = data.get("title")

    def __str__(self):
        if self.info is None:
            return f"`{self.url}`"
        return f"_{self.name}_"

    def plain_text(self):
        if self.info is None:
            return self.url
        if not self.name.isprintable():
            return self.url
        return self.name

    def get_filename(self):
        """Generate the filename of the video."""
        if self.info is None:
            raise errors.VideoInfoUnknown()
        return "./opusfiles/{}.opus".format(re.sub(r'[/\\?*"<>|!:]', "_", self.info.get("title", self.info["id"])))

    def ready_up(self):
        """Download the video."""
        # Skip download if it is already ready
        if self.is_ready:
            return
        # Retrieve info about the video
        self.get_info()
        # Check if the file to download already exists
        if os.path.exists(self.get_filename()):
            self.is_ready = True
            return
        # Download the file
        logger.info(f"Starting youtube_dl download of {repr(self)}")
        with youtube_dl.YoutubeDL({"noplaylist": True,
                                   "format": "best",
                                   "postprocessors": [{
                                       "key": 'FFmpegExtractAudio',
                                       "preferredcodec": 'opus'
                                   }],
                                   "outtmpl": self.get_filename(),
                                   "quiet": True}) as ytdl:
            ytdl.download([self.url])
        logger.info(f"Completed youtube_dl download of {repr(self)}")
        self.is_ready = True

    def make_audio_source(self):
        if not self.is_ready:
            raise errors.VideoIsNotReady()
        self.audio_source = discord.PCMVolumeTransformer(discord.FFmpegPCMAudio(self.get_filename(), **ffmpeg_settings))
        return self.audio_source


class LoopMode(enum.Enum):
    NORMAL = enum.auto()
    LOOP_QUEUE = enum.auto()
    LOOP_SINGLE = enum.auto()
    FOLLOW_SUGGESTIONS = enum.auto()
    AUTO_SHUFFLE = enum.auto()
    LOOPING_SHUFFLE = enum.auto()


class VideoQueue:
    """The queue of videos to be played."""

    def __init__(self):
        self.list: typing.List[Video] = []
        self.now_playing: typing.Optional[Video] = None
        self.loop_mode = LoopMode.NORMAL

    def __len__(self) -> int:
        return len(self.list)

    def __next__(self) -> Video:
        video = self.next_video()
        self.advance_queue()
        return video

    def __repr__(self) -> str:
        return f"<VideoQueue of length {len(self)}>"

    def add(self, video: Video, position: int = None) -> None:
        if position is None:
            self.list.append(video)
            return
        self.list.insert(position, video)

    def advance_queue(self):
        """Advance the queue to the next video."""
        if self.loop_mode == LoopMode.NORMAL:
            try:
                self.now_playing = self.list.pop(0)
            except IndexError:
                self.now_playing = None
        elif self.loop_mode == LoopMode.LOOP_QUEUE:
            self.add(self.list[0])
            self.now_playing = self.list.pop(0)
        elif self.loop_mode == LoopMode.LOOP_SINGLE:
            pass
        elif self.loop_mode == LoopMode.FOLLOW_SUGGESTIONS:
            if self.now_playing is None:
                self.now_playing = None
                return
            self.now_playing = self.now_playing.get_suggestion()
        elif self.loop_mode == LoopMode.AUTO_SHUFFLE:
            self.shuffle()
            try:
                self.now_playing = self.list.pop(0)
            except IndexError:
                self.now_playing = None
        elif self.loop_mode == LoopMode.LOOPING_SHUFFLE:
            self.shuffle()
            self.add(self.list[0])
            self.now_playing = self.list.pop(0)
    
    def next_video(self) -> typing.Optional[Video]:
        if len(self.list) == 0:
            return None
        return self.list[0]

    def shuffle(self):
        random.shuffle(self.list)

    def clear(self):
        self.list = []

    def find_video(self, name: str) -> typing.Optional[Video]:
        """Returns the first video with a certain name."""
        for video in self.list:
            if name in video.name:
                return video
        return None

    def not_ready_videos(self, limit: typing.Optional[int] = None):
        """Return the non-ready videos in the first limit positions of the queue."""
        video_list = []
        for video in (self.list[:limit] + ([self.now_playing] if self.now_playing else [])):
            if not video.is_ready:
                video_list.append(video)
        return video_list

    def __getitem__(self, index: int) -> Video:
        """Get an element from the list."""
        return self.list[index]


def escape(message: str):
    return message.replace("<", "&lt;").replace(">", "&gt;")


def command(func):
    """Decorator. Runs the function as a Discord command."""

    async def new_func(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str], *args,
                       **kwargs):
        if author is not None:
            self.sentry.user_context({
                "discord_id": author.id,
                "username": f"{author.name}#{author.discriminator}"
            })
        else:
            self.sentry.user_context({
                "source": "Telegram"
            })
        try:
            result = await func(self, channel=channel, author=author, params=params, *args, **kwargs)
        except Exception:
            ei = sys.exc_info()
            # noinspection PyBroadException
            try:
                await channel.send(f"☢ **ERRORE DURANTE L'ESECUZIONE DEL COMANDO {params[0]}**\n"
                                   f"Il comando è stato ignorato.\n"
                                   f"Una segnalazione di errore è stata automaticamente mandata a Steffo.\n\n"
                                   f"Dettagli dell'errore:\n"
                                   f"```python\n"
                                   f"{repr(ei[1])}\n"
                                   f"```")
            except Exception:
                pass
            self.sentry.captureException(exc_info=ei)
        else:
            return result

    return new_func


def requires_connected_voice_client(func):
    """Decorator. Ensures the voice client is connected before running the command."""

    async def new_func(self: "RoyalDiscordBot", channel: discord.TextChannel, author: discord.Member,
                       params: typing.List[str], *args, **kwargs):
        for voice_client in self.voice_clients:
            if voice_client.channel in self.main_guild.channels and voice_client.is_connected():
                break
        else:
            await channel.send("⚠️ Non sono connesso alla cv!\n"
                               "Fammi entrare scrivendo `!cv` mentre sei in chat vocale.")
            return
        return await func(self, channel=channel, author=author, params=params, *args, **kwargs)

    return new_func


def requires_rygdb(func, optional=False):
    async def new_func(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str], *args,
                       **kwargs):
        session = db.Session()
        dbuser = await loop.run_in_executor(executor,
                                            session.query(db.Discord)
                                            .filter_by(discord_id=author.id)
                                            .join(db.Royal)
                                            .first)
        await loop.run_in_executor(executor, session.close)
        if not optional and dbuser is None:
            await channel.send("⚠️ Devi essere registrato su Royalnet per poter utilizzare questo comando.")
            return
        return await func(self, channel=channel, author=author, params=params, dbuser=dbuser, *args, **kwargs)

    return new_func


class RoyalDiscordBot(discord.Client):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.main_channel: typing.Optional[discord.TextChannel] = None
        self.main_guild: typing.Optional[discord.Guild] = None
        self.commands = {
            "!ping": self.cmd_ping,
            "!cv": self.cmd_cv,
            "!summon": self.cmd_cv,
            "!play": self.cmd_play,
            "!alexaplay": self.cmd_play,
            "!okgoogleplay": self.cmd_play,
            "!heysiriplay": self.cmd_play,
            "!p": self.cmd_play,
            "!search": self.cmd_play,
            "!file": self.cmd_play,
            "!skip": self.cmd_skip,
            "!s": self.cmd_skip,
            "!next": self.cmd_skip,
            "!remove": self.cmd_remove,
            "!r": self.cmd_remove,
            "!cancel": self.cmd_remove,
            "!queue": self.cmd_queue,
            "!q": self.cmd_queue,
            "!shuffle": self.cmd_shuffle,
            "!clear": self.cmd_clear,
            "!register": self.cmd_register,
            "!radiomessages": self.cmd_radiomessages,
            "!yes": self.null,
            "!no": self.null,
            "!pause": self.cmd_pause,
            "!resume": self.cmd_resume,
            "!loop": self.cmd_mode,
            "!l": self.cmd_mode,
            "!mode": self.cmd_mode,
            "!m": self.cmd_mode
        }
        self.video_queue: VideoQueue = VideoQueue()
        self.load_config("config.ini")
        if self.sentry_token:
            self.sentry = raven.Client(self.sentry_token,
                                       release=raven.fetch_git_sha(os.path.dirname(__file__)),
                                       install_logging_hook=False,
                                       hook_libraries=[])
        else:
            logger.warning("Sentry not set, ignoring all calls to it.")
            self.sentry = Succ()
        self.inactivity_timer = 0

    # noinspection PyAttributeOutsideInit
    def load_config(self, filename):
        # Init the config reader
        config = configparser.ConfigParser()
        config.read(filename)
        # Token
        try:
            self.token = config["Discord"]["bot_token"]
        except (KeyError, ValueError):
            raise errors.InvalidConfigError("Missing Discord bot token.")
        # Main channels, will be fully loaded when ready
        try:
            self.main_guild_id = int(config["Discord"]["server_id"])
            self.main_channel_id = int(config["Discord"]["main_channel"])
        except (KeyError, ValueError):
            raise errors.InvalidConfigError("Missing main guild and channel ids.")
        # Max enqueable video duration
        # Defined in the YoutubeDLVideo class
        # Max videos to predownload
        try:
            self.max_videos_to_predownload = int(config["Video"]["cache_size"])
        except (KeyError, ValueError):
            logger.warning("Max videos to predownload is not set, setting it to infinity.")
            self.max_videos_to_predownload = None
        # Max time to ready a video
        try:
            self.max_video_ready_time = int(config["Video"]["max_ready_time"])
        except (KeyError, ValueError):
            logger.warning("Max time to ready a video is not set, setting it to infinity. ")
            self.max_video_ready_time = math.inf
        # Radio messages
        try:
            if config["Discord"]["radio_messages_enabled"] == "True":
                self.radio_messages = ["https://www.youtube.com/watch?v=3-yeK1Ck4yk",
                                       "https://youtu.be/YcR7du_A1Vc",
                                       "https://clyp.it/byg3i52l"]
                try:
                    self.radio_messages_every = int(config["Discord"]["radio_messages_every"])
                    self.radio_messages_next_in = self.radio_messages_every
                except (KeyError, ValueError):
                    logger.warning("Radio messages config error, disabling them.")
                    self.radio_messages = []
                    self.radio_messages_every = math.inf
                    self.radio_messages_next_in = math.inf
            else:
                logger.info("Radio messages are force-disabled.")
                self.radio_messages = []
                self.radio_messages_every = math.inf
                self.radio_messages_next_in = math.inf
        except (KeyError, ValueError):
            logger.warning("Radio messages config error, disabling them.")
            self.radio_messages = []
            self.radio_messages_every = math.inf
            self.radio_messages_next_in = math.inf
        # Activity reporting
        try:
            self.activity_report_sample_time = int(config["Discord"]["activityreport_sample_time"])
        except (KeyError, ValueError):
            logger.warning("Activity reporting config error, disabling it.")
            self.activity_report_sample_time = math.inf
        # Sentry error reporting
        try:
            self.sentry_token = config["Sentry"]["token"]
        except (KeyError, ValueError):
            logger.warning("Sentry client config error, disabling it.")
            self.sentry_token = None
        # Easter eggs
        try:
            if config["Discord"]["song_text_easter_eggs_enabled"] == "True":
                self.song_text_easter_eggs = {
                    "despacito": ":arrow_forward: this is so sad. alexa play {song}",
                    "faded": ":arrow_forward: Basta Garf, lasciami ascoltare {song}",
                    "ligma": ":arrow_forward: What is ligma? {song}!",
                    "sugma": ":arrow_forward: What is sugma? {song}!",
                    "sugondese": ":arrow_forward: What is sugondese? {song}!",
                    "bofa": ":arrow_forward: What is bofa? {song}!",
                    "updog": ":arrow_forward: What's up, dog? {song}!",
                    "sayo-nara": ":arrow_forward: I gently open the door. {song} awaits me inside.",
                    "monika": ":arrow_forward: Just Monika. Just Monika. Just {song}.",
                    "country road": ":arrow_forward: Take me home, to {song}, the place I belong!",
                    "never gonna give you up": ":arrow_forward: Rickrolling in 2018. Enjoy {song}!",
                    "september": ":arrow_forward: Do you remember? {song}.",
                    "homestuck": ":arrow_forward: > Enter song name. {song}",
                    "undertale": ":arrow_forward: Howdy! I'm Flowey! Listen to this friendly song: {song}",
                    "pumped up kicks": ":arrow_forward: Non metterti mica in testa strane idee ascoltando {song}...",
                    "jesus": ":arrow_forward: Respawn in 3 giorni. Intanto, ascolta {song}.",
                    "through The fire And flames": ":arrow_forward: Fai {song} su osu!, se ne sei capace!",
                    "slow clap": ":arrow_forward: :clap: :clap: :clap: {song} :clap: :clap: :clap:",
                    "pub scrubs": ":arrow_forward: MAV COME BACK WE MISS {song}!",
                    "alleluia": ":arrow_forward: Wah. Waaaah. Waluigi tiime: {song}",
                    "wah": ":arrow_forward: Wah. Waaaah. Waluigi tiime: {song}",
                    "waluigi": ":arrow_forward: Wah. Waaaah. Waluigi tiime: {song}",
                    "nyan cat": ":arrow_forward: Meow! :3 {song}",
                    "dragonborn": ":arrow_forward: FUS RO {song}!",
                    "dovahkiin": ":arrow_forward: FUS RO {song}!",
                    "initial d": ":arrow_forward: Guarda mamma sto driftando sul balcone di Balu grazie a {song}!",
                    "persona": ":arrow_forward: You'll never see {song} comiiiiing!",
                    "flamingo": ":arrow_forward: How many {song} do you have to eat?",
                    "linkin park": ":arrow_forward: Crawling in my {song}!",
                    "magicite": "⚠️ Warning: {song} contiene numerosi bug. E' ora in riproduzione.",
                    "papers please": ":arrow_forward: Glory to Arstotzka! {song}!",
                    "we are number one": ":arrow_forward: Now paying respect to Robbie Rotten: {song}",
                    "jump up superstar": ":arrow_forward: Is {song} the Tengen Toppa Guren Lagann opening?",
                    "the world revolving": ":arrow_forward: CHAOS! CHAOS! I CAN DO {song}!",
                    "deltarune": ":arrow_forward: You hug Ralsei. A strange music starts playing: {song}",
                    "song of unhealing": ":arrow_forward: BEN {song}",
                    "police academy": ":arrow_forward: {song} - freedom.png",
                    "super smash bros. ultimate": ":arrow_forward: Re-awaken the undying light with {song}!",
                    "powerwolf": ":arrow_forward: Spaggia, ma non ti sei un po' stancato di {song}?",
                    "eurobeat": ":arrow_forward: Nemesis approva la scelta di {song}. Ben fatto, amico.",
                    "k/da": ":arrow_forward: Che noia...\n"
                            "Non ci si può nemmeno divertire con {song} che c'è qualcuno che se ne lamenta.\n"
                            "La prossima volta, metti qualcosa di diverso, per piacere.",
                    "youtube rewind": ":arrow_forward: Perchè ti vuoi così male?"
                                      " Sigh, ascolta, discutere con te è inutile."
                                      " Ti lascio qui {song}. Richiamami quando sarà tutto finito."
                }
            else:
                self.song_text_easter_eggs = {}
        except (KeyError, ValueError):
            logger.warning("Song text easter egg information not found, defaulting to disabled.")
            self.song_text_easter_eggs = {}

    # noinspection PyAsyncCall
    async def on_ready(self):
        # Get the main guild
        self.main_guild = self.get_guild(self.main_guild_id)
        if not isinstance(self.main_guild, discord.Guild):
            raise errors.InvalidConfigError("The main guild does not exist!")
        # Get the main channel
        self.main_channel = self.get_channel(self.main_channel_id)
        if not isinstance(self.main_channel, discord.TextChannel):
            raise errors.InvalidConfigError("The main channel is not a TextChannel!")
        # Show yourself!
        await self.change_presence(status=discord.Status.online, activity=None)
        logger.info("Bot is ready!")
        # Start the bot tasks
        asyncio.ensure_future(self.queue_predownload_videos())
        asyncio.ensure_future(self.queue_play_next_video())
        asyncio.ensure_future(self.inactivity_countdown())
        asyncio.ensure_future(self.activity_task())

    async def on_message(self, message: discord.Message):
        if message.channel != self.main_channel or message.author.bot:
            return
        self.sentry.user_context({
            "discord": {
                "discord_id": message.author.id,
                "name": message.author.name,
                "discriminator": message.author.discriminator
            }
        })
        if not message.content.startswith("!"):
            await message.channel.send(f"⚠️ In questa chat sono consentiti solo comandi per il bot.\n"
                                       f"Riinvia il tuo messaggio in un altro canale!")
            await message.delete()
            return
        data = message.content.split(" ")
        if data[0] not in self.commands:
            await message.channel.send("⚠️ Comando non riconosciuto.")
            return
        logger.debug(f"Received command: {message.content}")
        self.sentry.extra_context({
            "command": data[0],
            "message": message
        })
        self.inactivity_timer = 3600
        await self.commands[data[0]](channel=message.channel,
                                     author=message.author,
                                     params=data)

    async def on_error(self, event_method, *args, **kwargs):
        ei = sys.exc_info()
        logger.critical(f"Critical error: {repr(ei)}")
        # noinspection PyBroadException
        try:
            await self.main_channel.send(f"☢️ **ERRORE CRITICO NELL'EVENTO** `{event_method}`!\n"
                                         f"Il bot si è chiuso e si dovrebbe riavviare entro qualche minuto.\n"
                                         f"Una segnalazione di errore è stata automaticamente mandata a Steffo.\n\n"
                                         f"Dettagli dell'errore:\n"
                                         f"```python\n"
                                         f"{repr(ei)}\n"
                                         f"```")
            await self.change_presence(status=discord.Status.invisible)
            await self.close()
        except Exception:
            logger.error(f"Double critical error: {sys.exc_info()}")
        loop.stop()
        self.sentry.captureException(exc_info=ei)
        exit(1)

    async def feed_pipe(self, connection):
        await self.wait_until_ready()
        while True:
            msg = await loop.run_in_executor(executor, connection.recv)
            logger.debug(f"Received from the Telegram-Discord pipe: {msg}")
            full_cv = (msg == "get cv full")
            if msg.startswith("get cv"):
                discord_members = list(self.main_guild.members)
                channels = {0: None}
                members_in_channels = {0: []}
                message = ""
                # Find all the channels
                for member in discord_members:
                    if member.voice is not None:
                        channel = members_in_channels.get(member.voice.channel.id)
                        if channel is None:
                            members_in_channels[member.voice.channel.id] = list()
                            channel = members_in_channels[member.voice.channel.id]
                            channels[member.voice.channel.id] = member.voice.channel
                        channel.append(member)
                    else:
                        members_in_channels[0].append(member)
                # Edit the message, sorted by channel
                for channel in sorted(channels, key=lambda c: -c):
                    members_in_channels[channel].sort(key=lambda x: x.nick if x.nick is not None else x.name)
                    if channel == 0:
                        message += "<b>Non in chat vocale:</b>\n"
                    else:
                        message += f"<b>In #{escape(channels[channel].name)}:</b>\n"
                    for member in members_in_channels[channel]:
                        # Ignore not-connected non-notable members
                        if not full_cv and channel == 0 and len(member.roles) < 2:
                            continue
                        # Ignore offline members
                        if member.status == discord.Status.offline and member.voice is None:
                            continue
                        # Online status emoji
                        if member.bot:
                            message += "🤖 "
                        elif member.status == discord.Status.online:
                            message += "🔵 "
                        elif member.status == discord.Status.idle:
                            message += "⚫️ "
                        elif member.status == discord.Status.dnd:
                            message += "🔴 "
                        elif member.status == discord.Status.offline:
                            message += "⚪️ "
                        # Voice
                        if channel != 0:
                            # Voice status
                            if member.voice.self_deaf:
                                message += f"🔇 "
                            elif member.voice.self_mute:
                                message += f"🔈 "
                            else:
                                message += f"🔊 "
                        # Nickname
                        if member.nick is not None:
                            message += escape(member.nick)
                        else:
                            message += escape(member.name)
                        # Game or stream
                        if member.activity is not None:
                            if member.activity.type == discord.ActivityType.playing:
                                message += f" | 🎮 {escape(member.activity.name)}"
                                # Rich presence
                                try:
                                    if member.activity.state is not None:
                                        message += f" ({escape(member.activity.state)}" \
                                            f" | {escape(member.activity.details)})"
                                except AttributeError:
                                    pass
                            elif member.activity.type == discord.ActivityType.streaming:
                                message += f" | 📡 [{escape(member.activity.name)}]({escape(member.activity.url)})"
                            elif member.activity.type == discord.ActivityType.listening:
                                message += f" | 🎧 {escape(member.activity.name)}"
                            elif member.activity.type == discord.ActivityType.watching:
                                message += f" | 📺 {escape(member.activity.name)}"
                        message += "\n"
                    message += "\n"
                connection.send(message)
                logger.debug(f"Answered successfully cvlist request.")
            elif msg.startswith("!"):
                data = msg.split(" ")
                if data[0] not in self.commands:
                    connection.send("error")
                    continue
                logger.debug(f"Received command: {msg}")
                await self.main_channel.send(f"{msg}\n"
                                             f"_(da Telegram)_")
                await self.commands[data[0]](channel=self.main_channel,
                                             author=None,
                                             params=data)
                connection.send("success")

    async def queue_predownload_videos(self):
        while True:
            await asyncio.sleep(1)
            # Might have some problems with del
            for index, video in enumerate(self.video_queue.not_ready_videos(self.max_videos_to_predownload)):
                try:
                    with async_timeout.timeout(self.max_video_ready_time):
                        await loop.run_in_executor(executor, video.ready_up)
                except asyncio.TimeoutError:
                    logger.warning(
                        f"Video {repr(video)} took more than {self.max_video_ready_time} to download, skipping...")
                    await self.main_channel.send(
                        f"⚠️ La preparazione di {video} ha richiesto più di {self.max_video_ready_time} secondi,"
                        f" pertanto è stato rimosso dalla coda.")
                    del self.video_queue.list[index]
                    continue
                except Exception as e:
                    self.sentry.user_context({
                        "discord": {
                            "discord_id": video.enqueuer.id,
                            "name": video.enqueuer.name,
                            "discriminator": video.enqueuer.discriminator
                        }
                    })
                    self.sentry.extra_context({
                        "video": video.plain_text()
                    })
                    self.sentry.captureException()
                    logger.error(f"Uncaught video download error: {e}")
                    await self.main_channel.send(f"⚠️ E' stato incontrato un errore durante il download di "
                                                 f"{str(video)}, quindi è stato rimosso dalla coda.\n\n"
                                                 f"```python\n"
                                                 f"{str(e.args)}"
                                                 f"```")
                    del self.video_queue.list[index]
                    continue

    async def queue_play_next_video(self):
        await self.wait_until_ready()
        while True:
            await asyncio.sleep(1)
            for voice_client in self.voice_clients:
                # Do not add play videos if something else is playing!
                if not voice_client.is_connected():
                    continue
                if voice_client.is_playing():
                    continue
                if voice_client.is_paused():
                    continue
                # Ensure the next video is ready
                next_video = self.video_queue.next_video()
                if next_video is None or not next_video.is_ready:
                    continue
                # Advance the queue
                self.video_queue.advance_queue()
                # Try to generate an AudioSource
                if self.video_queue.now_playing is None:
                    continue
                audio_source = self.video_queue.now_playing.make_audio_source()
                # Start playing the AudioSource
                logger.info(f"Started playing {self.video_queue.now_playing.plain_text()}.")
                voice_client.play(audio_source)
                # Update the voice_client activity
                activity = discord.Activity(name=self.video_queue.now_playing.plain_text(),
                                            type=discord.ActivityType.listening)
                logger.debug("Updating bot presence...")
                await self.change_presence(status=discord.Status.online, activity=activity)
                # Record the played song in the database
                if self.video_queue.now_playing.enqueuer is not None:
                    logger.debug(f"Adding {self.video_queue.now_playing.plain_text()} to db.PlayedMusic...")
                    try:
                        session = db.Session()
                        enqueuer = await loop.run_in_executor(executor, session.query(db.Discord)
                                                              .filter_by(
                            discord_id=self.video_queue.now_playing.enqueuer.id)
                                                              .one_or_none)
                        played_music = db.PlayedMusic(enqueuer=enqueuer,
                                                      filename=self.video_queue.now_playing.database_text(),
                                                      timestamp=datetime.datetime.now())
                        session.add(played_music)
                        await loop.run_in_executor(executor, session.commit)
                        await loop.run_in_executor(executor, session.close)
                    except sqlalchemy.exc.OperationalError:
                        pass
                # Send a message in chat
                for key in self.song_text_easter_eggs:
                    if key in self.video_queue.now_playing.name.lower():
                        await self.main_channel.send(
                            self.song_text_easter_eggs[key].format(song=str(self.video_queue.now_playing)))
                        break
                else:
                    await self.main_channel.send(
                        f":arrow_forward: Ora in riproduzione: {str(self.video_queue.now_playing)}")

    async def inactivity_countdown(self):
        while True:
            await asyncio.sleep(1)
            if self.inactivity_timer > 0:
                self.inactivity_timer -= 1
                continue
            for voice_client in self.voice_clients:
                if voice_client.is_connected():
                    logger.info("Disconnecting due to inactivity.")
                    await voice_client.disconnect()
                    await self.change_presence(status=discord.Status.online, activity=None)
                    await self.main_channel.send("💤 Mi sono disconnesso dalla cv per inattività.")

    async def create_activityreport(self):
        logger.debug("Fetching Discord users...")
        discord_users = list(self.main_guild.members)
        online_members_count = 0
        ingame_members_count = 0
        cv_count = 0
        cv_members_count = 0
        non_empty_channels = []
        for member in discord_users:
            if member.bot:
                continue
            if member.voice is not None and member.voice.channel != self.main_guild.afk_channel:
                cv_count += 1
                if member.voice.channel.id not in non_empty_channels:
                    non_empty_channels.append(member.voice.channel.id)
            if len(member.roles) >= 2:
                if member.voice is not None and member.voice.channel != self.main_guild.afk_channel:
                    cv_members_count += 1
                if member.status != discord.Status.offline and member.status != discord.Status.idle:
                    online_members_count += 1
                if member.activity is not None and member.activity.type == discord.ActivityType.playing:
                    ingame_members_count += 1
        logger.debug("Creating and committing db.ActivityReport...")
        session = db.Session()
        activityreport = db.ActivityReport(timestamp=datetime.datetime.now(),
                                           discord_members_online=online_members_count,
                                           discord_members_ingame=ingame_members_count,
                                           discord_cv=cv_count,
                                           discord_members_cv=cv_members_count,
                                           discord_channels_used=len(non_empty_channels))
        session.add(activityreport)
        await loop.run_in_executor(executor, session.commit)
        await loop.run_in_executor(executor, session.close)
        logger.info("ActivityReport created.")

    async def activity_task(self):
        await self.wait_until_ready()
        if self.activity_report_sample_time == math.inf:
            return
        while True:
            await self.create_activityreport()
            logger.debug(f"Waiting {self.activity_report_sample_time} seconds before the next record.")
            await asyncio.sleep(self.activity_report_sample_time)

    async def add_video_from_url(self, url: str, index: typing.Optional[int] = None, enqueuer: discord.Member = None):
        # Retrieve info
        logger.debug(f"Retrieving info for {url}.")
        async with self.main_channel.typing():
            with youtube_dl.YoutubeDL({"quiet": True,
                                       "ignoreerrors": True,
                                       "simulate": True}) as ytdl:
                info = await loop.run_in_executor(executor,
                                                  functools.partial(ytdl.extract_info, url=url, download=False))
            if info is None:
                logger.debug(f"No video found at {url}.")
                await self.main_channel.send(f"⚠️ Non è stato trovato nessun video all'URL `{url}`,"
                                             f" pertanto non è stato aggiunto alla coda.")
                return
            if "entries" in info:
                logger.debug(f"Playlist detected at {url}.")
                for entry in info["entries"]:
                    self.video_queue.add(YoutubeDLVideo(entry["webpage_url"], enqueuer=enqueuer), index)
                return
            logger.debug(f"Single video detected at {url}.")
            self.video_queue.add(YoutubeDLVideo(url, enqueuer=enqueuer), index)

    # noinspection PyUnusedLocal
    @command
    async def null(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        pass

    # noinspection PyUnusedLocal
    @command
    async def cmd_ping(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        await channel.send(f"Pong!")

    # noinspection PyUnusedLocal
    @command
    async def cmd_register(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        session = db.Session()
        if len(params) < 1:
            await channel.send("⚠️ Non hai specificato un username!\n"
                               "Sintassi corretta: `!register <username_ryg>`")
            return
        try:
            # noinspection PyTypeChecker
            d = db.Discord.create(session,
                                  royal_username=params[0],
                                  discord_user=author)
        except errors.AlreadyExistingError:
            await channel.send("⚠️ Il tuo account Discord è già collegato a un account RYG "
                               "o l'account RYG che hai specificato è già collegato a un account Discord.")
            return
        session.add(d)
        session.commit()
        session.close()
        await channel.send("✅ Sincronizzazione completata!")

    # noinspection PyUnusedLocal
    @command
    async def cmd_cv(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        """Summon the bot in the author's voice channel."""
        if author is None:
            await channel.send("⚠️ Questo comando richiede un autore.")
            return
        if author.voice is None:
            await channel.send("⚠️ Non sei in nessun canale!")
            return
        if author.voice.channel == self.main_guild.afk_channel:
            await channel.send("⚠️ Non posso connettermi al canale AFK!")
            return
        if author.voice.channel.bitrate < 64000:
            await channel.send("ℹ️ Sei in un canale con un bitrate ridotto.\n"
                               "L'utilizzo del bot in quel canale ignorerà il limite di bitrate e potrebbe causare lag"
                               " o eccessivo consumo di dati.\n"
                               "Se vuoi procedere comunque, scrivi `!yes`.")
            try:
                await self.wait_for("message", check=lambda m: m.content == "!yes", timeout=10.0)
            except asyncio.TimeoutError:
                return
        # Check if there's already a connected client
        for voice_client in self.voice_clients:
            if voice_client.channel in self.main_guild.channels and voice_client.is_connected():
                logger.info(f"Moving to {author.voice.channel.name}.")
                await voice_client.move_to(author.voice.channel)
                await channel.send(f"⤵️ Mi sono spostato in <#{author.voice.channel.id}>.")
                break
        else:
            logger.info(f"Connecting to {author.voice.channel.name}.")
            await author.voice.channel.connect()
            await channel.send(f"⤵️ Mi sono connesso in <#{author.voice.channel.id}>.")

    # noinspection PyUnusedLocal
    @command
    @requires_connected_voice_client
    async def cmd_play(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        if len(params) < 2:
            await channel.send("⚠️ Non hai specificato una canzone da riprodurre!\n"
                               "Sintassi: `!play <url|ricercayoutube|nomefile>`")
            return
        self.radio_messages_next_in -= 1
        if self.radio_messages_next_in <= 0:
            radio_message = random.sample(self.radio_messages, 1)[0]
            # pycharm are you drunk
            # noinspection PyAttributeOutsideInit
            self.radio_messages_next_in = self.radio_messages_every
            await self.add_video_from_url(radio_message)
            await channel.send(f"📻 Aggiunto un messaggio radio, disattiva con `!radiomessages off`.")
            logger.info(f"Radio message added to the queue.")
        # Parse the parameter as URL
        url = re.match(r"(?:https?://|ytsearch[0-9]*:|scsearch[0-9]*:).*", " ".join(params[1:]).strip("<>"))
        if url is not None:
            # This is a url
            await self.add_video_from_url(url.group(0), enqueuer=author)
            await channel.send(f"✅ Video aggiunto alla coda.")
            logger.debug(f"Added {url} to the queue as URL.")
            return
        # Search the parameter on youtube
        search = " ".join(params[1:])
        # This is a search
        await self.add_video_from_url(url=f"ytsearch:{search}", enqueuer=author)
        await channel.send(f"✅ Video aggiunto alla coda.")
        logger.debug(f"Added ytsearch:{search} to the queue.")

    # noinspection PyUnusedLocal
    @command
    @requires_connected_voice_client
    async def cmd_skip(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        for voice_client in self.voice_clients:
            if voice_client.is_playing():
                voice_client.stop()
                await channel.send(f"⏩ Video saltato.")
                logger.debug(f"A song was skipped.")
                break
        else:
            await channel.send("⚠️ Non c'è nessun video in riproduzione.")

    # noinspection PyUnusedLocal
    @command
    @requires_connected_voice_client
    async def cmd_remove(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        if len(self.video_queue) == 0:
            await channel.send("⚠️ Non c'è nessun video in coda.")
            return
        if len(params) == 1:
            index = len(self.video_queue) - 1
        else:
            try:
                index = int(params[1]) - 1
            except ValueError:
                await channel.send("⚠️ Il numero inserito non è valido.\n"
                                   "Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
                return
        if len(params) < 3:
            if abs(index) >= len(self.video_queue):
                await channel.send("⚠️ Il numero inserito non corrisponde a nessun video nella playlist.\n"
                                   "Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
                return
            video = self.video_queue.list.pop(index)
            await channel.send(f":regional_indicator_x: {str(video)} è stato rimosso dalla coda.")
            logger.debug(f"Removed from queue: {video.plain_text()}")
            return
        try:
            start = int(params[1]) - 1
        except ValueError:
            await channel.send("⚠️ Il numero iniziale inserito non è valido.\n"
                               "Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
            return
        if start >= len(self.video_queue):
            await channel.send("⚠️ Il numero iniziale inserito non corrisponde a nessun video nella"
                               " playlist.\n"
                               "Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
            return
        try:
            end = int(params[2]) - 2
        except ValueError:
            await channel.send("⚠️ Il numero finale inserito non è valido.\n"
                               "Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
            return
        if end >= len(self.video_queue):
            await channel.send("⚠️ Il numero finale inserito non corrisponde a nessun video nella"
                               " playlist.\n"
                               "Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
            return
        if start > end:
            await channel.send("⚠️ Il numero iniziale è maggiore del numero finale.\n"
                               "Sintassi: `!remove [numerovideoiniziale] [numerovideofinale]`")
            return
        del self.video_queue.list[start:end]
        await channel.send(f":regional_indicator_x: {end - start} video rimossi dalla coda.")
        logger.debug(f"Removed from queue {end - start} videos.")

    # noinspection PyUnusedLocal
    @command
    async def cmd_queue(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        msg = ""
        if self.video_queue.loop_mode == LoopMode.NORMAL:
            msg += "Modalità attuale: :arrow_right: **Nessuna ripetizione**\n"
        elif self.video_queue.loop_mode == LoopMode.LOOP_QUEUE:
            msg += "Modalità attuale: :repeat: **Ripeti intera coda**\n"
        elif self.video_queue.loop_mode == LoopMode.LOOP_SINGLE:
            msg += "Modalità attuale: :repeat_one: **Ripeti canzone singola**\n"
        elif self.video_queue.loop_mode == LoopMode.FOLLOW_SUGGESTIONS:
            msg += "Modalità attuale: :arrows_clockwise: **Continua con video suggeriti**\n"
        elif self.video_queue.loop_mode == LoopMode.AUTO_SHUFFLE:
            msg += "Modalità attuale: :twisted_rightwards_arrows: **Video casuale dalla coda**\n"
        elif self.video_queue.loop_mode == LoopMode.LOOPING_SHUFFLE:
            msg += "Modalità attuale: :arrows_counterclockwise: **Video casuali infiniti dalla coda**\n"
        msg += "**Video in coda:**\n"
        if self.video_queue.now_playing is None:
            msg += ":cloud: _nessuno_"
        else:
            msg += f":arrow_forward: {str(self.video_queue.now_playing)}\n"
        if self.video_queue.loop_mode == LoopMode.NORMAL:
            for index, video in enumerate(self.video_queue.list[:10]):
                msg += f"{number_emojis[index]} {str(video)}\n"
            if len(self.video_queue) > 10:
                msg += f"più altri {len(self.video_queue) - 10} video!"
        elif self.video_queue.loop_mode == LoopMode.LOOP_QUEUE:
            for index, video in enumerate(self.video_queue.list[:10]):
                msg += f"{number_emojis[index]} {str(video)}\n"
            if len(self.video_queue) > 10:
                msg += f"più altri {len(self.video_queue) - 10} video che si ripetono!"
            else:
                if len(self.video_queue) < 6:
                    count = len(self.video_queue)
                    while count < 10:
                        for index, video in enumerate(self.video_queue.list[:10]):
                            msg += f":asterisk: {str(video)}\n"
                        count += len(self.video_queue)
                msg += "e avanti così!"
        elif self.video_queue.loop_mode == LoopMode.LOOP_SINGLE:
            video = self.video_queue.now_playing
            for index in range(9):
                msg += f":asterisk: {str(video)}\n"
            msg += "all'infinito!"
        elif self.video_queue.loop_mode == LoopMode.FOLLOW_SUGGESTIONS:
            msg += ":rainbow:"
        elif self.video_queue.loop_mode == LoopMode.AUTO_SHUFFLE:
            for index, video in enumerate(self.video_queue.list[:10]):
                msg += f":hash: {str(video)}\n"
            if len(self.video_queue) > 10:
                msg += f"più altri {len(self.video_queue) - 10} video!"
        elif self.video_queue.loop_mode == LoopMode.LOOPING_SHUFFLE:
            for index, video in enumerate(self.video_queue.list[:10]):
                msg += f":hash: {str(video)}\n"
            if len(self.video_queue) > 10:
                msg += f"più altri {len(self.video_queue) - 10} video che si ripetono!"
            else:
                if len(self.video_queue) < 6:
                    count = len(self.video_queue)
                    while count < 10:
                        for index, video in enumerate(self.video_queue.list[:10]):
                            msg += f":asterisk: {str(video)}\n"
                        count += len(self.video_queue)
                msg += "a ripetizione casuale!"
        await channel.send(msg)

    # noinspection PyUnusedLocal
    @command
    @requires_connected_voice_client
    async def cmd_shuffle(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        if len(self.video_queue) == 0:
            await channel.send("⚠ Non ci sono video in coda!")
            return
        logger.info(f"The queue was shuffled by {author.name}#{author.discriminator}.")
        self.video_queue.shuffle()
        await channel.send("🔀 Shuffle completo!")

    # noinspection PyUnusedLocal
    @command
    @requires_connected_voice_client
    async def cmd_clear(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        if len(self.video_queue) == 0:
            await channel.send("⚠ Non ci sono video in coda!")
            return
        logger.info(f"The queue was cleared by {author.name}#{author.discriminator}.")
        self.video_queue.clear()
        await channel.send(":regional_indicator_x: Tutti i video in coda rimossi.")

    # noinspection PyUnusedLocal
    @command
    async def cmd_radiomessages(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        if not self.radio_messages:
            await channel.send("⚠ I messaggi radio sono stati disabilitati dall'amministratore del bot.")
            return
        if len(params) < 2:
            await channel.send("⚠ Sintassi del comando non valida.\n"
                               "Sintassi: `!radiomessages <on|off>`")
        else:
            if params[1].lower() == "on":
                self.radio_messages_next_in = self.radio_messages_every
            elif params[1].lower() == "off":
                # noinspection PyAttributeOutsideInit wtf
                self.radio_messages_next_in = math.inf
            else:
                await channel.send("⚠ Sintassi del comando non valida.\n"
                                   "Sintassi: `!radiomessages <on|off>`")
                return
        logger.info(f"Radio messages status to {'enabled' if self.radio_messages_next_in < math.inf else 'disabled'}.")
        await channel.send(
            f"📻 Messaggi radio **{'attivati' if self.radio_messages_next_in < math.inf else 'disattivati'}**.")

    # noinspection PyUnusedLocal
    @command
    @requires_connected_voice_client
    async def cmd_pause(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        for voice_client in self.voice_clients:
            if voice_client.is_playing():
                voice_client.pause()
                logger.debug(f"The audio stream was paused.")
                await channel.send(f"⏸ Riproduzione messa in pausa.\n"
                                   f"Riprendi con `!resume`.")

    # noinspection PyUnusedLocal
    @command
    @requires_connected_voice_client
    async def cmd_resume(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        for voice_client in self.voice_clients:
            if voice_client.is_paused():
                voice_client.resume()
                logger.debug(f"The audio stream was resumed.")
                await channel.send(f"⏯ Riproduzione ripresa.")

    # noinspection PyUnusedLocal
    @command
    @requires_connected_voice_client
    async def cmd_mode(self, channel: discord.TextChannel, author: discord.Member, params: typing.List[str]):
        if len(params) < 2:
            await channel.send("⚠ Sintassi del comando non valida.\n"
                               "Sintassi: `!mode <normal|repeat|loop|random|endless>`")
            return
        if params[1] == "normal":
            self.video_queue.loop_mode = LoopMode.NORMAL
            await channel.send("➡️ Modalità di coda impostata: **Nessuna ripetizione**")
        elif params[1] == "repeat":
            self.video_queue.loop_mode = LoopMode.LOOP_SINGLE
            await channel.send("🔂 Modalità di coda impostata: **Ripeti canzone singola**")
        elif params[1] == "loop":
            self.video_queue.loop_mode = LoopMode.LOOP_QUEUE
            await channel.send("🔁 Modalità di coda impostata: **Ripeti intera coda**")
        elif params[1] == "suggest":
            # self.video_queue.loop_mode = LoopMode.FOLLOW_SUGGESTIONS
            await channel.send("⚠️ La modalità **Continua con video suggeriti** non è ancora stata implementata.")
        elif params[1] == "random":
            self.video_queue.loop_mode = LoopMode.AUTO_SHUFFLE
            await channel.send("🔀 Modalità di coda impostata: **Video casuale dalla coda**")
        elif params[1] == "endless":
            self.video_queue.loop_mode = LoopMode.LOOPING_SHUFFLE
            await channel.send("🔄 Modalità di coda impostata: **Video casuali infiniti dalla coda**")
        else:
            await channel.send("⚠️ Sintassi del comando non valida.\n"
                               "Sintassi: `!loop <off|loop1|loopall|suggest|shuffle|loopshuffle>`")


def process(users_connection=None):
    logger.info("Initializing the bot...")
    bot = RoyalDiscordBot()
    if users_connection is not None:
        logger.info("Initializing Telegram-Discord connection...")
        asyncio.ensure_future(bot.feed_pipe(users_connection))
    logger.info("Logging in...")
    loop.run_until_complete(bot.login(bot.token, bot=True))
    logger.info("Connecting...")
    loop.run_until_complete(bot.connect())
    logger.info("Now stopping...")
    loop.run_until_complete(bot.main_channel.send("ℹ️ Spegnimento bot in corso..."))
    loop.run_until_complete(bot.logout())


if __name__ == "__main__":
    process()