LittlePaimon/Paimon_CloudGenshin/__init__.py

import json
import re
import uuid
from typing import Union
from pathlib import Path

from littlepaimon_utils.files import load_json, save_json
from nonebot import on_command, require, get_bot, logger
from nonebot.adapters.onebot.v11 import MessageEvent, Message, GroupMessageEvent
from nonebot.internal.matcher import Matcher
from nonebot.internal.params import ArgPlainText
from nonebot.params import CommandArg
from nonebot.plugin import PluginMetadata

from .data_source import get_Info, get_Notification, check_token
from ..utils.decorator import exception_handler

__plugin_meta__ = PluginMetadata(
    name="云原神",
    description="云原神相关功能模块",
    usage=(
        "云原神 绑定/bind : 绑定云原神的token\n"
        "云原神 信息/info: 查询云原神账户信息\n"
    ),
    extra={
        'type':    '工具',
        'range':   ['private', 'group'],
        "author":  "nicklly <1134741727@qq.com>",
        "version": "1.0.0",
    },
)

cloud_ys = on_command('云原神', priority=16, block=True)
rm_cloud_ys = on_command('云原神解绑', priority=16, block=True)
cloud_ys.__paimon_help__ = {
    "usage": "云原神",
    "introduce": "查询云原神账户信息, 绑定token进行签到",
    "priority": 95
}
rm_cloud_ys.__paimon_help__ = {
    "usage": "云原神解绑",
    "introduce": "解绑cookie并取消自动签到",
    "priority": 96
}
scheduler = require('nonebot_plugin_apscheduler').scheduler
uuid = str(uuid.uuid4())


@rm_cloud_ys.handle()
async def _handle(match: Matcher, args: Message = CommandArg()):
    if plan_text := args.extract_plain_text():
        match.set_arg('choice', plan_text)


@rm_cloud_ys.got('choice', prompt='是否要解绑token并取消自动签到？\n请输入 是/否 来进行操作')
async def _(event: Union[GroupMessageEvent, MessageEvent], choice: str = ArgPlainText('choice')):
    if choice == '是':
        user_id = str(event.user_id)
        data = load_json(Path() / 'data' / 'LittlePaimon' / 'CloudGenshin.json')
        try:
            del data[user_id]
            if scheduler.get_job(f'cloud_genshin_{user_id}'):
                scheduler.remove_job(f"cloud_genshin_{user_id}")
            save_json(data, Path() / 'data' / 'LittlePaimon' / 'CloudGenshin.json')
        except:
            await rm_cloud_ys.finish("你尚未绑定cookie！ 解绑失败", at_sender=True)

        await rm_cloud_ys.finish('token已解绑并取消自动签到~', at_sender=True)
    else:
        await rm_cloud_ys.finish()


async def auto_sign_cgn(user_id, data):
    token = data['token']
    uuid = data['uuid']

    if await check_token(uuid, token):
        """ 获取免费时间 """
        d2 = await get_Info(uuid, token)
        """ 解析签到返回信息 """
        if d2['data']['free_time']['free_time'] == data['limit']:
            await get_bot().send_private_msg(user_id=user_id, message='免费签到时长已达上限,无法继续签到')
        else:
            """ 取云原神签到信息 """
            d1 = await get_Notification(uuid, token)
            if not list(d1['data']['list']):
                logger.info(f'UID{data["uid"]} 已经签到云原神')
            else:
                signInfo = json.loads(d1['data']['list'][0]['msg'])
                if signInfo:
                    await get_bot().send_private_msg(
                        user_id=user_id,
                        message=f'签到成功~ {signInfo["msg"]}: {signInfo["num"]}分钟'
                    )
                else:
                    return
    else:
        await get_bot().send_private_msg(user_id=user_id, message='token已过期,请重新自行抓包并重新绑定')


@cloud_ys.handle()
@exception_handler()
async def _(event: Union[GroupMessageEvent, MessageEvent], msg: Message = CommandArg(), uuid=uuid):
    param = msg.extract_plain_text().strip()
    user_id = str(event.user_id)
    data = load_json(Path() / 'data' / 'LittlePaimon' / 'CloudGenshin.json')

    action = re.match(r'(?P<action>(信息|info)|(绑定|bind)|(签到|sign))', param)

    if event.message_type == 'guild':
        await cloud_ys.finish('派蒙提醒您: 该功能不支持群组或频道推送哦~', at_sender=True)

    if not param:
        message = f'亲爱的旅行者: {user_id}！ 欢迎使用云原神自动签到功能\n\n' \
                  '云原神指令:\n' \
                  '[绑定|bind] <cookie> - 绑定cookie到派蒙（cookie不能为空）\n' \
                  '[信息|info] - 查询当前绑定uid的剩余游戏时间\n' \
                  '[签到|sign] - 手动签到所绑定的uid（一般绑定之后默认开启自动签到）\n\n' \
                  '云原神解绑 - 解绑当前cookie并自动取消签到（此为单独指令）\n\n' \
                  '有关如何抓取token的方法:\n' \
                  '请前往 https://blog.ethreal.cn/archives/yysgettoken 查阅'
        await cloud_ys.finish(message, at_sender=True)
    else:
        if action is None:
            await cloud_ys.finish('派蒙提醒您: 指令参数错误辣!', at_sender=True)
        else:
            if action.group('action') in ['绑定', 'bind']:
                if len(param.split(" ")) == 1:
                    await cloud_ys.finish('派蒙提醒您: cookie捏？没有cookie的话.. 绑空气哦', at_sender=True)

                match = re.search(r'oi=\d+', param.split(" ")[1])
                if match:
                    uid = str(match.group()).split('=')[1]
                else:
                    await cloud_ys.finish('派蒙提醒您: cookie格式错误哦~ 请按照https://blog.ethreal.cn/archives/yysgettoken所写的方法获取cookie~')
                data[user_id] = {
                    'uid': uid,
                    'uuid': uuid,
                    'limit': 600,
                    'isFullTime': False,
                    'token': param.split(" ")[1]
                }

                if scheduler.get_job('cloud_genshin_' + user_id):
                    scheduler.remove_job("cloud_genshin_" + user_id)

                """ 保存数据 """
                save_json(data, Path() / 'data' / 'LittlePaimon' / 'CloudGenshin.json')
                """ 添加推送任务 """
                scheduler.add_job(
                    func=auto_sign_cgn,
                    trigger='cron',
                    hour=6,
                    id="cloud_genshin_" + user_id,
                    args=(user_id, data[user_id]),
                    misfire_grace_time=10
                )
                await cloud_ys.finish(f'派蒙提醒您: [UID:{uid}]已绑定成功辣~, 我将在每天6点为你自动签到哦!', at_sender=True)

            elif action.group('action') in ['信息', 'info']:

                token = data[user_id]['token']
                uid = data[user_id]['uid']
                uuid = data[user_id]['uuid']

                result = await get_Info(uuid, token)

                """ 米云币 """
                coins = result['data']['coin']['coin_num']
                """ 免费时间 """
                free_time = result['data']['free_time']['free_time']
                """ 畅玩卡 """
                card = result['data']['play_card']['short_msg']

                message = '======== UID: {0} ========\n' \
                          '剩余米云币: {1}\n' \
                          '剩余免费时间: {2}分钟\n' \
                          '畅玩卡状态: {3}'.format(uid, coins, free_time, card)
                await cloud_ys.finish(message)

            elif action.group('action') in ['签到', 'sign']:

                token = data[user_id]['token']
                uuid = data[user_id]['uuid']

                if await check_token(uuid, token):
                    d1 = await get_Notification(uuid, token)
                if not list(d1['data']['list']):
                    await cloud_ys.finish('派蒙提醒您: 你今天已签到了哦~', at_sender=True)
                else:
                    signInfo = json.loads(d1['data']['list'][0]['msg'])
                    if signInfo:
                        await cloud_ys.finish(f'派蒙提醒您: 旅行者!你已签到成功辣~ {signInfo["msg"]}: {signInfo["num"]}分钟', at_sender=True)


for user_id, data in load_json(Path() / 'data' / 'LittlePaimon' / 'CloudGenshin.json').items():
    scheduler.add_job(
        func=auto_sign_cgn,
        trigger='cron',
        hour=6,
        id="cloud_genshin_" + user_id,
        args=(user_id, data),
        misfire_grace_time=10
    )