# requires: pillow
# requires: wand
from .. import loader, utils
import io
from telethon.tl.types import DocumentAttributeFilename
import logging
from wand.image import Image
from PIL import Image as IM
# https://t.me/KeyZenD
# https://t.me/SomeScripts
logger = logging.getLogger(__name__)

def register(cb):
    cb(DistortNoApiMod())


@loader.tds
class DistortNoApiMod(loader.Module):
    """distorting images"""
    strings = {
        "name": "DistortNoApi"
    }

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

    @loader.sudo
    async def distortcmd(self, message):
        """.distort <reply to photo>
        .distort im
        .distort 50
        .distort 50 im
        .distort im 50
        im => кидает стикеры как фото
        50 => (от 0 до дохуя) процент сжатия"""
        if message.is_reply:
            reply_message = await message.get_reply_message()
            data, mime = await check_media(reply_message)
            if isinstance(data, bool):
                await utils.answer(message, "<code>Reply to image or stick!</code>")
                return
        else:
            await utils.answer(message, "<code>Reply to image or stick!</code>")
            return
        rescale_rate = 70
        a = utils.get_args(message)
        force_file = False
        if a:
            if 'im' in a:
                force_file = True
                a.remove('im')
            if len(a) > 0:
                if a[0].isdigit():
                    rescale_rate = int(a[0])
                    if rescale_rate <= 0:
                        rescale_rate = 70

        await message.edit("<code>D i s t o r t i n g . . .</code>")
        file = await message.client.download_media(data, bytes)
        file, img = io.BytesIO(file), io.BytesIO()
        img.name = 'img.png'
        IM.open(file).save(img, 'PNG')
        media = await distort(io.BytesIO(img.getvalue()), rescale_rate)
        out, im = io.BytesIO(), IM.open(media)
        if force_file:
            mime = 'png'
        out.name = f'out.{mime}'
        im.save(out, mime.upper())
        out.seek(0)
        await message.edit("<code>S e n d i n g . . .</code>")
        await message.client.send_file(message.to_id, out, reply_to=reply_message.id)

        await message.delete()

async def distort(file, rescale_rate):
    img = Image(file=file)
    x, y = img.size[0], img.size[1]
    popx = int(rescale_rate*(x//100))
    popy = int(rescale_rate*(y//100))
    img.liquid_rescale(popx, popy, delta_x=1, rigidity=0)
    img.resize(x, y)
    out = io.BytesIO()
    out.name = f'output.png'
    img.save(file=out)
    return io.BytesIO(out.getvalue())

async def check_media(reply_message):
    mime = None
    if reply_message and reply_message.media:
        if reply_message.photo:
            data = reply_message.photo
            mime = 'image/jpeg'
        elif reply_message.document:
            if DocumentAttributeFilename(file_name='AnimatedSticker.tgs') in reply_message.media.document.attributes:
                return False, mime
            if reply_message.gif or reply_message.video or reply_message.audio or reply_message.voice:
                return False, mime
            data = reply_message.media.document
            mime = reply_message.media.document.mime_type
            if 'image/' not in mime:
                return False, mime
        else:
            return False, mime
    else:
        return False, mime

    if not data or data is None:
        return False, mime
    else:
        mime = mime.split('/')[1]
        return data, mime