paopao-ce/internal/dao/sakila/comments.go

// Copyright 2023 ROC. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.

package sakila

import (
	"time"

	"github.com/alimy/tryst/cfg"
	"github.com/bitbus/sqlx"
	"github.com/rocboss/paopao-ce/internal/core"
	"github.com/rocboss/paopao-ce/internal/core/cs"
	"github.com/rocboss/paopao-ce/internal/core/ms"
	"github.com/rocboss/paopao-ce/internal/dao/sakila/auto/cc"
	"github.com/rocboss/paopao-ce/internal/dao/sakila/auto/pgc"
	"github.com/rocboss/paopao-ce/pkg/types"
)

var (
	_ core.CommentService       = (*commentSrv)(nil)
	_ core.CommentManageService = (*commentManageSrv)(nil)
)

type commentSrv struct {
	*sqlxSrv
	q *cc.Comment
}

type commentManageSrv struct {
	*sqlxSrv
	q *cc.CommentManage
}

func (s *commentSrv) GetComments(tweetId int64, style cs.StyleCommentType, limit int, offset int) (res []*ms.Comment, total int64, err error) {
	switch style {
	case cs.StyleCommentHots:
		if err = s.q.GetHotsComments.Select(&res, tweetId, limit, offset); err == nil {
			err = s.q.GetCommentCount.Get(&total, tweetId)
		}
	case cs.StyleCommentNewest:
		if err = s.q.GetNewestComments.Select(&res, tweetId, limit, offset); err == nil {
			err = s.q.GetCommentCount.Get(&total, tweetId)
		}
	case cs.StyleCommentDefault:
		fallthrough
	default:
		if err = s.q.GetDefaultComments.Select(&res, tweetId, limit, offset); err == nil {
			err = s.q.GetCommentCount.Get(&total, tweetId)
		}
	}
	return
}

func (s *commentSrv) GetCommentByID(id int64) (res *ms.Comment, err error) {
	res = &ms.Comment{}
	err = s.q.GetCommentById.Get(res, id)
	return
}

func (s *commentSrv) GetCommentReplyByID(id int64) (res *ms.CommentReply, err error) {
	res = &ms.CommentReply{}
	err = s.q.GetCommentReplyById.Get(res, id)
	return
}

func (s *commentSrv) GetCommentContentsByIDs(ids []int64) (res []*ms.CommentContent, err error) {
	if len(ids) == 0 {
		return nil, nil
	}
	err = s.db.InSelect(&res, s.q.GetCommentContentsByIds, ids)
	return res, err
}

func (s *commentSrv) GetCommentThumbsMap(userId int64, tweetId int64) (cs.CommentThumbsMap, cs.CommentThumbsMap, error) {
	if userId < 0 {
		return nil, nil, nil
	}
	commentThumbsList := cs.CommentThumbsList{}
	err := s.q.GetCommentThumbs.Select(&commentThumbsList, userId, tweetId)
	if err != nil {
		return nil, nil, err
	}
	commentThumbs, replyThumbs := make(cs.CommentThumbsMap), make(cs.CommentThumbsMap)
	for _, thumbs := range commentThumbsList {
		if thumbs.CommentType == 0 {
			commentThumbs[thumbs.CommentID] = thumbs
		} else {
			replyThumbs[thumbs.ReplyID] = thumbs
		}
	}
	return commentThumbs, replyThumbs, nil
}

func (s *commentSrv) GetCommentRepliesByID(ids []int64) ([]*ms.CommentReplyFormated, error) {
	if len(ids) == 0 {
		return nil, nil
	}
	replies := []*ms.CommentReply{}
	err := s.db.InSelect(&replies, s.q.GetCommmentRepliesByIds, ids)
	if err != nil {
		return nil, err
	}
	userIds := []int64{}
	for _, reply := range replies {
		userIds = append(userIds, reply.UserID, reply.AtUserID)
	}
	users := []*ms.UserFormated{}
	if len(userIds) > 0 {
		if err = s.db.InSelect(&users, s.q.GetUsersByIds, userIds); err != nil {
			return nil, err
		}
	}
	repliesFormated := []*ms.CommentReplyFormated{}
	for _, reply := range replies {
		replyFormated := reply.Format()
		for _, user := range users {
			if reply.UserID == user.ID {
				replyFormated.User = user
			}
			if reply.AtUserID == user.ID {
				replyFormated.AtUser = user
			}
		}
		repliesFormated = append(repliesFormated, replyFormated)
	}
	return repliesFormated, nil
}

// DeleteComment delete comment
// TODO: need delete related replies and update tweet comment count and comment content?
func (s *commentManageSrv) DeleteComment(r *ms.Comment) error {
	return s.db.Withx(func(tx *sqlx.Tx) error {
		now := time.Now().Unix()
		if _, err := tx.Stmtx(s.q.DeleteComment).Exec(now, r.ID); err != nil {
			return err
		}
		_, err := tx.Stmtx(s.q.DeleteCommentThumbs).Exec(now, r.UserID, r.PostID, r.ID)
		return err
	})
}

func (s *commentManageSrv) CreateComment(r *ms.Comment) (*ms.Comment, error) {
	res, err := s.q.CreateComment.Exec(r.PostID, r.UserID, r.IP, r.IPLoc, time.Now().Unix())
	if err != nil {
		return nil, err
	}
	id, err := res.LastInsertId()
	if err != nil {
		return nil, err
	}
	r.Model = &ms.Model{
		ID: id,
	}
	return r, nil
}

func (s *commentManageSrv) CreateCommentReply(r *ms.CommentReply) (*ms.CommentReply, error) {
	res, err := s.q.CreateCommentReply.Exec(r.CommentID, r.UserID, r.Content,
		r.AtUserID, r.IP, r.IPLoc, time.Now().Unix())
	if err != nil {
		return nil, err
	}
	id, err := res.LastInsertId()
	if err != nil {
		return nil, err
	}
	r.Model = &ms.Model{ID: id}
	return r, nil
}

func (s *commentManageSrv) DeleteCommentReply(r *ms.CommentReply) error {
	return s.db.Withx(func(tx *sqlx.Tx) error {
		now := time.Now().Unix()
		if _, err := tx.Stmtx(s.q.DeleteCommentReply).Exec(now, r.ID); err != nil {
			return err
		}
		_, err := tx.Stmtx(s.q.DeleteCommentThumbs).Exec(now, r.UserID, r.CommentID, r.ID)
		return err
	})
}

func (s *commentManageSrv) CreateCommentContent(r *ms.CommentContent) (*ms.CommentContent, error) {
	res, err := s.q.CreateCommentContent.Exec(r.CommentID, r.UserID, r.Content, r.Type, r.Sort, time.Now().Unix())
	if err != nil {
		return nil, err
	}
	id, err := res.LastInsertId()
	if err != nil {
		return nil, err
	}
	r.Model = &ms.Model{ID: id}
	return r, nil
}

func (s *commentManageSrv) ThumbsUpComment(userId int64, tweetId, commentId int64) error {
	return s.db.Withx(func(tx *sqlx.Tx) error {
		var (
			thumbsUpCount   int32 = 0
			thumbsDownCount int32 = 0
		)
		now := time.Now().Unix()
		commentThumbs := &ms.TweetCommentThumbs{}
		// 检查thumbs状态
		err := tx.Stmtx(s.q.GetTweetCommentThumb).Get(commentThumbs, userId, tweetId, commentId)
		if err == nil {
			switch {
			case commentThumbs.IsThumbsUp == types.Yes && commentThumbs.IsThumbsDown == types.No:
				thumbsUpCount, thumbsDownCount = -1, 0
			case commentThumbs.IsThumbsUp == types.No && commentThumbs.IsThumbsDown == types.No:
				thumbsUpCount, thumbsDownCount = 1, 0
			default:
				thumbsUpCount, thumbsDownCount = 1, -1
				commentThumbs.IsThumbsDown = types.No
			}
			commentThumbs.IsThumbsUp = 1 - commentThumbs.IsThumbsUp
			commentThumbs.ModifiedOn = now
			if _, err := tx.NamedStmt(s.q.UpdateThumbsUpdownComment).Exec(commentThumbs); err != nil {
				return err
			}
		} else {
			commentThumbs = &ms.TweetCommentThumbs{
				UserID:       userId,
				TweetID:      tweetId,
				CommentID:    commentId,
				IsThumbsUp:   types.Yes,
				IsThumbsDown: types.No,
				CommentType:  0,
				Model: &ms.Model{
					CreatedOn: now,
				},
			}
			thumbsUpCount, thumbsDownCount = 1, 0
			if _, err := tx.NamedStmt(s.q.CreateThumbsUpdownComment).Exec(commentThumbs); err != nil {
				return err
			}
		}
		// 更新comment thumbsUpCount
		_, err = tx.Stmtx(s.q.UpdateCommentThumbsCount).Exec(thumbsUpCount, thumbsDownCount, now, commentId)
		return err
	})
}

func (s *commentManageSrv) ThumbsDownComment(userId int64, tweetId, commentId int64) error {
	return s.db.Withx(func(tx *sqlx.Tx) error {
		var (
			thumbsUpCount   int32 = 0
			thumbsDownCount int32 = 0
		)
		now := time.Now().Unix()
		commentThumbs := &ms.TweetCommentThumbs{}
		// 检查thumbs状态
		err := tx.Stmtx(s.q.GetTweetCommentThumb).Get(commentThumbs, userId, tweetId, commentId)
		if err == nil {
			switch {
			case commentThumbs.IsThumbsDown == types.Yes:
				thumbsUpCount, thumbsDownCount = 0, -1
			case commentThumbs.IsThumbsDown == types.No && commentThumbs.IsThumbsUp == types.No:
				thumbsUpCount, thumbsDownCount = 0, 1
			default:
				thumbsUpCount, thumbsDownCount = -1, 1
				commentThumbs.IsThumbsUp = types.No

			}
			commentThumbs.IsThumbsDown = 1 - commentThumbs.IsThumbsDown
			commentThumbs.ModifiedOn = now
			if _, err := tx.NamedStmt(s.q.UpdateThumbsUpdownComment).Exec(commentThumbs); err != nil {
				return err
			}
		} else {
			commentThumbs = &ms.TweetCommentThumbs{
				UserID:       userId,
				TweetID:      tweetId,
				CommentID:    commentId,
				IsThumbsUp:   types.No,
				IsThumbsDown: types.Yes,
				CommentType:  0,
				Model: &ms.Model{
					CreatedOn: now,
				},
			}
			thumbsUpCount, thumbsDownCount = 0, 1
			if _, err := tx.NamedStmt(s.q.CreateThumbsUpdownComment).Exec(commentThumbs); err != nil {
				return err
			}
		}
		// 更新comment thumbsUpCount
		_, err = tx.Stmtx(s.q.UpdateCommentThumbsCount).Exec(thumbsUpCount, thumbsDownCount, now, commentId)
		return err
	})
}

func (s *commentManageSrv) ThumbsUpReply(userId int64, tweetId, commentId, replyId int64) error {
	return s.db.Withx(func(tx *sqlx.Tx) error {
		var (
			thumbsUpCount   int32 = 0
			thumbsDownCount int32 = 0
		)
		now := time.Now().Unix()
		commentThumbs := &ms.TweetCommentThumbs{}
		// 检查thumbs状态
		err := tx.Stmtx(s.q.GetCommentReplyThumb).Get(commentThumbs, userId, tweetId, commentId, replyId)
		if err == nil {
			switch {
			case commentThumbs.IsThumbsUp == types.Yes:
				thumbsUpCount, thumbsDownCount = -1, 0
			case commentThumbs.IsThumbsUp == types.No && commentThumbs.IsThumbsDown == types.No:
				thumbsUpCount, thumbsDownCount = 1, 0
			default:
				thumbsUpCount, thumbsDownCount = 1, -1
				commentThumbs.IsThumbsDown = types.No
			}
			commentThumbs.IsThumbsUp = 1 - commentThumbs.IsThumbsUp
			commentThumbs.ModifiedOn = now
			if _, err := tx.NamedStmt(s.q.UpdateThumbsUpdownComment).Exec(commentThumbs); err != nil {
				return err
			}
		} else {
			commentThumbs = &ms.TweetCommentThumbs{
				UserID:       userId,
				TweetID:      tweetId,
				CommentID:    commentId,
				ReplyID:      replyId,
				IsThumbsUp:   types.Yes,
				IsThumbsDown: types.No,
				CommentType:  1,
				Model: &ms.Model{
					CreatedOn: now,
				},
			}
			thumbsUpCount, thumbsDownCount = 1, 0
			if _, err := tx.NamedStmt(s.q.CreateThumbsUpdownComment).Exec(commentThumbs); err != nil {
				return err
			}
		}
		// 更新comment_reply thumbsUpCount
		_, err = tx.Stmtx(s.q.UpdateReplyThumbsCount).Exec(thumbsUpCount, thumbsDownCount, now, replyId)
		return err
	})
}

func (s *commentManageSrv) ThumbsDownReply(userId int64, tweetId, commentId, replyId int64) error {
	return s.db.Withx(func(tx *sqlx.Tx) error {
		var (
			thumbsUpCount   int32 = 0
			thumbsDownCount int32 = 0
		)
		now := time.Now().Unix()
		commentThumbs := &ms.TweetCommentThumbs{}
		// 检查thumbs状态
		err := tx.Stmtx(s.q.GetCommentReplyThumb).Get(commentThumbs, userId, tweetId, commentId, replyId)
		if err == nil {
			switch {
			case commentThumbs.IsThumbsDown == types.Yes:
				thumbsUpCount, thumbsDownCount = 0, -1
			case commentThumbs.IsThumbsUp == types.No && commentThumbs.IsThumbsDown == types.No:
				thumbsUpCount, thumbsDownCount = 0, 1
			default:
				thumbsUpCount, thumbsDownCount = -1, 1
				commentThumbs.IsThumbsUp = types.No
			}
			commentThumbs.IsThumbsDown = 1 - commentThumbs.IsThumbsDown
			commentThumbs.ModifiedOn = now
			if _, err := tx.NamedStmt(s.q.UpdateThumbsUpdownComment).Exec(commentThumbs); err != nil {
				return err
			}
		} else {
			commentThumbs = &ms.TweetCommentThumbs{
				UserID:       userId,
				TweetID:      tweetId,
				CommentID:    commentId,
				ReplyID:      replyId,
				IsThumbsUp:   types.No,
				IsThumbsDown: types.Yes,
				CommentType:  1,
				Model: &ms.Model{
					CreatedOn: now,
				},
			}
			thumbsUpCount, thumbsDownCount = 0, 1
			if _, err := tx.NamedStmt(s.q.CreateThumbsUpdownComment).Exec(commentThumbs); err != nil {
				return err
			}
		}
		// 更新comment_reply thumbsUpCount
		_, err = tx.Stmtx(s.q.UpdateReplyThumbsCount).Exec(thumbsUpCount, thumbsDownCount, now, replyId)
		return err
	})
}

func newCommentService(db *sqlx.DB) core.CommentService {
	return &commentSrv{
		sqlxSrv: newSqlxSrv(db),
		q:       ccBuild(db, cc.BuildComment),
	}
}

func newCommentManageService(db *sqlx.DB) (s core.CommentManageService) {
	cms := &commentManageSrv{
		sqlxSrv: newSqlxSrv(db),
		q:       ccBuild(db, cc.BuildCommentManage),
	}
	s = cms
	if cfg.Any("PostgreSQL", "PgSQL", "Postgres") {
		s = &pgcCommentManageSrv{
			commentManageSrv: cms,
			p:                pgcBuild(db, pgc.BuildCommentManage),
		}
	}
	return
}