from pyrogram import Client
from json.decoder import JSONDecodeError
from pagermaid.listener import listener
from pagermaid.utils import Message, client


@listener(command="zhrs",
          description="知乎热搜。")
async def zhrs(_: Client, message: Message):
    req = await client.get("https://tenapi.cn/zhihuresou/")
    if req.status_code == 200:
        try:
            data = req.json()
        except JSONDecodeError:
            await message.edit("出错了呜呜呜 ~ API 数据解析失败。")
            return
        res = '知乎实时热搜榜：\n'
        for i in range(10):
            res += f'\n{i + 1}.「<a href={data["list"][i]["url"]}>{data["list"][i]["query"]}</a>」'
        await message.edit(res)
    else:
        await message.edit("出错了呜呜呜 ~ 无法访问到 API 服务器 。")


@listener(command="wbrs",
          description="微博热搜。")
async def wbrs(_: Client, message: Message):
    req = await client.get("https://tenapi.cn/resou/")
    if req.status_code == 200:
        try:
            data = req.json()
        except JSONDecodeError:
            await message.edit("出错了呜呜呜 ~ API 数据解析失败。")
            return
        res = '微博实时热搜榜：\n'
        for i in range(10):
            res += f'\n{i + 1}.「<a href={data["list"][i]["url"]}>{data["list"][i]["name"]}</a>」  ' \
                   f'热度：{data["list"][i]["hot"]}'
        await message.edit(res)
    else:
        await message.edit("出错了呜呜呜 ~ 无法访问到 API 服务器 。")


@listener(command="dyrs",
          description="抖音热搜。")
async def dyrs(_: Client, message: Message):
    req = await client.get("https://tenapi.cn/douyinresou/")
    if req.status_code == 200:
        try:
            data = req.json()
        except JSONDecodeError:
            await message.edit("出错了呜呜呜 ~ API 数据解析失败。")
            return
        res = '抖音实时热搜榜：\n'
        for i in range(10):
            res += f'\n{i + 1}.「{data["list"][i]["name"]}」  热度：{data["list"][i]["hot"]}'
        await message.edit(res)
    else:
        await message.edit("出错了呜呜呜 ~ 无法访问到 API 服务器 。")