from threading import Thread

from discord import Intents, Client

from . import AiInstance


class DcClient:
    def __init__(self, _ai_instance: AiInstance):
        self._ai_instance = _ai_instance

        self._intents = Intents.default()
        self._intents.message_content = True
        self._discord_client = Client(intents=self._intents)

        @self._discord_client.event
        async def on_ready():
            print(f'We have logged in as {self._discord_client.user}')

        @self._discord_client.event
        async def on_message(message):
            if not self._ai_instance.discord.react_to_chat:
                return None

            if message.author == self._discord_client.user:
                return

            if message.content.isspace():
                await message.channel.send('### Empty message')
                return

            response = self._ai_instance.chat_text(message.author.name, message.content)
            if response is not None:
                await message.channel.send(response['content'])

    def connect(self):
        def start():
            self._discord_client.run(self._ai_instance.configuration.discord_token)

        thread = Thread(target=start)
        thread.start()

    def disconnect(self):
        self._discord_client.close()