1.6

backend/app/routers/capsules.py

@@ -0,0 +1,51 @@
+"""时光胶囊路由"""
+
+from fastapi import APIRouter, Depends, HTTPException
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.dependencies import get_db, get_current_user
+from app.models.user import User
+from app.schemas.capsule import CapsuleCreate
+from app.services.capsule_service import CapsuleService
+
+router = APIRouter()
+
+
+@router.get("/")
+async def list_capsules(
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """获取我的时光胶囊"""
+    service = CapsuleService(db)
+    return await service.get_capsules(user.id)
+
+
+@router.get("/{capsule_id}")
+async def get_capsule(
+    capsule_id: str,
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """获取单个胶囊"""
+    service = CapsuleService(db)
+    try:
+        return await service.get_capsule(user.id, capsule_id)
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+
+@router.post("/")
+async def create_capsule(
+    req: CapsuleCreate,
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """创建时光胶囊"""
+    service = CapsuleService(db)
+    try:
+        return await service.create_capsule(
+            user.id, req.recipient_id, req.title, req.content, req.unlock_at, req.mood
+        )
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))