#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
天翼订购页 1.html 接口：
- POST /api/submit_phone  传电话号码，完成滑块后返回 hn_userEquitys/getYanZhenMa/v2 响应
- POST /api/submit_code   传验证码，点击确认订购
"""
from __future__ import annotations

import threading
from typing import Any, Optional

from fastapi import FastAPI, HTTPException
from pydantic import BaseModel, Field

from tyyp_service import submit_phone as do_submit_phone, submit_code as do_submit_code

app = FastAPI(title="天翼订购页 API", description="1.html 手机号与验证码提交流程")

# 全局页面实例，用于 submit_phone 与 submit_code 之间复用
_page: Optional[Any] = None
_page_lock = threading.Lock()


def _get_or_create_page():
    global _page
    with _page_lock:
        if _page is None:
            from DrissionPage import ChromiumPage
            _page = ChromiumPage()
        return _page


class SubmitPhoneRequest(BaseModel):
    phone: str = Field(..., description="手机号码")
    port: int = Field(0, description="连接已有浏览器端口，0 表示新建")
    url: str = Field("http://yscnb.com/tyyp/1.html", description="目标页面 URL")


class SubmitPhoneResponse(BaseModel):
    success: bool = True
    data: Any = Field(..., description="hn_userEquitys/getYanZhenMa/v2 接口返回的响应体")


class SubmitCodeRequest(BaseModel):
    code: str = Field(..., description="短信验证码")


class SubmitCodeResponse(BaseModel):
    success: bool = True
    message: str = ""


@app.post("/api/submit_phone", response_model=SubmitPhoneResponse)
def api_submit_phone(req: SubmitPhoneRequest):
    """
    提交手机号：填写手机号、点击获取验证码、执行滑块验证，
    监听 hn_userEquitys/getYanZhenMa/v2，将其响应体原样返回。
    """
    global _page
    try:
        from DrissionPage import ChromiumOptions, ChromiumPage
        if req.port:
            co = ChromiumOptions().set_local_port(port=req.port)
            page = ChromiumPage(addr_or_opts=co)
            with _page_lock:
                _page = page  # 供 submit_code 复用
        else:
            page = _get_or_create_page()

        body = do_submit_phone(
            page=page,
            phone=req.phone,
            url=req.url,
            port=req.port,
        )
        return SubmitPhoneResponse(success=True, data=body)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@app.post("/api/submit_code", response_model=SubmitCodeResponse)
def api_submit_code(req: SubmitCodeRequest):
    """
    提交验证码：在已打开并处于验证码表单的页面上，
    填写验证码并点击确认订购按钮（img.btn-buy）。
    """
    try:
        page = _get_or_create_page()
        result = do_submit_code(page=page, code=req.code)
        return SubmitCodeResponse(success=True, message=result.get("message", "已点击确认订购"))
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8000)