92 lines
2.4 KiB
Go
92 lines
2.4 KiB
Go
|
package command
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"fmt"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"go-common/app/tool/saga/model"
|
|||
|
|
"go-common/app/tool/saga/service/notification"
|
|||
|
|
"go-common/library/log"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func (c *Command) runPlusOne(ctx context.Context, event *model.HookComment, repo *model.Repo) (err error) {
|
|||
|
|
var (
|
|||
|
|
author string
|
|||
|
|
url = event.ObjectAttributes.URL
|
|||
|
|
commit = event.MergeRequest.LastCommit.ID
|
|||
|
|
reviewer = event.User.UserName
|
|||
|
|
authorID = int(event.MergeRequest.AuthorID)
|
|||
|
|
sourceBranch = event.MergeRequest.SourceBranch
|
|||
|
|
targetBranch = event.MergeRequest.TargetBranch
|
|||
|
|
wip = event.MergeRequest.WorkInProgress
|
|||
|
|
)
|
|||
|
|
log.Info("runPlusOne start ...")
|
|||
|
|
if wip {
|
|||
|
|
c.gitlab.CreateMRNote(event.Project.ID, int(event.MergeRequest.IID), fmt.Sprintf("<pre>警告：当前MR处于WIP状态，请待开发结束后再review！</pre>"))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if author, err = c.gitlab.UserName(authorID); err != nil {
|
|||
|
|
log.Error("%+v", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log.Info("runPlusOne notification author: %s", author)
|
|||
|
|
if author != "" {
|
|||
|
|
go func() {
|
|||
|
|
notification.MailPlusOne(author, reviewer, url, commit, sourceBranch, targetBranch)
|
|||
|
|
}()
|
|||
|
|
go func() {
|
|||
|
|
notification.WechatPlusOne(c.dao, author, reviewer, url, commit, sourceBranch, targetBranch)
|
|||
|
|
}()
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func reviewedOwner(owners []string, reviewedUsers []string, username string) (isowner bool, reviewed bool) {
|
|||
|
|
for _, owner := range owners {
|
|||
|
|
if strings.EqualFold(owner, username) || strings.EqualFold(owner, "all") {
|
|||
|
|
return true, true
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
for _, owner := range owners {
|
|||
|
|
for _, user := range reviewedUsers {
|
|||
|
|
if strings.EqualFold(user, owner) {
|
|||
|
|
return false, true
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return false, false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (c *Command) reviewedUsers(projID int, mrIID int) (reviewedUsers []string, err error) {
|
|||
|
|
var awardUsers []string
|
|||
|
|
if reviewedUsers, err = c.gitlab.PlusUsernames(projID, mrIID); err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if awardUsers, err = c.gitlab.AwardEmojiUsernames(projID, mrIID); err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
OUTER:
|
|||
|
|
for _, au := range awardUsers {
|
|||
|
|
for _, mu := range reviewedUsers {
|
|||
|
|
if au == mu {
|
|||
|
|
continue OUTER
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
reviewedUsers = append(reviewedUsers, au)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func reviewedNum(reviewers []string, reviewedUsers []string) (num int) {
|
|||
|
|
for _, reviewer := range reviewers {
|
|||
|
|
for _, user := range reviewedUsers {
|
|||
|
|
if strings.EqualFold(user, reviewer) || strings.EqualFold(reviewer, "all") {
|
|||
|
|
num++
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|