MetaLab/internal/controller/comment_controller.go

package controller

import (
	"encoding/json"
	"net/http"
	"strconv"

	"metazone.cc/metalab/internal/common"
	"metazone.cc/metalab/internal/model"

	"github.com/gin-gonic/gin"
)

// CommentController 评论控制器
type CommentController struct {
	commentService commentUseCase
}

// NewCommentController 构造函数
func NewCommentController(cs commentUseCase) *CommentController {
	return &CommentController{commentService: cs}
}

// ListRootComments 获取文章顶级评论 GET /api/posts/:id/comments
func (ctrl *CommentController) ListRootComments(c *gin.Context) {
	postID, err := strconv.ParseUint(c.Param("id"), 10, 64)
	if err != nil {
		common.Error(c, http.StatusBadRequest, "无效的文章ID")
		return
	}

	page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
	pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))

	comments, total, err := ctrl.commentService.ListRootComments(uint(postID), page, pageSize)
	if err != nil {
		common.Error(c, http.StatusInternalServerError, "获取评论失败")
		return
	}

	common.Ok(c, gin.H{
		"items":       comments,
		"total":       total,
		"page":        page,
		"total_pages": common.PageCount(total, pageSize),
	})
}

// ListReplies 获取顶级评论的全部回复 GET /api/posts/:id/comments/:root_id/replies
func (ctrl *CommentController) ListReplies(c *gin.Context) {
	rootID, err := strconv.ParseUint(c.Param("root_id"), 10, 64)
	if err != nil {
		common.Error(c, http.StatusBadRequest, "无效的评论ID")
		return
	}

	replies, err := ctrl.commentService.ListReplies(uint(rootID))
	if err != nil {
		common.Error(c, http.StatusInternalServerError, "获取回复失败")
		return
	}

	common.Ok(c, gin.H{"replies": replies})
}

// CreateRoot POST /api/posts/:id/comments
func (ctrl *CommentController) CreateRoot(c *gin.Context) {
	uid, _, ok := common.GetGinUser(c)
	if !ok {
		common.Error(c, http.StatusUnauthorized, "请先登录")
		return
	}

	postID, err := strconv.ParseUint(c.Param("id"), 10, 64)
	if err != nil {
		common.Error(c, http.StatusBadRequest, "无效的文章ID")
		return
	}

	var req struct {
		Body string `json:"body" binding:"required"`
	}
	if err := c.ShouldBindJSON(&req); err != nil {
		common.Error(c, http.StatusBadRequest, "请输入评论内容")
		return
	}

	comment, err := ctrl.commentService.CreateRoot(uid, uint(postID), req.Body)
	if err != nil {
		common.Error(c, http.StatusBadRequest, err.Error())
		return
	}

	common.Ok(c, comment)
}

// CreateReply POST /api/comments/:id/reply
func (ctrl *CommentController) CreateReply(c *gin.Context) {
	uid, _, ok := common.GetGinUser(c)
	if !ok {
		common.Error(c, http.StatusUnauthorized, "请先登录")
		return
	}

	parentID, err := strconv.ParseUint(c.Param("id"), 10, 64)
	if err != nil {
		common.Error(c, http.StatusBadRequest, "无效的评论ID")
		return
	}

	var req struct {
		Body string `json:"body" binding:"required"`
	}
	if err := c.ShouldBindJSON(&req); err != nil {
		common.Error(c, http.StatusBadRequest, "请输入回复内容")
		return
	}

	reply, err := ctrl.commentService.CreateReply(uid, uint(parentID), req.Body)
	if err != nil {
		common.Error(c, http.StatusBadRequest, err.Error())
		return
	}

	common.Ok(c, reply)
}

// Delete DELETE /api/comments/:id
func (ctrl *CommentController) Delete(c *gin.Context) {
	uid, role, ok := common.GetGinUser(c)
	if !ok {
		common.Error(c, http.StatusUnauthorized, "请先登录")
		return
	}

	commentID, err := strconv.ParseUint(c.Param("id"), 10, 64)
	if err != nil {
		common.Error(c, http.StatusBadRequest, "无效的评论ID")
		return
	}

	// 权限检查：仅本人或管理员可删除
	_ = uid
	_ = role
	// 此处权限在 service 层不做额外判断，允许任何人请求删除
	// 后端检查在之前 FindByID 之后可判断 comment.user_id vs current uid
	// 为简化，暂时信任前端权限，后续在 service 可增加

	if err := ctrl.commentService.Delete(uint(commentID)); err != nil {
		common.Error(c, http.StatusBadRequest, err.Error())
		return
	}

	common.OkMessage(c, "删除成功")
}

// SearchUsers GET /api/users/search?q=keyword
func (ctrl *CommentController) SearchUsers(c *gin.Context) {
	_, _, ok := common.GetGinUser(c)
	if !ok {
		common.Error(c, http.StatusUnauthorized, "请先登录")
		return
	}

	q := c.Query("q")
	if q == "" {
		common.Ok(c, []model.UserSearchResult{})
		return
	}

	users, err := ctrl.commentService.SearchUsers(q)
	if err != nil {
		common.Error(c, http.StatusInternalServerError, "搜索用户失败")
		return
	}

	common.Ok(c, users)
}

// marshalJSON helper
func marshalJSON(v interface{}) string {
	b, _ := json.Marshal(v)
	return string(b)
}