from datetime import datetime from typing import Dict, Optional from arkowrapper import ArkoWrapper from simnet import StarRailClient, Region from simnet.errors import DataNotPublic, InvalidCookies, BadRequest as SimnetBadRequest from simnet.models.lab.record import Account from telegram import ReplyKeyboardMarkup, ReplyKeyboardRemove, TelegramObject, Update from telegram.ext import CallbackContext, ConversationHandler, filters from telegram.helpers import escape_markdown from core.basemodel import RegionEnum from core.plugin import Plugin, conversation, handler from core.services.cookies.models import CookiesDataBase as Cookies, CookiesStatusEnum from core.services.cookies.services import CookiesService from core.services.players.models import PlayersDataBase as Player, PlayerInfoSQLModel from core.services.players.services import PlayersService, PlayerInfoService from gram_core.services.devices import DevicesService from gram_core.services.devices.models import DevicesDataBase as Devices from modules.apihelper.models.genshin.cookies import CookiesModel from utils.log import logger __all__ = ("AccountCookiesPlugin",) class AccountIdNotFound(Exception): pass class AccountCookiesPluginDeviceData(TelegramObject): device_id: str = "" device_fp: str = "" device_name: Optional[str] = None class AccountCookiesPluginData(TelegramObject): region: RegionEnum = RegionEnum.NULL cookies: dict = {} account_id: int = 0 # player_id: int = 0 starrail_account: Optional[Account] = None device: Optional[AccountCookiesPluginDeviceData] = None def reset(self): self.region = RegionEnum.NULL self.cookies = {} self.account_id = 0 self.starrail_account = None self.device = None CHECK_SERVER, INPUT_COOKIES, COMMAND_RESULT = range(10100, 10103) class AccountCookiesPlugin(Plugin.Conversation): """Cookie绑定""" def __init__( self, players_service: PlayersService = None, cookies_service: CookiesService = None, player_info_service: PlayerInfoService = None, devices_service: DevicesService = None, ): self.cookies_service = cookies_service self.players_service = players_service self.player_info_service = player_info_service self.devices_service = devices_service # noinspection SpellCheckingInspection @staticmethod def parse_cookie(cookie: Dict[str, str]) -> Dict[str, str]: cookies = {} v1_keys = [ "ltoken", "ltuid", "account_id", "cookie_token", "stoken", "stuid", "login_uid", "login_ticket", "mid", ] v2_keys = ["ltoken_v2", "ltmid_v2", "ltuid_v2", "account_mid_v2", "cookie_token_v2", "account_id_v2"] for k in v1_keys + v2_keys: cookies[k] = cookie.get(k) return {k: v for k, v in cookies.items() if v is not None} @staticmethod def parse_headers(headers: Dict[str, str]) -> AccountCookiesPluginDeviceData: data = AccountCookiesPluginDeviceData() must_keys = {"x-rpc-device_id": 36, "x-rpc-device_fp": 13} optional_keys = ["x-rpc-device_name"] for k, v in must_keys.items(): if (k not in headers) or (not headers.get(k)): raise ValueError if len(headers.get(k)) != v: raise ValueError for k in optional_keys: if k not in headers: continue if headers.get(k) and len(headers.get(k)) > 64: raise ValueError data.device_id = headers.get("x-rpc-device_id") data.device_fp = headers.get("x-rpc-device_fp") data.device_name = headers.get("x-rpc-device_name") return data async def _parse_args(self, update: Update, context: CallbackContext) -> Optional[int]: args = self.get_args(context) account_cookies_plugin_data: AccountCookiesPluginData = context.chat_data.get("account_cookies_plugin_data") if len(args) < 2: return None regions = {"米游社": RegionEnum.HYPERION, "HoYoLab": RegionEnum.HOYOLAB} if args[0] not in regions: return None cookies = " ".join(args[1:]) account_cookies_plugin_data.region = regions[args[0]] if ret := await self.parse_cookies(update, context, cookies): return ret return await self.check_cookies(update, context) @conversation.entry_point @handler.command(command="setcookie", filters=filters.ChatType.PRIVATE, block=False) @handler.command(command="setcookies", filters=filters.ChatType.PRIVATE, block=False) @handler.command(command="start", filters=filters.Regex("set_cookie$"), block=False) async def command_start(self, update: Update, context: CallbackContext) -> int: user = update.effective_user message = update.effective_message logger.info("用户 %s[%s] 绑定账号命令请求", user.full_name, user.id) account_cookies_plugin_data: AccountCookiesPluginData = context.chat_data.get("account_cookies_plugin_data") if account_cookies_plugin_data is None: account_cookies_plugin_data = AccountCookiesPluginData() context.chat_data["account_cookies_plugin_data"] = account_cookies_plugin_data else: account_cookies_plugin_data.reset() if ret := await self._parse_args(update, context): return ret text = f'你好 {user.mention_markdown_v2()} {escape_markdown("！请选择要绑定的服务器！或回复退出取消操作")}' reply_keyboard = [["米游社", "HoYoLab"], ["退出"]] await message.reply_markdown_v2(text, reply_markup=ReplyKeyboardMarkup(reply_keyboard, one_time_keyboard=True)) return CHECK_SERVER @conversation.state(state=CHECK_SERVER) @handler.message(filters=filters.TEXT & ~filters.COMMAND, block=False) async def check_server(self, update: Update, context: CallbackContext) -> int: message = update.effective_message account_cookies_plugin_data: AccountCookiesPluginData = context.chat_data.get("account_cookies_plugin_data") if message.text == "退出": await message.reply_text("退出任务", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END if message.text == "米游社": region = RegionEnum.HYPERION bbs_name = "米游社" elif message.text == "HoYoLab": region = RegionEnum.HOYOLAB bbs_name = "HoYoLab" else: await message.reply_text("选择错误，请重新选择") return CHECK_SERVER account_cookies_plugin_data.region = region await message.reply_text(f"请输入{bbs_name}的Cookies！或回复退出取消操作", reply_markup=ReplyKeyboardRemove()) await message.reply_html("关于如何获取Cookies\n\nhttps://telegra.ph/paigramteam-bot-setcookies-10-02") return INPUT_COOKIES @conversation.state(state=INPUT_COOKIES) @handler.message(filters=filters.TEXT & ~filters.COMMAND, block=False) async def input_cookies(self, update: Update, context: CallbackContext) -> int: message = update.effective_message if message.text == "退出": await message.reply_text("退出任务", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END if ret := await self.parse_cookies(update, context, message.text): return ret return await self.check_cookies(update, context) async def parse_cookies(self, update: Update, context: CallbackContext, text: str) -> Optional[int]: user = update.effective_user message = update.effective_message account_cookies_plugin_data: AccountCookiesPluginData = context.chat_data.get("account_cookies_plugin_data") try: # cookie str to dict wrapped = ( ArkoWrapper(text.split(";")) .filter(lambda x: x != "") .map(lambda x: x.strip()) .map(lambda x: ((y := x.split("=", 1))[0], y[1])) ) cookie = {x[0]: x[1] for x in wrapped} cookies = self.parse_cookie(cookie) except (AttributeError, ValueError, IndexError) as exc: logger.info("用户 %s[%s] Cookies解析出现错误\ntext:%s", user.full_name, user.id, message.text) logger.debug("解析Cookies出现错误", exc_info=exc) await message.reply_text("解析Cookies出现错误，请检查是否正确", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END if account_cookies_plugin_data.region == RegionEnum.HYPERION: try: account_cookies_plugin_data.device = self.parse_headers(cookie) except ValueError: account_cookies_plugin_data.device = None await message.reply_text("解析 Devices 出现错误，可能无法绕过风控，查询操作将需要通过验证。") if not cookies: logger.info("用户 %s[%s] Cookies格式有误", user.full_name, user.id) await message.reply_text("Cookies格式有误，请检查后重新尝试绑定", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END account_cookies_plugin_data.cookies = cookies async def check_cookies(self, update: Update, context: CallbackContext) -> int: user = update.effective_user message = update.effective_message account_cookies_plugin_data: AccountCookiesPluginData = context.chat_data.get("account_cookies_plugin_data") cookies = CookiesModel(**account_cookies_plugin_data.cookies) if account_cookies_plugin_data.region == RegionEnum.HYPERION: region = Region.CHINESE elif account_cookies_plugin_data.region == RegionEnum.HOYOLAB: region = Region.OVERSEAS else: logger.error("用户 %s[%s] region 异常", user.full_name, user.id) await message.reply_text("数据错误", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END async with StarRailClient(cookies=cookies.to_dict(), region=region, lang="zh-cn") as client: check_cookie = cookies.check() if cookies.login_ticket is not None: try: cookies.cookie_token = await client.get_cookie_token_by_login_ticket() cookies.account_id = client.account_id cookies.ltuid = client.account_id logger.success("用户 %s[%s] 绑定时获取 cookie_token 成功", user.full_name, user.id) cookies.stoken = await client.get_stoken_by_login_ticket() logger.success("用户 %s[%s] 绑定时获取 stoken 成功", user.full_name, user.id) check_cookie = True except SimnetBadRequest as exc: logger.warning( "用户 %s[%s] 获取账号信息发生错误 [%s]%s", user.full_name, user.id, exc.ret_code, exc.original ) except Exception as exc: logger.error("绑定时获取新Cookie失败 [%s]", (str(exc))) finally: cookies.login_ticket = None cookies.login_uid = None if not check_cookie: await message.reply_text("检测到Cookie不完整，可能会出现问题。", reply_markup=ReplyKeyboardRemove()) if not cookies.stoken: await message.reply_text( "检测到缺少 stoken，请尝试添加 stoken 后重新绑定。", reply_markup=ReplyKeyboardRemove() ) return ConversationHandler.END if cookies.stoken and cookies.stoken.startswith("v2") and cookies.mid is None: await message.reply_text( "检测到缺少 mid，请尝试添加 mid 后重新绑定。", reply_markup=ReplyKeyboardRemove() ) return ConversationHandler.END try: if region == Region.CHINESE: cookies.stoken, cookies.mid = await client.get_stoken_v2_and_mid_by_by_stoken( cookies.stoken, cookies.account_id ) logger.success("用户 %s[%s] 绑定时获取 stoken_v2, mid 成功", user.full_name, user.id) cookies.cookie_token = await client.get_cookie_token_by_stoken(cookies.stoken, mid=cookies.mid) logger.success("用户 %s[%s] 绑定时获取 cookie_token 成功", user.full_name, user.id) cookies.ltoken = await client.get_ltoken_by_stoken() logger.success("用户 %s[%s] 绑定时获取 ltoken 成功", user.full_name, user.id) else: cookies_model = await client.get_all_token_by_stoken(cookies.stoken, cookies.account_id) cookies.stoken = cookies_model.stoken cookies.mid = cookies_model.mid cookies.ltoken = cookies_model.ltoken cookies.cookie_token = cookies_model.cookie_token logger.success( "用户 %s[%s] 绑定时获取 stoken_v2, mid, ltoken, cookie_token 成功", user.full_name, user.id ) except SimnetBadRequest as exc: logger.warning( "用户 %s[%s] 获取账号信息发生错误 [%s]%s", user.full_name, user.id, exc.ret_code, exc.original ) await message.reply_text("Stoken 无效，请重新绑定。", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END except UnicodeEncodeError: await message.reply_text("Stoken 非法，请重新绑定。", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END except ValueError as e: if "account_id" in str(e): await message.reply_text("account_id 未找到，请检查输入是否有误。") return ConversationHandler.END raise e try: if cookies.account_id is None: logger.info("正在尝试获取用户 %s[%s] account_id", user.full_name, user.id) account_info = await client.get_user_info() account_id = account_info.accident_id account_cookies_plugin_data.account_id = account_id if cookies.is_v2: cookies.set_v2_uid(account_id) else: cookies.set_uid(account_id) logger.success("获取用户 %s[%s] account_id[%s] 成功", user.full_name, user.id, account_id) else: account_cookies_plugin_data.account_id = client.account_id starrail_accounts = await client.get_starrail_accounts() except DataNotPublic: logger.info("用户 %s[%s] 账号疑似被注销", user.full_name, user.id) await message.reply_text("账号疑似被注销，请检查账号状态", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END except InvalidCookies: logger.info("用户 %s[%s] Cookies已经过期", user.full_name, user.id) await message.reply_text( "获取账号信息失败，返回Cookies已经过期，请尝试在无痕浏览器中登录获取Cookies。", reply_markup=ReplyKeyboardRemove(), ) return ConversationHandler.END except SimnetBadRequest as exc: logger.info( "用户 %s[%s] 获取账号信息发生错误 [%s]%s", user.full_name, user.id, exc.ret_code, exc.original ) await message.reply_text( f"获取账号信息发生错误，错误信息为 {exc.original}，请检查Cookie或者账号是否正常", reply_markup=ReplyKeyboardRemove(), ) return ConversationHandler.END except AccountIdNotFound: logger.info("用户 %s[%s] 无法获取账号ID", user.full_name, user.id) await message.reply_text("无法获取账号ID，请检查Cookie是否正常", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END except (AttributeError, ValueError) as exc: logger.warning("用户 %s[%s] Cookies错误", user.full_name, user.id) logger.debug("用户 %s[%s] Cookies错误", user.full_name, user.id, exc_info=exc) await message.reply_text("Cookies错误，请检查是否正确", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END if account_cookies_plugin_data.account_id is None: await message.reply_text("无法获取账号ID，请检查Cookie是否正确或请稍后重试") return ConversationHandler.END starrail_account: Optional[Account] = None level: int = 0 # todo : 多账号绑定 for temp in starrail_accounts: if temp.level >= level: # 获取账号等级最高的 level = temp.level starrail_account = temp if starrail_account is None: await message.reply_text("未找到星穹铁道账号，请确认账号信息无误。") return ConversationHandler.END account_cookies_plugin_data.starrail_account = starrail_account player_info = await self.players_service.get( user.id, player_id=starrail_account.uid, region=account_cookies_plugin_data.region ) if player_info: cookies_database = await self.cookies_service.get( user.id, player_info.account_id, account_cookies_plugin_data.region ) if cookies_database: await message.reply_text("警告，你已经绑定Cookie，如果继续操作会覆盖当前Cookie。") reply_keyboard = [["确认", "退出"]] await message.reply_text("获取角色基础信息成功，请检查是否正确！") logger.info( "用户 %s[%s] 获取账号 %s[%s] 信息成功", user.full_name, user.id, starrail_account.nickname, starrail_account.uid, ) text = ( f"*角色信息*\n" f"角色名称：{escape_markdown(starrail_account.nickname, version=2)}\n" f"角色等级：{starrail_account.level}\n" f"UID：`{starrail_account.uid}`\n" f"服务器名称：`{starrail_account.server_name}`\n" ) await message.reply_markdown_v2(text, reply_markup=ReplyKeyboardMarkup(reply_keyboard, one_time_keyboard=True)) account_cookies_plugin_data.cookies = cookies.to_dict() return COMMAND_RESULT async def update_devices(self, account_id: int, device: AccountCookiesPluginDeviceData): if not device: return device_model = await self.devices_service.get(account_id) if device_model: device_model.device_id = device.device_id device_model.device_fp = device.device_fp device_model.device_name = device.device_name device_model.is_valid = True await self.devices_service.update(device_model) else: device_model = Devices( account_id=account_id, device_id=device.device_id, device_fp=device.device_fp, device_name=device.device_name, is_valid=True, ) await self.devices_service.add(device_model) async def update_player(self, uid: int, genshin_account: Account, region: RegionEnum, account_id: int): player = await self.players_service.get(uid, player_id=genshin_account.uid, region=region) if player: player.account_id = account_id await self.players_service.update(player) else: player_model = Player( user_id=uid, account_id=account_id, player_id=genshin_account.uid, region=region, is_chosen=True, # todo 多账号 ) await self.update_player_info(player_model, genshin_account.nickname) await self.players_service.add(player_model) async def update_player_info(self, player: Player, nickname: str): player_info = await self.player_info_service.get(player) if player_info is None: player_info = PlayerInfoSQLModel( user_id=player.user_id, player_id=player.player_id, nickname=nickname, create_time=datetime.now(), is_update=True, ) # 不添加更新时间 await self.player_info_service.add(player_info) async def update_cookies(self, uid: int, account_id: int, region: RegionEnum, cookies: Dict): cookies_data_base = await self.cookies_service.get(uid, account_id, region) if cookies_data_base: cookies_data_base.data = cookies cookies_data_base.status = CookiesStatusEnum.STATUS_SUCCESS await self.cookies_service.update(cookies_data_base) else: cookies = Cookies( user_id=uid, account_id=account_id, data=cookies, region=region, status=CookiesStatusEnum.STATUS_SUCCESS, is_share=True, # todo 用户可以自行选择是否将Cookies加入公共池 ) await self.cookies_service.add(cookies) @conversation.state(state=COMMAND_RESULT) @handler.message(filters=filters.TEXT & ~filters.COMMAND, block=False) async def command_result(self, update: Update, context: CallbackContext) -> int: user = update.effective_user message = update.effective_message account_cookies_plugin_data: AccountCookiesPluginData = context.chat_data.get("account_cookies_plugin_data") if message.text == "退出": await message.reply_text("退出任务", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END if message.text == "确认": starrail_account = account_cookies_plugin_data.starrail_account await self.update_player( user.id, starrail_account, account_cookies_plugin_data.region, account_cookies_plugin_data.account_id ) await self.update_cookies( user.id, account_cookies_plugin_data.account_id, account_cookies_plugin_data.region, account_cookies_plugin_data.cookies, ) await self.update_devices(account_cookies_plugin_data.account_id, account_cookies_plugin_data.device) logger.info("用户 %s[%s] 绑定账号成功", user.full_name, user.id) await message.reply_text("保存成功", reply_markup=ReplyKeyboardRemove()) return ConversationHandler.END await message.reply_text("回复错误，请重新输入") return COMMAND_RESULT