import html
import re

from pyrogram.enums import ParseMode

from pagermaid.enums import Message, AsyncClient
from pagermaid.listener import listener


def clean_html(raw_html):
    return re.sub(re.compile(r"<.*?>"), "", raw_html)


def escape_definition(definition):
    for key, value in definition.items():
        if isinstance(value, str):
            definition[key] = html.escape(clean_html(value))
    return definition


@listener(command="pypi",
          description="Search PyPI packages",
          parameters="The query string")
async def pypi(message: Message, httpx: AsyncClient):
    if not message.arguments:
        return await message.edit("Please provide a query string")
    r = await httpx.get(f"https://pypi.org/pypi/{message.arguments}/json", follow_redirects=True)
    if r.status_code != 200:
        return await message.edit("Could not find the package")
    json = r.json()
    pypi_info = escape_definition(json["info"])
    text = """
<b><a href="{package_link}">{package_name}</a></b> by <i>{author_name} {author_email}</i>
平台：<b>{platform}</b>
版本：<b>{version}</b>
许可协议：<b>{license}</b>
摘要：<b>{summary}</b>""".format(
            package_link=f"https://pypi.org/pypi/{message.arguments}",
            package_name=pypi_info["name"],
            author_name=pypi_info["author"],
            author_email=f"&lt;{pypi_info['author_email']}&gt;"
            if pypi_info["author_email"]
            else "",
            platform=pypi_info["platform"] or "未指定",
            version=pypi_info["version"],
            license=pypi_info["license"] or "未指定",
            summary=pypi_info["summary"],
        )
    await message.edit(text, parse_mode=ParseMode.HTML, disable_web_page_preview=True)