#  Moon-Userbot - telegram userbot
#  Copyright (C) 2020-present Moon Userbot Organization
#
#  This program is free software: you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation, either version 3 of the License, or
#  (at your option) any later version.

#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.

#  You should have received a copy of the GNU General Public License
#  along with this program.  If not, see <https://www.gnu.org/licenses/>.

import os
from io import BytesIO

from pyrogram import Client, enums, filters, types
from utils.misc import modules_help, prefix
from utils.scripts import (
    format_exc,
    interact_with,
    interact_with_to_delete,
    resize_image,
    with_reply,
)


@Client.on_message(filters.command('kang', prefix) & filters.me)
@with_reply
async def kang(client: Client, message: types.Message):
    await message.edit('<b>Please wait...</b>')

    if len(message.command) < 2:
        await message.edit(
            f'<b>No arguments provided\nUsage: <code>{prefix}kang [pack]* [emoji]</code></b>',
        )
        return

    pack = message.command[1]
    if len(message.command) >= 3:
        emoji = message.command[2]
    else:
        emoji = '✨'

    await client.unblock_user('@stickers')
    await interact_with(await client.send_message('@stickers', '/cancel', parse_mode=enums.ParseMode.MARKDOWN))
    await interact_with(await client.send_message('@stickers', '/addsticker', parse_mode=enums.ParseMode.MARKDOWN))

    result = await interact_with(await client.send_message('@stickers', pack, parse_mode=enums.ParseMode.MARKDOWN))
    if '.TGS' in result.text:
        await message.edit("<b>Animated packs aren't supported</b>")
        return
    if 'StickerExample.psd' not in result.text:
        await message.edit(
            "<b>Stickerpack doesn't exitst. Create it using @Stickers bot (via /newpack command)</b>",
        )
        return

    try:
        path = await message.reply_to_message.download()
    except ValueError:
        await message.edit(
            "<b>Replied message doesn't contain any downloadable media</b>",
        )
        return

    resized = resize_image(path)
    if os.path.exists(path):
        os.remove(path)

    await interact_with(await client.send_document('@stickers', resized, parse_mode=enums.ParseMode.MARKDOWN))
    response = await interact_with(await client.send_message('@stickers', emoji, parse_mode=enums.ParseMode.MARKDOWN))
    if '/done' in response.text:
        # ok
        await interact_with(await client.send_message('@stickers', '/done', parse_mode=enums.ParseMode.MARKDOWN))
        await client.delete_messages('@stickers', interact_with_to_delete)
        await message.edit(
            f'<b>Sticker added to <a href=https://t.me/addstickers/{pack}>pack</a></b>',
        )
    else:
        await message.edit('<b>Something went wrong. Check history with @stickers</b>')
    interact_with_to_delete.clear()


@Client.on_message(filters.command(['stp', 's2p', 'stick2png'], prefix) & filters.me)
@with_reply
async def stick2png(client: Client, message: types.Message):
    try:
        await message.edit('<b>Downloading...</b>')

        path = await message.reply_to_message.download()
        with open(path, 'rb') as f:
            content = f.read()
        if os.path.exists(path):
            os.remove(path)

        file_io = BytesIO(content)
        file_io.name = 'sticker.png'

        await client.send_document(message.chat.id, file_io, parse_mode=enums.ParseMode.MARKDOWN)
    except Exception as e:
        await message.edit(format_exc(e))
    else:
        await message.delete()


@Client.on_message(filters.command(['resize'], prefix) & filters.me)
@with_reply
async def resize_cmd(client: Client, message: types.Message):
    try:
        await message.edit('<b>Downloading...</b>')

        path = await message.reply_to_message.download()
        resized = resize_image(path)
        resized.name = 'image.png'
        if os.path.exists(path):
            os.remove(path)

        await client.send_document(message.chat.id, resized, parse_mode=enums.ParseMode.MARKDOWN)
    except Exception as e:
        await message.edit(format_exc(e))
    else:
        await message.delete()


modules_help['stickers'] = {
    'kang [reply]* [pack]* [emoji]': 'Add sticker to defined pack',
    'stp [reply]*': 'Convert replied sticker to PNG',
    'resize [reply]*': 'Resize replied image to 512xN format',
}