import json
from pyrogram import Client

from pagermaid.listener import listener
from pagermaid.utils import Message, client


@listener(
    is_plugin=True, command="guess", description="能不能好好说话？ - 拼音首字母缩写释义工具（需要回复一句话）"
)
async def guess(_: Client, message: Message):
    text = message.arguments
    if not text:
        return await message.edit("请先输入一个缩写。")
    message = await message.edit("获取中 . . .")

    text = {"text": text}
    guess_json = (
        await client.post("https://lab.magiconch.com/api/nbnhhsh/guess", json=text)
    ).json()
    if len(guess_json) != 0:
        guess_res = []
        for num in range(len(guess_json)):
            guess_res1 = json.loads(json.dumps(guess_json[num]))
            guess_res1_name = guess_res1["name"]
            try:
                guess_res1_ans = ", ".join(guess_res1["trans"])
            except:
                try:
                    guess_res1_ans = ", ".join(guess_res1["inputting"])
                except:
                    guess_res1_ans = "尚未录入"
            guess_res.extend([f"词组：{guess_res1_name}" + "\n释义：" + guess_res1_ans])
        await message.edit("\n\n".join(guess_res))
    else:
        await message.edit("没有匹配到拼音首字母缩写")


@listener(is_plugin=True, command="wiki", description="查询维基百科词条", parameters="[词组]")
async def wiki(_: Client, message: Message):
    text = message.arguments
    if not text:
        return await message.edit("请先输入一个关键词。")
    message = await message.edit("获取中 . . .")
    try:
        req = await client.get(
            f"https://zh.wikipedia.org/w/api.php?action=query&list=search&format=json&formatversion=2&srsearch={text}"
        )

        wiki_json = json.loads(req.content.decode("utf-8"))
    except:
        return await message.edit("出错了呜呜呜 ~ 无法访问到维基百科。")
    try:
        if len(wiki_json["query"]["search"]) != 0:
            wiki_title = wiki_json["query"]["search"][0]["title"]
            wiki_content = (
                wiki_json["query"]["search"][0]["snippet"]
                .replace('<span class="searchmatch">', "**")
                .replace("</span>", "**")
            )
            wiki_time = (
                wiki_json["query"]["search"][0]["timestamp"]
                .replace("T", " ")
                .replace("Z", " ")
            )
            text = (
                (
                    f"词条： [{wiki_title}](https://zh.wikipedia.org/zh-cn/{wiki_title}"
                    + ")\n\n"
                    + wiki_content
                )
                + "...\n\n此词条最后修订于 "
            ) + wiki_time

        else:
            text = "没有匹配到相关词条"
    except KeyError:
        text = wiki_json["error"]["info"]
    await message.edit(text)