2017-12-04 00:14:26 +01:00
|
|
|
// Copyright 2017 The Gitea Authors. All rights reserved.
|
|
|
|
// Use of this source code is governed by a MIT-style
|
|
|
|
// license that can be found in the LICENSE file.
|
|
|
|
|
|
|
|
package models
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"fmt"
|
|
|
|
|
2021-09-19 13:49:59 +02:00
|
|
|
"code.gitea.io/gitea/models/db"
|
2021-12-10 02:27:50 +01:00
|
|
|
repo_model "code.gitea.io/gitea/models/repo"
|
2021-11-24 10:49:20 +01:00
|
|
|
user_model "code.gitea.io/gitea/models/user"
|
2017-12-05 21:57:01 +01:00
|
|
|
"code.gitea.io/gitea/modules/setting"
|
2019-08-15 16:46:21 +02:00
|
|
|
"code.gitea.io/gitea/modules/timeutil"
|
2017-12-05 21:57:01 +01:00
|
|
|
|
2019-06-23 17:22:43 +02:00
|
|
|
"xorm.io/builder"
|
2017-12-04 00:14:26 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
// Reaction represents a reactions on issues and comments.
|
|
|
|
type Reaction struct {
|
2020-11-10 23:37:11 +01:00
|
|
|
ID int64 `xorm:"pk autoincr"`
|
|
|
|
Type string `xorm:"INDEX UNIQUE(s) NOT NULL"`
|
|
|
|
IssueID int64 `xorm:"INDEX UNIQUE(s) NOT NULL"`
|
|
|
|
CommentID int64 `xorm:"INDEX UNIQUE(s)"`
|
|
|
|
UserID int64 `xorm:"INDEX UNIQUE(s) NOT NULL"`
|
|
|
|
OriginalAuthorID int64 `xorm:"INDEX UNIQUE(s) NOT NULL DEFAULT(0)"`
|
|
|
|
OriginalAuthor string `xorm:"INDEX UNIQUE(s)"`
|
2021-11-24 10:49:20 +01:00
|
|
|
User *user_model.User `xorm:"-"`
|
2020-01-15 12:14:07 +01:00
|
|
|
CreatedUnix timeutil.TimeStamp `xorm:"INDEX created"`
|
2017-12-04 00:14:26 +01:00
|
|
|
}
|
|
|
|
|
2021-09-19 13:49:59 +02:00
|
|
|
func init() {
|
|
|
|
db.RegisterModel(new(Reaction))
|
|
|
|
}
|
|
|
|
|
2017-12-04 00:14:26 +01:00
|
|
|
// FindReactionsOptions describes the conditions to Find reactions
|
|
|
|
type FindReactionsOptions struct {
|
2021-09-24 13:32:56 +02:00
|
|
|
db.ListOptions
|
2017-12-04 00:14:26 +01:00
|
|
|
IssueID int64
|
|
|
|
CommentID int64
|
2019-12-31 09:21:21 +01:00
|
|
|
UserID int64
|
|
|
|
Reaction string
|
2017-12-04 00:14:26 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
func (opts *FindReactionsOptions) toConds() builder.Cond {
|
2021-03-14 19:52:12 +01:00
|
|
|
// If Issue ID is set add to Query
|
|
|
|
cond := builder.NewCond()
|
2017-12-04 00:14:26 +01:00
|
|
|
if opts.IssueID > 0 {
|
|
|
|
cond = cond.And(builder.Eq{"reaction.issue_id": opts.IssueID})
|
|
|
|
}
|
2021-03-14 19:52:12 +01:00
|
|
|
// If CommentID is > 0 add to Query
|
|
|
|
// If it is 0 Query ignore CommentID to select
|
|
|
|
// If it is -1 it explicit search of Issue Reactions where CommentID = 0
|
2017-12-04 00:14:26 +01:00
|
|
|
if opts.CommentID > 0 {
|
|
|
|
cond = cond.And(builder.Eq{"reaction.comment_id": opts.CommentID})
|
2019-12-07 23:04:19 +01:00
|
|
|
} else if opts.CommentID == -1 {
|
|
|
|
cond = cond.And(builder.Eq{"reaction.comment_id": 0})
|
2017-12-04 00:14:26 +01:00
|
|
|
}
|
2019-12-31 09:21:21 +01:00
|
|
|
if opts.UserID > 0 {
|
2020-01-15 12:14:07 +01:00
|
|
|
cond = cond.And(builder.Eq{
|
|
|
|
"reaction.user_id": opts.UserID,
|
|
|
|
"reaction.original_author_id": 0,
|
|
|
|
})
|
2019-12-31 09:21:21 +01:00
|
|
|
}
|
|
|
|
if opts.Reaction != "" {
|
|
|
|
cond = cond.And(builder.Eq{"reaction.type": opts.Reaction})
|
|
|
|
}
|
2019-12-07 23:04:19 +01:00
|
|
|
|
2017-12-04 00:14:26 +01:00
|
|
|
return cond
|
|
|
|
}
|
|
|
|
|
2019-12-07 23:04:19 +01:00
|
|
|
// FindCommentReactions returns a ReactionList of all reactions from an comment
|
2021-12-15 06:39:34 +01:00
|
|
|
func FindCommentReactions(comment *Comment) (ReactionList, int64, error) {
|
2021-09-23 17:45:36 +02:00
|
|
|
return findReactions(db.GetEngine(db.DefaultContext), FindReactionsOptions{
|
2019-12-07 23:04:19 +01:00
|
|
|
IssueID: comment.IssueID,
|
2021-03-14 19:52:12 +01:00
|
|
|
CommentID: comment.ID,
|
|
|
|
})
|
2019-12-07 23:04:19 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// FindIssueReactions returns a ReactionList of all reactions from an issue
|
2021-12-15 06:39:34 +01:00
|
|
|
func FindIssueReactions(issue *Issue, listOptions db.ListOptions) (ReactionList, int64, error) {
|
2021-09-23 17:45:36 +02:00
|
|
|
return findReactions(db.GetEngine(db.DefaultContext), FindReactionsOptions{
|
2020-01-24 20:00:29 +01:00
|
|
|
ListOptions: listOptions,
|
|
|
|
IssueID: issue.ID,
|
|
|
|
CommentID: -1,
|
2019-12-07 23:04:19 +01:00
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-12-15 06:39:34 +01:00
|
|
|
func findReactions(e db.Engine, opts FindReactionsOptions) ([]*Reaction, int64, error) {
|
|
|
|
sess := e.
|
2020-01-24 20:00:29 +01:00
|
|
|
Where(opts.toConds()).
|
2019-12-18 14:07:36 +01:00
|
|
|
In("reaction.`type`", setting.UI.Reactions).
|
2020-01-24 20:00:29 +01:00
|
|
|
Asc("reaction.issue_id", "reaction.comment_id", "reaction.created_unix", "reaction.id")
|
|
|
|
if opts.Page != 0 {
|
2021-12-15 06:39:34 +01:00
|
|
|
sess = db.SetSessionPagination(sess, &opts)
|
2020-01-24 20:00:29 +01:00
|
|
|
|
|
|
|
reactions := make([]*Reaction, 0, opts.PageSize)
|
2021-12-15 06:39:34 +01:00
|
|
|
count, err := sess.FindAndCount(&reactions)
|
|
|
|
return reactions, count, err
|
2020-01-24 20:00:29 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
reactions := make([]*Reaction, 0, 10)
|
2021-12-15 06:39:34 +01:00
|
|
|
count, err := sess.FindAndCount(&reactions)
|
|
|
|
return reactions, count, err
|
2017-12-04 00:14:26 +01:00
|
|
|
}
|
|
|
|
|
2021-11-21 16:41:00 +01:00
|
|
|
func createReaction(e db.Engine, opts *ReactionOptions) (*Reaction, error) {
|
2017-12-04 00:14:26 +01:00
|
|
|
reaction := &Reaction{
|
|
|
|
Type: opts.Type,
|
|
|
|
UserID: opts.Doer.ID,
|
|
|
|
IssueID: opts.Issue.ID,
|
|
|
|
}
|
2019-12-31 09:21:21 +01:00
|
|
|
findOpts := FindReactionsOptions{
|
|
|
|
IssueID: opts.Issue.ID,
|
|
|
|
CommentID: -1, // reaction to issue only
|
|
|
|
Reaction: opts.Type,
|
|
|
|
UserID: opts.Doer.ID,
|
|
|
|
}
|
2017-12-04 00:14:26 +01:00
|
|
|
if opts.Comment != nil {
|
|
|
|
reaction.CommentID = opts.Comment.ID
|
2019-12-31 09:21:21 +01:00
|
|
|
findOpts.CommentID = opts.Comment.ID
|
|
|
|
}
|
|
|
|
|
2021-12-15 06:39:34 +01:00
|
|
|
existingR, _, err := findReactions(e, findOpts)
|
2019-12-31 09:21:21 +01:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
if len(existingR) > 0 {
|
|
|
|
return existingR[0], ErrReactionAlreadyExist{Reaction: opts.Type}
|
2017-12-04 00:14:26 +01:00
|
|
|
}
|
2019-12-31 09:21:21 +01:00
|
|
|
|
2017-12-04 00:14:26 +01:00
|
|
|
if _, err := e.Insert(reaction); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return reaction, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// ReactionOptions defines options for creating or deleting reactions
|
|
|
|
type ReactionOptions struct {
|
|
|
|
Type string
|
2021-11-24 10:49:20 +01:00
|
|
|
Doer *user_model.User
|
2017-12-04 00:14:26 +01:00
|
|
|
Issue *Issue
|
|
|
|
Comment *Comment
|
|
|
|
}
|
|
|
|
|
|
|
|
// CreateReaction creates reaction for issue or comment.
|
2019-12-31 09:21:21 +01:00
|
|
|
func CreateReaction(opts *ReactionOptions) (*Reaction, error) {
|
2019-12-07 23:04:19 +01:00
|
|
|
if !setting.UI.ReactionsMap[opts.Type] {
|
|
|
|
return nil, ErrForbiddenIssueReaction{opts.Type}
|
|
|
|
}
|
|
|
|
|
2021-11-21 16:41:00 +01:00
|
|
|
ctx, committer, err := db.TxContext()
|
|
|
|
if err != nil {
|
2017-12-04 00:14:26 +01:00
|
|
|
return nil, err
|
|
|
|
}
|
2021-11-21 16:41:00 +01:00
|
|
|
defer committer.Close()
|
2017-12-04 00:14:26 +01:00
|
|
|
|
2021-11-21 16:41:00 +01:00
|
|
|
reaction, err := createReaction(db.GetEngine(ctx), opts)
|
2017-12-04 00:14:26 +01:00
|
|
|
if err != nil {
|
2019-12-31 09:21:21 +01:00
|
|
|
return reaction, err
|
2017-12-04 00:14:26 +01:00
|
|
|
}
|
|
|
|
|
2021-11-21 16:41:00 +01:00
|
|
|
if err := committer.Commit(); err != nil {
|
2017-12-04 00:14:26 +01:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return reaction, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// CreateIssueReaction creates a reaction on issue.
|
2021-11-24 10:49:20 +01:00
|
|
|
func CreateIssueReaction(doer *user_model.User, issue *Issue, content string) (*Reaction, error) {
|
2017-12-04 00:14:26 +01:00
|
|
|
return CreateReaction(&ReactionOptions{
|
|
|
|
Type: content,
|
|
|
|
Doer: doer,
|
|
|
|
Issue: issue,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
// CreateCommentReaction creates a reaction on comment.
|
2021-11-24 10:49:20 +01:00
|
|
|
func CreateCommentReaction(doer *user_model.User, issue *Issue, comment *Comment, content string) (*Reaction, error) {
|
2017-12-04 00:14:26 +01:00
|
|
|
return CreateReaction(&ReactionOptions{
|
|
|
|
Type: content,
|
|
|
|
Doer: doer,
|
|
|
|
Issue: issue,
|
|
|
|
Comment: comment,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-09-19 13:49:59 +02:00
|
|
|
func deleteReaction(e db.Engine, opts *ReactionOptions) error {
|
2017-12-04 00:14:26 +01:00
|
|
|
reaction := &Reaction{
|
2021-01-22 03:56:19 +01:00
|
|
|
Type: opts.Type,
|
|
|
|
}
|
|
|
|
if opts.Doer != nil {
|
|
|
|
reaction.UserID = opts.Doer.ID
|
|
|
|
}
|
|
|
|
if opts.Issue != nil {
|
|
|
|
reaction.IssueID = opts.Issue.ID
|
2017-12-04 00:14:26 +01:00
|
|
|
}
|
|
|
|
if opts.Comment != nil {
|
|
|
|
reaction.CommentID = opts.Comment.ID
|
|
|
|
}
|
2020-01-15 12:14:07 +01:00
|
|
|
_, err := e.Where("original_author_id = 0").Delete(reaction)
|
2017-12-04 00:14:26 +01:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
// DeleteReaction deletes reaction for issue or comment.
|
|
|
|
func DeleteReaction(opts *ReactionOptions) error {
|
2021-11-21 16:41:00 +01:00
|
|
|
ctx, committer, err := db.TxContext()
|
|
|
|
if err != nil {
|
2017-12-04 00:14:26 +01:00
|
|
|
return err
|
|
|
|
}
|
2021-11-21 16:41:00 +01:00
|
|
|
defer committer.Close()
|
2017-12-04 00:14:26 +01:00
|
|
|
|
2021-11-21 16:41:00 +01:00
|
|
|
if err := deleteReaction(db.GetEngine(ctx), opts); err != nil {
|
2017-12-04 00:14:26 +01:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2021-11-21 16:41:00 +01:00
|
|
|
return committer.Commit()
|
2017-12-04 00:14:26 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// DeleteIssueReaction deletes a reaction on issue.
|
2021-11-24 10:49:20 +01:00
|
|
|
func DeleteIssueReaction(doer *user_model.User, issue *Issue, content string) error {
|
2017-12-04 00:14:26 +01:00
|
|
|
return DeleteReaction(&ReactionOptions{
|
|
|
|
Type: content,
|
|
|
|
Doer: doer,
|
|
|
|
Issue: issue,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
// DeleteCommentReaction deletes a reaction on comment.
|
2021-11-24 10:49:20 +01:00
|
|
|
func DeleteCommentReaction(doer *user_model.User, issue *Issue, comment *Comment, content string) error {
|
2017-12-04 00:14:26 +01:00
|
|
|
return DeleteReaction(&ReactionOptions{
|
|
|
|
Type: content,
|
|
|
|
Doer: doer,
|
|
|
|
Issue: issue,
|
|
|
|
Comment: comment,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2019-12-07 23:04:19 +01:00
|
|
|
// LoadUser load user of reaction
|
2021-11-24 10:49:20 +01:00
|
|
|
func (r *Reaction) LoadUser() (*user_model.User, error) {
|
2019-12-07 23:04:19 +01:00
|
|
|
if r.User != nil {
|
|
|
|
return r.User, nil
|
|
|
|
}
|
2021-11-24 10:49:20 +01:00
|
|
|
user, err := user_model.GetUserByIDCtx(db.DefaultContext, r.UserID)
|
2019-12-07 23:04:19 +01:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
r.User = user
|
|
|
|
return user, nil
|
|
|
|
}
|
|
|
|
|
2017-12-04 00:14:26 +01:00
|
|
|
// ReactionList represents list of reactions
|
|
|
|
type ReactionList []*Reaction
|
|
|
|
|
|
|
|
// HasUser check if user has reacted
|
|
|
|
func (list ReactionList) HasUser(userID int64) bool {
|
|
|
|
if userID == 0 {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
for _, reaction := range list {
|
2020-01-15 12:14:07 +01:00
|
|
|
if reaction.OriginalAuthor == "" && reaction.UserID == userID {
|
2017-12-04 00:14:26 +01:00
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
// GroupByType returns reactions grouped by type
|
|
|
|
func (list ReactionList) GroupByType() map[string]ReactionList {
|
2021-03-14 19:52:12 +01:00
|
|
|
reactions := make(map[string]ReactionList)
|
2017-12-04 00:14:26 +01:00
|
|
|
for _, reaction := range list {
|
|
|
|
reactions[reaction.Type] = append(reactions[reaction.Type], reaction)
|
|
|
|
}
|
|
|
|
return reactions
|
|
|
|
}
|
|
|
|
|
|
|
|
func (list ReactionList) getUserIDs() []int64 {
|
|
|
|
userIDs := make(map[int64]struct{}, len(list))
|
|
|
|
for _, reaction := range list {
|
2020-01-15 12:14:07 +01:00
|
|
|
if reaction.OriginalAuthor != "" {
|
|
|
|
continue
|
|
|
|
}
|
2017-12-04 00:14:26 +01:00
|
|
|
if _, ok := userIDs[reaction.UserID]; !ok {
|
|
|
|
userIDs[reaction.UserID] = struct{}{}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return keysInt64(userIDs)
|
|
|
|
}
|
|
|
|
|
2021-12-10 02:27:50 +01:00
|
|
|
func (list ReactionList) loadUsers(e db.Engine, repo *repo_model.Repository) ([]*user_model.User, error) {
|
2017-12-04 00:14:26 +01:00
|
|
|
if len(list) == 0 {
|
|
|
|
return nil, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
userIDs := list.getUserIDs()
|
2021-11-24 10:49:20 +01:00
|
|
|
userMaps := make(map[int64]*user_model.User, len(userIDs))
|
2017-12-04 00:14:26 +01:00
|
|
|
err := e.
|
|
|
|
In("id", userIDs).
|
|
|
|
Find(&userMaps)
|
|
|
|
if err != nil {
|
|
|
|
return nil, fmt.Errorf("find user: %v", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, reaction := range list {
|
2020-01-15 12:14:07 +01:00
|
|
|
if reaction.OriginalAuthor != "" {
|
2021-11-24 10:49:20 +01:00
|
|
|
reaction.User = user_model.NewReplaceUser(fmt.Sprintf("%s(%s)", reaction.OriginalAuthor, repo.OriginalServiceType.Name()))
|
2020-01-15 12:14:07 +01:00
|
|
|
} else if user, ok := userMaps[reaction.UserID]; ok {
|
2017-12-04 00:14:26 +01:00
|
|
|
reaction.User = user
|
|
|
|
} else {
|
2021-11-24 10:49:20 +01:00
|
|
|
reaction.User = user_model.NewGhostUser()
|
2017-12-04 00:14:26 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return valuesUser(userMaps), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// LoadUsers loads reactions' all users
|
2021-12-10 02:27:50 +01:00
|
|
|
func (list ReactionList) LoadUsers(repo *repo_model.Repository) ([]*user_model.User, error) {
|
2021-09-23 17:45:36 +02:00
|
|
|
return list.loadUsers(db.GetEngine(db.DefaultContext), repo)
|
2017-12-04 00:14:26 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// GetFirstUsers returns first reacted user display names separated by comma
|
|
|
|
func (list ReactionList) GetFirstUsers() string {
|
|
|
|
var buffer bytes.Buffer
|
2021-03-14 19:52:12 +01:00
|
|
|
rem := setting.UI.ReactionMaxUserNum
|
2017-12-04 00:14:26 +01:00
|
|
|
for _, reaction := range list {
|
|
|
|
if buffer.Len() > 0 {
|
|
|
|
buffer.WriteString(", ")
|
|
|
|
}
|
|
|
|
buffer.WriteString(reaction.User.DisplayName())
|
|
|
|
if rem--; rem == 0 {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return buffer.String()
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetMoreUserCount returns count of not shown users in reaction tooltip
|
|
|
|
func (list ReactionList) GetMoreUserCount() int {
|
|
|
|
if len(list) <= setting.UI.ReactionMaxUserNum {
|
|
|
|
return 0
|
|
|
|
}
|
|
|
|
return len(list) - setting.UI.ReactionMaxUserNum
|
|
|
|
}
|
2022-02-01 19:20:28 +01:00
|
|
|
|
|
|
|
// RemapExternalUser ExternalUserRemappable interface
|
|
|
|
func (r *Reaction) RemapExternalUser(externalName string, externalID, userID int64) error {
|
|
|
|
r.OriginalAuthor = externalName
|
|
|
|
r.OriginalAuthorID = externalID
|
|
|
|
r.UserID = userID
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetUserID ExternalUserRemappable interface
|
|
|
|
func (r *Reaction) GetUserID() int64 { return r.UserID }
|
|
|
|
|
|
|
|
// GetExternalName ExternalUserRemappable interface
|
|
|
|
func (r *Reaction) GetExternalName() string { return r.OriginalAuthor }
|
|
|
|
|
|
|
|
// GetExternalID ExternalUserRemappable interface
|
|
|
|
func (r *Reaction) GetExternalID() int64 { return r.OriginalAuthorID }
|