#             █ █ ▀ █▄▀ ▄▀█ █▀█ ▀
#             █▀█ █ █ █ █▀█ █▀▄ █
#              © Copyright 2022
#           https://t.me/hikariatama
#
# 🔒      Licensed under the GNU AGPLv3
# 🌐 https://www.gnu.org/licenses/agpl-3.0.html

# meta pic: https://static.dan.tatar/linter_icon.png
# meta banner: https://mods.hikariatama.ru/badges/linter.jpg
# meta developer: @hikarimods
# requires: black
# scope: hikka_only
# scope: hikka_min 1.2.10

import io
import logging
import re
from random import choice

import black
import requests
from telethon.tl.types import Message

from .. import loader, utils

logging.getLogger("blib2to3.pgen2.driver").setLevel(logging.ERROR)

URL = r"(?:(?:https?|ftp):\/\/)?[\w/\-?=%.]+\.[\w/\-&?=%.]+"

captions = [
    "Here is your new fresh linted code! Enjoy",
    "This was such a hard work to clean this code... Uff..",
    "Here we go!",
    "Glad to be your virtual code-cleaning-maid!",
    "Take this, master!",
]


@loader.tds
class PyLinterMod(loader.Module):
    """`Black` plugin wrapper for telegram"""

    strings = {"name": "PyLinter", "no_code": "🚫 <b>Please, specify code to lint</b>"}

    async def lintcmd(self, message: Message):
        """[code|reply] - Perform automatic lint to python code"""
        reply = await message.get_reply_message()
        args = utils.get_args_raw(message)
        media = message.media or (reply.media if reply else False)

        if media:
            try:
                args = (await self._client.download_file(media, bytes)).decode("utf-8")
            except TypeError:
                pass

        if not args:
            if not reply:
                await utils.answer(message, self.strings("no_code"))
                return

            args = reply.raw_text

        if re.match(URL, args):
            args = (await utils.run_sync(requests.get, args)).text

        lint_result = black.format_str(args, mode=black.Mode())

        if len(lint_result) < 2048:
            await utils.answer(
                message,
                f"<code>{utils.escape_html(lint_result)}</code>",
            )
            return

        file = io.BytesIO(args.encode("utf-8"))
        file.name = "lint_result.py"
        await self._client.send_file(
            message.peer_id,
            file,
            caption=(
                f"<i>{choice(captions)}</i>"
                f" <b>{utils.escape_html(utils.ascii_face())}</b>"
            ),
        )
        if message.out:
            await message.delete()