mirror of
https://github.com/xuthus83/LittlePaimon.git
synced 2024-10-21 16:27:15 +08:00
205 lines
8.4 KiB
Python
205 lines
8.4 KiB
Python
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
|
||
)
|