from time import sleep
from requests import get
from pagermaid import version
from pagermaid.listener import listener
from pagermaid.utils import alias_command, client


@listener(is_plugin=True, outgoing=True, command=alias_command("chp"),
          description="彩虹屁生成器。")
async def chp(context):
    await context.edit("获取中 . . .")
    status = False
    for _ in range(20):  # 最多尝试20次
        req = await client.get("https://api.shadiao.app/chp")
        if req.status_code == 200:
            res = req.json()
            await context.edit(res.get("data", {}).get("text", "什么都没有找到"), parse_mode='html', link_preview=False)
            status = True
            break
        else:
            continue
    if status == False:
        await context.edit("出错了呜呜呜 ~ 试了好多好多次都无法访问到 API 服务器 。")
        sleep(2)
        await context.delete()


@listener(is_plugin=True, outgoing=True, command=alias_command("djt"),
          description="毒鸡汤生成器。")
async def djt(context):
    await context.edit("获取中 . . .")
    status = False
    for _ in range(20):  # 最多尝试20次
        req = await client.get("https://api.shadiao.app/du")
        if req.status_code == 200:
            res = req.json()
            await context.edit(res.get("data", {}).get("text", "什么都没有找到"), parse_mode='html', link_preview=False)
            status = True
            break
        else:
            continue
    if status == False:
        await context.edit("出错了呜呜呜 ~ 试了好多好多次都无法访问到 API 服务器 。")
        sleep(2)
        await context.delete()


@listener(is_plugin=True, outgoing=True, command=alias_command("yxh"),
          description="营销号文案生成器（建议配合tts食用）。", parameters="<主体> <事件> <原因>")
async def yxh(context):
    try:
        await context.edit("生成中 . . .")
        text = f"{context.parameter[0]}{context.parameter[1]}是怎么回事呢？" \
               f"{context.parameter[0]}相信大家都很熟悉，" \
               f"但是{context.parameter[0]}{context.parameter[1]}是怎么回事呢，下面就让小编带大家一起了解吧。\n" \
               f"{context.parameter[0]}{context.parameter[1]}，其实就是{context.parameter[2]}，" \
               f"大家可能会很惊讶{context.parameter[0]}怎么会{context.parameter[1]}呢？但事实就是这样，" \
               f"小编也感到非常惊讶。\n这就是关于{context.parameter[0]}{context.parameter[1]}的事情了，" \
               f"大家有什么想法呢，欢迎在评论区告诉小编一起讨论哦！"
    except IndexError:
        await context.edit("使用方法：-yxh <主体> <事件> <原因>")
        return
    await context.edit(text)