119 lines
3.7 KiB
Python
119 lines
3.7 KiB
Python
|
from fastapi import Depends
|
|||
|
|
from fastapi.routing import APIRouter
|
|||
|
|
from database.user import Invite, User
|
|||
|
|
from database.creator import get_db_session
|
|||
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|||
|
|
from sqlalchemy import select
|
|||
|
|
from router.invite.model import (
|
|||
|
|
InviteRequest,
|
|||
|
|
RegisterRequest,
|
|||
|
|
InviteResponse,
|
|||
|
|
RegisterResponse,
|
|||
|
|
InviteTokenData,
|
|||
|
|
AuthMeData,
|
|||
|
|
)
|
|||
|
|
from provider.aufe.client import AUFEConnection
|
|||
|
|
from database.user import AuthME
|
|||
|
|
import secrets
|
|||
|
|
|
|||
|
|
invite_router = APIRouter(prefix="/api/v1/user")
|
|||
|
|
invite_tokens = []
|
|||
|
|
|
|||
|
|
|
|||
|
|
@invite_router.post("/veryfy_invite_code", summary="验证邀请码")
|
|||
|
|
async def verify_invite_code(
|
|||
|
|
data: InviteRequest,
|
|||
|
|
asyncsession: AsyncSession = Depends(get_db_session),
|
|||
|
|
) -> InviteResponse:
|
|||
|
|
"""
|
|||
|
|
验证邀请码
|
|||
|
|
:param data: InviteRequest
|
|||
|
|
:return: InviteResponse
|
|||
|
|
"""
|
|||
|
|
async with asyncsession as session:
|
|||
|
|
invite_code = data.invite_code
|
|||
|
|
invite = select(Invite).where(Invite.invite_code == invite_code)
|
|||
|
|
result = await session.execute(invite)
|
|||
|
|
invite_data = result.scalars().first()
|
|||
|
|
if invite_data:
|
|||
|
|
invite_token = secrets.token_urlsafe(128)
|
|||
|
|
invite_tokens.append(invite_token)
|
|||
|
|
return InviteResponse(
|
|||
|
|
code=200,
|
|||
|
|
message="邀请码验证成功",
|
|||
|
|
data=InviteTokenData(invite_token=invite_token),
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
return InviteResponse(
|
|||
|
|
code=400,
|
|||
|
|
message="邀请码无效或已过期",
|
|||
|
|
data=None,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@invite_router.post("/register", summary="注册新用户")
|
|||
|
|
async def register_user(
|
|||
|
|
data: RegisterRequest,
|
|||
|
|
asyncsession: AsyncSession = Depends(get_db_session),
|
|||
|
|
) -> RegisterResponse:
|
|||
|
|
"""
|
|||
|
|
注册新用户
|
|||
|
|
:param data: RegisterRequest
|
|||
|
|
:return: RegisterResponse
|
|||
|
|
"""
|
|||
|
|
async with asyncsession as session:
|
|||
|
|
userid = data.userid
|
|||
|
|
password = data.password
|
|||
|
|
easyconnect_password = data.easyconnect_password
|
|||
|
|
invite_token = data.invite_token
|
|||
|
|
if invite_token not in invite_tokens:
|
|||
|
|
return RegisterResponse(
|
|||
|
|
code=400,
|
|||
|
|
message="无效的邀请令牌",
|
|||
|
|
data=None,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 检查用户是否已存在
|
|||
|
|
existing_user = await session.execute(select(User).where(User.userid == userid))
|
|||
|
|
if existing_user.scalars().first():
|
|||
|
|
return RegisterResponse(
|
|||
|
|
code=400,
|
|||
|
|
message="用户已存在",
|
|||
|
|
data=None,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 检查连接
|
|||
|
|
vpn = AUFEConnection.create_or_get_connection("vpn.aufe.edu.cn", userid)
|
|||
|
|
if not await vpn.login(userid, easyconnect_password):
|
|||
|
|
return RegisterResponse(
|
|||
|
|
code=400,
|
|||
|
|
message="VPN登录失败，请检查用户名和密码",
|
|||
|
|
data=None,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if not await vpn.uaap_login(userid, password):
|
|||
|
|
return RegisterResponse(
|
|||
|
|
code=400,
|
|||
|
|
message="大学登录失败，请检查用户名和密码",
|
|||
|
|
data=None,
|
|||
|
|
)
|
|||
|
|
# 创建新用户
|
|||
|
|
|
|||
|
|
new_user = User(
|
|||
|
|
userid=userid,
|
|||
|
|
password=password,
|
|||
|
|
easyconnect_password=easyconnect_password,
|
|||
|
|
)
|
|||
|
|
session.add(new_user)
|
|||
|
|
await session.commit()
|
|||
|
|
authme_token = secrets.token_urlsafe(128)
|
|||
|
|
new_authme = AuthME(userid=userid, authme_token=authme_token)
|
|||
|
|
session.add(new_authme)
|
|||
|
|
await session.commit()
|
|||
|
|
invite_tokens.remove(invite_token)
|
|||
|
|
return RegisterResponse(
|
|||
|
|
code=200,
|
|||
|
|
message="注册成功",
|
|||
|
|
data=AuthMeData(authme_token=authme_token),
|
|||
|
|
)
|