107 lines
3.3 KiB
Go
107 lines
3.3 KiB
Go
package render
|
||
|
||
import (
|
||
"bbs-go/internal/models"
|
||
"bbs-go/internal/models/constants"
|
||
"bbs-go/internal/pkg/markdown"
|
||
"bbs-go/internal/services"
|
||
"html"
|
||
"strconv"
|
||
|
||
"github.com/mlogclub/simple/common/arrays"
|
||
"github.com/mlogclub/simple/web"
|
||
)
|
||
|
||
func BuildComment(comment *models.Comment) *models.CommentResponse {
|
||
return doBuildComment(comment, nil, true, true)
|
||
}
|
||
|
||
func BuildComments(comments []models.Comment, currentUser *models.User, isBuildReplies, isBuildQuote bool) []models.CommentResponse {
|
||
if len(comments) == 0 {
|
||
return nil
|
||
}
|
||
|
||
likedCommentIds := getLikedCommentIds(comments, currentUser)
|
||
|
||
var ret []models.CommentResponse
|
||
for _, comment := range comments {
|
||
item := doBuildComment(&comment, currentUser, isBuildReplies, isBuildQuote)
|
||
item.Liked = arrays.Contains(comment.Id, likedCommentIds)
|
||
ret = append(ret, *item)
|
||
}
|
||
return ret
|
||
}
|
||
|
||
func getLikedCommentIds(comments []models.Comment, currentUser *models.User) (likedCommentIds []int64) {
|
||
if currentUser == nil || len(comments) == 0 {
|
||
return
|
||
}
|
||
var commentIds []int64
|
||
for _, comment := range comments {
|
||
commentIds = append(commentIds, comment.Id)
|
||
}
|
||
likedCommentIds = services.UserLikeService.IsLiked(currentUser.Id, constants.EntityComment, commentIds)
|
||
return
|
||
}
|
||
|
||
// doBuildComment 渲染评论
|
||
// isBuildReplies 是否渲染评论的二级回复，一级评论时要设置为true，其他时候都为false
|
||
// isBuildQuote 是否渲染评论的引用，二级回复时要设置为true，其他时候都为false
|
||
func doBuildComment(comment *models.Comment, currentUser *models.User, isBuildReplies, isBuildQuote bool) *models.CommentResponse {
|
||
if comment == nil {
|
||
return nil
|
||
}
|
||
|
||
ret := &models.CommentResponse{
|
||
Id: comment.Id,
|
||
User: BuildUserInfoDefaultIfNull(comment.UserId),
|
||
EntityType: comment.EntityType,
|
||
EntityId: comment.EntityId,
|
||
QuoteId: comment.QuoteId,
|
||
LikeCount: comment.LikeCount,
|
||
CommentCount: comment.CommentCount,
|
||
ContentType: comment.ContentType,
|
||
IpLocation: comment.IpLocation,
|
||
Status: comment.Status,
|
||
CreateTime: comment.CreateTime,
|
||
}
|
||
|
||
if comment.Status == constants.StatusOk {
|
||
if comment.ContentType == constants.ContentTypeMarkdown {
|
||
content := markdown.ToHTML(comment.Content)
|
||
ret.Content = handleHtmlContent(content)
|
||
} else if comment.ContentType == constants.ContentTypeHtml {
|
||
ret.Content = handleHtmlContent(comment.Content)
|
||
} else {
|
||
ret.Content = html.EscapeString(comment.Content)
|
||
}
|
||
ret.ImageList = BuildImageList(comment.ImageList)
|
||
} else {
|
||
ret.Content = "内容已删除"
|
||
}
|
||
|
||
if isBuildReplies && comment.CommentCount > 0 {
|
||
var repliesLimit int64 = 3
|
||
replies, nextCursor, _ := services.CommentService.GetReplies(comment.Id, 0, int(repliesLimit))
|
||
//var replyResults []model.CommentResponse
|
||
//for _, reply := range replies {
|
||
// replyResults = append(replyResults, *doBuildComment(&reply, false, true))
|
||
//}
|
||
replyResults := BuildComments(replies, currentUser, false, true)
|
||
ret.Replies = &web.CursorResult{
|
||
Results: replyResults,
|
||
Cursor: strconv.FormatInt(nextCursor, 10),
|
||
HasMore: comment.CommentCount > repliesLimit,
|
||
}
|
||
}
|
||
|
||
if isBuildQuote && comment.QuoteId > 0 {
|
||
quote := doBuildComment(services.CommentService.Get(comment.QuoteId), currentUser, false, false)
|
||
if quote != nil {
|
||
ret.Quote = quote
|
||
}
|
||
}
|
||
|
||
return ret
|
||
}
|