main.py 55 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637
  1. """
  2. 主应用入口
  3. """
  4. import json
  5. import logging
  6. import urllib.parse
  7. import hashlib
  8. import uuid
  9. import random
  10. from typing import Optional, List, Dict, Any
  11. from fastapi import FastAPI, WebSocket, HTTPException, Request, Query
  12. from fastapi.responses import RedirectResponse, PlainTextResponse
  13. from fastapi.middleware.cors import CORSMiddleware
  14. from pydantic import BaseModel
  15. from contextlib import asynccontextmanager
  16. import httpx
  17. from xml.etree import ElementTree as ET
  18. from sqlalchemy import text, select, update
  19. from sqlalchemy.orm import selectinload
  20. from config import settings, get_dashscope_domain
  21. from websocket_handler import WebSocketHandler
  22. from database import init_database, close_database, db_manager
  23. from services.redis_cache_service import RedisCacheService
  24. from services.user_service import user_service
  25. from services.wechat_jsapi_service import WechatJSAPIService
  26. from services.energy_stone_service import energy_stone_service
  27. from models_db import VerificationCode, UserAvatar, UserBackgroundImage
  28. # 配置日志
  29. logging.basicConfig(
  30. level=getattr(logging, settings.LOG_LEVEL.upper(), logging.INFO),
  31. format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
  32. datefmt='%Y-%m-%d %H:%M:%S'
  33. )
  34. logger = logging.getLogger(__name__)
  35. @asynccontextmanager
  36. async def lifespan(app: FastAPI):
  37. """应用生命周期管理"""
  38. # 启动时初始化数据库(包含MySQL和Redis,自动启动健康检查任务)
  39. try:
  40. await init_database()
  41. # 获取Redis客户端(已配置健康检查),用于验证连通性
  42. from database import db_manager
  43. await db_manager.get_redis()
  44. # 创建Redis缓存服务,按需获取最新连接
  45. app.state.redis_cache = RedisCacheService(db_manager.get_redis)
  46. except Exception as e:
  47. logger.error(f"❌ 数据库初始化失败: {e}")
  48. app.state.redis_cache = None
  49. # 🔥 启动HTTP连接保活任务(保持连接活跃,避免长时间不交互后响应变慢)
  50. try:
  51. from http_pool import http_pool
  52. from ai_service import AIService
  53. # 创建AIService实例用于保活(共享连接池)
  54. ai_service = AIService()
  55. http_pool.start_keepalive(ai_service=ai_service)
  56. except Exception as e:
  57. logger.warning(f"⚠️ HTTP连接保活任务启动失败: {e}")
  58. yield
  59. # 关闭时清理资源
  60. await close_database()
  61. # 🔥 关闭HTTP连接池(类似SQLAlchemy的Engine.dispose())
  62. from http_pool import http_pool
  63. http_pool.close()
  64. app = FastAPI(
  65. title="Lyrria Backend API",
  66. lifespan=lifespan
  67. )
  68. # 配置CORS
  69. app.add_middleware(
  70. CORSMiddleware,
  71. allow_origins=["*"], # 生产环境应该限制具体域名
  72. allow_credentials=True,
  73. allow_methods=["*"],
  74. allow_headers=["*"],
  75. )
  76. @app.get("/")
  77. async def root():
  78. """健康检查"""
  79. return {"status": "ok", "message": "Lyrria Backend API"}
  80. class CreateUserRequest(BaseModel):
  81. """创建用户请求"""
  82. username: str = ""
  83. bot_name: Optional[str] = None
  84. class UpdateUsernameRequest(BaseModel):
  85. """更新用户名请求"""
  86. username: str
  87. class UpdateBotNameRequest(BaseModel):
  88. """更新机器人名字请求"""
  89. bot_name: str
  90. class RegisterPhoneRequest(BaseModel):
  91. """手机号注册请求"""
  92. phone: str
  93. username: str = ""
  94. class RegisterEmailRequest(BaseModel):
  95. """邮箱注册请求"""
  96. email: str
  97. username: str = ""
  98. class LoginWechatRequest(BaseModel):
  99. """微信登录请求"""
  100. openid: str
  101. username: str = ""
  102. class LoginQQRequest(BaseModel):
  103. """QQ登录请求"""
  104. openid: str
  105. username: str = ""
  106. class LoginPhoneRequest(BaseModel):
  107. """手机号登录请求"""
  108. phone: str
  109. class LoginEmailRequest(BaseModel):
  110. """邮箱登录请求"""
  111. email: str
  112. class SendCodeRequest(BaseModel):
  113. """发送验证码请求"""
  114. phone: Optional[str] = None
  115. email: Optional[str] = None
  116. send_type: str # "domestic_sms", "international_sms", "email"
  117. class VerifyCodeRequest(BaseModel):
  118. """验证验证码请求"""
  119. phone: Optional[str] = None
  120. email: Optional[str] = None
  121. code: str
  122. send_type: str # "domestic_sms", "international_sms", "email"
  123. class UserUIStateResponse(BaseModel):
  124. """用户前端可视化状态"""
  125. user_id: str
  126. current_page: Optional[str] = None
  127. visited_routes: List[str] = []
  128. has_shown_tap_to_chat: bool = False
  129. has_shown_cancel_tooltip: bool = False
  130. has_shown_send_tooltip: bool = False
  131. has_shown_progress_bar_tooltip: bool = False
  132. has_shown_mission_stone_modal: bool = False
  133. has_shown_conversation_stone_modal: bool = False
  134. has_clicked_mission_stone: bool = False
  135. has_clicked_conversation_stone: bool = False
  136. avatar_c_unlocked: bool = False
  137. has_shared_avatar: bool = False
  138. current_avatar_id: Optional[str] = None
  139. class UpdateUserUIStateRequest(BaseModel):
  140. """部分更新用户前端可视化状态"""
  141. current_page: Optional[str] = None
  142. visited_routes: Optional[List[str]] = None
  143. has_shown_tap_to_chat: Optional[bool] = None
  144. has_shown_cancel_tooltip: Optional[bool] = None
  145. has_shown_send_tooltip: Optional[bool] = None
  146. has_shown_progress_bar_tooltip: Optional[bool] = None
  147. has_shown_mission_stone_modal: Optional[bool] = None
  148. has_shown_conversation_stone_modal: Optional[bool] = None
  149. has_clicked_mission_stone: Optional[bool] = None
  150. has_clicked_conversation_stone: Optional[bool] = None
  151. avatar_c_unlocked: Optional[bool] = None
  152. has_shared_avatar: Optional[bool] = None
  153. current_avatar_id: Optional[str] = None
  154. # 分享图片响应模型
  155. class ShareImageItem(BaseModel):
  156. id: str
  157. url: str
  158. title: Optional[str] = None
  159. sort_order: Optional[int] = None
  160. is_active: Optional[bool] = None
  161. SHARE_IMAGES: List[ShareImageItem] = [
  162. ShareImageItem(id="share-1", url="https://lyrria-pro.oss-cn-beijing.aliyuncs.com/share-1.png"),
  163. ShareImageItem(id="share-2", url="https://lyrria-pro.oss-cn-beijing.aliyuncs.com/share-2.png"),
  164. ]
  165. class AvatarGifSet(BaseModel):
  166. default: str
  167. positive: str
  168. negative: str
  169. talking: str
  170. AVATAR_GIF_CONFIG = {
  171. "avatar_default": AvatarGifSet(
  172. default="https://lyrria-pro.oss-cn-beijing.aliyuncs.com/01-Default.gif",
  173. positive="https://lyrria-pro.oss-cn-beijing.aliyuncs.com/04-Positive-emotion.gif",
  174. negative="https://lyrria-pro.oss-cn-beijing.aliyuncs.com/05-Negative-emotion.gif",
  175. talking="https://lyrria-pro.oss-cn-beijing.aliyuncs.com/06-talking.gif",
  176. ),
  177. "avatar_registered_cap": AvatarGifSet(
  178. default="https://lyrria-pro.oss-cn-beijing.aliyuncs.com/01-Default-Registered.gif",
  179. positive="https://lyrria-pro.oss-cn-beijing.aliyuncs.com/04-Positive-emotion-Registered.gif",
  180. negative="https://lyrria-pro.oss-cn-beijing.aliyuncs.com/05-Negative-emotion-Registered.gif",
  181. talking="https://lyrria-pro.oss-cn-beijing.aliyuncs.com/06-talking-Registered.gif",
  182. ),
  183. }
  184. @app.post("/api/users/create")
  185. async def create_user(request: CreateUserRequest):
  186. """
  187. 创建游客用户
  188. Returns:
  189. {
  190. "user_id": "uuid",
  191. "username": "",
  192. "user_type": "guest",
  193. "bot_name": "机器人名字(可选)"
  194. }
  195. """
  196. try:
  197. user = await user_service.create_guest_user(
  198. username=request.username,
  199. bot_name=request.bot_name
  200. )
  201. return {
  202. "success": True,
  203. "data": user
  204. }
  205. except Exception as e:
  206. raise HTTPException(status_code=500, detail=f"创建用户失败: {str(e)}")
  207. @app.get("/api/users/{user_id}")
  208. async def get_user(user_id: str):
  209. """
  210. 获取用户信息
  211. Args:
  212. user_id: 用户ID
  213. Returns:
  214. {
  215. "success": true,
  216. "data": {
  217. "user_id": "uuid",
  218. "username": "",
  219. "user_type": "guest",
  220. "bot_name": "机器人名字(可选)",
  221. ...
  222. }
  223. }
  224. """
  225. try:
  226. user = await user_service.get_user(user_id)
  227. if user:
  228. return {
  229. "success": True,
  230. "data": user
  231. }
  232. else:
  233. raise HTTPException(status_code=404, detail="用户不存在")
  234. except HTTPException:
  235. raise
  236. except Exception as e:
  237. raise HTTPException(status_code=500, detail=f"获取用户信息失败: {str(e)}")
  238. @app.get("/api/users/{user_id}/energy-stones/count")
  239. async def get_energy_stone_counts(user_id: str):
  240. """
  241. 获取用户的能量石数量
  242. Args:
  243. user_id: 用户ID
  244. Returns:
  245. {
  246. "success": true,
  247. "data": {
  248. "mission_count": 0,
  249. "conversation_count": 0
  250. }
  251. }
  252. """
  253. try:
  254. if not energy_stone_service.db:
  255. await energy_stone_service.set_db_manager(db_manager)
  256. counts = await energy_stone_service.get_stone_counts(user_id)
  257. return {
  258. "success": True,
  259. "data": counts
  260. }
  261. except Exception as e:
  262. raise HTTPException(status_code=500, detail=f"获取能量石数量失败: {str(e)}")
  263. @app.get("/api/users/{user_id}/avatars")
  264. async def get_user_avatars(user_id: str):
  265. """
  266. 获取用户已拥有的头像列表
  267. Args:
  268. user_id: 用户ID
  269. Returns:
  270. {
  271. "success": true,
  272. "data": {
  273. "avatar_ids": ["avatar_default", "avatar_registered_cap", ...]
  274. }
  275. }
  276. """
  277. try:
  278. async with db_manager.get_session() as session:
  279. result = await session.execute(
  280. select(UserAvatar.avatar_id).where(UserAvatar.user_id == user_id)
  281. )
  282. avatar_ids = [row[0] for row in result.fetchall()]
  283. return {
  284. "success": True,
  285. "data": {
  286. "avatar_ids": avatar_ids
  287. }
  288. }
  289. except Exception as e:
  290. raise HTTPException(status_code=500, detail=f"获取用户头像信息失败: {str(e)}")
  291. @app.get("/api/users/{user_id}/background-images")
  292. async def get_user_background_images(user_id: str):
  293. """
  294. 获取用户已拥有的背景图片列表
  295. Args:
  296. user_id: 用户ID
  297. Returns:
  298. {
  299. "success": true,
  300. "data": {
  301. "image_ids": ["default", "happynewyear", ...]
  302. }
  303. }
  304. """
  305. try:
  306. async with db_manager.get_session() as session:
  307. result = await session.execute(
  308. select(UserBackgroundImage.image_id).where(UserBackgroundImage.user_id == user_id)
  309. )
  310. image_ids = [row[0] for row in result.fetchall()]
  311. return {
  312. "success": True,
  313. "data": {
  314. "image_ids": image_ids
  315. }
  316. }
  317. except Exception as e:
  318. raise HTTPException(status_code=500, detail=f"获取用户背景图片信息失败: {str(e)}")
  319. @app.get("/api/avatar-gifs")
  320. async def get_avatar_gifs() -> Dict[str, Any]:
  321. """
  322. 获取头像 GIF 配置
  323. - 优先读取数据库 avatar_gifs 表(is_active=1,按 sort_order、id 排序)
  324. - 字段期望:code, default_url, positive_url, negative_url, talking_url
  325. - 失败/无数据时返回静态兜底配置
  326. """
  327. try:
  328. from database import db_manager
  329. async with db_manager.get_session() as session:
  330. result = await session.execute(
  331. text(
  332. """
  333. SELECT id, code, default_url, positive_url, negative_url, talking_url, is_active
  334. FROM avatar_gifs
  335. WHERE is_active = 1
  336. ORDER BY sort_order ASC, id ASC
  337. """
  338. )
  339. )
  340. rows = result.mappings().all()
  341. if rows:
  342. data: Dict[str, Dict[str, str]] = {}
  343. for row in rows:
  344. code = row.get("code") or row.get("id")
  345. if not code:
  346. continue
  347. data[str(code)] = {
  348. "default": row.get("default_url"),
  349. "positive": row.get("positive_url"),
  350. "negative": row.get("negative_url"),
  351. "talking": row.get("talking_url"),
  352. }
  353. if data:
  354. return {"success": True, "data": data}
  355. logger.warning("avatar_gifs 表无记录,返回兜底静态配置")
  356. except Exception as exc:
  357. logger.error(f"获取头像 GIF 配置失败,使用兜底: {exc}")
  358. return {"success": True, "data": {k: v.model_dump() for k, v in AVATAR_GIF_CONFIG.items()}}
  359. @app.get("/api/share-images")
  360. async def get_share_images() -> Dict[str, Any]:
  361. """
  362. 获取分享图片列表
  363. - 优先从数据库 share_images 表读取(is_active=1,按 sort_order、id 排序)
  364. - 失败或无数据时返回兜底静态列表
  365. """
  366. try:
  367. from database import db_manager
  368. async with db_manager.get_session() as session:
  369. result = await session.execute(
  370. text(
  371. """
  372. SELECT id, code, url, title, sort_order, is_active
  373. FROM share_images
  374. WHERE is_active = 1
  375. ORDER BY sort_order ASC, id ASC
  376. """
  377. )
  378. )
  379. rows = result.mappings().all()
  380. if rows:
  381. data = [
  382. {
  383. "id": str(row.get("code") or row.get("id")),
  384. "url": row.get("url"),
  385. "title": row.get("title"),
  386. "sort_order": row.get("sort_order"),
  387. "is_active": bool(row.get("is_active")) if row.get("is_active") is not None else None,
  388. }
  389. for row in rows
  390. ]
  391. return {"success": True, "data": data}
  392. logger.warning("share_images 表无记录,返回兜底静态列表")
  393. except Exception as exc:
  394. logger.error(f"获取分享图片失败,使用兜底: {exc}")
  395. return {"success": True, "data": [item.model_dump() for item in SHARE_IMAGES]}
  396. @app.put("/api/users/{user_id}/username")
  397. async def update_username(user_id: str, request: UpdateUsernameRequest):
  398. """
  399. 更新用户名
  400. Args:
  401. user_id: 用户ID
  402. request: 更新请求(包含username)
  403. Returns:
  404. {
  405. "success": true,
  406. "message": "更新成功"
  407. }
  408. """
  409. try:
  410. success = await user_service.update_username(user_id, request.username)
  411. if success:
  412. return {
  413. "success": True,
  414. "message": "用户名更新成功"
  415. }
  416. else:
  417. raise HTTPException(status_code=500, detail="更新用户名失败")
  418. except Exception as e:
  419. raise HTTPException(status_code=500, detail=f"更新用户名失败: {str(e)}")
  420. @app.put("/api/users/{user_id}/bot-name")
  421. async def update_bot_name(user_id: str, request: UpdateBotNameRequest):
  422. """
  423. 更新用户给机器人取的名字
  424. Args:
  425. user_id: 用户ID
  426. request: 更新请求(包含bot_name)
  427. Returns:
  428. {
  429. "success": true,
  430. "message": "更新成功"
  431. }
  432. """
  433. import asyncio
  434. import time
  435. start_time = time.time()
  436. try:
  437. # 🔥 添加超时保护(5秒),避免长时间等待
  438. success = await asyncio.wait_for(
  439. user_service.update_bot_name(user_id, request.bot_name),
  440. timeout=5.0
  441. )
  442. duration = (time.time() - start_time) * 1000
  443. if success:
  444. return {
  445. "success": True,
  446. "message": "机器人名字更新成功"
  447. }
  448. else:
  449. raise HTTPException(
  450. status_code=500,
  451. detail=f"更新机器人名字失败(耗时: {duration:.1f}ms)"
  452. )
  453. except asyncio.TimeoutError:
  454. duration = (time.time() - start_time) * 1000
  455. logger.error(f"❌ 更新机器人名字超时: user_id={user_id}, 耗时={duration:.1f}ms")
  456. raise HTTPException(
  457. status_code=504,
  458. detail=f"更新机器人名字超时(耗时: {duration:.1f}ms),请稍后重试"
  459. )
  460. except HTTPException:
  461. raise
  462. except Exception as e:
  463. duration = (time.time() - start_time) * 1000
  464. logger.error(f"❌ 更新机器人名字异常: user_id={user_id}, error={e}, 耗时={duration:.1f}ms")
  465. raise HTTPException(
  466. status_code=500,
  467. detail=f"更新机器人名字失败: {str(e)}(耗时: {duration:.1f}ms)"
  468. )
  469. @app.post("/api/auth/register/phone")
  470. async def register_with_phone(request: RegisterPhoneRequest):
  471. """
  472. 手机号注册
  473. Returns:
  474. {
  475. "success": true,
  476. "data": {
  477. "user_id": "uuid",
  478. "username": "",
  479. "phone": "",
  480. "user_type": "registered",
  481. ...
  482. }
  483. }
  484. """
  485. try:
  486. user = await user_service.register_with_phone(
  487. phone=request.phone,
  488. username=request.username
  489. )
  490. return {
  491. "success": True,
  492. "data": user
  493. }
  494. except ValueError as e:
  495. raise HTTPException(status_code=400, detail=str(e))
  496. except Exception as e:
  497. raise HTTPException(status_code=500, detail=f"注册失败: {str(e)}")
  498. @app.post("/api/auth/register/email")
  499. async def register_with_email(request: RegisterEmailRequest):
  500. """
  501. 邮箱注册
  502. Returns:
  503. {
  504. "success": true,
  505. "data": {
  506. "user_id": "uuid",
  507. "username": "",
  508. "email": "",
  509. "user_type": "registered",
  510. ...
  511. }
  512. }
  513. """
  514. try:
  515. user = await user_service.register_with_email(
  516. email=request.email,
  517. username=request.username
  518. )
  519. return {
  520. "success": True,
  521. "data": user
  522. }
  523. except ValueError as e:
  524. raise HTTPException(status_code=400, detail=str(e))
  525. except Exception as e:
  526. raise HTTPException(status_code=500, detail=f"注册失败: {str(e)}")
  527. @app.post("/api/auth/login/wechat")
  528. async def login_with_wechat(request: LoginWechatRequest):
  529. """
  530. 微信登录/注册
  531. Returns:
  532. {
  533. "success": true,
  534. "data": {
  535. "user_id": "uuid",
  536. "username": "",
  537. "wechat_openid": "",
  538. "user_type": "registered",
  539. ...
  540. }
  541. }
  542. """
  543. try:
  544. user = await user_service.login_with_wechat(
  545. openid=request.openid,
  546. username=request.username
  547. )
  548. return {
  549. "success": True,
  550. "data": user
  551. }
  552. except Exception as e:
  553. raise HTTPException(status_code=500, detail=f"登录失败: {str(e)}")
  554. @app.post("/api/auth/login/qq")
  555. async def login_with_qq(request: LoginQQRequest):
  556. """
  557. QQ登录/注册
  558. Returns:
  559. {
  560. "success": true,
  561. "data": {
  562. "user_id": "uuid",
  563. "username": "",
  564. "qq_openid": "",
  565. "user_type": "registered",
  566. ...
  567. }
  568. }
  569. """
  570. try:
  571. user = await user_service.login_with_qq(
  572. openid=request.openid,
  573. username=request.username
  574. )
  575. return {
  576. "success": True,
  577. "data": user
  578. }
  579. except Exception as e:
  580. raise HTTPException(status_code=500, detail=f"登录失败: {str(e)}")
  581. @app.post("/api/auth/login/phone")
  582. async def login_with_phone(request: LoginPhoneRequest):
  583. """
  584. 手机号登录
  585. Returns:
  586. {
  587. "success": true,
  588. "data": {
  589. "user_id": "uuid",
  590. "phone": "",
  591. ...
  592. }
  593. }
  594. """
  595. try:
  596. user = await user_service.login_with_phone(request.phone)
  597. if user:
  598. return {
  599. "success": True,
  600. "data": user
  601. }
  602. else:
  603. raise HTTPException(status_code=404, detail="用户不存在,请先注册")
  604. except HTTPException:
  605. raise
  606. except Exception as e:
  607. raise HTTPException(status_code=500, detail=f"登录失败: {str(e)}")
  608. @app.post("/api/auth/login/email")
  609. async def login_with_email(request: LoginEmailRequest):
  610. """
  611. 邮箱登录
  612. Returns:
  613. {
  614. "success": true,
  615. "data": {
  616. "user_id": "uuid",
  617. "email": "",
  618. ...
  619. }
  620. }
  621. """
  622. try:
  623. user = await user_service.login_with_email(request.email)
  624. if user:
  625. return {
  626. "success": True,
  627. "data": user
  628. }
  629. else:
  630. raise HTTPException(status_code=404, detail="用户不存在,请先注册")
  631. except HTTPException:
  632. raise
  633. except Exception as e:
  634. raise HTTPException(status_code=500, detail=f"登录失败: {str(e)}")
  635. @app.post("/api/auth/send-code")
  636. async def send_code(request: SendCodeRequest):
  637. """
  638. 生成并保存验证码
  639. 当用户点击 send code 按钮时调用此接口,生成临时验证码并保存到数据库
  640. Args:
  641. request: 发送验证码请求
  642. - phone: 手机号(发送类型为短信时必填)
  643. - email: 邮箱(发送类型为邮箱时必填)
  644. - send_type: 发送类型 ("domestic_sms", "international_sms", "email")
  645. Returns:
  646. {
  647. "success": true,
  648. "data": {
  649. "code_id": "验证码ID",
  650. "message": "验证码已生成,等待发送"
  651. }
  652. }
  653. """
  654. try:
  655. # 验证参数
  656. if request.send_type in ["domestic_sms", "international_sms"]:
  657. if not request.phone:
  658. raise HTTPException(status_code=400, detail="手机号不能为空")
  659. elif request.send_type == "email":
  660. if not request.email:
  661. raise HTTPException(status_code=400, detail="邮箱不能为空")
  662. else:
  663. raise HTTPException(status_code=400, detail="无效的发送类型")
  664. # 生成4位数字验证码
  665. code = str(random.randint(1000, 9999))
  666. code_id = str(uuid.uuid4())
  667. # 保存到数据库
  668. async with db_manager.get_session() as session:
  669. # 🔥 将同一手机号/邮箱的旧验证码标记为无效
  670. # 确保同一时间只有一个有效的验证码
  671. if request.send_type in ["domestic_sms", "international_sms"]:
  672. # 将同一手机号的旧验证码标记为无效
  673. await session.execute(
  674. update(VerificationCode)
  675. .where(
  676. VerificationCode.phone == request.phone,
  677. VerificationCode.send_type == request.send_type,
  678. VerificationCode.is_valid == True
  679. )
  680. .values(is_valid=False)
  681. )
  682. elif request.send_type == "email":
  683. # 将同一邮箱的旧验证码标记为无效
  684. await session.execute(
  685. update(VerificationCode)
  686. .where(
  687. VerificationCode.email == request.email,
  688. VerificationCode.send_type == request.send_type,
  689. VerificationCode.is_valid == True
  690. )
  691. .values(is_valid=False)
  692. )
  693. # 创建新的验证码
  694. verification_code = VerificationCode(
  695. code_id=code_id,
  696. phone=request.phone,
  697. email=request.email,
  698. code=code,
  699. send_type=request.send_type,
  700. is_sent=False,
  701. is_valid=True,
  702. )
  703. session.add(verification_code)
  704. await session.commit()
  705. logger.info(f"✅ 验证码已生成: code_id={code_id}, send_type={request.send_type}, phone={request.phone}, email={request.email}")
  706. return {
  707. "success": True,
  708. "data": {
  709. "code_id": code_id,
  710. "message": "验证码已生成,等待发送"
  711. }
  712. }
  713. except HTTPException:
  714. raise
  715. except Exception as e:
  716. logger.error(f"❌ 生成验证码失败: {e}")
  717. raise HTTPException(status_code=500, detail=f"生成验证码失败: {str(e)}")
  718. @app.post("/api/auth/verify-code")
  719. async def verify_code(request: VerifyCodeRequest):
  720. """
  721. 验证验证码
  722. Args:
  723. request: 验证验证码请求
  724. - phone: 手机号(发送类型为短信时必填)
  725. - email: 邮箱(发送类型为邮箱时必填)
  726. - code: 验证码
  727. - send_type: 发送类型 ("domestic_sms", "international_sms", "email")
  728. Returns:
  729. {
  730. "success": true,
  731. "data": {
  732. "valid": true/false,
  733. "message": "验证成功" 或 "验证码错误"
  734. }
  735. }
  736. """
  737. try:
  738. # 验证参数
  739. if request.send_type in ["domestic_sms", "international_sms"]:
  740. if not request.phone:
  741. raise HTTPException(status_code=400, detail="手机号不能为空")
  742. elif request.send_type == "email":
  743. if not request.email:
  744. raise HTTPException(status_code=400, detail="邮箱不能为空")
  745. else:
  746. raise HTTPException(status_code=400, detail="无效的发送类型")
  747. if not request.code:
  748. raise HTTPException(status_code=400, detail="验证码不能为空")
  749. # 查询有效的验证码
  750. async with db_manager.get_session() as session:
  751. if request.send_type in ["domestic_sms", "international_sms"]:
  752. result = await session.execute(
  753. select(VerificationCode)
  754. .where(
  755. VerificationCode.phone == request.phone,
  756. VerificationCode.send_type == request.send_type,
  757. VerificationCode.code == request.code,
  758. VerificationCode.is_valid == True
  759. )
  760. .order_by(VerificationCode.created_at.desc())
  761. .limit(1)
  762. )
  763. else:
  764. result = await session.execute(
  765. select(VerificationCode)
  766. .where(
  767. VerificationCode.email == request.email,
  768. VerificationCode.send_type == request.send_type,
  769. VerificationCode.code == request.code,
  770. VerificationCode.is_valid == True
  771. )
  772. .order_by(VerificationCode.created_at.desc())
  773. .limit(1)
  774. )
  775. verification_code = result.scalar_one_or_none()
  776. if verification_code:
  777. # 验证成功,将验证码标记为无效(一次性使用)
  778. verification_code.is_valid = False
  779. await session.commit()
  780. logger.info(f"✅ 验证码验证成功: phone={request.phone}, email={request.email}, send_type={request.send_type}")
  781. return {
  782. "success": True,
  783. "data": {
  784. "valid": True,
  785. "message": "验证成功"
  786. }
  787. }
  788. else:
  789. logger.warning(f"❌ 验证码验证失败: phone={request.phone}, email={request.email}, code={request.code}")
  790. return {
  791. "success": True,
  792. "data": {
  793. "valid": False,
  794. "message": "验证码错误或已失效"
  795. }
  796. }
  797. except HTTPException:
  798. raise
  799. except Exception as e:
  800. logger.error(f"❌ 验证验证码失败: {e}")
  801. raise HTTPException(status_code=500, detail=f"验证验证码失败: {str(e)}")
  802. @app.get("/api/users/{user_id}/ui-state")
  803. async def get_user_ui_state(user_id: str) -> Dict[str, Any]:
  804. """
  805. 获取用户前端可视化状态(页面、路由、提示状态)
  806. """
  807. try:
  808. state = await user_service.get_user_ui_state(user_id)
  809. if not state:
  810. # 没有记录时返回默认值
  811. default_state = UserUIStateResponse(
  812. user_id=user_id,
  813. current_page=None,
  814. visited_routes=[],
  815. )
  816. return {"success": True, "data": default_state.dict()}
  817. return {"success": True, "data": state}
  818. except Exception as e: # noqa: BLE001
  819. raise HTTPException(status_code=500, detail=f"获取用户UI状态失败: {str(e)}") from e
  820. @app.put("/api/users/{user_id}/ui-state")
  821. async def update_user_ui_state(
  822. user_id: str,
  823. request: UpdateUserUIStateRequest,
  824. ) -> Dict[str, Any]:
  825. """
  826. 更新用户前端可视化状态(部分字段)
  827. """
  828. try:
  829. updated = await user_service.upsert_user_ui_state(
  830. user_id=user_id,
  831. current_page=request.current_page,
  832. visited_routes=request.visited_routes,
  833. has_shown_tap_to_chat=request.has_shown_tap_to_chat,
  834. has_shown_cancel_tooltip=request.has_shown_cancel_tooltip,
  835. has_shown_send_tooltip=request.has_shown_send_tooltip,
  836. has_shown_progress_bar_tooltip=request.has_shown_progress_bar_tooltip,
  837. has_shown_mission_stone_modal=request.has_shown_mission_stone_modal,
  838. has_shown_conversation_stone_modal=request.has_shown_conversation_stone_modal,
  839. has_clicked_mission_stone=request.has_clicked_mission_stone,
  840. has_clicked_conversation_stone=request.has_clicked_conversation_stone,
  841. avatar_c_unlocked=request.avatar_c_unlocked,
  842. current_avatar_id=request.current_avatar_id,
  843. )
  844. if not updated:
  845. raise HTTPException(status_code=500, detail="更新用户UI状态失败")
  846. return {"success": True, "data": updated}
  847. except HTTPException:
  848. raise
  849. except Exception as e: # noqa: BLE001
  850. raise HTTPException(status_code=500, detail=f"更新用户UI状态失败: {str(e)}") from e
  851. # ==================== OAuth 授权端点 ====================
  852. @app.get("/api/auth/wechat/authorize")
  853. async def wechat_authorize(request: Request):
  854. """
  855. 获取微信授权URL
  856. 根据微信官方文档:https://developers.weixin.qq.com/doc/oplatform/Website_App/WeChat_Login/Wechat_Login.html
  857. - 使用 qrconnect 端点进行网站应用授权
  858. - scope 必须为 snsapi_login
  859. - redirect_uri 需要 urlEncode 处理
  860. - state 用于防止 CSRF 攻击
  861. Returns:
  862. 重定向到微信授权页面
  863. """
  864. if not settings.WECHAT_OPEN_APP_ID or not settings.WECHAT_REDIRECT_URI:
  865. raise HTTPException(status_code=500, detail="微信OAuth未配置,请检查环境变量 WECHAT_OPEN_APP_ID 和 WECHAT_REDIRECT_URI")
  866. # 生成state参数(用于防止CSRF攻击)
  867. import secrets
  868. state = secrets.token_urlsafe(32)
  869. # 将 state 存储到 Redis(如果可用),用于回调时验证
  870. # TTL 设置为 10 分钟(code 的有效期也是 10 分钟)
  871. redis_cache = getattr(request.app.state, 'redis_cache', None)
  872. if redis_cache:
  873. try:
  874. redis_client = await redis_cache._get_redis()
  875. await redis_client.setex(f"oauth:wechat:state:{state}", 600, "1") # 10分钟过期
  876. except Exception as e:
  877. logger.warning(f"存储 state 到 Redis 失败,将跳过验证: {e}")
  878. # 构建微信授权URL(使用开放平台的AppID)
  879. # redirect_uri 会自动通过 urlencode 进行编码
  880. params = {
  881. "appid": settings.WECHAT_OPEN_APP_ID, # 使用开放平台的AppID
  882. "redirect_uri": settings.WECHAT_REDIRECT_URI, # urlencode 会自动编码
  883. "response_type": "code",
  884. "scope": "snsapi_login", # 网站应用必须使用 snsapi_login
  885. "state": state,
  886. }
  887. # 根据官方文档格式:https://open.weixin.qq.com/connect/qrconnect?appid=APPID&redirect_uri=REDIRECT_URI&response_type=code&scope=SCOPE&state=STATE#wechat_redirect
  888. auth_url = f"https://open.weixin.qq.com/connect/qrconnect?{urllib.parse.urlencode(params)}#wechat_redirect"
  889. return RedirectResponse(url=auth_url)
  890. @app.get("/api/auth/wechat/callback")
  891. async def wechat_callback(
  892. request: Request,
  893. code: str = Query(..., description="微信授权码"),
  894. state: str = Query(None, description="状态参数")
  895. ):
  896. """
  897. 微信OAuth回调处理
  898. 根据微信官方文档流程:
  899. 1. 验证 state 参数(防止 CSRF 攻击)
  900. 2. 通过 code 换取 access_token
  901. 3. 使用 openid 进行登录/注册
  902. Args:
  903. code: 微信返回的授权码(有效期10分钟,只能使用一次)
  904. state: 状态参数(用于验证请求合法性)
  905. Returns:
  906. 重定向到前端页面(带token)
  907. """
  908. try:
  909. if not settings.WECHAT_OPEN_APP_ID or not settings.WECHAT_OPEN_APP_SECRET:
  910. raise HTTPException(status_code=500, detail="微信OAuth未配置,请检查环境变量 WECHAT_OPEN_APP_ID 和 WECHAT_OPEN_APP_SECRET")
  911. # 1. 验证 state 参数(防止 CSRF 攻击)
  912. if state:
  913. redis_cache = getattr(request.app.state, 'redis_cache', None)
  914. if redis_cache:
  915. try:
  916. redis_client = await redis_cache._get_redis()
  917. state_key = f"oauth:wechat:state:{state}"
  918. exists = await redis_client.exists(state_key)
  919. if not exists:
  920. logger.warning(f"State 验证失败: {state} (可能已过期或被篡改)")
  921. return RedirectResponse(
  922. url=f"{settings.FRONTEND_URL}/register?error=wechat_auth_failed&reason=invalid_state"
  923. )
  924. # 验证成功后删除 state(一次性使用)
  925. await redis_client.delete(state_key)
  926. except Exception as e:
  927. logger.warning(f"State 验证异常: {e},继续处理")
  928. # 2. 通过code获取access_token(使用开放平台的AppID和AppSecret)
  929. # 根据官方文档:https://api.weixin.qq.com/sns/oauth2/access_token
  930. token_url = "https://api.weixin.qq.com/sns/oauth2/access_token"
  931. token_params = {
  932. "appid": settings.WECHAT_OPEN_APP_ID, # 使用开放平台的AppID
  933. "secret": settings.WECHAT_OPEN_APP_SECRET, # 使用开放平台的AppSecret
  934. "code": code,
  935. "grant_type": "authorization_code",
  936. }
  937. async with httpx.AsyncClient() as client:
  938. token_response = await client.get(token_url, params=token_params)
  939. token_data = token_response.json()
  940. if "errcode" in token_data:
  941. logger.error(f"微信获取access_token失败: {token_data}")
  942. return RedirectResponse(
  943. url=f"{settings.FRONTEND_URL}/register?error=wechat_auth_failed"
  944. )
  945. access_token = token_data.get("access_token")
  946. openid = token_data.get("openid")
  947. # refresh_token 可用于刷新 access_token,当前实现暂不保存
  948. if not access_token or not openid:
  949. logger.error(f"微信返回数据不完整: {token_data}")
  950. return RedirectResponse(
  951. url=f"{settings.FRONTEND_URL}/register?error=wechat_auth_failed"
  952. )
  953. # 3. 使用openid登录/注册
  954. # 注意:当前使用 snsapi_login scope,只能获取 openid
  955. # 如需获取用户昵称、头像等信息,需要:
  956. # - 将 scope 改为 snsapi_login,snsapi_userinfo
  957. # - 调用 /sns/userinfo 接口获取用户信息
  958. user = await user_service.login_with_wechat(openid=openid)
  959. # 4. 重定向到前端,携带用户信息
  960. frontend_url = f"{settings.FRONTEND_URL}/register?oauth=wechat&user_id={user['user_id']}&username={urllib.parse.quote(user['username'])}"
  961. return RedirectResponse(url=frontend_url)
  962. except Exception as e:
  963. logger.error(f"微信OAuth回调处理失败: {e}")
  964. return RedirectResponse(
  965. url=f"{settings.FRONTEND_URL}/register?error=wechat_auth_failed"
  966. )
  967. @app.get("/api/auth/qq/authorize")
  968. async def qq_authorize(request: Request):
  969. """
  970. 获取QQ授权URL
  971. Returns:
  972. 重定向到QQ授权页面
  973. """
  974. if not settings.QQ_APP_ID or not settings.QQ_REDIRECT_URI:
  975. raise HTTPException(status_code=500, detail="QQ OAuth未配置,请检查环境变量")
  976. # 生成state参数(用于防止CSRF攻击)
  977. import secrets
  978. state = secrets.token_urlsafe(32)
  979. # 将 state 存储到 Redis 以便回调校验
  980. redis_cache = getattr(request.app.state, 'redis_cache', None)
  981. if redis_cache:
  982. try:
  983. redis_client = await redis_cache._get_redis()
  984. await redis_client.setex(f"oauth:qq:state:{state}", 600, "1")
  985. except Exception as e:
  986. logger.warning(f"存储 QQ state 失败,将跳过校验: {e}")
  987. # 确保回调地址格式正确(去除末尾斜杠,确保与QQ应用配置完全一致)
  988. redirect_uri = settings.QQ_REDIRECT_URI.rstrip('/')
  989. # 构建QQ授权URL
  990. # 注意:QQ开放平台要求redirect_uri必须与配置的回调域完全匹配
  991. params = {
  992. "response_type": "code",
  993. "client_id": settings.QQ_APP_ID,
  994. "redirect_uri": redirect_uri, # 使用处理后的回调地址
  995. "state": state,
  996. "scope": "get_user_info", # QQ授权范围
  997. }
  998. # URL编码参数(确保特殊字符正确编码)
  999. auth_url = f"https://graph.qq.com/oauth2.0/authorize?{urllib.parse.urlencode(params)}"
  1000. # 调试日志:记录实际发送的授权URL(隐藏敏感信息)
  1001. logger.info(f"🔍 QQ授权URL生成 - AppID: {settings.QQ_APP_ID[:8]}..., 回调地址: {redirect_uri}")
  1002. return RedirectResponse(url=auth_url)
  1003. @app.get("/api/qq/mp/verify")
  1004. async def qq_callback(
  1005. request: Request,
  1006. code: str = Query(..., description="QQ授权码"),
  1007. state: str = Query(None, description="状态参数")
  1008. ):
  1009. """
  1010. QQ OAuth回调处理
  1011. Args:
  1012. code: QQ返回的授权码
  1013. state: 状态参数
  1014. Returns:
  1015. 重定向到前端页面(带token)
  1016. """
  1017. try:
  1018. # 0. 验证 state 防止 CSRF
  1019. if state:
  1020. redis_cache = getattr(request.app.state, 'redis_cache', None)
  1021. if redis_cache:
  1022. try:
  1023. redis_client = await redis_cache._get_redis()
  1024. state_key = f"oauth:qq:state:{state}"
  1025. exists = await redis_client.exists(state_key)
  1026. if not exists:
  1027. logger.warning(f"QQ state 验证失败: {state}")
  1028. return RedirectResponse(
  1029. url=f"{settings.FRONTEND_URL}/register?error=qq_auth_failed&reason=invalid_state"
  1030. )
  1031. await redis_client.delete(state_key)
  1032. except Exception as state_error:
  1033. logger.warning(f"QQ state 验证异常: {state_error},继续流程")
  1034. if not settings.QQ_APP_ID or not settings.QQ_APP_KEY:
  1035. raise HTTPException(status_code=500, detail="QQ OAuth未配置")
  1036. # 小工具:解析 QQ JSONP 或 querystring 响应
  1037. def parse_qq_payload(payload_text: str) -> dict:
  1038. cleaned = (payload_text or "").strip()
  1039. if not cleaned:
  1040. return {}
  1041. if cleaned.startswith("callback("):
  1042. cleaned = cleaned.split("callback(", 1)[1]
  1043. cleaned = cleaned.rsplit(");", 1)[0]
  1044. try:
  1045. return json.loads(cleaned)
  1046. except json.JSONDecodeError:
  1047. parsed = {}
  1048. for part in cleaned.split("&"):
  1049. if "=" in part:
  1050. key, value = part.split("=", 1)
  1051. parsed[key] = value
  1052. return parsed
  1053. # 1. 通过code获取access_token
  1054. # 确保回调地址格式正确(与授权时保持一致)
  1055. redirect_uri = settings.QQ_REDIRECT_URI.rstrip('/')
  1056. token_url = "https://graph.qq.com/oauth2.0/token"
  1057. token_params = {
  1058. "grant_type": "authorization_code",
  1059. "client_id": settings.QQ_APP_ID,
  1060. "client_secret": settings.QQ_APP_KEY,
  1061. "code": code,
  1062. "redirect_uri": redirect_uri, # 使用处理后的回调地址
  1063. "fmt": "json",
  1064. }
  1065. logger.info(f"🔍 QQ获取token - 回调地址: {redirect_uri}")
  1066. async with httpx.AsyncClient() as client:
  1067. token_response = await client.get(token_url, params=token_params)
  1068. token_text = token_response.text
  1069. try:
  1070. token_data = token_response.json()
  1071. except ValueError:
  1072. token_data = parse_qq_payload(token_text)
  1073. if "error" in token_data:
  1074. logger.error(f"QQ获取access_token失败: {token_data}")
  1075. return RedirectResponse(
  1076. url=f"{settings.FRONTEND_URL}/register?error=qq_auth_failed&reason=token"
  1077. )
  1078. access_token = token_data.get("access_token")
  1079. if not access_token:
  1080. logger.error(f"QQ返回数据不完整: {token_data}")
  1081. return RedirectResponse(
  1082. url=f"{settings.FRONTEND_URL}/register?error=qq_auth_failed"
  1083. )
  1084. # 2. 通过access_token获取openid
  1085. openid_url = "https://graph.qq.com/oauth2.0/me"
  1086. openid_params = {
  1087. "access_token": access_token,
  1088. "unionid": 1,
  1089. "fmt": "json",
  1090. }
  1091. async with httpx.AsyncClient() as client:
  1092. openid_response = await client.get(openid_url, params=openid_params)
  1093. openid_data = parse_qq_payload(openid_response.text)
  1094. if "error" in openid_data:
  1095. logger.error(f"QQ获取openid失败: {openid_data}")
  1096. return RedirectResponse(
  1097. url=f"{settings.FRONTEND_URL}/register?error=qq_auth_failed"
  1098. )
  1099. openid = openid_data.get("openid")
  1100. unionid = openid_data.get("unionid")
  1101. if not openid:
  1102. logger.error(f"QQ返回openid为空: {openid_data}")
  1103. return RedirectResponse(
  1104. url=f"{settings.FRONTEND_URL}/register?error=qq_auth_failed"
  1105. )
  1106. # 2.1 可选:拉取用户信息用于填充昵称
  1107. nickname = ""
  1108. user_info = {}
  1109. user_info_url = "https://graph.qq.com/user/get_user_info"
  1110. user_info_params = {
  1111. "access_token": access_token,
  1112. "oauth_consumer_key": settings.QQ_APP_ID,
  1113. "openid": openid,
  1114. "fmt": "json",
  1115. }
  1116. try:
  1117. async with httpx.AsyncClient() as client:
  1118. user_info_resp = await client.get(
  1119. user_info_url,
  1120. params=user_info_params,
  1121. )
  1122. user_info = user_info_resp.json()
  1123. if user_info.get("ret") == 0:
  1124. nickname = user_info.get("nickname", "") or ""
  1125. else:
  1126. logger.warning(
  1127. f"QQ用户信息获取失败: {user_info}"
  1128. )
  1129. except Exception as info_error:
  1130. logger.warning(f"拉取QQ用户信息异常: {info_error}")
  1131. # 3. 使用openid登录/注册
  1132. user = await user_service.login_with_qq(
  1133. openid=openid,
  1134. username=nickname,
  1135. unionid=unionid,
  1136. )
  1137. # 4. 重定向到前端,携带用户信息
  1138. frontend_url = f"{settings.FRONTEND_URL}/register?oauth=qq&user_id={user['user_id']}&username={urllib.parse.quote(user['username'])}"
  1139. return RedirectResponse(url=frontend_url)
  1140. except Exception as e:
  1141. logger.error(f"QQ OAuth回调处理失败: {e}")
  1142. return RedirectResponse(
  1143. url=f"{settings.FRONTEND_URL}/register?error=qq_auth_failed"
  1144. )
  1145. @app.get("/api/wechat/jsapi/config")
  1146. async def get_wechat_jsapi_config(
  1147. url: str = Query(..., description="当前页面URL(用于生成签名)")
  1148. ):
  1149. """
  1150. 获取微信JS-SDK配置(包含签名)
  1151. Args:
  1152. url: 当前页面完整URL(前端传入,用于生成签名)
  1153. Returns:
  1154. {
  1155. "success": true,
  1156. "data": {
  1157. "appId": "微信AppID",
  1158. "timestamp": 1234567890,
  1159. "nonceStr": "随机字符串",
  1160. "signature": "SHA1签名"
  1161. }
  1162. }
  1163. """
  1164. try:
  1165. if not settings.WECHAT_MP_APP_ID:
  1166. logger.error("微信公众平台AppID未配置,请检查环境变量 WECHAT_MP_APP_ID")
  1167. raise HTTPException(status_code=500, detail="微信公众平台AppID未配置,请检查环境变量 WECHAT_MP_APP_ID")
  1168. # 记录接收到的URL(用于调试)
  1169. logger.info(f"📥 收到微信JS-SDK配置请求: url={url[:100]}...")
  1170. # 获取Redis客户端(用于缓存jsapi_ticket)
  1171. from database import db_manager
  1172. redis_getter = db_manager.get_redis
  1173. # 创建微信JS-SDK服务
  1174. jsapi_service = WechatJSAPIService(redis_getter=redis_getter)
  1175. # 获取配置
  1176. config = await jsapi_service.get_jsapi_config(url)
  1177. if not config:
  1178. logger.error("获取微信JS-SDK配置失败(返回None)")
  1179. raise HTTPException(status_code=500, detail="获取微信JS-SDK配置失败")
  1180. logger.info(f"✅ 微信JS-SDK配置生成成功: appId={config.get('appId', 'N/A')[:10]}..., timestamp={config.get('timestamp')}")
  1181. return {
  1182. "success": True,
  1183. "data": config
  1184. }
  1185. except HTTPException:
  1186. raise
  1187. except Exception as e:
  1188. logger.error(f"获取微信JS-SDK配置异常: {e}", exc_info=True)
  1189. raise HTTPException(status_code=500, detail=f"获取配置失败: {str(e)}")
  1190. @app.get("/api/qq/share/config")
  1191. async def get_qq_share_config():
  1192. """
  1193. 获取QQ分享配置(包含AppID和应用名称)
  1194. Returns:
  1195. {
  1196. "success": true,
  1197. "data": {
  1198. "appId": "QQ AppID",
  1199. "appName": "应用名称"
  1200. }
  1201. }
  1202. """
  1203. try:
  1204. if not settings.QQ_APP_ID:
  1205. logger.warning("QQ AppID未配置,返回空配置")
  1206. return {
  1207. "success": True,
  1208. "data": {
  1209. "appId": "",
  1210. "appName": settings.QQ_APP_NAME
  1211. }
  1212. }
  1213. logger.info(f"📥 收到QQ分享配置请求: appId={settings.QQ_APP_ID[:8]}..., appName={settings.QQ_APP_NAME}")
  1214. return {
  1215. "success": True,
  1216. "data": {
  1217. "appId": settings.QQ_APP_ID,
  1218. "appName": settings.QQ_APP_NAME
  1219. }
  1220. }
  1221. except Exception as e:
  1222. logger.error(f"获取QQ分享配置异常: {e}", exc_info=True)
  1223. raise HTTPException(status_code=500, detail=f"获取配置失败: {str(e)}")
  1224. # ==================== 微信公众平台服务器配置 ====================
  1225. @app.get("/api/wechat/mp/verify")
  1226. async def wechat_mp_verify(
  1227. signature: str = Query(..., description="微信加密签名"),
  1228. timestamp: str = Query(..., description="时间戳"),
  1229. nonce: str = Query(..., description="随机数"),
  1230. echostr: str = Query(..., description="随机字符串")
  1231. ):
  1232. """
  1233. 微信公众平台服务器验证接口(GET)
  1234. 用于微信公众平台配置服务器URL时的验证
  1235. Args:
  1236. signature: 微信加密签名
  1237. timestamp: 时间戳
  1238. nonce: 随机数
  1239. echostr: 随机字符串
  1240. Returns:
  1241. 验证成功返回echostr,失败返回空字符串
  1242. """
  1243. if not settings.WECHAT_MP_TOKEN:
  1244. logger.error("微信公众平台Token未配置")
  1245. return PlainTextResponse("", status_code=500)
  1246. try:
  1247. # 1. 将token、timestamp、nonce按字典序排序
  1248. token = settings.WECHAT_MP_TOKEN
  1249. tmp_arr = [token, timestamp, nonce]
  1250. tmp_arr.sort()
  1251. # 2. 拼接字符串并SHA1加密
  1252. tmp_str = ''.join(tmp_arr)
  1253. tmp_str = hashlib.sha1(tmp_str.encode('utf-8')).hexdigest()
  1254. # 3. 与signature比较
  1255. if tmp_str == signature:
  1256. logger.info("✅ 微信公众平台服务器验证成功")
  1257. return PlainTextResponse(echostr)
  1258. else:
  1259. logger.warning(f"⚠️ 微信公众平台服务器验证失败: signature不匹配")
  1260. return PlainTextResponse("", status_code=403)
  1261. except Exception as e:
  1262. logger.error(f"❌ 微信公众平台服务器验证异常: {e}")
  1263. return PlainTextResponse("", status_code=500)
  1264. @app.post("/api/wechat/mp/message")
  1265. async def wechat_mp_message(request: Request):
  1266. """
  1267. 微信公众平台消息推送接口(POST)
  1268. 接收微信公众平台推送的消息(文本、事件等)
  1269. Returns:
  1270. XML格式的响应消息
  1271. """
  1272. if not settings.WECHAT_MP_TOKEN:
  1273. logger.error("微信公众平台Token未配置")
  1274. return PlainTextResponse("", status_code=500)
  1275. try:
  1276. # 1. 获取请求参数(用于验证签名)
  1277. signature = request.query_params.get("signature", "")
  1278. timestamp = request.query_params.get("timestamp", "")
  1279. nonce = request.query_params.get("nonce", "")
  1280. # 2. 验证签名
  1281. token = settings.WECHAT_MP_TOKEN
  1282. tmp_arr = [token, timestamp, nonce]
  1283. tmp_arr.sort()
  1284. tmp_str = ''.join(tmp_arr)
  1285. tmp_str = hashlib.sha1(tmp_str.encode('utf-8')).hexdigest()
  1286. if tmp_str != signature:
  1287. logger.warning(f"⚠️ 微信公众平台消息签名验证失败")
  1288. return PlainTextResponse("", status_code=403)
  1289. # 3. 读取XML消息体
  1290. body = await request.body()
  1291. xml_data = body.decode('utf-8')
  1292. # 4. 解析XML
  1293. root = ET.fromstring(xml_data)
  1294. msg_type = root.find('MsgType').text if root.find('MsgType') is not None else ''
  1295. from_user = root.find('FromUserName').text if root.find('FromUserName') is not None else ''
  1296. to_user = root.find('ToUserName').text if root.find('ToUserName') is not None else ''
  1297. logger.info(f"📨 收到微信公众平台消息: type={msg_type}, from={from_user}")
  1298. # 5. 处理不同类型的消息
  1299. if msg_type == 'text':
  1300. # 文本消息
  1301. content = root.find('Content').text if root.find('Content') is not None else ''
  1302. logger.info(f"📝 文本消息内容: {content}")
  1303. # 回复文本消息(示例:自动回复)
  1304. reply_content = "感谢您的消息!"
  1305. reply_xml = f"""<xml>
  1306. <ToUserName><![CDATA[{from_user}]]></ToUserName>
  1307. <FromUserName><![CDATA[{to_user}]]></FromUserName>
  1308. <CreateTime>{int(__import__('time').time())}</CreateTime>
  1309. <MsgType><![CDATA[text]]></MsgType>
  1310. <Content><![CDATA[{reply_content}]]></Content>
  1311. </xml>"""
  1312. return PlainTextResponse(reply_xml, media_type="application/xml")
  1313. elif msg_type == 'event':
  1314. # 事件消息
  1315. event = root.find('Event').text if root.find('Event') is not None else ''
  1316. logger.info(f"🎯 事件消息: {event}")
  1317. # 处理关注/取消关注事件
  1318. if event == 'subscribe':
  1319. reply_content = "欢迎关注!"
  1320. elif event == 'unsubscribe':
  1321. reply_content = "感谢您的关注!"
  1322. else:
  1323. reply_content = "收到事件消息"
  1324. reply_xml = f"""<xml>
  1325. <ToUserName><![CDATA[{from_user}]]></ToUserName>
  1326. <FromUserName><![CDATA[{to_user}]]></FromUserName>
  1327. <CreateTime>{int(__import__('time').time())}</CreateTime>
  1328. <MsgType><![CDATA[text]]></MsgType>
  1329. <Content><![CDATA[{reply_content}]]></Content>
  1330. </xml>"""
  1331. return PlainTextResponse(reply_xml, media_type="application/xml")
  1332. else:
  1333. # 其他类型消息,返回空响应
  1334. logger.info(f"ℹ️ 未处理的消息类型: {msg_type}")
  1335. return PlainTextResponse("success")
  1336. except Exception as e:
  1337. logger.error(f"❌ 处理微信公众平台消息异常: {e}")
  1338. return PlainTextResponse("success") # 返回success避免微信重试
  1339. @app.websocket("/ws")
  1340. async def websocket_endpoint(websocket: WebSocket):
  1341. """WebSocket端点"""
  1342. # 从查询参数获取user_id(Lyrria_user_id)
  1343. user_id = websocket.query_params.get("user_id")
  1344. # 记录连接信息
  1345. import logging
  1346. logger = logging.getLogger(__name__)
  1347. logger.info(f"🔌 WebSocket连接: user_id={user_id}, query_params={dict(websocket.query_params)}")
  1348. # 🔥 传入Redis缓存服务和user_id
  1349. redis_cache = getattr(app.state, 'redis_cache', None)
  1350. handler = WebSocketHandler(websocket, user_id=user_id, redis_cache=redis_cache)
  1351. await handler.handle_connection()
  1352. if __name__ == "__main__":
  1353. import uvicorn
  1354. import os
  1355. import socket
  1356. from pathlib import Path
  1357. # 🔥 DNS预解析(减少首次连接延迟)
  1358. dashscope_domain = get_dashscope_domain()
  1359. try:
  1360. socket.getaddrinfo(dashscope_domain, 443)
  1361. except Exception as e:
  1362. pass
  1363. # SSL证书路径配置
  1364. # 优先使用环境变量配置的证书路径(生产环境)
  1365. ssl_keyfile = settings.SSL_KEYFILE
  1366. ssl_certfile = settings.SSL_CERTFILE
  1367. # 如果环境变量未配置,使用本地开发证书
  1368. if not ssl_keyfile or not ssl_certfile:
  1369. ssl_dir = Path(__file__).parent.parent / "ssl"
  1370. ssl_keyfile = str(ssl_dir / "key.pem")
  1371. ssl_certfile = str(ssl_dir / "cert.pem")
  1372. # 检查证书是否存在
  1373. if ssl_keyfile and ssl_certfile:
  1374. if not os.path.exists(ssl_keyfile) or not os.path.exists(ssl_certfile):
  1375. ssl_keyfile = None
  1376. ssl_certfile = None
  1377. uvicorn.run(
  1378. "main:app",
  1379. host=settings.host,
  1380. port=settings.port,
  1381. reload=settings.DEBUG,
  1382. ssl_keyfile=ssl_keyfile,
  1383. ssl_certfile=ssl_certfile,
  1384. )