import asyncio

from pykeyboard import InlineKeyboard
from pyrogram import filters
from misskaty.helper.http import http
from misskaty import app
from misskaty.vars import COMMAND_HANDLER


async def getData(chat_id, message_id, GetWord, CurrentPage):
    UDJson = (await http.get(f"https://api.urbandictionary.com/v0/define?term={GetWord}")).json()

    if "list" not in UDJson:
        CNMessage = await app.send_message(chat_id=chat_id, reply_to_message_id=message_id, text=(f"Word: {GetWord}\n" "Results: Sorry could not find any matching results!"))
        await asyncio.sleep(5)
        await CNMessage.delete()
        return
    try:
        index = int(CurrentPage - 1)
        PageLen = len(UDJson["list"])

        UDReasult = f"**Definition of {GetWord}**\n" f"{UDJson['list'][index]['definition']}\n\n" "**📌 Examples**\n" f"__{UDJson['list'][index]['example']}__"

        UDFReasult = "".join(i for i in UDReasult if i not in "[]")

        return (UDFReasult, PageLen)

    except IndexError or KeyError:
        CNMessage = await app.send_message(chat_id=chat_id, reply_to_message_id=message_id, text=(f"Word: {GetWord}\n" "Results: Sorry could not find any matching results!"))
        await asyncio.sleep(5)
        await CNMessage.delete()


@app.on_message(filters.command(["ud"], COMMAND_HANDLER))
async def urbanDictionary(client, message):
    if not message.from_user:
        return
    message_id = message.id
    chat_id = message.chat.id
    GetWord = " ".join(message.command[1:])
    if not GetWord:
        message = await message.chat.ask("Now give any word for query!", identifier=(message.from_user.id, message.from_user.id, None))
        GetWord = message.text

    CurrentPage = 1
    UDReasult, PageLen = await getData(chat_id, message_id, GetWord, CurrentPage)

    keyboard = InlineKeyboard()
    keyboard.paginate(PageLen, CurrentPage, "pagination_urban#{number}" + f"#{GetWord}")
    await message.reply(text=f"{UDReasult}", reply_markup=keyboard)


@app.on_callback_query(filters.create(lambda _, __, query: "pagination_urban#" in query.data))
async def ud_callback(client, callback_query):
    message_id = callback_query.message.id
    chat_id = callback_query.message.chat.id
    CurrentPage = int(callback_query.data.split("#")[1])
    GetWord = callback_query.data.split("#")[2]

    try:
        UDReasult, PageLen = await getData(chat_id, message_id, GetWord, CurrentPage)
    except TypeError:
        return

    keyboard = InlineKeyboard()
    keyboard.paginate(PageLen, CurrentPage, "pagination_urban#{number}" + f"#{GetWord}")
    await app.edit_message_text(chat_id=chat_id, message_id=message_id, text=UDReasult, reply_markup=keyboard)