05c7f8a5eb
Add rich message segment support enabling structured content (text, images, votes, mentions) in posts and comments. Includes segments field in models, DTOs, handlers, and services. Also adds @mention notification processing and migrates vote data to embedded segments.
324 lines
8.5 KiB
Go
324 lines
8.5 KiB
Go
package handler
|
||
|
||
import (
|
||
"encoding/json"
|
||
"strconv"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
|
||
"with_you/internal/dto"
|
||
"with_you/internal/model"
|
||
"with_you/internal/pkg/cursor"
|
||
"with_you/internal/pkg/response"
|
||
"with_you/internal/service"
|
||
)
|
||
|
||
// CommentHandler 评论处理器
|
||
type CommentHandler struct {
|
||
commentService *service.CommentService
|
||
}
|
||
|
||
// NewCommentHandler 创建评论处理器
|
||
func NewCommentHandler(commentService *service.CommentService) *CommentHandler {
|
||
return &CommentHandler{
|
||
commentService: commentService,
|
||
}
|
||
}
|
||
|
||
// Create 创建评论
|
||
func (h *CommentHandler) Create(c *gin.Context) {
|
||
userID := c.GetString("user_id")
|
||
if userID == "" {
|
||
response.Unauthorized(c, "")
|
||
return
|
||
}
|
||
|
||
type CreateRequest struct {
|
||
PostID string `json:"post_id" binding:"required"`
|
||
Content string `json:"content"`
|
||
Segments model.MessageSegments `json:"segments"`
|
||
ParentID *string `json:"parent_id"`
|
||
Images []string `json:"images"`
|
||
}
|
||
|
||
var req CreateRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
response.BadRequest(c, err.Error())
|
||
return
|
||
}
|
||
|
||
// 验证：评论必须有内容或图片或segments
|
||
if req.Content == "" && len(req.Images) == 0 && len(req.Segments) == 0 {
|
||
response.BadRequest(c, "评论内容或图片不能同时为空")
|
||
return
|
||
}
|
||
|
||
// 如果没有content但有segments，从segments提取文本
|
||
content := req.Content
|
||
if content == "" && len(req.Segments) > 0 {
|
||
content = dto.ExtractTextContent(req.Segments)
|
||
}
|
||
|
||
// 将图片列表转换为JSON字符串
|
||
var imagesJSON string
|
||
if len(req.Images) > 0 {
|
||
imagesBytes, _ := json.Marshal(req.Images)
|
||
imagesJSON = string(imagesBytes)
|
||
}
|
||
|
||
comment, err := h.commentService.Create(c.Request.Context(), req.PostID, userID, content, req.Segments, req.ParentID, imagesJSON, req.Images)
|
||
if err != nil {
|
||
response.InternalServerError(c, "failed to create comment")
|
||
return
|
||
}
|
||
|
||
response.Success(c, dto.ConvertCommentToResponse(comment, false))
|
||
}
|
||
|
||
// GetByID 获取单条评论详情
|
||
func (h *CommentHandler) GetByID(c *gin.Context) {
|
||
userID := c.GetString("user_id")
|
||
id := c.Param("id")
|
||
|
||
comment, err := h.commentService.GetByID(c.Request.Context(), id)
|
||
if err != nil {
|
||
response.NotFound(c, "comment not found")
|
||
return
|
||
}
|
||
|
||
resp := dto.ConvertCommentToResponse(comment, h.commentService.IsLiked(c.Request.Context(), id, userID))
|
||
response.Success(c, resp)
|
||
}
|
||
|
||
// GetByPostID 获取帖子评论
|
||
func (h *CommentHandler) GetByPostID(c *gin.Context) {
|
||
userID := c.GetString("user_id")
|
||
postID := c.Param("id")
|
||
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
||
pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
|
||
|
||
comments, total, err := h.commentService.GetByPostID(c.Request.Context(), postID, page, pageSize)
|
||
if err != nil {
|
||
response.InternalServerError(c, "failed to get comments")
|
||
return
|
||
}
|
||
|
||
// 转换为响应结构，检查每个评论的点赞状态
|
||
commentResponses := dto.ConvertCommentsToResponseWithUser(comments, userID, h.commentService)
|
||
|
||
response.Paginated(c, commentResponses, total, page, pageSize)
|
||
}
|
||
|
||
// GetReplies 获取回复
|
||
func (h *CommentHandler) GetReplies(c *gin.Context) {
|
||
userID := c.GetString("user_id")
|
||
parentID := c.Param("id")
|
||
|
||
comments, err := h.commentService.GetReplies(c.Request.Context(), parentID)
|
||
if err != nil {
|
||
response.InternalServerError(c, "failed to get replies")
|
||
return
|
||
}
|
||
|
||
// 转换为响应结构，检查每个回复的点赞状态
|
||
commentResponses := dto.ConvertCommentsToResponseWithUser(comments, userID, h.commentService)
|
||
|
||
response.Success(c, commentResponses)
|
||
}
|
||
|
||
// GetRepliesByRootID 根据根评论ID分页获取回复（扁平化）
|
||
func (h *CommentHandler) GetRepliesByRootID(c *gin.Context) {
|
||
userID := c.GetString("user_id")
|
||
rootID := c.Param("id")
|
||
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
||
pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "10"))
|
||
|
||
replies, total, err := h.commentService.GetRepliesByRootID(c.Request.Context(), rootID, page, pageSize)
|
||
if err != nil {
|
||
response.InternalServerError(c, "failed to get replies")
|
||
return
|
||
}
|
||
|
||
// 转换为响应结构，检查每个回复的点赞状态
|
||
replyResponses := dto.ConvertCommentsToResponseWithUser(replies, userID, h.commentService)
|
||
|
||
response.Paginated(c, replyResponses, total, page, pageSize)
|
||
}
|
||
|
||
// Update 更新评论
|
||
func (h *CommentHandler) Update(c *gin.Context) {
|
||
userID := c.GetString("user_id")
|
||
if userID == "" {
|
||
response.Unauthorized(c, "")
|
||
return
|
||
}
|
||
|
||
id := c.Param("id")
|
||
|
||
comment, err := h.commentService.GetByID(c.Request.Context(), id)
|
||
if err != nil {
|
||
response.NotFound(c, "comment not found")
|
||
return
|
||
}
|
||
|
||
if comment.UserID != userID {
|
||
response.Forbidden(c, "cannot update others' comment")
|
||
return
|
||
}
|
||
|
||
type UpdateRequest struct {
|
||
Content string `json:"content" binding:"required"`
|
||
}
|
||
|
||
var req UpdateRequest
|
||
if err := c.ShouldBindJSON(&req); err != nil {
|
||
response.BadRequest(c, err.Error())
|
||
return
|
||
}
|
||
|
||
comment.Content = req.Content
|
||
|
||
err = h.commentService.Update(c.Request.Context(), comment)
|
||
if err != nil {
|
||
response.InternalServerError(c, "failed to update comment")
|
||
return
|
||
}
|
||
|
||
response.Success(c, dto.ConvertCommentToResponse(comment, false))
|
||
}
|
||
|
||
// Delete 删除评论
|
||
func (h *CommentHandler) Delete(c *gin.Context) {
|
||
userID := c.GetString("user_id")
|
||
if userID == "" {
|
||
response.Unauthorized(c, "")
|
||
return
|
||
}
|
||
|
||
id := c.Param("id")
|
||
|
||
comment, err := h.commentService.GetByID(c.Request.Context(), id)
|
||
if err != nil {
|
||
response.NotFound(c, "comment not found")
|
||
return
|
||
}
|
||
|
||
if comment.UserID != userID {
|
||
response.Forbidden(c, "cannot delete others' comment")
|
||
return
|
||
}
|
||
|
||
err = h.commentService.Delete(c.Request.Context(), id)
|
||
if err != nil {
|
||
response.InternalServerError(c, "failed to delete comment")
|
||
return
|
||
}
|
||
|
||
response.SuccessWithMessage(c, "comment deleted", nil)
|
||
}
|
||
|
||
// Like 点赞评论
|
||
func (h *CommentHandler) Like(c *gin.Context) {
|
||
userID := c.GetString("user_id")
|
||
if userID == "" {
|
||
response.Unauthorized(c, "")
|
||
return
|
||
}
|
||
|
||
id := c.Param("id")
|
||
|
||
err := h.commentService.Like(c.Request.Context(), id, userID)
|
||
if err != nil {
|
||
response.InternalServerError(c, "failed to like comment")
|
||
return
|
||
}
|
||
|
||
response.SuccessWithMessage(c, "liked", nil)
|
||
}
|
||
|
||
// Unlike 取消点赞评论
|
||
func (h *CommentHandler) Unlike(c *gin.Context) {
|
||
userID := c.GetString("user_id")
|
||
if userID == "" {
|
||
response.Unauthorized(c, "")
|
||
return
|
||
}
|
||
|
||
id := c.Param("id")
|
||
|
||
err := h.commentService.Unlike(c.Request.Context(), id, userID)
|
||
if err != nil {
|
||
response.InternalServerError(c, "failed to unlike comment")
|
||
return
|
||
}
|
||
|
||
response.SuccessWithMessage(c, "unliked", nil)
|
||
}
|
||
|
||
// GetByPostIDByCursor 游标分页获取帖子评论
|
||
func (h *CommentHandler) GetByPostIDByCursor(c *gin.Context) {
|
||
userID := c.GetString("user_id")
|
||
postID := c.Param("id")
|
||
|
||
// 解析游标分页请求
|
||
req := h.parseCursorRequest(c)
|
||
|
||
// 调用游标分页服务
|
||
result, err := h.commentService.GetCommentsByCursor(c.Request.Context(), postID, req)
|
||
if err != nil {
|
||
response.InternalServerError(c, "failed to get comments")
|
||
return
|
||
}
|
||
|
||
// 转换为响应结构，检查每个评论的点赞状态
|
||
commentResponses := dto.ConvertCommentsToResponseWithUser(result.Items, userID, h.commentService)
|
||
|
||
// 构建游标分页响应
|
||
cursorResp := &dto.CommentCursorPageResponse{
|
||
Items: commentResponses,
|
||
NextCursor: result.NextCursor,
|
||
PrevCursor: result.PrevCursor,
|
||
HasMore: result.HasMore,
|
||
}
|
||
|
||
response.Success(c, cursorResp)
|
||
}
|
||
|
||
// GetRepliesByRootIDByCursor 游标分页获取根评论的回复
|
||
func (h *CommentHandler) GetRepliesByRootIDByCursor(c *gin.Context) {
|
||
userID := c.GetString("user_id")
|
||
rootID := c.Param("id")
|
||
|
||
// 解析游标分页请求
|
||
req := h.parseCursorRequest(c)
|
||
|
||
// 调用游标分页服务
|
||
result, err := h.commentService.GetRepliesByCursor(c.Request.Context(), rootID, req)
|
||
if err != nil {
|
||
response.InternalServerError(c, "failed to get replies")
|
||
return
|
||
}
|
||
|
||
// 转换为响应结构，检查每个回复的点赞状态
|
||
replyResponses := dto.ConvertCommentsToResponseWithUser(result.Items, userID, h.commentService)
|
||
|
||
// 构建游标分页响应
|
||
cursorResp := &dto.CommentCursorPageResponse{
|
||
Items: replyResponses,
|
||
NextCursor: result.NextCursor,
|
||
PrevCursor: result.PrevCursor,
|
||
HasMore: result.HasMore,
|
||
}
|
||
|
||
response.Success(c, cursorResp)
|
||
}
|
||
|
||
// parseCursorRequest 解析游标分页请求参数
|
||
func (h *CommentHandler) parseCursorRequest(c *gin.Context) *cursor.PageRequest {
|
||
cursorStr := c.Query("cursor")
|
||
direction := cursor.Direction(c.DefaultQuery("direction", "forward"))
|
||
pageSize, _ := strconv.Atoi(c.DefaultQuery("page_size", "20"))
|
||
|
||
return cursor.NewPageRequest(cursorStr, direction, pageSize)
|
||
}
|