mirror of
https://github.com/PaiGramTeam/SIMNet.git
synced 2024-11-26 15:59:33 +00:00
87 lines
3.2 KiB
Python
87 lines
3.2 KiB
Python
import json
|
|
import re
|
|
import time
|
|
from typing import Optional
|
|
|
|
from httpx import Headers
|
|
|
|
from simnet.client.base import BaseClient
|
|
from simnet.client.routes import URL
|
|
from simnet.errors import RegionNotSupported
|
|
from simnet.utils.enum_ import Region
|
|
|
|
|
|
class VerifyClient(BaseClient):
|
|
"""VerifyClient component."""
|
|
|
|
HOST = URL("https://api-takumi-record.mihoyo.com")
|
|
VERIFICATION_HOST = URL("https://api.geetest.com")
|
|
CREATE_VERIFICATION_URL = HOST / "/game_record/app/card/wapi/createVerification"
|
|
VERIFY_VERIFICATION_URL = HOST / "/game_record/app/card/wapi/verifyVerification"
|
|
AJAX_URL = VERIFICATION_HOST / "/ajax.php"
|
|
|
|
async def create_verification(self, is_high: bool = False):
|
|
"""Create a verification challenge.
|
|
|
|
Args:
|
|
is_high (bool): Ifs the challenge for a high risk device.
|
|
|
|
Returns:
|
|
str: The challenge.
|
|
"""
|
|
if self.region != Region.CHINESE:
|
|
raise RegionNotSupported("This method is only available for the Chinese region.")
|
|
params = {"is_high": "true" if is_high else "false"}
|
|
return await self.request_lab(self.CREATE_VERIFICATION_URL, params=params)
|
|
|
|
async def verify_verification(self, challenge: str, validate: str):
|
|
"""Verify a verification challenge.
|
|
|
|
Args:
|
|
challenge (str): The challenge code.
|
|
validate (str): The validate code.
|
|
|
|
Returns:
|
|
str: The verification result.
|
|
"""
|
|
if self.region != Region.CHINESE:
|
|
raise RegionNotSupported("This method is only available for the Chinese region.")
|
|
data = {"geetest_challenge": challenge, "geetest_validate": validate, "geetest_seccode": f"{validate}|jordan"}
|
|
return await self.request_lab(self.VERIFY_VERIFICATION_URL, data=data)
|
|
|
|
async def request_verify_ajax(self, referer: str, gt: str, challenge: str) -> Optional[str]:
|
|
"""Get the ajax validate code.
|
|
|
|
Args:
|
|
referer (str): The web referer.
|
|
gt (str): The gt challenge.
|
|
challenge (str): The challenge code.
|
|
|
|
Returns:
|
|
Optional[str]: The ajax validate code.
|
|
"""
|
|
if self.region != Region.CHINESE:
|
|
raise RegionNotSupported("This method is only available for the Chinese region.")
|
|
params = {
|
|
"gt": gt,
|
|
"challenge": challenge,
|
|
"lang": "zh-cn",
|
|
"pt": 3,
|
|
"client_type": "web_mobile",
|
|
"callback": f"geetest_{int(time.time() * 1000)}",
|
|
}
|
|
headers = Headers()
|
|
headers["user-agent"] = self.user_agent
|
|
headers["x-rpc-app_version"] = self.app_version
|
|
headers["x-rpc-client_type"] = self.client_type
|
|
headers["x-rpc-device_id"] = self.get_device_id()
|
|
headers["x-rpc-device_fp"] = self.get_device_fp()
|
|
headers["referer"] = referer
|
|
response = await self.request("GET", self.AJAX_URL, params=params, headers=headers)
|
|
text = response.text
|
|
json_data = re.findall(r"^.*?\((\{.*?)\)$", text)[0]
|
|
data = json.loads(json_data)
|
|
if "success" in data["status"] and "success" in data["data"]["result"]:
|
|
return data["data"]["validate"]
|
|
return None
|