1.9

backend/app/routers/friends.py

@@ -144,3 +144,76 @@ async def remove_friend(
     service = FriendService(db)
     await service.remove_friend(user.id, friend_id)
     return {"success": True, "message": "已删除好友"}
+
+
+# ============ 好友分组/标签 ============
+
+@router.get("/tags/list")
+async def list_tags(
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """获取我的好友标签"""
+    from app.models.friend_tag import FriendTag
+    result = await db.execute(
+        select(FriendTag).where(FriendTag.user_id == user.id).order_by(FriendTag.created_at)
+    )
+    return [{"id": t.id, "name": t.name} for t in result.scalars().all()]
+
+
+@router.post("/tags/create")
+async def create_tag(
+    body: dict,
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """创建好友标签"""
+    from app.models.friend_tag import FriendTag
+    import uuid
+    tag = FriendTag(id=str(uuid.uuid4()), user_id=user.id, name=body.get("name", "")[:30])
+    db.add(tag)
+    await db.flush()
+    return {"id": tag.id, "name": tag.name}
+
+
+@router.delete("/tags/{tag_id}")
+async def delete_tag(
+    tag_id: str,
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """删除好友标签"""
+    from app.models.friend_tag import FriendTag
+    from sqlalchemy import delete as sql_delete
+    await db.execute(sql_delete(FriendTag).where(FriendTag.id == tag_id, FriendTag.user_id == user.id))
+    return {"success": True}
+
+
+@router.put("/{friend_id}/tags")
+async def assign_tags(
+    friend_id: str,
+    body: dict,
+    user: User = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    """给好友分配标签（覆盖式）"""
+    from app.models.friend_tag import FriendTag, FriendTagAssignment
+    from app.models.friend import Friend
+    from sqlalchemy import delete as sql_delete
+    # 找到该好友关系记录
+    fr = await db.execute(
+        select(Friend).where(Friend.user_id == user.id, Friend.friend_user_id == friend_id)
+    )
+    friendship = fr.scalars().first()
+    if not friendship:
+        raise HTTPException(status_code=404, detail="好友不存在")
+    # 清空旧分配
+    await db.execute(
+        sql_delete(FriendTagAssignment).where(FriendTagAssignment.friend_id == friendship.id)
+    )
+    # 新分配
+    tag_ids = body.get("tag_ids", [])
+    import uuid
+    for tid in tag_ids:
+        db.add(FriendTagAssignment(id=str(uuid.uuid4()), friend_id=friendship.id, tag_id=tid))
+    return {"success": True}