from api.core.Gzip import register_gzip_request import json from fastapi import APIRouter, Body, Request, Response from api.core.Auth import createToken from api.core.Response import response from api.core.Redis import get_redis_info, get_redis_info_async from api.dp import AsyncRedisDep, RedisDep, AltersDep from core.basic.equipment import get_equipment from core.character.adventure import get_adv_info, get_adv_list from core.basic.character import createCharacter from core.open.helper import get_user_info_by_id, get_user_info_by_role_id, get_user_list_by_id, getRoleJobCode from core.open.skillTree import get_skill_tree_info from api.core.exception import ResponseException from api.utils.skill_detail import SKILL_DETAIL_NORMALIZE_VERSION, normalize_skill_detail, skill_detail_for_level from api.utils.openapi_data import openapi_locale_path from api.utils.character_response import character_response_encoding, get_character_response_bytes router = APIRouter() register_gzip_request(router) @router.get('/adventure') def get_adventure_info(request: Request, redis: RedisDep, name: str = None): adventure_info = get_redis_info(redis, 'dcalc:adventure', get_adv_list) if name: total = [item for job in adventure_info for item in job.get('children', [])] code = getRoleJobCode(name) adventure_info = next((x for x in total if (x['title'] == name or x['name'] == code) and x['open']), None) return response(data=adventure_info) @router.get('/token/get/{alter}') def getToken(redis: RedisDep, alter: str, request: Request, version: str = None, equVersion: str = '0'): if get_adv_info(alter) is None: raise ResponseException('无效的职业') token = createToken(alter, equVersion, redis) return response(data=token) @router.get('/character') def get_character_info(request: Request, state: AltersDep, redis: RedisDep): def get_character(): return createCharacter(state.alter, state.equVersion).getInfo() def load_info(): return get_redis_info(redis, f'dcalc:character:{state.alter}:{state.equVersion}', get_character) encoding = character_response_encoding(request.headers.get('accept-encoding')) content = get_character_response_bytes(redis, state.alter, state.equVersion, encoding, load_info) headers = {} if encoding == 'gzip': headers['Content-Encoding'] = 'gzip' headers['Vary'] = 'Accept-Encoding' return Response(content=content, media_type='application/json', headers=headers) @router.get('/info/oath/{suitId}/{skillId}') def get_oath_info(state: AltersDep, suitId: str, skillId: str, rarity: str, lv: int, redis: RedisDep): def get_info(): skill = None version = state.equVersion equ = get_equipment(version) oath = next((x for x in equ.oath_suits if x.suitId == suitId and x.rarity == rarity and str(x.level) == str(lv)), None) if oath is not None: oathSkill = next((x for x in oath.skills if x.skillId == skillId), None) if oathSkill is not None: skill = {**oathSkill.__dict__, 'point': oath.points.split(',')[lv - 1], 'imageUrl': oath.imageUrl} return skill info = get_redis_info(redis, f'dcalc:oath:{suitId}:{rarity}:{lv}:{skillId}', get_info) return response(data=info) @router.get('/skill/{skillId}/{level}') def get_skill_info(state: AltersDep, skillId: str, level: int, redis: RedisDep): """ 获取技能信息 """ alter = state.alter jobId = alter.split('.')[1] jobGrowId = alter.split('.')[2].replace('_carry', '') skill_info = {} key = f'openapi:{SKILL_DETAIL_NORMALIZE_VERSION}:{jobId}:{jobGrowId}:{skillId}' try: def get_skill_info(): # This function should retrieve the skill info based on job and jobGrow # For now, we return a placeholder dictionary with open(openapi_locale_path('cn', jobId, jobGrowId, 'skillDetail', f'{skillId}.json'), encoding='utf-8') as f: skill_data = json.load(f) return normalize_skill_detail(skill_data) skill_info = get_redis_info(redis, key, get_skill_info) skill_info = skill_detail_for_level(skill_info, level) except FileNotFoundError: return response(code=404, message=f'技能信息未找到: {jobId} {jobGrowId} {skillId}', data=None) return response(data=skill_info) @router.get('/dnfhelper/{uid}/list') async def get_dnfhelper_list(uid: str, redis: AsyncRedisDep): cache_key = f'dnfhelper:{uid}' async def get_list(): return await get_user_list_by_id(uid) result = await get_redis_info_async(redis, cache_key, get_list, expire=12 * 60 * 60) return response(data=result) # return response(code=500, message='Failed to fetch DNFHelper list', data=None) @router.get('/dnfhelper/{uid}/{id}/') async def get_dnfhelper_detail(uid: str, id: str, state: AltersDep, redis: AsyncRedisDep): cache_key = f'dnfhelper:{uid}:{id}' async def get_detail(): return await get_user_info_by_role_id(uid, id, state) result = await get_redis_info_async(redis, cache_key, get_detail, expire=12 * 60 * 60) return response(data=result) # return response(code=500, message='Failed to fetch DNFHelper detail', data=None) @router.get('/dnfhelper/{uid}/') async def get_info_by_dnfhelper(uid: str, state: AltersDep): try: result = await get_user_info_by_id(uid, state) return response(data=result) except Exception as e: if isinstance(e, ResponseException): return response(code=500, message=str(e), data=None) return response(code=500, message='Failed to fetch DNFHelper info', data=None) @router.post('/skillTree/') def get_skill_by_code(state: AltersDep, code=Body(None)): try: skills = get_skill_tree_info(code.get('code', '')) if skills.get('job', {}).get('class', '') != state.alter.replace('_carry', ''): return response(code=500, message='技能加点代码职业与登录职业不匹配', data=None) return response(data=skills) except Exception as e: if isinstance(e, ResponseException): return response(code=500, message=str(e), data=None) return response(code=500, message='解析技能加点失败', data=None)