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))

backend/app/routers/leaves.py

@@ -0,0 +1,46 @@
+"""心情叶路由"""
+
+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.leaf import LeafUpdate
+from app.services.leaf_service import LeafService
+
+router = APIRouter()
+
+
+@router.get("/today")
+async def get_today_leaf(
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """获取或创建今日心情叶"""
+    service = LeafService(db)
+    return await service.get_or_create_today(user.id)
+
+
+@router.put("/{leaf_id}")
+async def update_leaf(
+    leaf_id: str,
+    req: LeafUpdate,
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """更新今日叶子的心情和备注"""
+    service = LeafService(db)
+    try:
+        return await service.update_leaf(user.id, leaf_id, req.mood, req.note)
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+
+@router.get("/collection")
+async def get_collection(
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """获取叶子收藏"""
+    service = LeafService(db)
+    return await service.get_collection(user.id)

backend/app/routers/trees.py

@@ -0,0 +1,45 @@
+"""好友之树路由"""
+
+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.services.tree_service import TreeService
+
+router = APIRouter()
+
+
+@router.get("/")
+async def list_trees(
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """获取所有好友的树"""
+    service = TreeService(db)
+    return await service.get_all_trees(user.id)
+
+
+@router.get("/{friend_id}")
+async def get_tree(
+    friend_id: str,
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """获取与某好友的树"""
+    service = TreeService(db)
+    return await service.get_tree(user.id, friend_id)
+
+
+@router.post("/{friend_id}/water")
+async def water_tree(
+    friend_id: str,
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """浇水"""
+    service = TreeService(db)
+    try:
+        return await service.water(user.id, friend_id)
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))