# 导入库
from typing import Annotated
from fastapi import APIRouter, Depends
from fastapi.security import OAuth2PasswordRequestForm
from sqlmodel.ext.asyncio.session import AsyncSession

from model import database
from model.response import TokenResponse
from services import session as session_service
from pkg import utils

Router = APIRouter(tags=["令牌 session"])

# FastAPI 登录路由 / FastAPI login route
@Router.post(
    path="/api/token",
    summary="获取访问令牌",
    description="使用用户名和密码获取访问令牌",
    response_model=TokenResponse,
    response_description="访问令牌"
)
async def login_for_access_token(
    form_data: Annotated[OAuth2PasswordRequestForm, Depends()],
    session: Annotated[AsyncSession, Depends(database.Database.get_session)],
) -> TokenResponse:
    token_response = await session_service.login_for_access_token(
        session=session,
        username=form_data.username,
        password=form_data.password,
    )
    if not token_response:
        utils.raise_unauthorized("Incorrect username or password")

    return token_response