from wbb import app
from wbb.utils.inlinefuncs import (
    google_search_func, urban_func, translate_func,
    alive_function, webss, shortify, wall_func,
    saavn_func, deezer_func, inline_help_func,
    torrent_func, youtube_func, lyrics_func,
    paste_func
)
from pyrogram import filters

__MODULE__ = "Inline"
__HELP__ = '''```+===============================================================+
|            Query            |           Function              |
+===============================================================+
|  alive                      |  Check Bot's Stats.             |
|  tr [LANGUAGE_CODE] [QUERY] |  Translate Text.                |
|  ud      [QUERY]            |  Urban Dictionary Query.        |
|  google  [QUERY]            |  Google Search.                 |
|  webss   [URL]              |  Take Screenshot Of A Website.  |
|  bitly   [URL]              |  Shorten A Link.                |
|  wall    [QUERY]            |  Find Wallpapers.               |
|  saavn   [SONG_NAME]        |  Get Songs From Saavn.          |
|  deezer  [SONG_NAME]        |  Get Songs From Deezer.         |
|  yt      [Query]            |  Youtube Search.                |
|  torrent [QUERY]            |  Torrent Search.                |
|  lyrics  [QUERY]            |  Lyrics Search.                 |
|  paste   [TEXT]             |  Pase Text On Pastebin. <=500   |
+===============================================================+
```'''


@app.on_message(filters.command("inline"))
async def inline_help(_, message):
    await app.send_message(message.chat.id, text=__HELP__)


@app.on_inline_query()
async def inline_query_handler(client, query):
    try:
        text = query.query.lower()
        answers = []
        if text.strip() == '':
            answerss = await inline_help_func(__HELP__)
            await client.answer_inline_query(
                query.id,
                results=answerss,
                cache_time=10
            )
            return
        elif text.split()[0] == "alive":
            answerss = await alive_function(answers)
            await client.answer_inline_query(
                query.id,
                results=answerss,
                cache_time=10
            )
        elif text.split()[0] == "tr":
            lang = text.split()[1]
            tex = text.split(None, 2)[2]
            answerss = await translate_func(answers, lang, tex)
            await client.answer_inline_query(
                query.id,
                results=answerss,
            )
        elif text.split()[0] == "ud":
            tex = text.split(None, 1)[1]
            answerss = await urban_func(answers, tex)
            await client.answer_inline_query(
                query.id,
                results=answerss,
            )
        elif text.split()[0] == "google":
            tex = text.split(None, 1)[1]
            answerss = await google_search_func(answers, tex)
            await client.answer_inline_query(
                query.id,
                results=answerss,
            )
        elif text.split()[0] == "webss":
            tex = text.split(None, 1)[1]
            answerss = await webss(tex)
            await client.answer_inline_query(
                query.id,
                results=answerss,
                cache_time=2
            )
        elif text.split()[0] == "bitly":
            tex = text.split(None, 1)[1]
            answerss = await shortify(tex)
            await client.answer_inline_query(
                query.id,
                results=answerss,
                cache_time=2
            )

        elif text.split()[0] == "wall":
            tex = text.split(None, 1)[1]
            answerss = await wall_func(answers, tex)
            await client.answer_inline_query(
                query.id,
                results=answerss
            )

        elif text.split()[0] == "saavn":
            tex = text.split(None, 1)[1]
            answerss = await saavn_func(answers, tex)
            await client.answer_inline_query(
                query.id,
                results=answerss
            )

        elif text.split()[0] == "deezer":
            tex = text.split(None, 1)[1]
            answerss = await deezer_func(answers, tex)
            await client.answer_inline_query(
                query.id,
                results=answerss
            )

        elif text.split()[0] == "torrent":
            tex = text.split(None, 1)[1]
            answerss = await torrent_func(answers, tex)
            await client.answer_inline_query(
                query.id,
                results=answerss,
            )

        elif text.split()[0] == "yt":
            tex = text.split(None, 1)[1]
            answerss = await youtube_func(answers, tex)
            await client.answer_inline_query(
                query.id,
                results=answerss
            )

        elif text.split()[0] == "lyrics":
            tex = text.split(None, 1)[1]
            answerss = await lyrics_func(answers, tex)
            await client.answer_inline_query(
                query.id,
                results=answerss
            )

        elif text.split()[0] == "paste":
            tex = text.split(None, 1)[1]
            answerss = await paste_func(answers, tex)
            await client.answer_inline_query(
                query.id,
                results=answerss,
                cache_time=2
            )

    except (IndexError, TypeError, KeyError, ValueError):
        return