mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-12-11 11:06:29 +00:00
134 lines
3 KiB
Go
134 lines
3 KiB
Go
package notify
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
|
|
"github.com/Bugagazavr/go-gitlab-client"
|
|
"github.com/drone/drone/plugin/remote/gitlab"
|
|
"github.com/drone/drone/shared/model"
|
|
)
|
|
|
|
type Gitlab struct {
|
|
SkipVerify bool `yaml:"skip_verify,omitempty"`
|
|
Started bool `yaml:"on_started,omitempty"`
|
|
Success bool `yaml:"on_success,omitempty"`
|
|
Failure bool `yaml:"on_failure,omitempty"`
|
|
}
|
|
|
|
type GitlabClient interface {
|
|
SendRepoCommitComment(id string, sha string, body string) (*gogitlab.CommitComment, error)
|
|
}
|
|
|
|
const (
|
|
StatusPending = ":raised_hand:"
|
|
StatusSuccess = ":thumbsup:"
|
|
StatusFailure = ":thumbsdown:"
|
|
StatusError = ":exclamation:"
|
|
)
|
|
|
|
const (
|
|
DescPending = "this build is pending"
|
|
DescSuccess = "the build was successful"
|
|
DescFailure = "the build failed"
|
|
DescError = "oops, something went wrong"
|
|
)
|
|
|
|
const (
|
|
PRMasterBranch = "master"
|
|
PRBadToMerge = " -> bad to merge"
|
|
PRGoodToMerge = " -> good to merge"
|
|
)
|
|
|
|
// Send uses the Gitlab repository API to comment the commit
|
|
func (g *Gitlab) Send(context *model.Request) error {
|
|
if !g.isRequested(context) {
|
|
return nil
|
|
}
|
|
|
|
return g.send(
|
|
context,
|
|
gitlab.NewClient(fmt.Sprintf("http://%s", context.Repo.Host), context.User.Access, g.SkipVerify),
|
|
)
|
|
}
|
|
|
|
func (g *Gitlab) isRequested(context *model.Request) bool {
|
|
if context.Repo.Remote != model.RemoteGitlab {
|
|
return false
|
|
}
|
|
|
|
switch context.Commit.Status {
|
|
case model.StatusStarted:
|
|
if !g.Started {
|
|
return false
|
|
}
|
|
case model.StatusSuccess:
|
|
if !g.Success {
|
|
return false
|
|
}
|
|
case model.StatusFailure, model.StatusError, model.StatusKilled:
|
|
if !g.Failure {
|
|
return false
|
|
}
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func (g *Gitlab) send(context *model.Request, client GitlabClient) error {
|
|
msg := fmt.Sprintf(
|
|
"[%s](%s) %s%s",
|
|
getDesc(context.Commit.Status),
|
|
getBuildUrl(context),
|
|
getStatus(context.Commit.Status),
|
|
getMergeRequestComment(context.Commit.Branch, context.Commit.Status),
|
|
)
|
|
|
|
_, err := client.SendRepoCommitComment(strconv.FormatInt(context.Repo.ID, 10), context.Commit.Sha, msg)
|
|
|
|
return err
|
|
}
|
|
|
|
// getStatus converts a Drone status to a Gitlab status.
|
|
func getStatus(status string) string {
|
|
switch status {
|
|
case model.StatusEnqueue, model.StatusStarted:
|
|
return StatusPending
|
|
case model.StatusSuccess:
|
|
return StatusSuccess
|
|
case model.StatusFailure:
|
|
return StatusFailure
|
|
case model.StatusError, model.StatusKilled:
|
|
return StatusError
|
|
default:
|
|
return StatusError
|
|
}
|
|
}
|
|
|
|
// getDesc generates a description message for the comment based on the status.
|
|
func getDesc(status string) string {
|
|
switch status {
|
|
case model.StatusEnqueue, model.StatusStarted:
|
|
return DescPending
|
|
case model.StatusSuccess:
|
|
return DescSuccess
|
|
case model.StatusFailure:
|
|
return DescFailure
|
|
case model.StatusError, model.StatusKilled:
|
|
return DescError
|
|
default:
|
|
return DescError
|
|
}
|
|
}
|
|
|
|
func getMergeRequestComment(branch, status string) string {
|
|
if branch != PRMasterBranch {
|
|
switch status {
|
|
case model.StatusSuccess:
|
|
return PRGoodToMerge
|
|
case model.StatusFailure:
|
|
return PRBadToMerge
|
|
}
|
|
}
|
|
return ""
|
|
}
|