初始提交

2026-07-24 09:43:10 +08:00
commit b5b4721456
793 changed files with 116305 additions and 0 deletions
@@ -0,0 +1,192 @@
+import uuid
+from datetime import datetime, timedelta
+from typing import Annotated
+
+import jwt
+from fastapi import Depends, Request, Response
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from common.annotation.cache_annotation import ApiCache, ApiCacheEvict
+from common.annotation.log_annotation import Log
+from common.annotation.rate_limit_annotation import ApiRateLimit, ApiRateLimitPreset
+from common.aspect.db_seesion import DBSessionDependency
+from common.aspect.pre_auth import CurrentUserDependency
+from common.constant import ApiGroup, ApiNamespace
+from common.enums import BusinessType, RedisInitKeyConfig
+from common.router import APIRouterPro
+from common.vo import CrudResponseModel, DataResponseModel, DynamicResponseModel, ResponseBaseModel
+from config.env import AppConfig, JwtConfig
+from module_admin.entity.vo.login_vo import LoginToken, RouterModel, Token, UserLogin, UserRegister
+from module_admin.entity.vo.user_vo import CurrentUserModel, EditUserModel
+from module_admin.service.login_service import CustomOAuth2PasswordRequestForm, LoginService, oauth2_scheme
+from module_admin.service.user_service import UserService
+from utils.log_util import logger
+from utils.response_util import ResponseUtil
+
+login_controller = APIRouterPro(order_num=1, tags=['登录模块'])
+
+
+@login_controller.post(
+    '/login',
+    summary='登录接口',
+    description='用于用户登录',
+    response_model=DynamicResponseModel[LoginToken] | Token,
+)
+@ApiRateLimit(namespace=ApiNamespace.LOGIN, preset=ApiRateLimitPreset.ANON_AUTH_LOGIN)
+@ApiCacheEvict(namespaces=ApiGroup.LOGIN_SUCCESS_MUTATION)
+@Log(title='用户登录', business_type=BusinessType.OTHER, log_type='login')
+async def login(
+    request: Request,
+    form_data: Annotated[CustomOAuth2PasswordRequestForm, Depends()],
+    query_db: Annotated[AsyncSession, DBSessionDependency()],
+) -> Response:
+    captcha_enabled = (
+        await request.app.state.redis.get(f'{RedisInitKeyConfig.SYS_CONFIG.key}:sys.account.captchaEnabled') == 'true'
+    )
+    user = UserLogin(
+        userName=form_data.username,
+        password=form_data.password,
+        code=form_data.code,
+        uuid=form_data.uuid,
+        loginInfo=form_data.login_info,
+        captchaEnabled=captcha_enabled,
+    )
+    result = await LoginService.authenticate_user(request, query_db, user)
+    access_token_expires = timedelta(minutes=JwtConfig.jwt_expire_minutes)
+    session_id = str(uuid.uuid4())
+    access_token = await LoginService.create_access_token(
+        data={
+            'user_id': str(result[0].user_id),
+            'user_name': result[0].user_name,
+            'dept_name': result[1].dept_name if result[1] else None,
+            'session_id': session_id,
+            'login_info': user.login_info,
+        },
+        expires_delta=access_token_expires,
+    )
+    if AppConfig.app_same_time_login:
+        await request.app.state.redis.set(
+            f'{RedisInitKeyConfig.ACCESS_TOKEN.key}:{session_id}',
+            access_token,
+            ex=timedelta(minutes=JwtConfig.jwt_redis_expire_minutes),
+        )
+    else:
+        # 此方法可实现同一账号同一时间只能登录一次
+        await request.app.state.redis.set(
+            f'{RedisInitKeyConfig.ACCESS_TOKEN.key}:{result[0].user_id}',
+            access_token,
+            ex=timedelta(minutes=JwtConfig.jwt_redis_expire_minutes),
+        )
+    await UserService.edit_user_services(
+        query_db, EditUserModel(userId=result[0].user_id, loginDate=datetime.now(), type='status')
+    )
+    logger.info('登录成功')
+    # 判断请求是否来自于api文档，如果是返回指定格式的结果，用于修复api文档认证成功后token显示undefined的bug
+    request_from_swagger = request.headers.get('referer').endswith('docs') if request.headers.get('referer') else False
+    request_from_redoc = request.headers.get('referer').endswith('redoc') if request.headers.get('referer') else False
+    if request_from_swagger or request_from_redoc:
+        return {'access_token': access_token, 'token_type': 'Bearer'}
+    return ResponseUtil.success(msg='登录成功', dict_content={'token': access_token})
+
+
+@login_controller.get(
+    '/getInfo',
+    summary='获取用户信息接口',
+    description='用于获取当前登录用户的信息',
+    response_model=DynamicResponseModel[CurrentUserModel],
+)
+@ApiCache(namespace=ApiNamespace.LOGIN_USER_INFO)
+async def get_login_user_info(
+    request: Request, current_user: Annotated[CurrentUserModel, CurrentUserDependency()]
+) -> Response:
+    logger.info('获取成功')
+
+    return ResponseUtil.success(model_content=current_user)
+
+
+@login_controller.get(
+    '/getRouters',
+    summary='获取用户路由接口',
+    description='用于获取当前登录用户的路由信息',
+    response_model=DataResponseModel[list[RouterModel]],
+)
+@ApiCache(namespace=ApiNamespace.LOGIN_USER_ROUTERS)
+async def get_login_user_routers(
+    request: Request,
+    current_user: Annotated[CurrentUserModel, CurrentUserDependency()],
+    query_db: Annotated[AsyncSession, DBSessionDependency()],
+) -> Response:
+    logger.info('获取成功')
+    user_routers = await LoginService.get_current_user_routers(current_user.user.user_id, query_db)
+
+    return ResponseUtil.success(data=user_routers)
+
+
+@login_controller.post(
+    '/register',
+    summary='注册接口',
+    description='用于用户注册',
+    response_model=DataResponseModel[CrudResponseModel],
+)
+@ApiRateLimit(namespace=ApiNamespace.REGISTER, preset=ApiRateLimitPreset.ANON_AUTH_REGISTER)
+@ApiCacheEvict(namespaces=ApiGroup.USER_ENTITY_MUTATION)
+async def register_user(
+    request: Request,
+    user_register: UserRegister,
+    query_db: Annotated[AsyncSession, DBSessionDependency()],
+) -> Response:
+    user_register_result = await LoginService.register_user_services(request, query_db, user_register)
+    logger.info(user_register_result.message)
+
+    return ResponseUtil.success(data=user_register_result, msg=user_register_result.message)
+
+
+# @login_controller.post("/getSmsCode", response_model=SmsCode)
+# async def get_sms_code(request: Request, user: ResetUserModel, query_db: AsyncSession = DBSessionDependency()):
+#     try:
+#         sms_result = await LoginService.get_sms_code_services(request, query_db, user)
+#         if sms_result.is_success:
+#             logger.info('获取成功')
+#             return ResponseUtil.success(data=sms_result)
+#         else:
+#             logger.warning(sms_result.message)
+#             return ResponseUtil.failure(msg=sms_result.message)
+#     except Exception as e:
+#         logger.exception(e)
+#         return ResponseUtil.error(msg=str(e))
+#
+#
+# @login_controller.post("/forgetPwd", response_model=CrudResponseModel)
+# async def forget_user_pwd(request: Request, forget_user: ResetUserModel, query_db: AsyncSession = DBSessionDependency()):
+#     try:
+#         forget_user_result = await LoginService.forget_user_services(request, query_db, forget_user)
+#         if forget_user_result.is_success:
+#             logger.info(forget_user_result.message)
+#             return ResponseUtil.success(data=forget_user_result, msg=forget_user_result.message)
+#         else:
+#             logger.warning(forget_user_result.message)
+#             return ResponseUtil.failure(msg=forget_user_result.message)
+#     except Exception as e:
+#         logger.exception(e)
+#         return ResponseUtil.error(msg=str(e))
+
+
+@login_controller.post(
+    '/logout',
+    summary='退出登录接口',
+    description='用于用户退出登录',
+    response_model=ResponseBaseModel,
+)
+@ApiCacheEvict(namespaces=ApiGroup.LOGOUT_MUTATION)
+async def logout(request: Request, token: Annotated[str | None, Depends(oauth2_scheme)]) -> Response:
+    payload = jwt.decode(
+        token, JwtConfig.jwt_secret_key, algorithms=[JwtConfig.jwt_algorithm], options={'verify_exp': False}
+    )
+    if AppConfig.app_same_time_login:
+        token_id: str = payload.get('session_id')
+    else:
+        token_id: str = payload.get('user_id')
+    await LoginService.logout_services(request, token_id)
+    logger.info('退出成功')
+
+    return ResponseUtil.success(msg='退出成功')