import logging
import os
import re
from threading import Lock

import discord
from dotenv import load_dotenv

import data
import jokes

logging.basicConfig(level=logging.INFO)

LIL_STINKY_ID = 704043422276780072

LOGGER = logging.getLogger(__name__)


class RoboPage(discord.Client):
    db_path: str = 'messages.db'

    def __init__(self, *args, **kwargs):
        super(RoboPage, self).__init__(*args, **kwargs)
        attrs = filter(lambda n: n.endswith('Joke') and not n.startswith('Joke'), dir(jokes))
        attrs = map(lambda n: getattr(jokes, n)(), attrs)
        self.jokes = list(attrs)
        self.lock = Lock()
        self.most_regex = re.compile("^who is the most (?P<emoji>\w+)(?: in the past (?P<days>\d+) days)?\??$",
                                     re.IGNORECASE)
        self.leaderboard_regex = re.compile('^most (?P<emoji>\w+) leaderboard$', re.IGNORECASE)

    def run(self):
        return super().run(os.getenv('DISCORD_TOKEN'))

    async def handle_ready(self):
        async def alive():
            channel: discord.TextChannel = discord.utils.get(self.get_all_channels(), name='robotics-facility')
            await channel.send(f"I'm aliiiiiive {discord.utils.get(self.emojis, name='kaylon')}")

        self.data: data.MsgData = await data.MsgData.create(
            client=self,
            limit=3000,
            # limit=20,
            days=30,
        )
        self.data.to_sql('messages.db')
        LOGGER.info(f'{self.data.msgs.shape[0]} messages total')
        # await alive()

    async def handle_message(self, message):
        await self.data.add_msg(message)

        if message.author != self.user:
            if (m := self.leaderboard_regex.match(message.content)) is not None:
                try:
                    await message.reply(self.data.emoji_leaderboard(emoji_name=m.group('emoji'), days=14))
                except KeyError as e:
                    await message.reply(f"I couldn't find any {m.group('emoji')} reactions. Leave me alone!")

            elif (m := self.most_regex.match(message.content)) is not None:
                days = m.group('days') or 14
                try:
                    await message.reply(
                        await self.data.biggest_single(client=self, emoji=m.group('emoji'), days=int(days))
                    )
                except IndexError as e:
                    await message.reply('NObody')

            else:
                for joke in self.jokes:
                    if (scan_res := joke.scan(message)):
                        print(f'{joke.__class__.__name__} detected:\n{message.content}\n{scan_res}')
                        await joke.respond(message, self, scan_res)


if __name__ == '__main__':
    load_dotenv()

    client = RoboPage()


    @client.event
    async def on_ready():
        # print(len(list(client.get_all_members())))
        await client.handle_ready()
        # print('\n'.join(client.data.reactions.index.get_level_values(1).drop_duplicates().sort_values()))


    @client.event
    async def on_message(message: discord.Message):
        await client.handle_message(message)


    @client.event
    async def on_raw_reaction_add(payload):
        LOGGER.info(payload)
        await client.data.update_reaction(payload=payload, client=client)


    @client.event
    async def on_raw_reaction_remove(payload):
        LOGGER.info(payload)
        await client.data.update_reaction(payload=payload, client=client)


    client.run()