"""
                              _
__   _____  ___  ___ ___   __| | ___ _ __
\ \ / / __|/ _ \/ __/ _ \ / _` |/ _ \ '__|
 \ V /\__ \  __/ (_| (_) | (_| |  __/ |
  \_/ |___/\___|\___\___/ \__,_|\___|_|

  Copyleft 2022 t.me/vsecoder
  This program is free software; you can redistribute it and/or modify

"""
# meta developer: @vsecoder_m
# meta pic: https://img.icons8.com/office/344/administrator-male--v1.png
# meta banner: https://chojuu.vercel.app/api/banner?img=https://img.icons8.com/office/344/administrator-male--v1.png&title=Profilemod&description=Telegram%20Profile%20Statistic

__version__ = (0, 0, 1)

import logging
from .. import loader, utils  # type: ignore
from telethon import functions
import imgkit  # type: ignore
import base64
import requests

logger = logging.getLogger(__name__)


@loader.tds
class Profilemod(loader.Module):
    """Module for get beautiful picture profile statistic"""

    strings = {"name": "Profilemod"}

    def __init__(self):
        self.config = loader.ModuleConfig(
            loader.ConfigValue(
                "background",
                "https://chojuu.vercel.app/api/banner?img=https://img.icons8.com/office/344/administrator-male--v1.png&title=.&description=.",
                "Url to background (540x220 is perfect)",
                validator=loader.validators.Link(),
            ),
            loader.ConfigValue(
                "html_template",
                "https://raw.githubusercontent.com/vsecoder/hikka_modules/main/assets/profile.html",
                "link to html template (if you don't know how, don't touch!)",
                validator=loader.validators.Link(),
            ),
        )
        self.name = self.strings["name"]

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

    @loader.unrestricted
    @loader.ratelimit
    async def profilecmd(self, message):
        """
        - get
        """
        chats, channels, bots, users = 0, 0, 0, 0
        message = await utils.answer(message, "Geting profile info...")
        async for dialog in self._client.iter_dialogs(ignore_migrated=True):
            if dialog.is_group:
                chats += 1
            elif dialog.is_channel:
                channels += 1
            elif dialog.entity.bot:
                bots += 1
            else:
                users += 1

        options = {"crop-w": 540, "crop-h": 220, "encoding": "UTF-8"}

        me = await self._client.get_me()
        desc = await self._client(functions.users.GetFullUserRequest(me.id))

        message = await utils.answer(message, "Downloading profile photo...")
        await self._client.download_profile_photo("me", "profile.jpg")
        message = await utils.answer(message, "Converting profile photo...")
        base64EncodedStr = base64.b64encode(open("profile.jpg", "rb").read()).decode(
            "utf-8"
        )

        message = await utils.answer(message, "Formating info to template...")
        with open("profile.html", "w") as f:
            template = requests.get(self.config["html_template"]).text
            f.write(
                template.format(
                    self.config["background"],
                    base64EncodedStr,
                    f"@{me.username}",
                    chats,
                    channels,
                    users,
                    bots,
                    desc.full_user.about,
                )
            )

        message = await utils.answer(message, "Converting to image...")
        imgkit.from_file("profile.html", "profile.jpg", options=options)
        message = await utils.answer(message, "Complete:")

        await self._client.send_file(
            utils.get_chat_id(message),
            open("profile.jpg", "rb"),
        )