from fastapi import APIRouter

from apps.jd.schemas.models import SmsCodeRequest, SmsLoginRequest
from apps.jd.services.login import LoginSpider
from observability.logging import get_logger_with_trace

router = APIRouter(prefix="/jd/sms", tags=["短信认证"])

logger = get_logger_with_trace(__name__)

@router.post("/code")
async def get_code(request_data: SmsCodeRequest):
    """发送短信验证码"""
    phone = request_data.phone

    res = LoginSpider(phone_=phone).run_send_code()
    logger.info(f"发送验证码返回：{res}")

    return res


@router.post("/login")
async def sms_login(request_data: SmsLoginRequest):
    """短信登录"""
    phone = request_data.phone
    ck = request_data.ck
    code = request_data.code
    s_token = request_data.s_token
    jd_risk_token_id = request_data.jd_risk_token_id
    rsa_modulus = request_data.rsa_modulus

    res = LoginSpider(phone_=phone).run_get_ck(
        ck=ck,
        code=code,
        s_token=s_token,
        jd_risk_token_id=jd_risk_token_id,
        rsa_modulus=rsa_modulus,
    )
    logger.info(f"短信登录返回：{res}")

    return res