d3be20ce85
- Multi-threaded account registration - Auto email verification via freemail API - Auto NSFW/Unhinged mode activation - Temporary email cleanup after registration
165 lines
4.7 KiB
Python
165 lines
4.7 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Optional, Dict, Any
|
|
|
|
from curl_cffi import requests
|
|
|
|
DEFAULT_USER_AGENT = (
|
|
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
|
|
"AppleWebKit/537.36 (KHTML, like Gecko) "
|
|
"Chrome/120.0.0.0 Safari/537.36"
|
|
)
|
|
|
|
|
|
class NsfwSettingsService:
|
|
"""开启 NSFW 相关设置（线程安全，无全局状态）。"""
|
|
|
|
def __init__(self, cf_clearance: str = ""):
|
|
self.cf_clearance = (cf_clearance or "").strip()
|
|
|
|
def enable_nsfw(
|
|
self,
|
|
sso: str,
|
|
sso_rw: str,
|
|
impersonate: str,
|
|
user_agent: Optional[str] = None,
|
|
cf_clearance: Optional[str] = None,
|
|
timeout: int = 15,
|
|
) -> Dict[str, Any]:
|
|
"""
|
|
启用 always_show_nsfw_content。
|
|
返回: {
|
|
ok: bool,
|
|
hex_reply: str,
|
|
status_code: int | None,
|
|
grpc_status: str | None,
|
|
error: str | None
|
|
}
|
|
"""
|
|
if not sso:
|
|
return {
|
|
"ok": False,
|
|
"hex_reply": "",
|
|
"status_code": None,
|
|
"grpc_status": None,
|
|
"error": "缺少 sso",
|
|
}
|
|
if not sso_rw:
|
|
return {
|
|
"ok": False,
|
|
"hex_reply": "",
|
|
"status_code": None,
|
|
"grpc_status": None,
|
|
"error": "缺少 sso-rw",
|
|
}
|
|
|
|
url = "https://grok.com/auth_mgmt.AuthManagement/UpdateUserFeatureControls"
|
|
|
|
cookies = {
|
|
"sso": sso,
|
|
"sso-rw": sso_rw,
|
|
}
|
|
clearance = (cf_clearance if cf_clearance is not None else self.cf_clearance).strip()
|
|
if clearance:
|
|
cookies["cf_clearance"] = clearance
|
|
|
|
headers = {
|
|
"content-type": "application/grpc-web+proto",
|
|
"origin": "https://grok.com",
|
|
"referer": "https://grok.com/?_s=data",
|
|
"x-grpc-web": "1",
|
|
"user-agent": user_agent or DEFAULT_USER_AGENT,
|
|
}
|
|
|
|
data = (
|
|
b"\x00\x00\x00\x00"
|
|
b"\x20"
|
|
b"\x0a\x02\x10\x01"
|
|
b"\x12\x1a"
|
|
b"\x0a\x18"
|
|
b"always_show_nsfw_content"
|
|
)
|
|
|
|
try:
|
|
response = requests.post(
|
|
url,
|
|
headers=headers,
|
|
cookies=cookies,
|
|
data=data,
|
|
impersonate=impersonate or "chrome120",
|
|
timeout=timeout,
|
|
)
|
|
hex_reply = response.content.hex()
|
|
grpc_status = response.headers.get("grpc-status")
|
|
|
|
error = None
|
|
ok = response.status_code == 200 and (grpc_status in (None, "0"))
|
|
if response.status_code == 403:
|
|
error = "403 Forbidden"
|
|
elif response.status_code != 200:
|
|
error = f"HTTP {response.status_code}"
|
|
elif grpc_status not in (None, "0"):
|
|
error = f"gRPC {grpc_status}"
|
|
|
|
return {
|
|
"ok": ok,
|
|
"hex_reply": hex_reply,
|
|
"status_code": response.status_code,
|
|
"grpc_status": grpc_status,
|
|
"error": error,
|
|
}
|
|
except Exception as e:
|
|
return {
|
|
"ok": False,
|
|
"hex_reply": "",
|
|
"status_code": None,
|
|
"grpc_status": None,
|
|
"error": str(e),
|
|
}
|
|
|
|
def enable_unhinged(
|
|
self,
|
|
sso: str,
|
|
impersonate: str = "chrome120",
|
|
user_agent: Optional[str] = None,
|
|
timeout: int = 30,
|
|
) -> Dict[str, Any]:
|
|
"""
|
|
使用帖子方法开启 Unhinged 模式（二次验证）。
|
|
"""
|
|
import struct
|
|
|
|
url = "https://grok.com/auth_mgmt.AuthManagement/UpdateUserFeatureControls"
|
|
|
|
headers = {
|
|
"accept": "*/*",
|
|
"content-type": "application/grpc-web+proto",
|
|
"origin": "https://grok.com",
|
|
"referer": "https://grok.com/",
|
|
"user-agent": user_agent or DEFAULT_USER_AGENT,
|
|
"x-grpc-web": "1",
|
|
"x-user-agent": "connect-es/2.1.1",
|
|
"cookie": f"sso={sso}; sso-rw={sso}"
|
|
}
|
|
|
|
payload = bytes([0x08, 0x01, 0x10, 0x01])
|
|
data = b'\x00' + struct.pack('>I', len(payload)) + payload
|
|
|
|
try:
|
|
response = requests.post(
|
|
url,
|
|
headers=headers,
|
|
data=data,
|
|
impersonate=impersonate,
|
|
timeout=timeout,
|
|
)
|
|
return {
|
|
"ok": response.status_code == 200,
|
|
"status_code": response.status_code,
|
|
}
|
|
except Exception as e:
|
|
return {
|
|
"ok": False,
|
|
"error": str(e),
|
|
}
|