devplacepy/devplacepy/routers/comments.py

import logging
from typing import Annotated
from datetime import datetime, timezone
from fastapi import APIRouter, Request, Form
from fastapi.responses import RedirectResponse
from devplacepy.database import get_table, resolve_object_url, delete_engagement
from devplacepy.attachments import link_attachments, delete_target_attachments
from devplacepy.content import is_owner
from devplacepy.utils import generate_uid, require_user, create_mention_notifications, create_notification, award_rewards, XP_COMMENT
from devplacepy.models import CommentForm
from devplacepy.responses import action_result, wants_json, json_error

logger = logging.getLogger(__name__)
router = APIRouter()


@router.post("/create")
async def create_comment(request: Request, data: Annotated[CommentForm, Form()]):
    user = require_user(request)
    content = data.content.strip()
    target_uid = data.target_uid or data.post_uid
    target_type = data.target_type
    parent_uid = data.parent_uid

    redirect_url = resolve_object_url(target_type, target_uid)

    comment_uid = generate_uid()
    insert = {
        "uid": comment_uid,
        "target_uid": target_uid,
        "target_type": target_type,
        "user_uid": user["uid"],
        "content": content,
        "parent_uid": parent_uid or None,
        "created_at": datetime.now(timezone.utc).isoformat(),
    }
    if target_type == "post":
        insert["post_uid"] = target_uid
    get_table("comments").insert(insert)

    if data.attachment_uids:
        link_attachments(data.attachment_uids, "comment", comment_uid)

    award_rewards(user["uid"], XP_COMMENT, "First Comment")

    comment_url = f"{redirect_url}#comment-{comment_uid}"

    if target_type == "post":
        if parent_uid:
            parent = get_table("comments").find_one(uid=parent_uid)
            if parent and parent["user_uid"] != user["uid"]:
                create_notification(parent["user_uid"], "reply", f"{user['username']} replied to your comment", user["uid"], comment_url)
        else:
            posts = get_table("posts")
            post = posts.find_one(uid=target_uid)
            if not post:
                post = posts.find_one(slug=target_uid)
            if post and post["user_uid"] != user["uid"]:
                create_notification(post["user_uid"], "comment", f"{user['username']} commented on your post", user["uid"], comment_url)

    create_mention_notifications(content, user["uid"], comment_url)
    logger.info(f"Comment by {user['username']} on {target_type} {target_uid}")
    return action_result(request, comment_url, data={"uid": comment_uid, "url": comment_url})


@router.post("/delete/{comment_uid}")
async def delete_comment(request: Request, comment_uid: str):
    user = require_user(request)
    comments = get_table("comments")
    comment = comments.find_one(uid=comment_uid)
    if not is_owner(comment, user):
        if wants_json(request):
            return json_error(403, "Not allowed")
        return RedirectResponse(url="/feed", status_code=302)
    target_type = comment.get("target_type", "post")
    target_uid = comment.get("target_uid") or comment.get("post_uid", "")
    delete_target_attachments("comment", comment_uid)
    get_table("votes").delete(target_uid=comment_uid, target_type="comment")
    delete_engagement("comment", [comment_uid])
    comments.delete(id=comment["id"])
    logger.info(f"Comment {comment_uid} deleted by {user['username']}")
    redirect_url = resolve_object_url(target_type, target_uid)
    return action_result(request, redirect_url)