110 lines
3.4 KiB
Python
110 lines
3.4 KiB
Python
from fastapi import Depends
|
||
from fastapi.routing import APIRouter
|
||
from database.user import User
|
||
from database.creator import get_db_session
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from sqlalchemy import select
|
||
from router.login.model import (
|
||
LoginRequest,
|
||
LoginResponse,
|
||
AuthmeResponse,
|
||
AuthmeStatusData,
|
||
)
|
||
from router.invite.model import AuthMeData
|
||
from provider.aufe.client import AUFEConnection
|
||
from provider.loveac.authme import manage_user_tokens, generate_device_id, fetch_user_by_token, AuthmeRequest
|
||
import secrets
|
||
|
||
login_router = APIRouter(prefix="/api/v1/user")
|
||
|
||
|
||
@login_router.post("/login", summary="用户登录")
|
||
async def login_user(
|
||
data: LoginRequest, asyncsession: AsyncSession = Depends(get_db_session)
|
||
) -> LoginResponse:
|
||
"""
|
||
用户登录
|
||
:param data: LoginRequest
|
||
:return: LoginResponse
|
||
"""
|
||
async with asyncsession as session:
|
||
userid = data.userid
|
||
password = data.password
|
||
easyconnect_password = data.easyconnect_password
|
||
|
||
# 检查用户是否存在
|
||
existing_user = await session.execute(select(User).where(User.userid == userid))
|
||
user = existing_user.scalars().first()
|
||
if not user:
|
||
return LoginResponse(
|
||
code=400,
|
||
message="用户不存在",
|
||
data=None,
|
||
)
|
||
|
||
# 检查连接
|
||
vpn = AUFEConnection.create_or_get_connection("vpn.aufe.edu.cn", userid)
|
||
# 检查连接是否已经存在，避免重复登录
|
||
if not vpn.login_status():
|
||
if not await vpn.login(userid, easyconnect_password):
|
||
return LoginResponse(
|
||
code=400,
|
||
message="VPN登录失败，请检查用户名和密码",
|
||
data=None,
|
||
)
|
||
if not vpn.uaap_login_status():
|
||
if not await vpn.uaap_login(userid, password):
|
||
return LoginResponse(
|
||
code=400,
|
||
message="大学登录失败，请检查用户名和密码",
|
||
data=None,
|
||
)
|
||
|
||
# 生成新的token和设备ID
|
||
authme_token = secrets.token_urlsafe(128)
|
||
device_id = generate_device_id()
|
||
|
||
# 使用新的token管理系统
|
||
await manage_user_tokens(userid, authme_token, device_id, session)
|
||
|
||
return LoginResponse(
|
||
code=200,
|
||
message="登录成功",
|
||
data=AuthMeData(authme_token=authme_token),
|
||
)
|
||
|
||
|
||
@login_router.post("/authme", summary="验证登录状态")
|
||
async def check_auth_status(
|
||
data: AuthmeRequest, asyncsession: AsyncSession = Depends(get_db_session)
|
||
) -> AuthmeResponse:
|
||
"""
|
||
验证token是否有效，返回登录状态
|
||
:param data: AuthmeRequest
|
||
:return: AuthmeResponse
|
||
"""
|
||
try:
|
||
# 使用已有的fetch_user_by_token函数验证token
|
||
user = await fetch_user_by_token(data, asyncsession)
|
||
|
||
return AuthmeResponse(
|
||
code=200,
|
||
message="验证成功",
|
||
data=AuthmeStatusData(
|
||
is_logged_in=True,
|
||
userid=user.userid
|
||
),
|
||
)
|
||
except Exception as e:
|
||
# token无效或其他错误
|
||
return AuthmeResponse(
|
||
code=401,
|
||
message="token无效或已过期",
|
||
data=AuthmeStatusData(
|
||
is_logged_in=False,
|
||
userid=None
|
||
),
|
||
)
|
||
|
||
|