hyf-backend/th_agenter/api/endpoints/auth.py

"""Authentication endpoints."""

from datetime import timedelta
from fastapi import APIRouter, Depends, HTTPException, status, Request
from fastapi.security import OAuth2PasswordRequestForm
from sqlalchemy.orm import Session

from ...core.config import get_settings
from ...db.database import DrSession, get_session
from ...services.auth import AuthService
from ...services.user import UserService
from ...schemas.user import UserResponse, UserCreate, LoginResponse
from utils.util_schemas import Token, LoginRequest
from loguru import logger
from utils.util_exceptions import HxfResponse

router = APIRouter()
settings = get_settings()

@router.post("/register", response_model=UserResponse, summary="注册新用户")
async def register(
    request_user_data: UserCreate,
    session: DrSession = Depends(get_session)
):
    """注册新用户"""    
    user_service = UserService(session)
    session.desc = f"START: 注册用户 {request_user_data.email}"
    if await user_service.get_user_by_email(request_user_data.email):
        raise HTTPException(
            status_code=status.HTTP_400_BAD_REQUEST,
            detail=f"邮箱 {request_user_data.email} 已被注册，请使用其他邮箱注册!!!"
        )
    
    if await user_service.get_user_by_username(request_user_data.username):
        raise HTTPException(
            status_code=status.HTTP_400_BAD_REQUEST,
            detail=f"用户名 {request_user_data.username} 已被注册，请使用其他用户名注册!!!"
        )
    
    user = await user_service.create_user(request_user_data)
    response = UserResponse.model_validate(user, from_attributes=True)
    return HxfResponse(response)

@router.post("/login", response_model=LoginResponse, summary="邮箱与密码登录")
async def login(
    login_data: LoginRequest,
    session: DrSession = Depends(get_session)
):
    """邮箱与密码登录"""
    # Authenticate user by email
    session.desc = f"START: 用户 {login_data.email} 尝试登录" 
    user = await AuthService.authenticate_user_by_email(session, login_data.email, login_data.password)
    if not user:
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
            detail=f"邮箱 {login_data.email} 或密码错误，请检查后重试!!!",
            headers={"WWW-Authenticate": "Bearer"},
        )
    
    # Create access token
    access_token_expires = timedelta(minutes=settings.security.access_token_expire_minutes)
    access_token = await AuthService.create_access_token(
        session, data={"sub": user.username}, expires_delta=access_token_expires
    )
    session.desc = f"用户 {user.username} 登录成功"
    
    response = LoginResponse(
        access_token=access_token,
        token_type="bearer",
        expires_in=settings.security.access_token_expire_minutes * 60,
        user=UserResponse.model_validate(user, from_attributes=True)
    )
    return HxfResponse(response)

@router.post("/login-oauth", response_model=Token, summary="用户通过用户名和密码登录 (OAuth2 兼容)")
async def login_oauth(
    form_data: OAuth2PasswordRequestForm = Depends(),
    session: DrSession = Depends(get_session)
):
    """用户通过用户名和密码登录 (OAuth2 兼容)"""
    session.desc = f"START: 用户 {form_data.username} 尝试 OAuth2 登录"
    user = await AuthService.authenticate_user(session, form_data.username, form_data.password)
    if not user:
        session.desc = f"用户 {form_data.username} 尝试 OAuth2 登录失败"
        raise HTTPException(
            status_code=status.HTTP_401_UNAUTHORIZED,
            detail="Incorrect username or password",
            headers={"WWW-Authenticate": "Bearer"},
        )
    
    # Create access token
    access_token_expires = timedelta(minutes=settings.security.access_token_expire_minutes)
    access_token = await AuthService.create_access_token(
        session, data={"sub": user.username}, expires_delta=access_token_expires
    )
    session.desc = f"用户 {user.username} OAuth2 登录成功"
    
    return HxfResponse(
        {
            "access_token": access_token,
            "token_type": "bearer",
            "expires_in": settings.security.access_token_expire_minutes * 60
        }
    )

@router.post("/refresh", response_model=Token, summary="刷新访问token")
async def refresh_token(
    current_user = Depends(AuthService.get_current_user),
    session: DrSession = Depends(get_session)
):
    """刷新访问 token"""
    # Create new access token
    access_token_expires = timedelta(minutes=settings.security.access_token_expire_minutes)
    access_token = await AuthService.create_access_token(
        session, data={"sub": current_user.username}, expires_delta=access_token_expires
    )
    
    response = Token(
        access_token=access_token,
        token_type="bearer",
        expires_in=settings.security.access_token_expire_minutes * 60
    )
    return HxfResponse(response)

@router.get("/me", response_model=UserResponse, summary="获取当前用户信息")
async def get_current_user_info(
    current_user = Depends(AuthService.get_current_user)
):
    """获取当前用户信息"""
    response = UserResponse.model_validate(current_user, from_attributes=True)
    return HxfResponse(response)