117 lines
3.6 KiB
Python
117 lines
3.6 KiB
Python
from fastapi import Depends, HTTPException
|
||
from database.creator import get_db_session
|
||
from provider.loveac.authme import fetch_user_by_token
|
||
from provider.aufe.isim import ISIMClient
|
||
from provider.aufe.client import AUFEConnection
|
||
from database.user import User
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
from typing import Dict
|
||
from database.isim import ISIMRoomBinding
|
||
from sqlalchemy import select
|
||
|
||
|
||
# 全局ISIM客户端池
|
||
_isim_clients: Dict[str, ISIMClient] = {}
|
||
|
||
def get_cached_isim_client(user_id: str) -> ISIMClient:
|
||
"""
|
||
获取缓存的ISIM客户端
|
||
|
||
Args:
|
||
user_id: 用户ID
|
||
|
||
Returns:
|
||
ISIMClient: 缓存的ISIM客户端实例，如果未找到则返回None
|
||
"""
|
||
return _isim_clients.get(user_id)
|
||
|
||
def cache_isim_client(user_id: str, client: ISIMClient) -> None:
|
||
"""
|
||
缓存ISIM客户端
|
||
|
||
Args:
|
||
user_id: 用户ID
|
||
client: ISIM客户端实例
|
||
"""
|
||
_isim_clients[user_id] = client
|
||
|
||
|
||
async def get_isim_client(
|
||
user: User = Depends(fetch_user_by_token),
|
||
session: AsyncSession = Depends(get_db_session),
|
||
) -> ISIMClient:
|
||
from loguru import logger
|
||
"""
|
||
获取ISIM客户端实例
|
||
|
||
Args:
|
||
user: 用户对象（通过认证令牌获取）
|
||
|
||
Returns:
|
||
ISIMClient: ISIM客户端实例
|
||
|
||
Raises:
|
||
HTTPException: 认证失败时抛出
|
||
"""
|
||
if not user:
|
||
raise HTTPException(status_code=400, detail="无效的令牌或用户不存在")
|
||
|
||
# 首先检查是否已有缓存的ISIM客户端
|
||
cached_client = get_cached_isim_client(user.userid)
|
||
if cached_client:
|
||
# 检查缓存的客户端是否仍然有效
|
||
try:
|
||
if cached_client.is_session_valid():
|
||
from loguru import logger
|
||
logger.info(f"复用缓存的ISIM客户端: user_id={user.userid}")
|
||
return cached_client
|
||
except Exception as e:
|
||
from loguru import logger
|
||
logger.warning(f"缓存的ISIM客户端无效，将重新创建: {str(e)}")
|
||
|
||
# 创建或获取VPN连接
|
||
aufe = AUFEConnection.create_or_get_connection("vpn.aufe.edu.cn", user.userid)
|
||
|
||
# 检查VPN登录状态
|
||
if not aufe.login_status():
|
||
userid = user.userid
|
||
easyconnect_password = user.easyconnect_password
|
||
if not await aufe.login(userid, easyconnect_password):
|
||
raise HTTPException(
|
||
status_code=400,
|
||
detail="VPN登录失败，请检查用户名和密码",
|
||
)
|
||
|
||
# 检查UAAP登录状态
|
||
if not aufe.uaap_login_status():
|
||
userid = user.userid
|
||
password = user.password
|
||
if not await aufe.uaap_login(userid, password):
|
||
raise HTTPException(
|
||
status_code=400,
|
||
detail="大学登录失败，请检查用户名和密码",
|
||
)
|
||
|
||
# 创建新的ISIM客户端
|
||
isim_client = ISIMClient(aufe)
|
||
|
||
|
||
result_query = await session.execute(
|
||
select(ISIMRoomBinding).where(ISIMRoomBinding.userid == user.userid)
|
||
)
|
||
binding_record = result_query.scalars().first()
|
||
if binding_record:
|
||
logger.info(f"找到用户({user.userid})绑定记录，进行启动再绑定")
|
||
await isim_client.bind_room(
|
||
building_code=binding_record.building_code,
|
||
floor_code=binding_record.floor_code,
|
||
room_code=binding_record.room_code,
|
||
)
|
||
|
||
# 缓存客户端
|
||
cache_isim_client(user.userid, isim_client)
|
||
|
||
logger.info(f"创建并缓存新的ISIM客户端: user_id={user.userid}")
|
||
|
||
return isim_client
|