清理代码

core/sentinel_handler.py

@@ -0,0 +1,95 @@
+"""Sentinel 反爬机制处理器"""
+
+from typing import Optional, Dict, Any
+from utils.logger import logger
+from utils.fingerprint import BrowserFingerprint
+
+
+class SentinelHandler:
+    """Sentinel 反爬机制处理器"""
+
+    SENTINEL_API = "https://chatgpt.com/_next/static/chunks/sentinel.js"
+    SENTINEL_TOKEN_ENDPOINT = "https://api.openai.com/sentinel/token"
+
+    def __init__(self, session):
+        from core.session import OAISession
+        self.session: OAISession = session
+        self.oai_did = session.oai_did
+        self.fingerprint = BrowserFingerprint(session_id=self.oai_did)
+        self._solver = None
+        logger.info("SentinelHandler initialized")
+
+    def _get_solver(self):
+        """延迟初始化 Sentinel 求解器"""
+        if self._solver is None:
+            try:
+                from core.sentinel import SentinelSolver
+                self._solver = SentinelSolver(self.fingerprint)
+                logger.debug("SentinelSolver initialized")
+            except ImportError as e:
+                logger.error(f"Failed to import SentinelSolver: {e}")
+                raise ImportError("Sentinel solver not found. Check core/sentinel/ directory.") from e
+        return self._solver
+
+    async def get_token(self, flow: str = "username_password_create", **kwargs) -> Dict[str, Any]:
+        """获取 Sentinel Token"""
+        logger.info(f"Generating Sentinel token for flow='{flow}'")
+
+        try:
+            solver = self._get_solver()
+            token_dict = solver.generate_requirements_token()
+            token_dict["flow"] = flow
+            token_dict["id"] = self.oai_did
+
+            if "p" not in token_dict or not token_dict["p"]:
+                raise ValueError("Generated token missing 'p' field")
+
+            logger.success(f"Sentinel token generated: {str(token_dict)[:50]}...")
+            return token_dict
+
+        except ImportError as e:
+            logger.error(f"Sentinel solver not available: {e}")
+            raise NotImplementedError(f"Sentinel solver import failed: {e}") from e
+
+        except Exception as e:
+            logger.exception(f"Failed to generate Sentinel token: {e}")
+            raise RuntimeError(f"Sentinel token generation failed: {e}") from e
+
+    async def solve_enforcement(self, enforcement_config: Dict[str, Any]) -> str:
+        """解决 enforcement 挑战（PoW + Turnstile）"""
+        logger.info("Solving enforcement challenge...")
+
+        try:
+            solver = self._get_solver()
+            token_json = solver.solve_enforcement(enforcement_config)
+            logger.success(f"Enforcement solved: {token_json[:50]}...")
+            return token_json
+
+        except Exception as e:
+            logger.exception(f"Failed to solve enforcement: {e}")
+            raise RuntimeError(f"Enforcement solving failed: {e}") from e
+
+    async def verify_token(self, token: str) -> bool:
+        """验证 Sentinel Token 格式"""
+        if not token or token == "placeholder_token":
+            return False
+        if not token.startswith("gAAAAA"):
+            logger.warning(f"Invalid token format: {token[:20]}...")
+            return False
+        return True
+
+    def get_sentinel_script(self) -> Optional[str]:
+        """获取 Sentinel JavaScript 代码"""
+        try:
+            response = self.session.get(self.SENTINEL_API)
+            if response.status_code == 200:
+                logger.info(f"Sentinel script fetched: {len(response.text)} bytes")
+                return response.text
+            logger.error(f"Failed to fetch Sentinel script: {response.status_code}")
+            return None
+        except Exception as e:
+            logger.error(f"Error fetching Sentinel script: {e}")
+            return None
+
+
+__all__ = ["SentinelHandler"]