mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-23 00:01:45 +00:00
371 lines
18 KiB
Python
371 lines
18 KiB
Python
|
import asyncio
|
|||
|
import datetime
|
|||
|
import random
|
|||
|
import time
|
|||
|
from enum import Enum
|
|||
|
from json import JSONDecodeError
|
|||
|
from typing import Optional, Tuple, List
|
|||
|
|
|||
|
from aiohttp import ClientConnectorError
|
|||
|
from genshin import Game, GenshinException, AlreadyClaimed, Client, InvalidCookies
|
|||
|
from genshin.utility import recognize_genshin_server
|
|||
|
from httpx import AsyncClient, TimeoutException
|
|||
|
from telegram import InlineKeyboardButton, InlineKeyboardMarkup
|
|||
|
from telegram.constants import ParseMode
|
|||
|
from telegram.error import Forbidden, BadRequest
|
|||
|
from telegram.ext import CallbackContext
|
|||
|
|
|||
|
from core.config import config
|
|||
|
from core.dependence.redisdb import RedisDB
|
|||
|
from core.plugin import Plugin
|
|||
|
from core.services.cookies import CookiesService
|
|||
|
from core.services.sign.models import SignStatusEnum
|
|||
|
from core.services.sign.services import SignServices
|
|||
|
from core.services.users.services import UserService
|
|||
|
from modules.apihelper.client.components.verify import Verify
|
|||
|
from plugins.tools.genshin import GenshinHelper
|
|||
|
from utils.log import logger
|
|||
|
|
|||
|
|
|||
|
class SignJobType(Enum):
|
|||
|
START = 1
|
|||
|
REDO = 2
|
|||
|
|
|||
|
|
|||
|
class SignSystemException(Exception):
|
|||
|
def __init__(self, message: str):
|
|||
|
self.message = message
|
|||
|
super().__init__()
|
|||
|
|
|||
|
|
|||
|
class NeedChallenge(Exception):
|
|||
|
def __init__(self, uid: int, gt: str = "", challenge: str = ""):
|
|||
|
super().__init__()
|
|||
|
self.uid = uid
|
|||
|
self.gt = gt
|
|||
|
self.challenge = challenge
|
|||
|
|
|||
|
|
|||
|
class SignSystem(Plugin):
|
|||
|
REFERER = (
|
|||
|
"https://webstatic.mihoyo.com/bbs/event/signin-ys/index.html?"
|
|||
|
"bbs_auth_required=true&act_id=e202009291139501&utm_source=bbs&utm_medium=mys&utm_campaign=icon"
|
|||
|
)
|
|||
|
|
|||
|
def __init__(
|
|||
|
self,
|
|||
|
redis: RedisDB,
|
|||
|
user_service: UserService,
|
|||
|
cookies_service: CookiesService,
|
|||
|
sign_service: SignServices,
|
|||
|
genshin_helper: GenshinHelper,
|
|||
|
):
|
|||
|
self.cookies_service = cookies_service
|
|||
|
self.user_service = user_service
|
|||
|
self.sign_service = sign_service
|
|||
|
self.genshin_helper = genshin_helper
|
|||
|
self.cache = redis.client
|
|||
|
self.qname = "plugin:sign:"
|
|||
|
self.verify = Verify()
|
|||
|
|
|||
|
async def get_challenge(self, uid: int) -> Tuple[Optional[str], Optional[str]]:
|
|||
|
data = await self.cache.get(f"{self.qname}{uid}")
|
|||
|
if not data:
|
|||
|
return None, None
|
|||
|
data = data.decode("utf-8").split("|")
|
|||
|
return data[0], data[1]
|
|||
|
|
|||
|
async def set_challenge(self, uid: int, gt: str, challenge: str):
|
|||
|
await self.cache.set(f"{self.qname}{uid}", f"{gt}|{challenge}")
|
|||
|
await self.cache.expire(f"{self.qname}{uid}", 10 * 60)
|
|||
|
|
|||
|
async def get_challenge_button(
|
|||
|
self,
|
|||
|
bot_username: str,
|
|||
|
uid: int,
|
|||
|
user_id: int,
|
|||
|
gt: Optional[str] = None,
|
|||
|
challenge: Optional[str] = None,
|
|||
|
callback: bool = True,
|
|||
|
) -> Optional[InlineKeyboardMarkup]:
|
|||
|
if not config.pass_challenge_user_web:
|
|||
|
return None
|
|||
|
if challenge and gt:
|
|||
|
await self.set_challenge(uid, gt, challenge)
|
|||
|
if not challenge or not gt:
|
|||
|
gt, challenge = await self.get_challenge(uid)
|
|||
|
if not challenge or not gt:
|
|||
|
return None
|
|||
|
if callback:
|
|||
|
data = f"sign|{user_id}|{uid}"
|
|||
|
return InlineKeyboardMarkup([[InlineKeyboardButton("请尽快点我进行手动验证", callback_data=data)]])
|
|||
|
url = (
|
|||
|
f"{config.pass_challenge_user_web}?"
|
|||
|
f"username={bot_username}&command=sign>={gt}&challenge={challenge}&uid={uid}"
|
|||
|
)
|
|||
|
return InlineKeyboardMarkup([[InlineKeyboardButton("请尽快点我进行手动验证", url=url)]])
|
|||
|
|
|||
|
async def recognize(self, gt: str, challenge: str, referer: str = None) -> Optional[str]:
|
|||
|
if not referer:
|
|||
|
referer = self.REFERER
|
|||
|
if not gt or not challenge:
|
|||
|
return None
|
|||
|
pass_challenge_params = {
|
|||
|
"gt": gt,
|
|||
|
"challenge": challenge,
|
|||
|
"referer": referer,
|
|||
|
}
|
|||
|
if config.pass_challenge_app_key:
|
|||
|
pass_challenge_params["appkey"] = config.pass_challenge_app_key
|
|||
|
headers = {
|
|||
|
"Accept": "*/*",
|
|||
|
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) "
|
|||
|
"Chrome/107.0.0.0 Safari/537.36",
|
|||
|
}
|
|||
|
try:
|
|||
|
async with AsyncClient(headers=headers) as client:
|
|||
|
resp = await client.post(
|
|||
|
config.pass_challenge_api,
|
|||
|
params=pass_challenge_params,
|
|||
|
timeout=60,
|
|||
|
)
|
|||
|
logger.debug("recognize 请求返回:%s", resp.text)
|
|||
|
data = resp.json()
|
|||
|
status = data.get("status")
|
|||
|
if status != 0:
|
|||
|
logger.error("recognize 解析错误:[%s]%s", data.get("code"), data.get("msg"))
|
|||
|
if data.get("code", 0) != 0:
|
|||
|
raise RuntimeError
|
|||
|
logger.info("recognize 解析成功")
|
|||
|
return data["data"]["validate"]
|
|||
|
except JSONDecodeError:
|
|||
|
logger.warning("recognize 请求 JSON 解析失败")
|
|||
|
except TimeoutException as exc:
|
|||
|
logger.warning("recognize 请求超时")
|
|||
|
raise exc
|
|||
|
except KeyError:
|
|||
|
logger.warning("recognize 请求数据错误")
|
|||
|
except RuntimeError:
|
|||
|
logger.warning("recognize 请求失败")
|
|||
|
return None
|
|||
|
|
|||
|
async def start_sign(
|
|||
|
self,
|
|||
|
client: Client,
|
|||
|
challenge: Optional[str] = None,
|
|||
|
validate: Optional[str] = None,
|
|||
|
is_sleep: bool = False,
|
|||
|
is_raise: bool = False,
|
|||
|
title: Optional[str] = "签到结果",
|
|||
|
) -> str:
|
|||
|
if is_sleep:
|
|||
|
if recognize_genshin_server(client.uid) in ("cn_gf01", "cn_qd01"):
|
|||
|
await asyncio.sleep(random.randint(10, 300)) # nosec
|
|||
|
else:
|
|||
|
await asyncio.sleep(random.randint(0, 3)) # nosec
|
|||
|
try:
|
|||
|
rewards = await client.get_monthly_rewards(game=Game.GENSHIN, lang="zh-cn")
|
|||
|
except GenshinException as error:
|
|||
|
logger.warning("UID[%s] 获取签到信息失败,API返回信息为 %s", client.uid, str(error))
|
|||
|
if is_raise:
|
|||
|
raise error
|
|||
|
return f"获取签到信息失败,API返回信息为 {str(error)}"
|
|||
|
try:
|
|||
|
daily_reward_info = await client.get_reward_info(game=Game.GENSHIN, lang="zh-cn") # 获取签到信息失败
|
|||
|
except GenshinException as error:
|
|||
|
logger.warning("UID[%s] 获取签到状态失败,API返回信息为 %s", client.uid, str(error))
|
|||
|
if is_raise:
|
|||
|
raise error
|
|||
|
return f"获取签到状态失败,API返回信息为 {str(error)}"
|
|||
|
if not daily_reward_info.signed_in:
|
|||
|
try:
|
|||
|
if validate:
|
|||
|
logger.info("UID[%s] 正在尝试通过验证码\nchallenge[%s]\nvalidate[%s]", client.uid, challenge, validate)
|
|||
|
request_daily_reward = await client.request_daily_reward(
|
|||
|
"sign",
|
|||
|
method="POST",
|
|||
|
game=Game.GENSHIN,
|
|||
|
lang="zh-cn",
|
|||
|
challenge=challenge,
|
|||
|
validate=validate,
|
|||
|
)
|
|||
|
logger.debug("request_daily_reward 返回 %s", request_daily_reward)
|
|||
|
if request_daily_reward and request_daily_reward.get("success", 0) == 1:
|
|||
|
# 尝试通过 ajax 请求绕过签到
|
|||
|
gt = request_daily_reward.get("gt", "")
|
|||
|
challenge = request_daily_reward.get("challenge", "")
|
|||
|
logger.warning("UID[%s] 触发验证码\ngt[%s]\nchallenge[%s]", client.uid, gt, challenge)
|
|||
|
validate = await self.verify.ajax(
|
|||
|
referer=self.REFERER,
|
|||
|
gt=gt,
|
|||
|
challenge=challenge,
|
|||
|
)
|
|||
|
if validate:
|
|||
|
logger.success("ajax 通过验证成功\nchallenge[%s]\nvalidate[%s]", challenge, validate)
|
|||
|
request_daily_reward = await client.request_daily_reward(
|
|||
|
"sign",
|
|||
|
method="POST",
|
|||
|
game=Game.GENSHIN,
|
|||
|
lang="zh-cn",
|
|||
|
challenge=challenge,
|
|||
|
validate=validate,
|
|||
|
)
|
|||
|
logger.debug("request_daily_reward 返回 %s", request_daily_reward)
|
|||
|
if request_daily_reward and request_daily_reward.get("success", 0) == 1:
|
|||
|
logger.warning("UID[%s] 触发验证码\nchallenge[%s]", client.uid, challenge)
|
|||
|
raise NeedChallenge(
|
|||
|
uid=client.uid,
|
|||
|
gt=request_daily_reward.get("gt", ""),
|
|||
|
challenge=request_daily_reward.get("challenge", ""),
|
|||
|
)
|
|||
|
elif config.pass_challenge_app_key:
|
|||
|
# 如果无法绕过 检查配置文件是否配置识别 API 尝试请求绕过
|
|||
|
# 注意 需要重新获取没有进行任何请求的 Challenge
|
|||
|
logger.info("UID[%s] 正在使用 recognize 重新请求签到", client.uid)
|
|||
|
_request_daily_reward = await client.request_daily_reward(
|
|||
|
"sign",
|
|||
|
method="POST",
|
|||
|
game=Game.GENSHIN,
|
|||
|
lang="zh-cn",
|
|||
|
)
|
|||
|
logger.debug("request_daily_reward 返回\n%s", _request_daily_reward)
|
|||
|
if _request_daily_reward and _request_daily_reward.get("success", 0) == 1:
|
|||
|
_gt = _request_daily_reward.get("gt", "")
|
|||
|
_challenge = _request_daily_reward.get("challenge", "")
|
|||
|
logger.info("UID[%s] 创建验证码\ngt[%s]\nchallenge[%s]", client.uid, _gt, _challenge)
|
|||
|
_validate = await self.recognize(_gt, _challenge)
|
|||
|
if _validate:
|
|||
|
logger.success("recognize 通过验证成功\nchallenge[%s]\nvalidate[%s]", _challenge, _validate)
|
|||
|
request_daily_reward = await client.request_daily_reward(
|
|||
|
"sign",
|
|||
|
method="POST",
|
|||
|
game=Game.GENSHIN,
|
|||
|
lang="zh-cn",
|
|||
|
challenge=_challenge,
|
|||
|
validate=_validate,
|
|||
|
)
|
|||
|
if request_daily_reward and request_daily_reward.get("success", 0) == 1:
|
|||
|
logger.warning("UID[%s] 触发验证码\nchallenge[%s]", client.uid, _challenge)
|
|||
|
gt = request_daily_reward.get("gt", "")
|
|||
|
challenge = request_daily_reward.get("challenge", "")
|
|||
|
logger.success("UID[%s] 创建验证成功\ngt[%s]\nchallenge[%s]", client.uid, gt, challenge)
|
|||
|
raise NeedChallenge(
|
|||
|
uid=client.uid,
|
|||
|
gt=gt,
|
|||
|
challenge=challenge,
|
|||
|
)
|
|||
|
logger.success("UID[%s] 通过 recognize 签到成功", client.uid)
|
|||
|
else:
|
|||
|
request_daily_reward = await client.request_daily_reward(
|
|||
|
"sign", method="POST", game=Game.GENSHIN, lang="zh-cn"
|
|||
|
)
|
|||
|
gt = request_daily_reward.get("gt", "")
|
|||
|
challenge = request_daily_reward.get("challenge", "")
|
|||
|
logger.success("UID[%s] 创建验证成功\ngt[%s]\nchallenge[%s]", client.uid, gt, challenge)
|
|||
|
raise NeedChallenge(uid=client.uid, gt=gt, challenge=challenge)
|
|||
|
else:
|
|||
|
request_daily_reward = await client.request_daily_reward(
|
|||
|
"sign", method="POST", game=Game.GENSHIN, lang="zh-cn"
|
|||
|
)
|
|||
|
gt = request_daily_reward.get("gt", "")
|
|||
|
challenge = request_daily_reward.get("challenge", "")
|
|||
|
logger.success("UID[%s] 创建验证成功\ngt[%s]\nchallenge[%s]", client.uid, gt, challenge)
|
|||
|
raise NeedChallenge(uid=client.uid, gt=gt, challenge=challenge)
|
|||
|
else:
|
|||
|
logger.success("UID[%s] 签到成功", client.uid)
|
|||
|
except TimeoutException as error:
|
|||
|
logger.warning("UID[%s] 签到请求超时", client.uid)
|
|||
|
if is_raise:
|
|||
|
raise error
|
|||
|
return "签到失败了呜呜呜 ~ 服务器连接超时 服务器熟啦 ~ "
|
|||
|
except AlreadyClaimed as error:
|
|||
|
logger.warning("UID[%s] 已经签到", client.uid)
|
|||
|
if is_raise:
|
|||
|
raise error
|
|||
|
result = "今天旅行者已经签到过了~"
|
|||
|
except GenshinException as error:
|
|||
|
logger.warning("UID %s 签到失败,API返回信息为 %s", client.uid, str(error))
|
|||
|
if is_raise:
|
|||
|
raise error
|
|||
|
return f"获取签到状态失败,API返回信息为 {str(error)}"
|
|||
|
else:
|
|||
|
result = "OK"
|
|||
|
else:
|
|||
|
logger.info("UID[%s] 已经签到", client.uid)
|
|||
|
result = "今天旅行者已经签到过了~"
|
|||
|
logger.info("UID[%s] 签到结果 %s", client.uid, result)
|
|||
|
reward = rewards[daily_reward_info.claimed_rewards - (1 if daily_reward_info.signed_in else 0)]
|
|||
|
today = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
|||
|
cn_timezone = datetime.timezone(datetime.timedelta(hours=8))
|
|||
|
now = datetime.datetime.now(cn_timezone)
|
|||
|
missed_days = now.day - daily_reward_info.claimed_rewards
|
|||
|
if not daily_reward_info.signed_in:
|
|||
|
missed_days -= 1
|
|||
|
message = (
|
|||
|
f"#### {title} ####\n"
|
|||
|
f"时间:{today} (UTC+8)\n"
|
|||
|
f"UID: {client.uid}\n"
|
|||
|
f"今日奖励: {reward.name} × {reward.amount}\n"
|
|||
|
f"本月漏签次数:{missed_days}\n"
|
|||
|
f"签到结果: {result}"
|
|||
|
)
|
|||
|
return message
|
|||
|
|
|||
|
async def do_sign_job(self, context: CallbackContext, job_type: SignJobType):
|
|||
|
include_status: List[SignStatusEnum] = [
|
|||
|
SignStatusEnum.STATUS_SUCCESS,
|
|||
|
SignStatusEnum.TIMEOUT_ERROR,
|
|||
|
SignStatusEnum.NEED_CHALLENGE,
|
|||
|
]
|
|||
|
if job_type == SignJobType.START:
|
|||
|
title = "自动签到"
|
|||
|
elif job_type == SignJobType.REDO:
|
|||
|
title = "自动重新签到"
|
|||
|
else:
|
|||
|
raise ValueError
|
|||
|
sign_list = await self.sign_service.get_all()
|
|||
|
for sign_db in sign_list:
|
|||
|
if sign_db.status not in include_status:
|
|||
|
continue
|
|||
|
user_id = sign_db.user_id
|
|||
|
try:
|
|||
|
client = await self.genshin_helper.get_genshin_client(user_id)
|
|||
|
text = await self.start_sign(client, is_sleep=True, is_raise=True, title=title)
|
|||
|
except InvalidCookies:
|
|||
|
text = "自动签到执行失败,Cookie无效"
|
|||
|
sign_db.status = SignStatusEnum.INVALID_COOKIES
|
|||
|
except AlreadyClaimed:
|
|||
|
text = "今天旅行者已经签到过了~"
|
|||
|
sign_db.status = SignStatusEnum.ALREADY_CLAIMED
|
|||
|
except GenshinException as exc:
|
|||
|
text = f"自动签到执行失败,API返回信息为 {str(exc)}"
|
|||
|
sign_db.status = SignStatusEnum.GENSHIN_EXCEPTION
|
|||
|
except ClientConnectorError:
|
|||
|
text = "签到失败了呜呜呜 ~ 服务器连接超时 服务器熟啦 ~ "
|
|||
|
sign_db.status = SignStatusEnum.TIMEOUT_ERROR
|
|||
|
except NeedChallenge:
|
|||
|
text = "签到失败,触发验证码风控,自动签到自动关闭"
|
|||
|
sign_db.status = SignStatusEnum.NEED_CHALLENGE
|
|||
|
except Exception as exc:
|
|||
|
logger.error("执行自动签到时发生错误 user_id[%s] Message[%s]", user_id, exc.message)
|
|||
|
text = "签到失败了呜呜呜 ~ 执行自动签到时发生错误"
|
|||
|
else:
|
|||
|
sign_db.status = SignStatusEnum.STATUS_SUCCESS
|
|||
|
if sign_db.chat_id < 0:
|
|||
|
text = f'<a href="tg://user?id={sign_db.user_id}">NOTICE {sign_db.user_id}</a>\n\n{text}'
|
|||
|
try:
|
|||
|
await context.bot.send_message(sign_db.chat_id, text, parse_mode=ParseMode.HTML)
|
|||
|
except BadRequest as exc:
|
|||
|
logger.error("执行自动签到时发生错误 user_id[%s] Message[%s]", user_id, exc.message)
|
|||
|
sign_db.status = SignStatusEnum.BAD_REQUEST
|
|||
|
except Forbidden as exc:
|
|||
|
logger.error("执行自动签到时发生错误 user_id[%s] message[%s]", user_id, exc.message)
|
|||
|
sign_db.status = SignStatusEnum.FORBIDDEN
|
|||
|
except Exception as exc:
|
|||
|
logger.error("执行自动签到时发生错误 user_id[%s]", user_id, exc_info=exc)
|
|||
|
continue
|
|||
|
else:
|
|||
|
sign_db.status = SignStatusEnum.STATUS_SUCCESS
|
|||
|
sign_db.time_updated = datetime.datetime.now()
|
|||
|
await self.sign_service.update(sign_db)
|