import asyncio

from core import command
from loguru import logger
from pyrogram import Client
from pyrogram.errors import FloodWait, RPCError
from pyrogram.types import Message
from tools.helpers import delete_this, escape_markdown
from tools.sessions import session


@Client.on_message(command("diss"))
async def diss(_: Client, msg: Message):
    """喷人"""
    symbol = '💢 '
    api = 'https://zuan.shabi.workers.dev/'
    await msg.edit_text(f"{symbol}It's preparating.")
    await get_api(api=api, msg=msg)


@Client.on_message(command('tg'))
async def tg(_: Client, msg: Message):
    """舔狗"""
    symbol = '👅 '
    api = 'http://ovooa.com/API/tgrj/api.php'
    await msg.edit_text(f"{symbol}It's preparating.")
    await get_api(api=api, msg=msg)


async def get_api(api: str, msg: Message) -> None:
    for _ in range(10):
        try:
            resp = await session.get(api, timeout=5.5)
            if resp.status == 200:
                text = escape_markdown(await resp.text())
            else:
                resp.raise_for_status()
        except Exception as e:
            logger.error(e)
            continue
        words = f"{msg.reply_to_message.from_user.mention(style='md')} {text}" \
            if msg.reply_to_message and msg.reply_to_message.from_user else text
        try:
            await msg.edit_text(words, parse_mode='md')
        except FloodWait as e:
            await asyncio.sleep(e.x)
            await msg.edit_text(words, parse_mode='md')
        except RPCError as e:
            logger.error(e)
        await logger.complete()
        return
    # Failed to get api text
    await delete_this(msg)
    res = await msg.edit_text('😤 Rest for a while.')
    await asyncio.sleep(3)
    await delete_this(res)