feat(relationship): implement relationship(friends, blocks) api (close #6)

app/router/__init__.py

@@ -4,6 +4,7 @@ from . import (  # pyright: ignore[reportUnusedImport]  # noqa: F401
     beatmap,
     beatmapset,
     me,
+    relationship,
 )
 from .api_router import router as api_router
 from .auth import router as auth_router

app/router/relationship.py

@@ -0,0 +1,107 @@
+from __future__ import annotations
+
+from typing import Literal
+
+from app.database import User as DBUser
+from app.database.relationship import Relationship, RelationshipResp, RelationshipType
+from app.dependencies.database import get_db
+from app.dependencies.user import get_current_user
+
+from .api_router import router
+
+from fastapi import Depends, HTTPException, Query
+from sqlmodel import select
+from sqlmodel.ext.asyncio.session import AsyncSession
+
+
+@router.get("/{type}", tags=["relationship"], response_model=list[RelationshipResp])
+async def get_relationship(
+    type: Literal["friends", "blocks"],
+    current_user: DBUser = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    if type == "friends":
+        relationship_type = RelationshipType.FOLLOW
+    else:
+        relationship_type = RelationshipType.BLOCK
+    relationships = await db.exec(
+        select(Relationship).where(
+            Relationship.user_id == current_user.id,
+            Relationship.type == relationship_type,
+        )
+    )
+    return [await RelationshipResp.from_db(db, rel) for rel in relationships]
+
+
+@router.post("/{type}", tags=["relationship"], response_model=RelationshipResp)
+async def add_relationship(
+    type: Literal["friends", "blocks"],
+    target: int = Query(),
+    current_user: DBUser = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    if type == "blocks":
+        relationship_type = RelationshipType.BLOCK
+    else:
+        relationship_type = RelationshipType.FOLLOW
+    if target == current_user.id:
+        raise HTTPException(422, "Cannot add relationship to yourself")
+    relationship = (
+        await db.exec(
+            select(Relationship).where(
+                Relationship.user_id == current_user.id,
+                Relationship.target_id == target,
+            )
+        )
+    ).first()
+    if relationship:
+        relationship.type = relationship_type
+        # 这里原来如何是 block 也会修改为 follow
+        # 与 ppy/osu-web 的行为保持一致
+    else:
+        relationship = Relationship(
+            user_id=current_user.id,
+            target_id=target,
+            type=relationship_type,
+        )
+        db.add(relationship)
+    if relationship.type == RelationshipType.BLOCK:
+        target_relationship = (
+            await db.exec(
+                select(Relationship).where(
+                    Relationship.user_id == target,
+                    Relationship.target_id == current_user.id,
+                )
+            )
+        ).first()
+        if target_relationship and target_relationship.type == RelationshipType.FOLLOW:
+            await db.delete(target_relationship)
+    await db.commit()
+    await db.refresh(relationship)
+    return await RelationshipResp.from_db(db, relationship)
+
+
+@router.delete("/{type}/{target}", tags=["relationship"])
+async def delete_relationship(
+    type: Literal["friends", "blocks"],
+    target: int,
+    current_user: DBUser = Depends(get_current_user),
+    db: AsyncSession = Depends(get_db),
+):
+    relationship_type = (
+        RelationshipType.BLOCK if type == "blocks" else RelationshipType.FOLLOW
+    )
+    relationship = (
+        await db.exec(
+            select(Relationship).where(
+                Relationship.user_id == current_user.id,
+                Relationship.target_id == target,
+            )
+        )
+    ).first()
+    if not relationship:
+        raise HTTPException(404, "Relationship not found")
+    if relationship.type != relationship_type:
+        raise HTTPException(422, "Relationship type mismatch")
+    await db.delete(relationship)
+    await db.commit()