108 lines
3.5 KiB
Python
108 lines
3.5 KiB
Python
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()
|