likes.py

from fastapi import APIRouter, Depends, HTTPException, Query, status
from sqlalchemy.orm import Session

from database import get_db
from models import Post as PostModel, Like as LikeModel
from schemas import Like, LikeRequest, Error

router = APIRouter(prefix="/posts/{postId}/likes", tags=["Likes"])


@router.post("", response_model=Like, status_code=status.HTTP_201_CREATED)
def like_post(postId: str, like_request: LikeRequest, db: Session = Depends(get_db)):
    """게시물 좋아요"""
    post = db.query(PostModel).filter(PostModel.id == postId).first()
    if not post:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail={"message": "리소스를 찾을 수 없습니다", "details": "지정된 ID의 게시물이 존재하지 않습니다"}
        )
    
    # Check if already liked
    existing_like = db.query(LikeModel).filter(
        LikeModel.post_id == postId,
        LikeModel.username == like_request.username
    ).first()
    
    if existing_like:
        raise HTTPException(
            status_code=status.HTTP_400_BAD_REQUEST,
            detail={"message": "잘못된 요청", "details": "이미 좋아요를 눌렀습니다"}
        )
    
    new_like = LikeModel(
        post_id=postId,
        username=like_request.username
    )
    db.add(new_like)
    db.commit()
    db.refresh(new_like)
    
    return {
        "postId": new_like.post_id,
        "username": new_like.username,
        "createdAt": new_like.created_at
    }


@router.delete("", status_code=status.HTTP_204_NO_CONTENT)
def unlike_post(postId: str, username: str = Query(...), db: Session = Depends(get_db)):
    """게시물 좋아요 취소"""
    like = db.query(LikeModel).filter(
        LikeModel.post_id == postId,
        LikeModel.username == username
    ).first()
    
    if not like:
        raise HTTPException(
            status_code=status.HTTP_404_NOT_FOUND,
            detail={"message": "리소스를 찾을 수 없습니다", "details": "좋아요를 찾을 수 없습니다"}
        )
    
    db.delete(like)
    db.commit()
    return None