# Ultroid - UserBot
# Copyright (C) 2020 TeamUltroid
#
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
# PLease read the GNU Affero General Public License in
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>

import re
from telethon.tl.types import InputWebDocument as wb
from . import get_string, async_searcher, in_pattern, InlinePlugin, async_searcher, asst

__doc__ = f"""
✘ Commands Available -
• `@{asst.username} winget <query>`
    Searches for the query on Winget and returns the results.
"""
from telethon.tl.custom import Button

@in_pattern("winget", owner=True)
async def search_winget(event):
    QUERY = event.text.split(maxsplit=1)
    try:
        query = QUERY[1]
    except IndexError:
        await event.answer(
            [], switch_pm=get_string("instu_3"), switch_pm_param="start"
        )
        return
    le = "https://api.winget.run/v2/packages?ensureContains=true&partialMatch=true&take=20&query=" + query.replace(" ", "+")
    ct = await async_searcher(le, re_json=True)
    out = []
    for on in ct["Packages"]:
        data = on["Latest"]
        name = data["Name"]
        homep = data.get("Homepage")
        text = f"> **{name}**\n - {data['Description']}\n\n`winget install {on['Id']}`\n\n**Version:** `{on['Versions'][0]}`\n"
        text += "**Tags:**" + " ".join([f"#{_}" for _ in data["Tags"]])
        if homep:
          text += f"\n\n{homep}"
        out.append(
            await event.builder.article(
                title=name, description=data["Description"], url=homep, text=text, buttons=Button.switch_inline("Search Again", "winget", same_peer=True)
            )
        )
    uppar = "|> Winget Results" if out else "No Results Found :("
    await event.answer(out, switch_pm=uppar, switch_pm_param="start", cache_time=3000)


InlinePlugin.update({"Search Winget": "winget telegram"})