import random
from io import BytesIO

from pyrogram import Client, enums, filters
from pyrogram.types import Message
from utils.misc import modules_help, prefix
from utils.scripts import import_library

requests = import_library('requests')
PIL = import_library('PIL', 'pillow')

from PIL import Image, ImageDraw, ImageFont


@Client.on_message(filters.command(['dem'], prefix) & filters.me)
async def demotivator(client: Client, message: Message):
    await message.edit('<code>Process of demotivation...</code>', parse_mode=enums.ParseMode.HTML)
    font = requests.get('https://github.com/The-MoonTg-project/files/blob/main/Times%20New%20Roman.ttf?raw=true')
    f = font.content
    template_dem = requests.get('https://raw.githubusercontent.com/files/main/demotivator.png')
    if message.reply_to_message:
        words = ['random', 'text', 'typing', 'fuck']
        if message.reply_to_message.photo:
            donwloads = await client.download_media(message.reply_to_message.photo.file_id)
            photo = Image.open(f'{donwloads}')
            resize_photo = photo.resize((469, 312))
            text = message.text.split(' ', maxsplit=1)[1] if len(message.text.split()) > 1 else random.choice(words)
            im = Image.open(BytesIO(template_dem.content))
            im.paste(resize_photo, (65, 48))
            text_font = ImageFont.truetype(BytesIO(f), 22)
            text_draw = ImageDraw.Draw(im)
            text_draw.multiline_text((299, 412), text, font=text_font, fill=(255, 255, 255), anchor='ms')
            im.save(f'downloads/{message.id}.png')
            await message.reply_to_message.reply_photo(f'downloads/{message.id}.png')
            await message.delete()
        elif message.reply_to_message.sticker:
            if not message.reply_to_message.sticker.is_animated:
                donwloads = await client.download_media(message.reply_to_message.sticker.file_id)
                photo = Image.open(f'{donwloads}')
                resize_photo = photo.resize((469, 312))
                text = message.text.split(' ', maxsplit=1)[1] if len(message.text.split()) > 1 else random.choice(words)
                im = Image.open(BytesIO(template_dem.content))
                im.paste(resize_photo, (65, 48))
                text_font = ImageFont.truetype(BytesIO(f), 22)
                text_draw = ImageDraw.Draw(im)
                text_draw.multiline_text((299, 412), text, font=text_font, fill=(255, 255, 255), anchor='ms')
                im.save(f'downloads/{message.id}.png')
                await message.reply_to_message.reply_photo(f'downloads/{message.id}.png')
                await message.delete()
            else:
                await message.edit(
                    '<b>Animated stickers are not supported</b>',
                    parse_mode=enums.ParseMode.HTML,
                )
        else:
            await message.edit(
                '<b>Need to answer the photo/sticker</b>',
                parse_mode=enums.ParseMode.HTML,
            )
    else:
        await message.edit('<b>Need to answer the photo/sticker</b>', parse_mode=enums.ParseMode.HTML)


modules_help['demotivator'] = {'dem [text]*': 'Reply to the picture to make a demotivator out of it'}