import asyncio

from nio import MatrixRoom, Event, AsyncClient
from nio.crypto import ENCRYPTION_ENABLED

from pydantic import Field

from matrix_bot.bot import MatrixBot
from matrix_bot.client import MatrixClient
from matrix_bot.callbacks import properly_fail
from matrix_bot.eventparser import MessageEventParser, ignore_when_not_concerned, EventNotConcerned
from tchap_bot.config import Config

from markov import parle


class ConfigProxy(Config):
    https_proxy: str = Field("", description="Proxy URL")

env_config = ConfigProxy()

# le décorateur @properly_fail va permettre à la commande de laisser un message d'erreur si la commande plante et
# d'envoyer le message que le bot n'est plus en train d'écrire
# la fonction va être appelée dans tous les cas, le décorateur @ignore_when_not_concerned
# permet de laisser event_parser gérer le cas où la commande n'est pas concernée
@properly_fail
@ignore_when_not_concerned
async def repond(room: MatrixRoom, message: Event, matrix_client: MatrixClient):
    # on initialise un event_parser pour décider à quel message cette commande va répondre
    event_parser = MessageEventParser(room=room, event=message, matrix_client=matrix_client)
    # il ne va pas répondre à ses propres messages
    event_parser.do_not_accept_own_message()

    # ne répond qu'en message direct ou dans un salon s'il est interpellé avec @identifiant.du.bot dans le corps du message
    if len(room.users) > 2 and not (message.formatted_body and matrix_client.user_id in message.formatted_body):
        raise EventNotConcerned

    reponse = parle()
    # il envoie l'information qu'il est en train d'écrire
    await matrix_client.room_typing(room.room_id)
    await asyncio.sleep(len(reponse) / 30)
    # il envoie le message
    await matrix_client.send_text_message(room.room_id, reponse, "m.notice")

tchap_bot = MatrixBot(
    env_config.matrix_home_server,
    env_config.matrix_bot_username,
    env_config.matrix_bot_password,
    proxy=env_config.https_proxy,
    use_functions=True,
    ssl=True
)

tchap_bot.matrix_client.matrix_config.encryption_enabled = True
tchap_bot.matrix_client.matrix_config.ignore_unverified_devices = True
tchap_bot.matrix_client.matrix_config.join_on_invite = True

tchap_bot.callbacks.register_on_message_event(repond)
tchap_bot.run()