添加虎牙自动注册

core/huya/auto_register.py

@@ -0,0 +1,199 @@
+"""虎牙手机号自动注册流程。"""
+
+from __future__ import annotations
+
+import threading
+import time
+from dataclasses import dataclass
+from datetime import datetime
+
+from core.sms_provider import SmsLine, SmsProviderClient
+
+from .login import HuyaLoginError
+from .sms_login import login_huya_sms, normalize_huya_phone, send_huya_sms_code
+
+
+@dataclass
+class HuyaAutoRegisterResult:
+    """单个手机号自动注册结果。"""
+
+    phone: str
+    provider: str
+    success: bool
+    status: str
+    message: str
+    cookie: str = ""
+    code: str = ""
+    sdid: str = ""
+    normalized_phone: str = ""
+    attempts: int = 0
+
+
+def _sleep_or_stop(stop_event: threading.Event | None, seconds: float) -> bool:
+    """等待一小段时间；返回 False 表示收到停止信号。"""
+    if seconds <= 0:
+        return not (stop_event and stop_event.is_set())
+    if stop_event:
+        return not stop_event.wait(seconds)
+    time.sleep(seconds)
+    return True
+
+
+def register_huya_with_sms_line(
+    item: SmsLine,
+    wait_seconds: float = 180,
+    poll_interval: float = 5,
+    stop_event: threading.Event | None = None,
+) -> HuyaAutoRegisterResult:
+    """使用固定手机号和接码地址完成虎牙短信注册/登录。"""
+    phone = item.phone.strip()
+    normalized_phone = normalize_huya_phone(phone)
+    wait_seconds = max(15.0, float(wait_seconds or 180))
+    poll_interval = max(1.0, float(poll_interval or 5))
+
+    if stop_event and stop_event.is_set():
+        return HuyaAutoRegisterResult(
+            phone=phone,
+            provider=item.provider,
+            success=False,
+            status="stopped",
+            message="已停止",
+            normalized_phone=normalized_phone,
+        )
+
+    sent_at = datetime.now()
+    try:
+        code_result = send_huya_sms_code(phone=phone)
+    except HuyaLoginError as exc:
+        return HuyaAutoRegisterResult(
+            phone=phone,
+            provider=item.provider,
+            success=False,
+            status="error",
+            message=str(exc),
+            normalized_phone=normalized_phone,
+        )
+    except Exception as exc:
+        return HuyaAutoRegisterResult(
+            phone=phone,
+            provider=item.provider,
+            success=False,
+            status="error",
+            message=f"发送虎牙短信失败: {exc}",
+            normalized_phone=normalized_phone,
+        )
+
+    if not code_result.success or not code_result.state:
+        return HuyaAutoRegisterResult(
+            phone=phone,
+            provider=item.provider,
+            success=False,
+            status="error",
+            message=code_result.message or "发送虎牙短信失败",
+            sdid=code_result.sdid,
+            normalized_phone=normalized_phone,
+        )
+
+    client = SmsProviderClient()
+    deadline = time.monotonic() + wait_seconds
+    attempts = 0
+    last_message = "等待验证码"
+
+    while time.monotonic() < deadline:
+        if stop_event and stop_event.is_set():
+            return HuyaAutoRegisterResult(
+                phone=phone,
+                provider=item.provider,
+                success=False,
+                status="stopped",
+                message="已停止",
+                sdid=code_result.sdid,
+                normalized_phone=normalized_phone,
+                attempts=attempts,
+            )
+
+        attempts += 1
+        poll_result = client.poll(item, after_time=sent_at)
+        last_message = poll_result.message or last_message
+        if poll_result.has_code:
+            try:
+                login_result = login_huya_sms(
+                    authcode=poll_result.code,
+                    state=code_result.state,
+                    phone=phone,
+                )
+            except HuyaLoginError as exc:
+                return HuyaAutoRegisterResult(
+                    phone=phone,
+                    provider=item.provider,
+                    success=False,
+                    status="error",
+                    message=str(exc),
+                    code=poll_result.code,
+                    sdid=code_result.sdid,
+                    normalized_phone=normalized_phone,
+                    attempts=attempts,
+                )
+            except Exception as exc:
+                return HuyaAutoRegisterResult(
+                    phone=phone,
+                    provider=item.provider,
+                    success=False,
+                    status="error",
+                    message=f"提交虎牙短信登录失败: {exc}",
+                    code=poll_result.code,
+                    sdid=code_result.sdid,
+                    normalized_phone=normalized_phone,
+                    attempts=attempts,
+                )
+
+            if login_result.success and login_result.cookie:
+                return HuyaAutoRegisterResult(
+                    phone=phone,
+                    provider=item.provider,
+                    success=True,
+                    status="success",
+                    message="注册/登录成功",
+                    cookie=login_result.cookie,
+                    code=poll_result.code,
+                    sdid=login_result.sdid or code_result.sdid,
+                    normalized_phone=normalized_phone,
+                    attempts=attempts,
+                )
+            return HuyaAutoRegisterResult(
+                phone=phone,
+                provider=item.provider,
+                success=False,
+                status="error",
+                message=login_result.message or "虎牙短信登录失败",
+                code=poll_result.code,
+                sdid=login_result.sdid or code_result.sdid,
+                normalized_phone=normalized_phone,
+                attempts=attempts,
+            )
+
+        remaining = deadline - time.monotonic()
+        if remaining <= 0:
+            break
+        if not _sleep_or_stop(stop_event, min(poll_interval, remaining)):
+            return HuyaAutoRegisterResult(
+                phone=phone,
+                provider=item.provider,
+                success=False,
+                status="stopped",
+                message="已停止",
+                sdid=code_result.sdid,
+                normalized_phone=normalized_phone,
+                attempts=attempts,
+            )
+
+    return HuyaAutoRegisterResult(
+        phone=phone,
+        provider=item.provider,
+        success=False,
+        status="error",
+        message=f"等待验证码超时: {last_message}",
+        sdid=code_result.sdid,
+        normalized_phone=normalized_phone,
+        attempts=attempts,
+    )