mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-16 12:02:16 +00:00
97 lines
4.1 KiB
Python
97 lines
4.1 KiB
Python
from typing import Tuple, Optional
|
||
|
||
from genshin import Region, GenshinException
|
||
from telegram import Update, WebAppInfo, KeyboardButton, ReplyKeyboardMarkup
|
||
from telegram.ext import CallbackContext, filters
|
||
|
||
from core.base.redisdb import RedisDB
|
||
from core.baseplugin import BasePlugin
|
||
from core.config import config
|
||
from core.cookies import CookiesService
|
||
from core.cookies.error import CookiesNotFoundError
|
||
from core.plugin import Plugin, handler
|
||
from core.user import UserService
|
||
from core.user.error import UserNotFoundError
|
||
from modules.apihelper.client.components.verify import Verify
|
||
from modules.apihelper.error import ResponseException
|
||
from utils.decorators.error import error_callable
|
||
from utils.decorators.restricts import restricts
|
||
from utils.helpers import get_genshin_client
|
||
from utils.log import logger
|
||
|
||
|
||
class VerificationSystem:
|
||
def __init__(self, redis: RedisDB = None):
|
||
self.cache = redis.client
|
||
self.qname = "plugin:verification:"
|
||
|
||
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)
|
||
|
||
|
||
class VerificationPlugins(Plugin, BasePlugin):
|
||
def __init__(self, user_service: UserService = None, cookies_service: CookiesService = None, redis: RedisDB = None):
|
||
self.cookies_service = cookies_service
|
||
self.user_service = user_service
|
||
self.system = VerificationSystem(redis)
|
||
|
||
@handler.command("verify", filters=filters.ChatType.PRIVATE, block=False)
|
||
@restricts(restricts_time=60)
|
||
@error_callable
|
||
async def verify(self, update: Update, context: CallbackContext) -> None:
|
||
user = update.effective_user
|
||
message = update.effective_message
|
||
logger.info(f"用户 %s[%s] 发出verify命令", user.full_name, user.id)
|
||
try:
|
||
client = await get_genshin_client(user.id)
|
||
if client.region != Region.CHINESE:
|
||
await message.reply_text("非法用户")
|
||
return
|
||
except UserNotFoundError:
|
||
await message.reply_text("用户未找到")
|
||
return
|
||
except CookiesNotFoundError:
|
||
await message.reply_text("检测到用户为UID绑定，无需认证")
|
||
return
|
||
is_high: bool = False
|
||
verification = Verify(cookies=client.cookie_manager.cookies)
|
||
if not context.args:
|
||
try:
|
||
await client.get_genshin_notes()
|
||
except GenshinException as exc:
|
||
if exc.retcode == 1034:
|
||
is_high = True
|
||
else:
|
||
raise exc
|
||
else:
|
||
await message.reply_text("账户正常，无需认证")
|
||
return
|
||
try:
|
||
data = await verification.create(is_high=is_high)
|
||
challenge = data["challenge"]
|
||
gt = data["gt"]
|
||
logger.success("用户 %s[%s] 创建验证成功\ngt[%s]\nchallenge[%s]", user.full_name, user.id, gt, challenge)
|
||
except ResponseException as exc:
|
||
logger.warning("用户 %s[%s] 创建验证失效 API返回 [%s]%s", user.full_name, user.id, exc.code, exc.message)
|
||
await message.reply_text(f"创建验证失败 错误信息为 [{exc.code}]{exc.message} 请稍后重试")
|
||
return
|
||
await self.system.set_challenge(client.uid, gt, challenge)
|
||
url = f"{config.pass_challenge_user_web}/webapp?username={context.bot.username}&command=verify>={gt}&challenge={challenge}&uid={client.uid}"
|
||
await message.reply_text(
|
||
"请尽快在10秒内完成手动验证\n或发送 /web_cancel 取消操作",
|
||
reply_markup=ReplyKeyboardMarkup.from_button(
|
||
KeyboardButton(
|
||
text="点我手动验证",
|
||
web_app=WebAppInfo(url=url),
|
||
)
|
||
),
|
||
)
|