from mautrix.types import EventType, MessageType
from mautrix.crypto import attachments

from maubot import Plugin, MessageEvent
from maubot.handlers import event


class HtmlFile(Plugin):
    @event.on(EventType.ROOM_MESSAGE)
    async def event_handler(self, event: MessageEvent) -> None:

        # self.log.debug(f"received: {event}")

        if not event.content.msgtype == MessageType.FILE:
            return

        if event.content.info.mimetype == "text/html":
            if event.content.url:
                file_content = (await self.client.download_media(url = event.content.url)).decode("utf-8")
            else:
                enc_file = event.content.file
                ciphertext = await self.client.download_media(url = enc_file.url)
                file_content = attachments.decrypt_attachment(ciphertext, enc_file.key.key, enc_file.hashes["sha256"], enc_file.iv).decode("utf-8")

            await event.reply(file_content, allow_html = True)