refactor: 后端从Node.js重写为Python FastAPI

server/app/routers/story.py

@@ -0,0 +1,221 @@
+"""
+故事相关API路由
+"""
+import random
+from fastapi import APIRouter, Depends, Query, HTTPException
+from sqlalchemy.ext.asyncio import AsyncSession
+from sqlalchemy import select, update, func, distinct
+from typing import Optional
+from pydantic import BaseModel
+
+from app.database import get_db
+from app.models.story import Story, StoryNode, StoryChoice
+
+router = APIRouter()
+
+
+# ========== 请求/响应模型 ==========
+class LikeRequest(BaseModel):
+    like: bool
+
+
+class RewriteRequest(BaseModel):
+    ending_name: str
+    ending_content: str
+    prompt: str
+
+
+# ========== API接口 ==========
+
+@router.get("")
+async def get_stories(
+    category: Optional[str] = Query(None),
+    featured: bool = Query(False),
+    limit: int = Query(20, ge=1, le=100),
+    offset: int = Query(0, ge=0),
+    db: AsyncSession = Depends(get_db)
+):
+    """获取故事列表"""
+    query = select(Story).where(Story.status == 1)
+    
+    if category:
+        query = query.where(Story.category == category)
+    if featured:
+        query = query.where(Story.is_featured == True)
+    
+    query = query.order_by(Story.is_featured.desc(), Story.play_count.desc())
+    query = query.limit(limit).offset(offset)
+    
+    result = await db.execute(query)
+    stories = result.scalars().all()
+    
+    data = [{
+        "id": s.id,
+        "title": s.title,
+        "cover_url": s.cover_url,
+        "description": s.description,
+        "category": s.category,
+        "play_count": s.play_count,
+        "like_count": s.like_count,
+        "is_featured": s.is_featured
+    } for s in stories]
+    
+    return {"code": 0, "data": data}
+
+
+@router.get("/hot")
+async def get_hot_stories(
+    limit: int = Query(10, ge=1, le=50),
+    db: AsyncSession = Depends(get_db)
+):
+    """获取热门故事"""
+    query = select(Story).where(Story.status == 1).order_by(Story.play_count.desc()).limit(limit)
+    result = await db.execute(query)
+    stories = result.scalars().all()
+    
+    data = [{
+        "id": s.id,
+        "title": s.title,
+        "cover_url": s.cover_url,
+        "description": s.description,
+        "category": s.category,
+        "play_count": s.play_count,
+        "like_count": s.like_count
+    } for s in stories]
+    
+    return {"code": 0, "data": data}
+
+
+@router.get("/categories")
+async def get_categories(db: AsyncSession = Depends(get_db)):
+    """获取分类列表"""
+    query = select(distinct(Story.category)).where(Story.status == 1)
+    result = await db.execute(query)
+    categories = [row[0] for row in result.all()]
+    
+    return {"code": 0, "data": categories}
+
+
+@router.get("/{story_id}")
+async def get_story_detail(story_id: int, db: AsyncSession = Depends(get_db)):
+    """获取故事详情（含节点和选项）"""
+    # 获取故事
+    result = await db.execute(select(Story).where(Story.id == story_id, Story.status == 1))
+    story = result.scalar_one_or_none()
+    
+    if not story:
+        raise HTTPException(status_code=404, detail="故事不存在")
+    
+    # 获取节点
+    nodes_result = await db.execute(
+        select(StoryNode).where(StoryNode.story_id == story_id).order_by(StoryNode.sort_order)
+    )
+    nodes = nodes_result.scalars().all()
+    
+    # 获取选项
+    choices_result = await db.execute(
+        select(StoryChoice).where(StoryChoice.story_id == story_id).order_by(StoryChoice.sort_order)
+    )
+    choices = choices_result.scalars().all()
+    
+    # 组装节点和选项
+    nodes_map = {}
+    for node in nodes:
+        nodes_map[node.node_key] = {
+            "id": node.id,
+            "node_key": node.node_key,
+            "content": node.content,
+            "speaker": node.speaker,
+            "background_image": node.background_image,
+            "character_image": node.character_image,
+            "bgm": node.bgm,
+            "is_ending": node.is_ending,
+            "ending_name": node.ending_name,
+            "ending_score": node.ending_score,
+            "ending_type": node.ending_type,
+            "choices": []
+        }
+    
+    for choice in choices:
+        # 找到对应的节点
+        for node in nodes:
+            if node.id == choice.node_id and node.node_key in nodes_map:
+                nodes_map[node.node_key]["choices"].append({
+                    "text": choice.text,
+                    "nextNodeKey": choice.next_node_key,
+                    "isLocked": choice.is_locked
+                })
+                break
+    
+    data = {
+        "id": story.id,
+        "title": story.title,
+        "cover_url": story.cover_url,
+        "description": story.description,
+        "category": story.category,
+        "author_id": story.author_id,
+        "play_count": story.play_count,
+        "like_count": story.like_count,
+        "is_featured": story.is_featured,
+        "nodes": nodes_map
+    }
+    
+    return {"code": 0, "data": data}
+
+
+@router.post("/{story_id}/play")
+async def record_play(story_id: int, db: AsyncSession = Depends(get_db)):
+    """记录游玩"""
+    await db.execute(
+        update(Story).where(Story.id == story_id).values(play_count=Story.play_count + 1)
+    )
+    await db.commit()
+    return {"code": 0, "message": "记录成功"}
+
+
+@router.post("/{story_id}/like")
+async def toggle_like(story_id: int, request: LikeRequest, db: AsyncSession = Depends(get_db)):
+    """点赞/取消点赞"""
+    delta = 1 if request.like else -1
+    await db.execute(
+        update(Story).where(Story.id == story_id).values(like_count=Story.like_count + delta)
+    )
+    await db.commit()
+    return {"code": 0, "message": "点赞成功" if request.like else "取消点赞成功"}
+
+
+@router.post("/{story_id}/rewrite")
+async def ai_rewrite_ending(story_id: int, request: RewriteRequest, db: AsyncSession = Depends(get_db)):
+    """AI改写结局"""
+    if not request.prompt:
+        raise HTTPException(status_code=400, detail="请输入改写指令")
+    
+    # 获取故事信息
+    result = await db.execute(select(Story).where(Story.id == story_id))
+    story = result.scalar_one_or_none()
+    
+    # 模拟AI生成（后续替换为真实API调用）
+    templates = [
+        f"根据你的愿望「{request.prompt}」，故事有了新的发展...\n\n",
+        f"命运的齿轮开始转动，{request.prompt}...\n\n",
+        f"在另一个平行世界里，{request.prompt}成为了现实...\n\n"
+    ]
+    
+    template = random.choice(templates)
+    new_content = (
+        template +
+        "原本的结局被改写，新的故事在这里展开。\n\n" +
+        f"【AI改写提示】这是基于「{request.prompt}」生成的新结局。\n" +
+        "实际部署时，这里将由AI大模型根据上下文生成更精彩的内容。"
+    )
+    
+    return {
+        "code": 0,
+        "data": {
+            "content": new_content,
+            "speaker": "旁白",
+            "is_ending": True,
+            "ending_name": f"{request.ending_name}（改写版）",
+            "ending_type": "rewrite"
+        }
+    }