zhenxun_bot/plugins/genshin/query_user/bind/__init__.py

import json
from typing import Tuple

from nonebot import on_command
from nonebot.adapters.onebot.v11 import GroupMessageEvent, Message, MessageEvent
from nonebot.params import Command, CommandArg

from services.log import logger
from utils.depends import OneCommand
from utils.http_utils import AsyncHttpx
from utils.utils import is_number

from .._models import Genshin

__zx_plugin_name__ = "原神绑定"
__plugin_usage__ = """
usage：
    绑定原神uid等数据，cookie极为重要，请谨慎绑定
    ** 如果对拥有者不熟悉，并不建议添加cookie **
    该项目只会对cookie用于”米游社签到“，“原神玩家查询”，“原神便笺查询”
    指令：
        原神绑定uid [uid]
        原神绑定米游社id [mys_id]
        原神绑定cookie [cookie] # 该绑定请私聊
        原神解绑
        示例：原神绑定uid 92342233
    如果不明白怎么获取cookie请输入“原神绑定cookie”。
""".strip()
__plugin_des__ = "绑定自己的原神uid等"
__plugin_cmd__ = ["原神绑定uid [uid]", "原神绑定米游社id [mys_id]", "原神绑定cookie [cookie]", "原神解绑"]
__plugin_type__ = ("原神相关",)
__plugin_version__ = 0.1
__plugin_author__ = "HibiKier"
__plugin_settings__ = {
    "level": 5,
    "default_status": True,
    "limit_superuser": False,
    "cmd": ["原神绑定"],
}

bind = on_command(
    "原神绑定uid", aliases={"原神绑定米游社id", "原神绑定cookie"}, priority=5, block=True
)

unbind = on_command("原神解绑", priority=5, block=True)

web_Api = "https://api-takumi.mihoyo.com"
bbs_Cookie_url = "https://webapi.account.mihoyo.com/Api/cookie_accountinfo_by_loginticket?login_ticket={}"
bbs_Cookie_url2 = (
    web_Api
    + "/auth/api/getMultiTokenByLoginTicket?login_ticket={}&token_types=3&uid={}"
)


@bind.handle()
async def _(event: MessageEvent, cmd: str = OneCommand(), arg: Message = CommandArg()):
    msg = arg.extract_plain_text().strip()
    user = await Genshin.get_or_none(user_qq=event.user_id)
    if cmd in ["原神绑定uid", "原神绑定米游社id"]:
        if not is_number(msg):
            await bind.finish("uid/id必须为纯数字！", at_senders=True)
        msg = int(msg)
    if cmd == "原神绑定uid":
        if user:
            await bind.finish(f"您已绑定过uid：{user.uid}，如果希望更换uid，请先发送原神解绑")
        if await Genshin.get_or_none(user_qq=event.user_id, uid=msg):
            await bind.finish("添加失败，该uid可能已存在...")
        user = await Genshin.create(user_qq=event.user_id, uid=msg)
        _x = f"已成功添加原神uid：{msg}"
    elif cmd == "原神绑定米游社id":
        if not user:
            await bind.finish("请先绑定原神uid..")
        user.mys_id = int(msg)
        _x = f"已成功为uid：{user.uid} 设置米游社id：{msg}"
    else:
        if not msg:
            await bind.finish(
                """私聊发送！！
            1.以无痕模式打开浏览器（Edge请新建InPrivate窗口）
            2.打开http://bbs.mihoyo.com/ys/ 并登陆
            3.登陆后打开http://user.mihoyo.com/进行登陆
            4.按下F12，打开控制台，输入以下命令：
            var cookie=document.cookie;var ask=confirm('Cookie:'+cookie+'\\n\\nDo you want to copy the cookie to the clipboard?');if(ask==true){copy(cookie);msg=cookie}else{msg='Cancel'}
            5.私聊发送：原神绑定cookie 刚刚复制的cookie"""
            )
        if isinstance(event, GroupMessageEvent):
            await bind.finish("请立即撤回你的消息并私聊发送！")
        if not user:
            await bind.finish("请先绑定原神uid..")
        if msg.startswith('"') or msg.startswith("'"):
            msg = msg[1:]
        if msg.endswith('"') or msg.endswith("'"):
            msg = msg[:-1]
        cookie = msg
        # 用: 代替=, ,代替;
        cookie = '{"' + cookie.replace("=", '": "').replace("; ", '","') + '"}'
        # print(cookie)
        cookie_json = json.loads(cookie)
        # print(cookie_json)
        if "login_ticket" not in cookie_json:
            await bind.finish("请发送正确完整的cookie！")
        user.cookie = str(msg)
        login_ticket = cookie_json["login_ticket"]
        # try:
        res = await AsyncHttpx.get(url=bbs_Cookie_url.format(login_ticket))
        res.encoding = "utf-8"
        data = json.loads(res.text)
        # print(data)
        if "成功" in data["data"]["msg"]:
            stuid = str(data["data"]["cookie_info"]["account_id"])
            res = await AsyncHttpx.get(url=bbs_Cookie_url2.format(login_ticket, stuid))
            res.encoding = "utf-8"
            data = json.loads(res.text)
            stoken = data["data"]["list"][0]["token"]
            # await Genshin.set_cookie(uid, cookie)
            user.stoken = stoken
            user.stuid = stuid
            user.login_ticket = login_ticket
        # except Exception as e:
        #     await bind.finish("获取登陆信息失败，请检查cookie是否正确或更新cookie")
        elif data["data"]["msg"] == "登录信息已失效，请重新登录":
            await bind.finish("登录信息失效，请重新获取最新cookie进行绑定")
        _x = f"已成功为uid：{user.uid} 设置cookie"
    if isinstance(event, GroupMessageEvent):
        user.bind_group = event.group_id
    if user:
        await user.save(
            update_fields=[
                "mys_id",
                "cookie",
                "stoken",
                "stuid",
                "login_ticket",
                "bind_group",
            ]
        )
    await bind.send(_x)
    logger.info(
        f"(USER {event.user_id}, "
        f"GROUP {event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
        f" {cmd}：{msg}"
    )


@unbind.handle()
async def _(event: MessageEvent):
    await Genshin.filter(user_qq=event.user_id).delete()
    await unbind.send("用户数据删除成功...")
    logger.info(
        f"(USER {event.user_id}, GROUP "
        f"{event.group_id if isinstance(event, GroupMessageEvent) else 'private'})"
        f"原神解绑"
    )