from random import randint

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


@Client.on_message(filters.command(['aniq', 'aq'], prefix) & filters.me)
async def aniquotes_handler(client: Client, message: Message):
    if message.reply_to_message and message.reply_to_message.text:
        query = message.reply_to_message.text[:512]
    elif message.reply_to_message and message.reply_to_message.caption:
        query = message.reply_to_message.caption[:512]
    elif len(message.command) > 1:
        query = message.text.split(maxsplit=1)[1][:512]
    else:
        return await message.edit(
            '<b>[💮 Aniquotes] <i>Please enter text to create sticker.</i></b>',
            parse_mode=enums.ParseMode.HTML,
        )

    try:
        await message.delete()
        result = await client.get_inline_bot_results('@quotafbot', query)
        return await message.reply_inline_bot_result(
            query_id=result.query_id,
            result_id=result.results[randint(1, 2)].id,  # noqa: S311
            reply_to_message_id=(message.reply_to_message.id if message.reply_to_message else None),
        )
    except Exception as e:
        return await message.reply(
            f'<b>[💮 Aniquotes]</b>\n<code>{format_exc(e)}</code>',
            parse_mode=enums.ParseMode.HTML,
        )


modules_help['aniquotes'] = {
    'aq [text]': 'Create animated sticker with text',
}