LittlePaimon/Paimon_Info/__init__.py

import datetime
import random
import re
from asyncio import sleep
from collections import defaultdict

from littlepaimon_utils.tools import FreqLimiter
from nonebot import on_command, require, logger, get_bot
from nonebot.adapters.onebot.v11 import MessageEvent, Message, Bot, MessageSegment
from nonebot.params import CommandArg, Arg
from nonebot.permission import SUPERUSER
from nonebot.plugin import PluginMetadata
from nonebot.rule import to_me
from nonebot.typing import T_State

from .draw_abyss_info import draw_abyss_card
from .draw_daily_note import draw_daily_note_card
from .draw_month_info import draw_monthinfo_card
from .draw_player_card import draw_player_card, draw_all_chara_card, draw_chara_card
from .draw_role_card import draw_role_card
from .get_data import addStoken, get_bind_game, get_sign_info, sign, get_sign_list, get_abyss_data, get_daily_note_data, \
    get_enka_data
from .get_data import get_monthinfo_data, get_player_card_data, get_chara_detail_data, get_chara_skill_data
from ..utils.alias_handler import get_match_alias
from ..utils.auth_util import check_cookie
from ..utils.config import config
from ..utils.db_util import get_auto_sign, delete_auto_sign, get_last_query, get_private_stoken, update_private_stoken, \
    get_coin_auto_sign
from ..utils.db_util import insert_public_cookie, update_private_cookie, delete_cookie_cache, delete_private_cookie, \
    update_last_query, reset_public_cookie
from ..utils.db_util import update_note_remind2, update_note_remind, get_note_remind, delete_note_remind, \
    update_day_remind_count, get_private_cookie, add_auto_sign, get_all_query, add_coin_auto_sign, delete_coin_auto_sign
from ..utils.decorator import exception_handler
from ..utils.enka_util import PlayerInfo
from ..utils.message_util import MessageBuild as MsgBd
from ..utils.message_util import get_uid_in_msg, uid_userId_to_dict, replace_all, transform_uid, get_message_id

require('nonebot_plugin_apscheduler')
from nonebot_plugin_apscheduler import scheduler
from .get_coin import MihoyoBBSCoin

__plugin_meta__ = PluginMetadata(
    name="Paimon_Info",
    description="小派蒙的原神信息查询模块",
    usage=(
        "[ys (uid)]查看原神个人卡片(包含宝箱、探索度等数据)\n"
        "[ysa (uid)]查看所有公开的8角色的简略信息\n"
        "[ysc (uid) 角色名]查看公开的8角色的详细信息\n"
        "*绑定私人cookie之后就可以查看所有角色啦\n"
        "--------\n"
        "[ssbq/实时便签 (uid)]查询当前树脂、洞天宝钱、派遣状况等\n"
        "[ssbq (uid) 开启提醒(树脂数)/关闭提醒]开启/关闭树脂提醒，达到树脂数时会在群里艾特你\n"
        "*绑定私人cookie之后才能使用\n"
        "--------\n"
        "[sy/深渊查询/深境螺旋查询 (uid) (层数)]查询深渊战绩信息\n"
        "*绑定私人cookie之后才能查看层数具体阵容哦\n"
        "--------\n"
        "[mys签到]手动进行一次米游社原神签到\n"
        "[mys自动签到开启uid/关闭]开启米游社原神自动签到\n"
        "--------\n"
        "[myzj/每月札记/zj (uid) (月份)]查看该月份获得的原石、摩拉数\n"
        "*绑定私人cookie之后才能使用,只能查看最近3个月的记录,默认为本月\n"
        "--------\n"
        "[ysb cookie]绑定你的私人cookie以开启高级功能\n"
        "[删除ck]删除你的私人cookie\n"
        "[添加公共ck cookie]添加公共cookie以供大众查询*仅管理员\n"
        "--------\n"
        "[更新角色信息 uid]更新游戏内展柜8个角色的面板信息\n"
        "[ysd 角色名 uid]查看指定角色的详细面板信息\n"
        "--------\n"
        "[myb获取]手动进行一次米游币获取\n"
        "[myb自动获取开启uid/关闭]开启米游币原神自动获取\n"
        "--------\n"
        "[添加stoken+stoken]添加stoken"
    ),
    extra={
        'type':    '原神信息查询',
        'range':   ['private', 'group', 'guild'],
        "author":  "惜月 <277073121@qq.com>",
        "version": "0.1.3",
    },
)

sy = on_command('sy', aliases={'深渊信息', '深境螺旋信息'}, priority=7, block=True)
sy.__paimon_help__ = {
    "usage":     "sy[层数](uid)",
    "introduce": "查看深渊战绩信息",
    "priority":  6
}
ssbq = on_command('ssbq', aliases={'实时便笺', '实时便签', '当前树脂'}, priority=7, block=True)
ssbq.__paimon_help__ = {
    "usage":     "ssbq(uid)",
    "introduce": "*查看当前树脂、洞天宝钱、派遣状况等",
    "priority":  7
}
myzj = on_command('myzj', aliases={'札记信息', '每月札记'}, priority=7, block=True)
myzj.__paimon_help__ = {
    "usage":     "myzj[月份](uid)",
    "introduce": "*查看该月份获得的原石、摩拉数",
    "priority":  8
}
ys = on_command('ys', aliases={'原神卡片', '个人卡片'}, priority=7, block=True)
ys.__paimon_help__ = {
    "usage":     "ys(uid)",
    "introduce": "查看原神个人卡片(宝箱、探索度等)",
    "priority":  1
}
ysa = on_command('ysa', aliases={'角色背包'}, priority=7, block=True)
ysa.__paimon_help__ = {
    "usage":     "ysa(uid)",
    "introduce": "查看原神公开角色的简略信息",
    "priority":  2
}
ysc = on_command('ysc', aliases={'角色卡片'}, priority=7, block=True)
ysc.__paimon_help__ = {
    "usage":     "ysc<角色名>(uid)",
    "introduce": "查看原神指定角色的简略信息",
    "priority":  3
}
ysb = on_command('ysb', aliases={'原神绑定', '绑定cookie'}, priority=7, block=True)
ysb.__paimon_help__ = {
    "usage":     "ysb<cookie>",
    "introduce": "绑定私人cookie以开启更多功能",
    "priority":  99
}
mys_sign = on_command('mys_sign', aliases={'mys签到', '米游社签到'}, priority=7, block=True)
mys_sign_auto = on_command('mys_sign_auto', aliases={'mys自动签到', '米游社自动签到'}, priority=7, block=True)
mys_sign_auto.__paimon_help__ = {
    "usage":     "mys自动签到<on|off uid>",
    "introduce": "*米游社原神区自动签到奖励获取",
    "priority":  9
}
mys_sign_all = on_command('mys_sign_all', aliases={'全部重签'}, priority=1, permission=SUPERUSER, rule=to_me(), block=True)
update_all = on_command('update_all', aliases={'更新全部玩家'}, priority=1, permission=SUPERUSER, rule=to_me(), block=True)
add_public_ck = on_command('add_ck', aliases={'添加公共cookie', '添加公共ck'}, permission=SUPERUSER, priority=7, block=True)
delete_ck = on_command('delete_ck', aliases={'删除ck', '删除cookie'}, priority=7, block=True)
update_info = on_command('udi', aliases={'更新角色信息', '更新角色面板', '更新玩家信息'}, priority=6, block=True)
update_info.__paimon_help__ = {
    "usage":     "更新角色信息(uid)",
    "introduce": "更新游戏内展柜8个角色的面板信息，以便使用ysd指令",
    "priority":  5
}
role_info = on_command('角色面板', aliases={'角色详情', '角色信息', 'ysd'}, block=True, priority=7)
role_info.__paimon_help__ = {
    "usage":     "ysd<角色名>(uid)",
    "introduce": "查看指定角色的详细面板信息",
    "priority":  4
}
get_mys_coin = on_command('myb获取', aliases={'米游币获取', '获取米游币'}, priority=7, block=True)
get_mys_coin_auto = on_command('myb自动获取', aliases={'米游币自动获取', '自动获取米游币'}, priority=7, block=True)
get_mys_coin_auto.__paimon_help__ = {
    "usage":     "myb自动获取<on|off uid>",
    "introduce": "自动获取米游币",
    "priority":  10
}
add_stoken = on_command('添加stoken', priority=7, block=True)
add_stoken.__paimon_help__ = {
    "usage":     "添加stoken[stoken]",
    "introduce": "添加stoken以获取米游币",
    "priority":  99
}


@sy.handle()
@ys.handle()
@ysa.handle()
@ysc.handle()
async def _(event: MessageEvent, state: T_State, msg: Message = CommandArg()):
    state['use_cache'] = False if '-r' in msg.extract_plain_text() else True
    msg_text = msg.extract_plain_text().replace('-r', '').strip()
    if not msg:
        uid = await get_last_query(str(event.user_id))
        if uid:
            state['uid'] = uid
        state['user_id'] = str(event.user_id)
    else:
        uid_list = []
        check_uid = msg_text.split(' ')
        for check in check_uid:
            uid = re.search(r'(?P<uid>(1|2|5)\d{8})', check)
            if uid:
                uid_list.append(uid.group('uid'))
                msg_text = msg_text.replace(uid.group('uid'), '')
                state['user_id'] = str(event.user_id)
        if not uid_list:
            user_list = []
            for msg_seg in msg:
                if msg_seg.type == "at":
                    user_list.append(msg_seg.data['qq'])
            if len(user_list) == 1:
                uid = await get_last_query(user_list[0])
                if uid:
                    state['uid'] = uid
                    state['user_id'] = user_list[0]
            elif user_list:
                state['uid'] = [await get_last_query(user) for user in user_list]
                state['user_id'] = user_list
            else:
                state['user_id'] = str(event.user_id)
                uid = await get_last_query(str(event.user_id))
                if uid:
                    state['uid'] = uid
        else:
            state['uid'] = uid_list[0] if len(uid_list) == 1 else uid_list if uid_list else None
    if msg_text:
        state['msg'] = msg_text.strip()


@sy.got('uid', prompt='请把要查询的uid给派蒙哦~')
@exception_handler()
async def _(event: MessageEvent, state: T_State):
    uid = transform_uid(state['uid'])
    if uid:
        state['uid'] = uid
    else:
        await ysa.finish('这个uid不正确哦~，请检查一下', at_sender=True)
    if 'msg' not in state:
        floor = []
    else:
        floor = state['msg'].split(' ')
    true_floor = [int(f) for f in floor if f.isdigit() and (9 <= int(f) <= 12)]
    true_floor.sort()
    query_dict, total_result = uid_userId_to_dict(state['uid'], state['user_id'])
    for uid, user_id in query_dict.items():
        await update_last_query(user_id, uid)
        data = await get_abyss_data(user_id, uid, use_cache=state['use_cache'])
        if isinstance(data, str):
            total_result += MessageSegment.text(data + '\n')
        else:
            abyss_card = await draw_abyss_card(data, uid, true_floor)
            total_result += abyss_card
    await sy.finish(total_result)


@ssbq.handle()
@exception_handler()
async def ssbq_handler(event: MessageEvent, msg: Message = CommandArg()):
    if event.message_type == 'group':
        gid = str(event.group_id)
    else:
        gid = str(event.user_id)
    uid, msg, user_id, use_cache = await get_uid_in_msg(event, msg)
    if not uid:
        await ssbq.finish('请把要查的uid给派蒙哦!', at_sender=True)
    find_remind_enable = re.search(r'(?P<action>开启提醒|关闭提醒|删除提醒)\s*((?P<num>\d{1,3})|(?:.*))', msg)
    if find_remind_enable:
        if event.message_type == 'guild':
            await ssbq.finish('实时便签提醒功能暂时无法在频道内使用哦')
        if find_remind_enable.group('action') == '开启提醒':
            if find_remind_enable.group('num'):
                await update_note_remind2(user_id, uid, gid, 1, find_remind_enable.group('num'))
                await ssbq.finish(f'开启提醒成功,派蒙会在你的树脂达到{find_remind_enable.group("num")}时提醒你的', at_sender=True)
            else:
                await update_note_remind2(user_id, uid, gid, 1)
                await ssbq.finish('开启提醒成功', at_sender=True)
        elif find_remind_enable.group('action') == '关闭提醒':
            await update_note_remind2(user_id, uid, gid, 0)
            await ssbq.finish('关闭提醒成功', at_sender=True)
        elif find_remind_enable.group('action') == '删除提醒':
            await delete_note_remind(user_id, uid)
            await ssbq.finish('删除提醒成功', at_sender=True)
    else:
        data = await get_daily_note_data(uid)
        if isinstance(data, str):
            await ssbq.finish(data, at_sender=True)
        else:
            daily_note_card = await draw_daily_note_card(data, uid)
            await ssbq.finish(daily_note_card)


@myzj.handle()
@exception_handler()
async def myzj_handler(event: MessageEvent, msg: Message = CommandArg()):
    uid, msg, user_id, use_cache = await get_uid_in_msg(event, msg)
    # 札记只能查看最近3个月的，构造正则来获取月份
    month_now = datetime.datetime.now().month
    if month_now == 1:
        month_list = ['11', '12', '1']
    elif month_now == 2:
        month_list = ['12', '1', '2']
    else:
        month_list = [str(month_now - 2), str(month_now - 1), str(month_now)]
    find_month = '(?P<month>' + '|'.join(month_list) + ')'
    match = re.search(find_month, msg)
    month = match.group('month') if match else month_now
    data = await get_monthinfo_data(uid, month, use_cache=use_cache)
    if isinstance(data, str):
        await myzj.finish(data, at_sender=True)
    else:
        monthinfo_card = await draw_monthinfo_card(data)
        await myzj.finish(monthinfo_card)


@ys.got('uid', prompt='请把要查询的uid给派蒙哦~')
@exception_handler()
async def ys_handler(bot: Bot, event: MessageEvent, state: T_State):
    uid = transform_uid(state['uid'])
    if uid:
        state['uid'] = uid
    else:
        await ysa.finish('这个uid不正确哦~，请检查一下', at_sender=True)
    query_dict, total_result = uid_userId_to_dict(state['uid'], state['user_id'])
    for uid, user_id in query_dict.items():
        await update_last_query(user_id, uid)
        data = await get_player_card_data(user_id, uid, use_cache=state['use_cache'])
        if isinstance(data, str):
            total_result += MessageSegment.text(data + '\n')
        else:
            if event.message_type == 'group':
                user_info = await bot.get_group_member_info(group_id=event.group_id, user_id=int(user_id))
                nickname = user_info['card'] or user_info['nickname']
            else:
                nickname = event.sender.nickname
            chara_data = await get_chara_detail_data(user_id, uid, use_cache=state['use_cache'])
            chara_data = None if isinstance(chara_data, str) else chara_data
            player_card = await draw_player_card(data, chara_data, uid, nickname)
            total_result += player_card
    await ys.finish(total_result)


@ysa.got('uid', prompt='请把要查询的uid给派蒙哦~')
@exception_handler()
async def ysa_handler(event: MessageEvent, state: T_State):
    uid = transform_uid(state['uid'])
    if uid:
        state['uid'] = uid
    else:
        await ysa.finish('这个uid不正确哦~，请检查一下', at_sender=True)
    query_dict, total_result = uid_userId_to_dict(state['uid'], state['user_id'])
    for uid, user_id in query_dict.items():
        await update_last_query(user_id, uid)
        chara_data = await get_chara_detail_data(user_id, uid, use_cache=state['use_cache'])
        if isinstance(chara_data, str):
            total_result += MessageSegment.text(chara_data + '\n')
        else:
            player_card = await draw_all_chara_card(chara_data, uid)
            total_result += player_card
    await ysa.finish(total_result)


@ysc.got('uid', prompt='请把要查询的uid给派蒙哦~')
async def _(event: MessageEvent, state: T_State):
    uid = transform_uid(state['uid'])
    if uid:
        state['uid'] = uid
    else:
        await ysa.finish('这个uid不正确哦~，请检查一下', at_sender=True)


@ysc.got('msg', prompt='请把要查询的角色名给派蒙哦~')
async def _(event: MessageEvent, state: T_State):
    name = state['msg']
    if isinstance(name, Message):
        name = replace_all(name.extract_plain_text().strip().replace('ysc', ''), state['uid'])
        if name == 'q':
            await ysc.finish()
    match_alias = get_match_alias(name, 'roles', True)
    if len(match_alias) == 1:
        state['choice'] = match_alias
    else:
        if not match_alias:
            await ysc.finish(MsgBd.Text(f'没有找到叫{name}的角色哦~'), at_sender=True)
        elif 'choice' not in state:
            msg = f'你要找的角色是哪个呀：\n'
            # 将字典中每个键拼接到msg中，并添加序号
            msg += '\n'.join([f'{int(i) + 1}. {name}' for i, name in enumerate(match_alias)])
            await ysc.send(msg + '\n回答\"q\"可以取消查询', at_sender=True)
        state['match_alias'] = match_alias


@ysc.got('choice')
async def _(event: MessageEvent, state: T_State):
    choice = state['choice']
    if isinstance(choice, dict):
        # 获取字典的键
        role = list(choice.items())[0]
    else:
        match_alias = state['match_alias']
        choice = replace_all(choice.extract_plain_text().strip().replace('ysc', ''), state['uid'])

        if choice == 'q':
            await ysc.finish()
        if choice.isdigit() and (1 <= int(choice) <= len(match_alias)):
            role = list(match_alias.items())[int(choice) - 1]
        elif choice not in match_alias.keys():
            state['times'] = state['times'] + 1 if 'times' in state else 1
            if state['times'] == 1:
                await ysc.reject(f'请旅行者从上面的角色中选一个问派蒙\n回答\"q\"可以取消查询', at_sender=True)
            elif state['times'] == 2:
                await ysc.reject(f'别调戏派蒙啦，快选一个吧，不想问了请回答\"q\"！', at_sender=True)
            elif state['times'] >= 3:
                await ysc.finish(f'看来旅行者您有点神志不清哦(，下次再问派蒙吧' + MessageSegment.face(146), at_sender=True)
        else:
            role = [m for m in list(match_alias.items()) if m[0] == choice][0]
    query_dict, total_result = uid_userId_to_dict(state['uid'], state['user_id'])
    for uid, user_id in query_dict.items():
        await update_last_query(user_id, uid)
        chara_data = await get_chara_detail_data(user_id, uid, use_cache=state['use_cache'])
        if isinstance(chara_data, str):
            total_result += MessageSegment.text(chara_data + '\n')
        else:
            skill_data = await get_chara_skill_data(uid, role[1], use_cache=state['use_cache'])
            chara_card = await draw_chara_card(chara_data, skill_data, role, uid)
            total_result += chara_card
    await ysc.finish(total_result)


cookie_error_msg = '这个cookie无效哦，请旅行者确认是否正确\n1.ck要登录mys帐号后获取,且不能退出登录\n\n获取cookie的教程：\ndocs.qq.com/doc/DQ3JLWk1vQVllZ2Z1\n'


@ysb.handle()
@exception_handler()
async def ysb_handler(event: MessageEvent, msg: Message = CommandArg()):
    cookie = msg.extract_plain_text().strip()
    if cookie == '':
        res = '获取cookie的教程：\ndocs.qq.com/doc/DQ3JLWk1vQVllZ2Z1\n获取到后，添加派蒙好友私聊发送ysb接复制到的cookie就行啦~'
        await ysb.finish(res, at_sender=True)
    else:
        cookie_info, mys_id = await get_bind_game(cookie)
        if not cookie_info or cookie_info['retcode'] != 0:
            msg = cookie_error_msg
            if event.message_type != 'private':
                msg += '\n当前是在群聊里绑定，建议旅行者添加派蒙好友私聊绑定!'
            await ysb.finish(msg, at_sender=True)
        else:
            uid = nickname = None
            for data in cookie_info['data']['list']:
                if data['game_id'] == 2:
                    uid = data['game_role_id']
                    nickname = data['nickname']
                    break
            if uid:
                await update_private_cookie(user_id=str(event.user_id), uid=uid, mys_id=mys_id, cookie=cookie)
                await update_last_query(str(event.user_id), uid, 'uid')
                await delete_cookie_cache(uid, key='uid', all=False)
                msg = f'{nickname}绑定成功啦!使用ys/ysa等指令和派蒙互动吧!'
                if event.message_type != 'private':
                    msg += '\n当前是在群聊里绑定，建议旅行者把cookie撤回哦!'
                await ysb.finish(MsgBd.Text(msg), at_sender=True)


@add_public_ck.handle()
@exception_handler()
async def add_public_ck_handler(event: MessageEvent, msg: Message = CommandArg()):
    cookie = str(msg).strip()
    if await check_cookie(cookie):
        await insert_public_cookie(cookie)
        await add_public_ck.finish('公共cookie添加成功啦,派蒙开始工作!')
    else:
        await add_public_ck.finish(cookie_error_msg)


@delete_ck.handle()
@exception_handler()
async def delete_ck_handler(event: MessageEvent):
    await delete_private_cookie(str(event.user_id))
    await delete_ck.finish('派蒙把你的私人cookie都删除啦!', at_sender=True)


@mys_sign.handle()
@exception_handler()
async def mys_sign_handler(event: MessageEvent, msg: Message = CommandArg()):
    uid, msg, user_id, use_cache = await get_uid_in_msg(event, msg)
    sign_list = await get_sign_list()
    sign_info = await get_sign_info(uid)
    if isinstance(sign_info, str):
        await mys_sign.finish(sign_info, at_sender=True)
    elif sign_info['data']['is_sign']:
        sign_day = sign_info['data']['total_sign_day'] - 1
        await mys_sign.finish(
            f'你今天已经签过到了哦,获得的奖励为:\n{sign_list["data"]["awards"][sign_day]["name"]} * {sign_list["data"]["awards"][sign_day]["cnt"]}',
            at_sender=True)
    else:
        sign_day = sign_info['data']['total_sign_day']
        sign_action = await sign(uid)
        if isinstance(sign_action, str):
            await mys_sign.finish(sign_action, at_sender=True)
        else:
            await mys_sign.finish(
                f'签到成功,获得的奖励为:\n{sign_list["data"]["awards"][sign_day]["name"]} * {sign_list["data"]["awards"][sign_day]["cnt"]}',
                at_sender=True)


@mys_sign_auto.handle()
@exception_handler()
async def mys_sign_auto_handler(event: MessageEvent, msg: Message = CommandArg()):
    if event.message_type == 'group':
        remind_id = str(event.group_id)
    elif event.message_type == 'private':
        remind_id = 'q' + str(event.user_id)
    else:
        await mys_sign_auto.finish('自动签到功能暂时不支持频道使用哦')
    msg = str(msg).strip()
    find_uid = re.search(r'(?P<uid>(1|2|5)\d{8})', msg)
    if not find_uid:
        await mys_sign_auto.finish('请把正确的需要帮忙签到的uid给派蒙哦!', at_sender=True)
    else:
        uid = find_uid.group('uid')
        find_action = re.search(r'(?P<action>开启|启用|打开|关闭|禁用|on|off)', msg)
        if find_action:
            if find_action.group('action') in ['开启', '启用', '打开', 'on']:
                cookie = await get_private_cookie(uid, key='uid')
                if not cookie:
                    await mys_sign_auto.finish('你的该uid还没绑定cookie哦，先用ysb绑定吧!', at_sender=True)
                await add_auto_sign(str(event.user_id), uid, remind_id)
                await mys_sign_auto.finish('开启米游社自动签到成功,派蒙会在每日0点帮你签到', at_sender=True)
            elif find_action.group('action') in ['关闭', '禁用', 'off']:
                await delete_auto_sign(str(event.user_id), uid)
                await mys_sign_auto.finish('关闭米游社自动签到成功', at_sender=True)
        else:
            await mys_sign_auto.finish('指令错误，在后面加 开启/关闭 来使用哦', at_sender=True)


ud_lmt = FreqLimiter(300)
ud_p_lmt = FreqLimiter(15)


@update_info.handle()
async def _(event: MessageEvent, state: T_State, msg: Message = CommandArg()):
    uid = re.search(r'(?P<uid>(1|2|5)\d{8})', msg.extract_plain_text())
    if uid:
        state['uid'] = uid.group('uid')
    else:
        user = ''
        for msg_seg in msg:
            if msg_seg.type == "at":
                user = msg_seg.data['qq']
                break
        if user:
            uid = await get_last_query(str(user))
            if uid:
                state['uid'] = uid
        else:
            uid = await get_last_query(str(event.user_id))
            if uid:
                state['uid'] = uid
    if 'uid' in state and not ud_lmt.check(state['uid']):
        await update_info.finish(f'每个uid每5分钟才能更新一次信息，请稍等一下吧~(剩余{ud_lmt.left_time(state["uid"])}秒)')
    if not ud_p_lmt.check(get_message_id(event)):
        await update_info.finish(f'每个会话每15秒才能更新一次信息，请稍等一下吧~(剩余{ud_lmt.left_time(get_message_id(event))}秒)')


@update_info.got('uid', prompt='请把要更新的uid给派蒙哦~')
@exception_handler()
async def _(event: MessageEvent, uid: Message = Arg('uid')):
    uid = transform_uid(uid)
    if not uid:
        await update_info.finish('这好像不是一个正确的uid哦~，请检查一下', at_sender=True)
    await update_last_query(str(event.user_id), uid)

    await update_info.send('派蒙开始更新信息~请稍等哦~')
    enka_data = await get_enka_data(uid)
    if not enka_data:
        if uid[0] == '5':
            await update_info.finish('暂不支持B服账号哦~请等待开发者更新吧~')
        else:
            await update_info.finish('派蒙没有获取到该uid的信息哦，可能是接口服务出现问题，稍候再试吧~')
    ud_lmt.start_cd(uid, 300)
    ud_lmt.start_cd(get_message_id(event), 15)
    player_info = PlayerInfo(uid)
    player_info.set_player(enka_data['playerInfo'])
    if 'avatarInfoList' not in enka_data:
        player_info.save()
        await update_info.finish('你未在游戏中打开角色展柜，派蒙查不到~请打开5分钟后再试~')
    else:
        for role in enka_data['avatarInfoList']:
            player_info.set_role(role)
        player_info.save()
        role_list = list(player_info.get_update_roles_list().keys())
        await update_info.finish(f'uid{uid}更新完成~本次更新的角色有：\n' + ' '.join(role_list))


@role_info.handle()
async def _(event: MessageEvent, state: T_State, msg: Message = CommandArg()):
    uid = re.search(r'(?P<uid>(1|2|5)\d{8})', msg.extract_plain_text())
    if uid:
        state['uid'] = uid.group('uid')
        await update_last_query(str(event.user_id), uid.group('uid'))
    else:
        user = ''
        for msg_seg in msg:
            if msg_seg.type == "at":
                user = msg_seg.data['qq']
        if user:
            uid = await get_last_query(str(user))
            if uid:
                state['uid'] = uid
        else:
            uid = await get_last_query(str(event.user_id))
            if uid:
                state['uid'] = uid
    msg = msg.extract_plain_text().replace(state['uid'] if 'uid' in state else 'ysd', '').strip()
    if not msg:
        await role_info.finish('请把要查询角色名给派蒙哦~')
    if msg.startswith(('a', '全部', '所有', '查看')):
        state['role'] = 'all'
    else:
        match_alias = get_match_alias(msg, 'roles', True)
        if match_alias:
            state['role'] = match_alias if isinstance(match_alias, str) else tuple(match_alias.keys())[0]
        else:
            await role_info.finish(MsgBd.Text(f'哪有名为{msg}的角色啊，别拿派蒙开玩笑!'))


@role_info.got('uid', prompt='请把要查询的uid给派蒙哦~')
@exception_handler()
async def _(event: MessageEvent, state: T_State):
    uid = transform_uid(state['uid'])
    if uid:
        state['uid'] = uid
    else:
        await role_info.finish('这个uid不正确哦~，请检查一下', at_sender=True)
    uid = state['uid']
    role = state['role']
    player_info = PlayerInfo(uid)
    roles_list = player_info.get_roles_list()
    if role == 'all':
        if not roles_list:
            await role_info.finish('你在派蒙这里没有角色面板信息哦，先用 更新角色信息 命令获取吧~', at_sender=True)
        res = '目前已获取的角色面板有：\n'
        for r in roles_list:
            res += r
            res += ' ' if (roles_list.index(r) + 1) % 4 else '\n'
        await role_info.finish(res, at_sender=True)
    if role not in roles_list:
        await role_info.finish(MsgBd.Text(f'派蒙还没有你{role}的信息哦，先用 更新角色信息 命令更新吧~'), at_sender=True)
    else:
        role_data = player_info.get_roles_info(role)
        img = await draw_role_card(uid, role_data)
        await role_info.finish(img)


@mys_sign_all.handle()
async def sign_all():
    await auto_sign()


@update_all.handle()
async def _():
    res = await all_update()
    await update_all.finish(res)


@scheduler.scheduled_job('cron', hour=config.paimon_sign_hour, minute=config.paimon_sign_minute, misfire_grace_time=10)
async def auto_sign():
    data = await get_auto_sign()
    if data:
        ann = defaultdict(lambda: defaultdict(list))
        logger.info('---派蒙开始执行米游社自动签到---')
        sign_list = await get_sign_list()
        for user_id, uid, remind_id in data:
            await sleep(random.randint(3, 8))
            sign_result = await sign(uid)
            if not isinstance(sign_result, str):
                await sleep(1)
                sign_info = await get_sign_info(uid)
                sign_day = sign_info['data']['total_sign_day'] - 1
                if remind_id.startswith('q'):
                    await get_bot().send_private_msg(user_id=remind_id[1:],
                                                     message=f'你的uid{uid}自动签到成功!签到奖励为{sign_list["data"]["awards"][sign_day]["name"]}*{sign_list["data"]["awards"][sign_day]["cnt"]}')
                else:
                    ann[remind_id]['成功'].append(
                        f'.UID{uid}-{sign_list["data"]["awards"][sign_day]["name"]}*{sign_list["data"]["awards"][sign_day]["cnt"]}')
            else:
                await delete_auto_sign(user_id, uid)
                if remind_id.startswith('q'):
                    await get_bot().send_private_msg(user_id=remind_id[1:],
                                                     message=f'你的uid{uid}签到失败，请重新绑定cookie再开启自动签到')
                else:
                    ann[remind_id]['失败'].append(f'.UID{uid}')
        for group_id, content in ann.items():
            group_str = '米游社自动签到结果：\n'
            for type, ann_list in content.items():
                if ann_list:
                    group_str += f'签到{type}：\n'
                    for u in ann_list:
                        group_str += str(ann_list.index(u) + 1) + u + '\n'
            try:
                await get_bot().send_group_msg(group_id=group_id, message=group_str)
                await sleep(random.randint(3, 8))
            except Exception as e:
                logger.error(f'米游社签到结果发送失败：{e}')


@scheduler.scheduled_job('cron', hour=config.paimon_coin_hour, minute=config.paimon_coin_minute, misfire_grace_time=10)
async def coin_auto_sign():
    data = await get_coin_auto_sign()
    ann = defaultdict(lambda: defaultdict(list))
    if data:
        logger.info('---派蒙开始执行米游币自动获取---')
        for user_id, uid, remind_id in data:
            sk = await get_private_stoken(uid, key='uid')
            stoken = sk[0][4]
            get_coin_task = MihoyoBBSCoin(stoken)
            data = await get_coin_task.task_run()
            if get_coin_task.is_Right is False:
                await delete_coin_auto_sign(user_id, uid)
                if remind_id.startswith('q'):
                    await get_bot().send_private_msg(user_id=remind_id[1:],
                                                     message=f'你的uid{uid}米游币获取失败，请重新绑定cookie再开启')
                else:
                    ann[remind_id]['失败'].append(f'.UID{uid}')
            else:
                if remind_id.startswith('q'):
                    await get_bot().send_private_msg(user_id=remind_id[1:],
                                                     message=f'你的uid{uid}米游币自动获取成功')
                else:
                    ann[remind_id]['成功'].append(f'.UID{uid}')
        for group_id, content in ann.items():
            group_str = '米游币自动获取结果：\n'
            for type, ann_list in content.items():
                if ann_list:
                    group_str += f'{type}：\n'
                    for u in ann_list:
                        group_str += str(ann_list.index(u) + 1) + u + '\n'
            try:
                await get_bot().send_group_msg(group_id=group_id, message=group_str)
                await sleep(random.randint(3, 8))
            except Exception as e:
                logger.error(f'米游币自动获取结果发送失败：{e}')


@scheduler.scheduled_job('cron', minute=f'*/{config.paimon_check_interval}', misfire_grace_time=10)
async def check_note():
    now_time = datetime.datetime.now()
    start_time = datetime.datetime(now_time.year, now_time.month, now_time.day, config.paimon_remind_start, 0, 0)
    end_time = datetime.datetime(now_time.year, now_time.month, now_time.day, config.paimon_remind_end, 0, 0)
    if start_time < now_time < end_time:
        return
    data = await get_note_remind()
    if data:
        logger.info('---派蒙开始检查实时便签树脂提醒---')
        for user_id, uid, count, remind_group, enable, last_remind_time, today_remind_count in data:
            if last_remind_time:
                last_remind_time = datetime.datetime.strptime(last_remind_time, '%Y%m%d %H:%M:%S')
                if now_time - last_remind_time > datetime.timedelta(minutes=30):
                    time_e = True
                else:
                    time_e = False
            else:
                time_e = True
            if enable and ((
                                   today_remind_count and today_remind_count < config.paimon_remind_limit) or not today_remind_count) and time_e:
                now_data = await get_daily_note_data(uid)
                if isinstance(now_data, str):
                    try:
                        await delete_note_remind(user_id, uid)
                        if user_id == remind_group:
                            await get_bot().send_private_msg(user_id=user_id,
                                                             message=f'[CQ:at,qq={user_id}]你的cookie失效了哦,派蒙没办法帮你检查树脂,'
                                                                     f'请重新添加ck后再叫派蒙开启提醒')
                        else:
                            await get_bot().send_group_msg(group_id=remind_group,
                                                           message=f'[CQ:at,qq={user_id}]你的cookie失效了哦,派蒙没办法帮你检查树脂,'
                                                                   f'请重新添加ck后再叫派蒙开启提醒')
                    except Exception as e:
                        logger.error(f'---派蒙发送树脂提醒失败:{e}---')
                else:
                    if now_data['data']['current_resin'] >= count:
                        logger.info(f'---用户{user_id}的uid{uid}的树脂已经达到阈值了,发送提醒---')
                        if today_remind_count:
                            today_remind_count += 1
                        else:
                            today_remind_count = 1
                        now_time_str = now_time.strftime('%Y%m%d %H:%M:%S')
                        try:
                            await update_note_remind(user_id, uid, count, remind_group, enable, now_time_str,
                                                     today_remind_count)
                            if user_id == remind_group:
                                await get_bot().send_private_msg(user_id=user_id,
                                                                 message=f'[CQ:at,qq={user_id}]⚠️你的树脂已经达到了{now_data["data"]["current_resin"]},记得清理哦!⚠️')
                            else:
                                await get_bot().send_group_msg(group_id=remind_group,
                                                               message=f'[CQ:at,qq={user_id}]⚠️你的树脂已经达到了{now_data["data"]["current_resin"]},记得清理哦!⚠️')
                        except Exception as e:
                            logger.error(f'---派蒙发送树脂提醒失败:{e}---')
                await sleep(3)


@scheduler.scheduled_job('cron', hour=0, misfire_grace_time=10)
async def daily_update():
    logger.info('---清空今日树脂提醒限制---')
    await update_day_remind_count()
    logger.info('---清空今日cookie缓存---')
    await delete_cookie_cache(all=True)
    logger.info('---清空今日cookie限制记录---')
    await reset_public_cookie()


# @scheduler.scheduled_job('cron', hour=3, misfire_grace_time=10)
async def all_update():
    uid_list = await get_all_query()
    logger.info('派蒙开始更新用户角色信息，共{}个用户'.format(len(uid_list)))
    failed_time = 0
    for uid in uid_list:
        try:
            data = await get_enka_data(uid)
            if data:
                player_info = PlayerInfo(uid)
                player_info.set_player(data['playerInfo'])
                if 'avatarInfoList' in data:
                    for role in data['avatarInfoList']:
                        player_info.set_role(role)
                player_info.save()
                logger.info(f'---派蒙更新{uid}的角色信息成功---')
            await sleep(random.randint(8, 15))
        except Exception:
            failed_time += 1
            if failed_time > 5:
                break
    return '玩家信息uid更新共{}个，更新完成'.format(len(uid_list))


@get_mys_coin.handle()
@exception_handler()
async def get_mys_coin_handler(event: MessageEvent, msg: Message = CommandArg()): \
        # 获取UID
    uid, msg, user_id, use_cache = await get_uid_in_msg(event, msg)
    if not uid:
        await get_mys_coin.finish('没有找到你的uid哦')
    sk = await get_private_stoken(uid, key='uid')
    if not sk:
        await get_mys_coin.finish('请旅行者先添加cookie和stoken哦')
    cookie = sk[0][1]
    if not cookie:
        await get_mys_coin.finish('你的该uid还没绑定cookie哦，先用ysb绑定吧')
    stoken = sk[0][4]
    await get_mys_coin.send('开始执行米游币获取，请稍等哦~')
    get_coin_task = MihoyoBBSCoin(stoken)
    data = await get_coin_task.task_run()
    msg = "米游币获取完成\n" + data
    await get_mys_coin.finish(msg)


@get_mys_coin_auto.handle()
@exception_handler()
async def get_mys_coin_auto_handler(event: MessageEvent, msg: Message = CommandArg()):
    if event.message_type == 'group':
        remind_id = str(event.group_id)
    elif event.message_type == 'private':
        remind_id = 'q' + str(event.user_id)
    else:
        await get_mys_coin_auto.finish('米游币自动获取功能暂时不支持频道使用哦')
    msg = msg.extract_plain_text().strip()
    find_uid = re.search(r'(?P<uid>(1|2|5)\d{8})', msg)
    if not find_uid:
        await get_mys_coin_auto.finish('请把正确的需要帮忙获取的uid给派蒙哦!', at_sender=True)
    else:
        uid = find_uid.group('uid')
        find_action = re.search(r'(?P<action>开启|启用|打开|关闭|禁用|on|off)', msg)
        if find_action:
            if find_action.group('action') in ['开启', '启用', '打开', 'on']:
                sk = await get_private_stoken(uid, key='uid')
                stoken = sk[0][4]
                if not stoken:
                    await get_mys_coin_auto.finish('你的该uid还没绑定stoken哦，先用添加stoken绑定吧!', at_sender=True)
                await add_coin_auto_sign(str(event.user_id), uid, remind_id)
                await get_mys_coin_auto.finish('开启米游币自动获取成功,派蒙会在每日0点帮你签到', at_sender=True)
            elif find_action.group('action') in ['关闭', '禁用', 'off']:
                await delete_coin_auto_sign(str(event.user_id), uid)
                await get_mys_coin_auto.finish('关闭米游币自动获取成功', at_sender=True)
        else:
            await get_mys_coin_auto.finish('指令错误，在后面加 开启/关闭 来使用哦', at_sender=True)


@add_stoken.handle()
@exception_handler()
async def add_stoken_handler(event: MessageEvent, msg: Message = CommandArg()):
    stoken = msg.extract_plain_text().strip()
    if stoken == '':
        res = '获取stoken的教程：\ndocs.qq.com/doc/DQ3JLWk1vQVllZ2Z1\n获取到后，添加派蒙好友私聊发送ysb接复制到的cookie就行啦~'
        await add_stoken.finish(res, at_sender=True)
    else:
        uid = (await get_private_cookie(event.user_id, key='user_id'))[0][2]
        stoken, mys_id, stoken_info, m = await addStoken(stoken, uid)
        if not stoken_info and not mys_id:
            await add_stoken.finish(m)
        if not stoken_info or stoken_info['retcode'] != 0:
            msg = cookie_error_msg
            if event.message_type != 'private':
                msg += '\n当前是在群聊里绑定，建议旅行者添加派蒙好友私聊绑定!'
            await add_stoken.finish(msg, at_sender=True)
        else:
            if uid:
                await update_private_stoken(user_id=str(event.user_id), uid=uid, mys_id=mys_id, cookie='',
                                            stoken=stoken)
                await update_last_query(str(event.user_id), uid, 'uid')
                msg = f'stoken绑定成功啦!'
                if event.message_type != 'private':
                    msg += '\n当前是在群聊里绑定，建议旅行者把stoken撤回哦!'
                await add_stoken.finish(MsgBd.Text(msg), at_sender=True)