import io

from PIL import Image, ImageOps

from ubot import ldr


@ldr.add("chatid")
async def chatidgetter(event):
    if event.is_reply:
        reply = await event.get_reply_message()
        if reply.forward and reply.forward.channel_id:
            await event.reply(f"**Channel ID:** {reply.forward.channel_id}")
            return
        chat_id = reply.chat_id
    else:
        chat_id = event.chat_id

    await event.reply(f"**Chat ID:** {chat_id}")


@ldr.add("userid")
async def useridgetter(event):
    if event.is_reply:
        reply = await event.get_reply_message()
        user_id = reply.sender_id
    else:
        user_id = event.sender_id

    await event.reply(f"**User ID:** {user_id}")


@ldr.add("stickpng", help="Converts stickers to PNG files.")
async def stickertopng(event):
    sticker = await event.get_sticker()

    if not sticker:
        await event.reply("Reply to a sticker to get it as a PNG file!")
        return

    sticker_webp_io = io.BytesIO()
    await event.client.download_media(sticker, sticker_webp_io)
    await event.reply(file=await ldr.run_async(stickertopngsync, sticker_webp_io), force_document=True)


def stickertopngsync(sticker_webp_io):
    sticker_webp = Image.open(sticker_webp_io)
    sticker_png_io = io.BytesIO()
    sticker_webp.save(sticker_png_io, "PNG")
    sticker_png_io.name = "sticker.png"
    sticker_png_io.seek(0)

    return sticker_png_io


@ldr.add("stickflip", help="Flips stickers horizontally.")
async def flipsticker(event):
    sticker = await event.get_sticker()

    if not sticker:
        await event.reply("Reply to a sticker to flip that bitch!")
        return

    sticker_webp_io = io.BytesIO()
    await event.client.download_media(sticker, sticker_webp_io)
    await event.reply(file=await ldr.run_async(flipstickersync, sticker_webp_io))


def flipstickersync(sticker_webp_io):
    sticker_webp = Image.open(sticker_webp_io)
    sticker_webp = ImageOps.mirror(sticker_webp)
    sticker_flipped_io = io.BytesIO()
    sticker_webp.save(sticker_flipped_io, "WebP")
    sticker_flipped_io.name = "sticker.webp"
    sticker_flipped_io.seek(0)

    return sticker_flipped_io


@ldr.add("stickimg", help="Converts images to sticker sized PNG files.")
async def createsticker(event):
    data = await event.get_image()

    if not data:
        await event.reply("Reply to or caption an image to make it sticker-sized!")
        return

    image_io = io.BytesIO()
    await event.client.download_media(data, image_io)
    await event.reply(file=await ldr.run_async(createstickersync, image_io), force_document=True)


def createstickersync(image_io):
    sticker_png = Image.open(image_io)
    sticker_png = sticker_png.crop(sticker_png.getbbox())

    final_width = 512
    final_height = 512

    if sticker_png.width > sticker_png.height:
        final_height = 512 * (sticker_png.height / sticker_png.width)
    elif sticker_png.width < sticker_png.height:
        final_width = 512 * (sticker_png.width / sticker_png.height)

    sticker_png = ImageOps.fit(sticker_png, (int(final_width), int(final_height)))
    sticker_new_io = io.BytesIO()
    sticker_png.save(sticker_new_io, "PNG")
    sticker_new_io.name = "sticker.png"
    sticker_new_io.seek(0)

    return sticker_new_io