# .------.------.------.------.------.------.------.------.------.------.
# |D.--. |4.--. |N.--. |1.--. |3.--. |L.--. |3.--. |K.--. |0.--. |0.--. |
# | :/\: | :/\: | :(): | :/\: | :(): | :/\: | :(): | :/\: | :/\: | :/\: |
# | (__) | :\/: | ()() | (__) | ()() | (__) | ()() | :\/: | :\/: | :\/: |
# | '--'D| '--'4| '--'N| '--'1| '--'3| '--'L| '--'3| '--'K| '--'0| '--'0|
# `------`------`------`------`------`------`------`------`------`------'
#
#                     Copyright 2023 t.me/D4n13l3k00
#           Licensed under the Creative Commons CC BY-NC-ND 4.0
#
#                    Full license text can be found at:
#       https://creativecommons.org/licenses/by-nc-nd/4.0/legalcode
#
#                           Human-friendly one:
#            https://creativecommons.org/licenses/by-nc-nd/4.0

# meta developer: @D4n13l3k00


import io

import requests

from .. import loader  # type: ignore


def register(cb):
    cb(DeepAIMod())


class DeepAIMod(loader.Module):
    strings = {"name": "DeepAI"}

    def __init__(self):
        self.name = self.strings["name"]

    async def client_ready(self, client, db):
        self._db = db

    async def dai_set_tokencmd(self, m):
        self._db.set("deepai", "token", str(m.raw_text.split(" ", 1)[1]))
        await m.edit("[DeepAI] Токен установлен")

    async def cnsfwcmd(self, m):
        """.cnsfw <reply to photo/sticker> - Check on nsfw content by DeepAI"""
        reply = await m.get_reply_message()
        if not reply:
            await m.edit("<b>Reply to media</b>")
            return
        try:
            media = reply.media
        except Exception:
            await m.edit("<b>Only media</b>")
            return
        if token := self._db.get("deepai", "token", None):
            await m.edit("[DeepAI] Детектим nsfw...")
            photo = io.BytesIO()
            await m.client.download_media(media, photo)
            photo.seek(0)
            r = requests.post(
                "https://api.deepai.org/api/nsfw-detector",
                files={
                    "image": photo.read(),
                },
                headers={"api-key": token},
            )
            try:
                await m.edit(
                    "[DeepAI] Тут Nsfw на "
                    + str(round(r.json()["output"]["nsfw_score"] * 100, 1))
                    + "%"
                )
            except Exception:
                await m.edit(f"[DeepAI] {str(r.json())}")
        else:
            await m.edit(
                "[DeepAI] Укажите токен для работы с API\n<code>.dai_set_token TOKEN</code>"
            )