from os import getenv, listdir
import re

import discord
from typing import cast
from discord import app_commands
from discord.app_commands import Choice
from discord.ext import commands
from dotenv import load_dotenv

load_dotenv()


class MyClient(commands.Bot):
    def __init__(self):
        intents = discord.Intents.none()
        intents.guilds = True
        intents.messages = True
        intents.message_content = True
        super().__init__(intents=intents, command_prefix=".")

        self.MY_GUILDS = [discord.Object(id=int(guild_id)) for guild_id in getenv("GUILD_ID").split(',')]

        self.initial_extensions = ["cogs." + f[:-3] for f in listdir("./cogs") if
                                   f.endswith(".py") and f.__str__() != "__init__.py"]
        # self.add_listener(self.on_message)

    async def setup_hook(self):
        # cogs
        for extension in self.initial_extensions:
            await self.load_extension(extension)

        for guild in self.MY_GUILDS:
            self.tree.copy_global_to(guild=guild)
            await self.tree.sync(guild=guild)

    async def on_ready(self):
        team = cast(discord.Team, self.application.team)

        print(f'Logged in as {self.user} (ID: {self.user.id})')
        print(f"Discord version : {discord.__version__}")
        print(f"Admins : {', '.join([m.name for m in team.members])}")
        print('------')

    async def on_message(self, msg: discord.Message):
        immunise = msg.author.get_role(1336818572734890138) is not None

        if immunise:
            return

        # quoicoubot
        print(f"[{msg.channel}] {msg.author} : {msg.content}")
        quoi = re.compile("[qQ][uU][oO][iI][:\.!\?\\\"\'«»;\(\)\-\{\},~\\s]*$")
        if quoi.search(msg.content) is not None:
            await msg.reply("Quoicoubeh !")

        # ouistitibot
        oui = re.compile("[oO][uU][iI][:\.!\?\\\"\'«»;\(\)\-\{\},~\\s]*$")
        if oui.search(msg.content) is not None:
            await msg.reply("Ouistiti !")
 
        # factorielle
        fact = re.compile("([0-9]) factorielle")
        if fact.search(msg.content) is not None:
            await msg.reply("FACTORIELLE {} :banana: !".format(fact.findall(msg.content)[0]))


        # je suis
        je_suis = "je suis "
        if msg.content.lower().startswith(je_suis):  # détection
            words = msg.content[len(je_suis):].split(' ')
            new_pseudo = ""  # pseudo creation (do not split a word)
            for word in words:
                if (len(new_pseudo) + len(word) + 1) > 32:
                    break
                new_pseudo += " " + word
            if new_pseudo != "":
                member = msg.author
                try:
                    await member.edit(nick=new_pseudo, reason="Il est " + new_pseudo)
                except discord.Forbidden:
                    pass


if __name__ == "__main__":
    client = MyClient()
    client.run(getenv("TOKEN"))