70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
from fastapi import APIRouter, HTTPException, status
|
|
|
|
from app.config import settings
|
|
from app.schemas.sms import SmsSendRequest, SmsVerifyRequest, SmsResponse
|
|
from app.services.sms import generate_and_send_sms, verify_sms_code
|
|
|
|
router = APIRouter(prefix="/sms", tags=["短信验证码"])
|
|
|
|
|
|
def _validate_captcha_token(captcha_token: str | None) -> None:
|
|
if settings.SMS_MOCK:
|
|
return
|
|
if not (req_captcha := captcha_token):
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail="需要验证码",
|
|
)
|
|
|
|
from app.services.auth import decode_access_token
|
|
|
|
token_sub = decode_access_token(req_captcha)
|
|
if not token_sub or not token_sub.startswith("captcha:"):
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail="验证码无效",
|
|
)
|
|
|
|
|
|
@router.post(
|
|
"/send",
|
|
response_model=SmsResponse,
|
|
summary="发送短信验证码",
|
|
description="客户端发送短信验证码。scene=register 用于注册，scene=login 用于短信登录，scene=set_password 用于设置密码。正式环境会按配置校验图形验证码。",
|
|
)
|
|
async def send_sms_code(req: SmsSendRequest):
|
|
_validate_captcha_token(req.captcha_token)
|
|
|
|
try:
|
|
ok, code = await generate_and_send_sms(req.phone, req.scene.value)
|
|
except ValueError as exc:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail=str(exc),
|
|
) from exc
|
|
|
|
if not ok:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
detail="短信发送失败，请稍后重试",
|
|
)
|
|
|
|
message = f"验证码已发送:{code}" if settings.SMS_MOCK else "验证码已发送"
|
|
return SmsResponse(message=message, success=True)
|
|
|
|
|
|
@router.post(
|
|
"/verify",
|
|
response_model=SmsResponse,
|
|
summary="校验短信验证码",
|
|
description="校验指定手机号、场景下的短信验证码。业务接口一般会内部校验，本接口主要用于前端调试或单独校验。",
|
|
)
|
|
async def verify_sms(req: SmsVerifyRequest):
|
|
ok = await verify_sms_code(req.phone, req.code, req.scene.value)
|
|
if not ok:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_400_BAD_REQUEST,
|
|
detail="验证码错误或已过期",
|
|
)
|
|
return SmsResponse(message="验证成功", success=True)
|