forked from NYANDEV/forgejo
d6b97c8557
opening issues generates a webhook to discord that contains a url to the gitea api. the message title in discord is therefore referencing to the api instead of the issue itself.
502 lines
15 KiB
Go
502 lines
15 KiB
Go
// 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 (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"code.gitea.io/git"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
api "code.gitea.io/sdk/gitea"
|
|
)
|
|
|
|
type (
|
|
// DiscordEmbedFooter for Embed Footer Structure.
|
|
DiscordEmbedFooter struct {
|
|
Text string `json:"text"`
|
|
}
|
|
|
|
// DiscordEmbedAuthor for Embed Author Structure
|
|
DiscordEmbedAuthor struct {
|
|
Name string `json:"name"`
|
|
URL string `json:"url"`
|
|
IconURL string `json:"icon_url"`
|
|
}
|
|
|
|
// DiscordEmbedField for Embed Field Structure
|
|
DiscordEmbedField struct {
|
|
Name string `json:"name"`
|
|
Value string `json:"value"`
|
|
}
|
|
|
|
// DiscordEmbed is for Embed Structure
|
|
DiscordEmbed struct {
|
|
Title string `json:"title"`
|
|
Description string `json:"description"`
|
|
URL string `json:"url"`
|
|
Color int `json:"color"`
|
|
Footer DiscordEmbedFooter `json:"footer"`
|
|
Author DiscordEmbedAuthor `json:"author"`
|
|
Fields []DiscordEmbedField `json:"fields"`
|
|
}
|
|
|
|
// DiscordPayload represents
|
|
DiscordPayload struct {
|
|
Wait bool `json:"wait"`
|
|
Content string `json:"content"`
|
|
Username string `json:"username"`
|
|
AvatarURL string `json:"avatar_url"`
|
|
TTS bool `json:"tts"`
|
|
Embeds []DiscordEmbed `json:"embeds"`
|
|
}
|
|
|
|
// DiscordMeta contains the discord metadata
|
|
DiscordMeta struct {
|
|
Username string `json:"username"`
|
|
IconURL string `json:"icon_url"`
|
|
}
|
|
)
|
|
|
|
func color(clr string) int {
|
|
if clr != "" {
|
|
clr = strings.TrimLeft(clr, "#")
|
|
if s, err := strconv.ParseInt(clr, 16, 32); err == nil {
|
|
return int(s)
|
|
}
|
|
}
|
|
|
|
return 0
|
|
}
|
|
|
|
var (
|
|
successColor = color("1ac600")
|
|
warnColor = color("ffd930")
|
|
failedColor = color("ff3232")
|
|
)
|
|
|
|
// SetSecret sets the discord secret
|
|
func (p *DiscordPayload) SetSecret(_ string) {}
|
|
|
|
// JSONPayload Marshals the DiscordPayload to json
|
|
func (p *DiscordPayload) JSONPayload() ([]byte, error) {
|
|
data, err := json.MarshalIndent(p, "", " ")
|
|
if err != nil {
|
|
return []byte{}, err
|
|
}
|
|
return data, nil
|
|
}
|
|
|
|
func getDiscordCreatePayload(p *api.CreatePayload, meta *DiscordMeta) (*DiscordPayload, error) {
|
|
// created tag/branch
|
|
refName := git.RefEndName(p.Ref)
|
|
title := fmt.Sprintf("[%s] %s %s created", p.Repo.FullName, p.RefType, refName)
|
|
|
|
return &DiscordPayload{
|
|
Username: meta.Username,
|
|
AvatarURL: meta.IconURL,
|
|
Embeds: []DiscordEmbed{
|
|
{
|
|
Title: title,
|
|
URL: p.Repo.HTMLURL + "/src/" + refName,
|
|
Color: successColor,
|
|
Author: DiscordEmbedAuthor{
|
|
Name: p.Sender.UserName,
|
|
URL: setting.AppURL + p.Sender.UserName,
|
|
IconURL: p.Sender.AvatarURL,
|
|
},
|
|
},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func getDiscordDeletePayload(p *api.DeletePayload, meta *DiscordMeta) (*DiscordPayload, error) {
|
|
// deleted tag/branch
|
|
refName := git.RefEndName(p.Ref)
|
|
title := fmt.Sprintf("[%s] %s %s deleted", p.Repo.FullName, p.RefType, refName)
|
|
|
|
return &DiscordPayload{
|
|
Username: meta.Username,
|
|
AvatarURL: meta.IconURL,
|
|
Embeds: []DiscordEmbed{
|
|
{
|
|
Title: title,
|
|
URL: p.Repo.HTMLURL + "/src/" + refName,
|
|
Color: warnColor,
|
|
Author: DiscordEmbedAuthor{
|
|
Name: p.Sender.UserName,
|
|
URL: setting.AppURL + p.Sender.UserName,
|
|
IconURL: p.Sender.AvatarURL,
|
|
},
|
|
},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func getDiscordForkPayload(p *api.ForkPayload, meta *DiscordMeta) (*DiscordPayload, error) {
|
|
// fork
|
|
title := fmt.Sprintf("%s is forked to %s", p.Forkee.FullName, p.Repo.FullName)
|
|
|
|
return &DiscordPayload{
|
|
Username: meta.Username,
|
|
AvatarURL: meta.IconURL,
|
|
Embeds: []DiscordEmbed{
|
|
{
|
|
Title: title,
|
|
URL: p.Repo.HTMLURL,
|
|
Color: successColor,
|
|
Author: DiscordEmbedAuthor{
|
|
Name: p.Sender.UserName,
|
|
URL: setting.AppURL + p.Sender.UserName,
|
|
IconURL: p.Sender.AvatarURL,
|
|
},
|
|
},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func getDiscordPushPayload(p *api.PushPayload, meta *DiscordMeta) (*DiscordPayload, error) {
|
|
var (
|
|
branchName = git.RefEndName(p.Ref)
|
|
commitDesc string
|
|
)
|
|
|
|
var titleLink string
|
|
if len(p.Commits) == 1 {
|
|
commitDesc = "1 new commit"
|
|
titleLink = p.Commits[0].URL
|
|
} else {
|
|
commitDesc = fmt.Sprintf("%d new commits", len(p.Commits))
|
|
titleLink = p.CompareURL
|
|
}
|
|
if titleLink == "" {
|
|
titleLink = p.Repo.HTMLURL + "/src/" + branchName
|
|
}
|
|
|
|
title := fmt.Sprintf("[%s:%s] %s", p.Repo.FullName, branchName, commitDesc)
|
|
|
|
var text string
|
|
// for each commit, generate attachment text
|
|
for i, commit := range p.Commits {
|
|
text += fmt.Sprintf("[%s](%s) %s - %s", commit.ID[:7], commit.URL,
|
|
strings.TrimRight(commit.Message, "\r\n"), commit.Author.Name)
|
|
// add linebreak to each commit but the last
|
|
if i < len(p.Commits)-1 {
|
|
text += "\n"
|
|
}
|
|
}
|
|
|
|
return &DiscordPayload{
|
|
Username: meta.Username,
|
|
AvatarURL: meta.IconURL,
|
|
Embeds: []DiscordEmbed{
|
|
{
|
|
Title: title,
|
|
Description: text,
|
|
URL: titleLink,
|
|
Color: successColor,
|
|
Author: DiscordEmbedAuthor{
|
|
Name: p.Sender.UserName,
|
|
URL: setting.AppURL + p.Sender.UserName,
|
|
IconURL: p.Sender.AvatarURL,
|
|
},
|
|
},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func getDiscordIssuesPayload(p *api.IssuePayload, meta *DiscordMeta) (*DiscordPayload, error) {
|
|
var text, title string
|
|
var color int
|
|
url := fmt.Sprintf("%s/issues/%d", p.Repository.HTMLURL, p.Issue.Index)
|
|
switch p.Action {
|
|
case api.HookIssueOpened:
|
|
title = fmt.Sprintf("[%s] Issue opened: #%d %s", p.Repository.FullName, p.Index, p.Issue.Title)
|
|
text = p.Issue.Body
|
|
color = warnColor
|
|
case api.HookIssueClosed:
|
|
title = fmt.Sprintf("[%s] Issue closed: #%d %s", p.Repository.FullName, p.Index, p.Issue.Title)
|
|
color = failedColor
|
|
text = p.Issue.Body
|
|
case api.HookIssueReOpened:
|
|
title = fmt.Sprintf("[%s] Issue re-opened: #%d %s", p.Repository.FullName, p.Index, p.Issue.Title)
|
|
text = p.Issue.Body
|
|
color = warnColor
|
|
case api.HookIssueEdited:
|
|
title = fmt.Sprintf("[%s] Issue edited: #%d %s", p.Repository.FullName, p.Index, p.Issue.Title)
|
|
text = p.Issue.Body
|
|
color = warnColor
|
|
case api.HookIssueAssigned:
|
|
title = fmt.Sprintf("[%s] Issue assigned to %s: #%d %s", p.Repository.FullName,
|
|
p.Issue.Assignee.UserName, p.Index, p.Issue.Title)
|
|
text = p.Issue.Body
|
|
color = successColor
|
|
case api.HookIssueUnassigned:
|
|
title = fmt.Sprintf("[%s] Issue unassigned: #%d %s", p.Repository.FullName, p.Index, p.Issue.Title)
|
|
text = p.Issue.Body
|
|
color = warnColor
|
|
case api.HookIssueLabelUpdated:
|
|
title = fmt.Sprintf("[%s] Issue labels updated: #%d %s", p.Repository.FullName, p.Index, p.Issue.Title)
|
|
text = p.Issue.Body
|
|
color = warnColor
|
|
case api.HookIssueLabelCleared:
|
|
title = fmt.Sprintf("[%s] Issue labels cleared: #%d %s", p.Repository.FullName, p.Index, p.Issue.Title)
|
|
text = p.Issue.Body
|
|
color = warnColor
|
|
case api.HookIssueSynchronized:
|
|
title = fmt.Sprintf("[%s] Issue synchronized: #%d %s", p.Repository.FullName, p.Index, p.Issue.Title)
|
|
text = p.Issue.Body
|
|
color = warnColor
|
|
case api.HookIssueMilestoned:
|
|
title = fmt.Sprintf("[%s] Issue milestone: #%d %s", p.Repository.FullName, p.Index, p.Issue.Title)
|
|
text = p.Issue.Body
|
|
color = warnColor
|
|
case api.HookIssueDemilestoned:
|
|
title = fmt.Sprintf("[%s] Issue clear milestone: #%d %s", p.Repository.FullName, p.Index, p.Issue.Title)
|
|
text = p.Issue.Body
|
|
color = warnColor
|
|
}
|
|
|
|
return &DiscordPayload{
|
|
Username: meta.Username,
|
|
AvatarURL: meta.IconURL,
|
|
Embeds: []DiscordEmbed{
|
|
{
|
|
Title: title,
|
|
Description: text,
|
|
URL: url,
|
|
Color: color,
|
|
Author: DiscordEmbedAuthor{
|
|
Name: p.Sender.UserName,
|
|
URL: setting.AppURL + p.Sender.UserName,
|
|
IconURL: p.Sender.AvatarURL,
|
|
},
|
|
},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func getDiscordIssueCommentPayload(p *api.IssueCommentPayload, discord *DiscordMeta) (*DiscordPayload, error) {
|
|
title := fmt.Sprintf("#%d %s", p.Issue.Index, p.Issue.Title)
|
|
url := fmt.Sprintf("%s/issues/%d#%s", p.Repository.HTMLURL, p.Issue.Index, CommentHashTag(p.Comment.ID))
|
|
content := ""
|
|
var color int
|
|
switch p.Action {
|
|
case api.HookIssueCommentCreated:
|
|
title = "New comment: " + title
|
|
content = p.Comment.Body
|
|
color = successColor
|
|
case api.HookIssueCommentEdited:
|
|
title = "Comment edited: " + title
|
|
content = p.Comment.Body
|
|
color = warnColor
|
|
case api.HookIssueCommentDeleted:
|
|
title = "Comment deleted: " + title
|
|
url = fmt.Sprintf("%s/issues/%d", p.Repository.HTMLURL, p.Issue.Index)
|
|
content = p.Comment.Body
|
|
color = warnColor
|
|
}
|
|
|
|
return &DiscordPayload{
|
|
Username: discord.Username,
|
|
AvatarURL: discord.IconURL,
|
|
Embeds: []DiscordEmbed{
|
|
{
|
|
Title: title,
|
|
Description: content,
|
|
URL: url,
|
|
Color: color,
|
|
Author: DiscordEmbedAuthor{
|
|
Name: p.Sender.UserName,
|
|
URL: setting.AppURL + p.Sender.UserName,
|
|
IconURL: p.Sender.AvatarURL,
|
|
},
|
|
},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func getDiscordPullRequestPayload(p *api.PullRequestPayload, meta *DiscordMeta) (*DiscordPayload, error) {
|
|
var text, title string
|
|
var color int
|
|
switch p.Action {
|
|
case api.HookIssueOpened:
|
|
title = fmt.Sprintf("[%s] Pull request opened: #%d %s", p.Repository.FullName, p.Index, p.PullRequest.Title)
|
|
text = p.PullRequest.Body
|
|
color = warnColor
|
|
case api.HookIssueClosed:
|
|
if p.PullRequest.HasMerged {
|
|
title = fmt.Sprintf("[%s] Pull request merged: #%d %s", p.Repository.FullName, p.Index, p.PullRequest.Title)
|
|
color = successColor
|
|
} else {
|
|
title = fmt.Sprintf("[%s] Pull request closed: #%d %s", p.Repository.FullName, p.Index, p.PullRequest.Title)
|
|
color = failedColor
|
|
}
|
|
text = p.PullRequest.Body
|
|
case api.HookIssueReOpened:
|
|
title = fmt.Sprintf("[%s] Pull request re-opened: #%d %s", p.Repository.FullName, p.Index, p.PullRequest.Title)
|
|
text = p.PullRequest.Body
|
|
color = warnColor
|
|
case api.HookIssueEdited:
|
|
title = fmt.Sprintf("[%s] Pull request edited: #%d %s", p.Repository.FullName, p.Index, p.PullRequest.Title)
|
|
text = p.PullRequest.Body
|
|
color = warnColor
|
|
case api.HookIssueAssigned:
|
|
list, err := MakeAssigneeList(&Issue{ID: p.PullRequest.ID})
|
|
if err != nil {
|
|
return &DiscordPayload{}, err
|
|
}
|
|
title = fmt.Sprintf("[%s] Pull request assigned to %s: #%d %s", p.Repository.FullName,
|
|
list, p.Index, p.PullRequest.Title)
|
|
text = p.PullRequest.Body
|
|
color = successColor
|
|
case api.HookIssueUnassigned:
|
|
title = fmt.Sprintf("[%s] Pull request unassigned: #%d %s", p.Repository.FullName, p.Index, p.PullRequest.Title)
|
|
text = p.PullRequest.Body
|
|
color = warnColor
|
|
case api.HookIssueLabelUpdated:
|
|
title = fmt.Sprintf("[%s] Pull request labels updated: #%d %s", p.Repository.FullName, p.Index, p.PullRequest.Title)
|
|
text = p.PullRequest.Body
|
|
color = warnColor
|
|
case api.HookIssueLabelCleared:
|
|
title = fmt.Sprintf("[%s] Pull request labels cleared: #%d %s", p.Repository.FullName, p.Index, p.PullRequest.Title)
|
|
text = p.PullRequest.Body
|
|
color = warnColor
|
|
case api.HookIssueSynchronized:
|
|
title = fmt.Sprintf("[%s] Pull request synchronized: #%d %s", p.Repository.FullName, p.Index, p.PullRequest.Title)
|
|
text = p.PullRequest.Body
|
|
color = warnColor
|
|
case api.HookIssueMilestoned:
|
|
title = fmt.Sprintf("[%s] Pull request milestone: #%d %s", p.Repository.FullName, p.Index, p.PullRequest.Title)
|
|
text = p.PullRequest.Body
|
|
color = warnColor
|
|
case api.HookIssueDemilestoned:
|
|
title = fmt.Sprintf("[%s] Pull request clear milestone: #%d %s", p.Repository.FullName, p.Index, p.PullRequest.Title)
|
|
text = p.PullRequest.Body
|
|
color = warnColor
|
|
}
|
|
|
|
return &DiscordPayload{
|
|
Username: meta.Username,
|
|
AvatarURL: meta.IconURL,
|
|
Embeds: []DiscordEmbed{
|
|
{
|
|
Title: title,
|
|
Description: text,
|
|
URL: p.PullRequest.HTMLURL,
|
|
Color: color,
|
|
Author: DiscordEmbedAuthor{
|
|
Name: p.Sender.UserName,
|
|
URL: setting.AppURL + p.Sender.UserName,
|
|
IconURL: p.Sender.AvatarURL,
|
|
},
|
|
},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func getDiscordRepositoryPayload(p *api.RepositoryPayload, meta *DiscordMeta) (*DiscordPayload, error) {
|
|
var title, url string
|
|
var color int
|
|
switch p.Action {
|
|
case api.HookRepoCreated:
|
|
title = fmt.Sprintf("[%s] Repository created", p.Repository.FullName)
|
|
url = p.Repository.HTMLURL
|
|
color = successColor
|
|
case api.HookRepoDeleted:
|
|
title = fmt.Sprintf("[%s] Repository deleted", p.Repository.FullName)
|
|
color = warnColor
|
|
}
|
|
|
|
return &DiscordPayload{
|
|
Username: meta.Username,
|
|
AvatarURL: meta.IconURL,
|
|
Embeds: []DiscordEmbed{
|
|
{
|
|
Title: title,
|
|
URL: url,
|
|
Color: color,
|
|
Author: DiscordEmbedAuthor{
|
|
Name: p.Sender.UserName,
|
|
URL: setting.AppURL + p.Sender.UserName,
|
|
IconURL: p.Sender.AvatarURL,
|
|
},
|
|
},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
func getDiscordReleasePayload(p *api.ReleasePayload, meta *DiscordMeta) (*DiscordPayload, error) {
|
|
var title, url string
|
|
var color int
|
|
switch p.Action {
|
|
case api.HookReleasePublished:
|
|
title = fmt.Sprintf("[%s] Release created", p.Release.TagName)
|
|
url = p.Release.URL
|
|
color = successColor
|
|
case api.HookReleaseUpdated:
|
|
title = fmt.Sprintf("[%s] Release updated", p.Release.TagName)
|
|
url = p.Release.URL
|
|
color = successColor
|
|
case api.HookReleaseDeleted:
|
|
title = fmt.Sprintf("[%s] Release deleted", p.Release.TagName)
|
|
url = p.Release.URL
|
|
color = successColor
|
|
}
|
|
|
|
return &DiscordPayload{
|
|
Username: meta.Username,
|
|
AvatarURL: meta.IconURL,
|
|
Embeds: []DiscordEmbed{
|
|
{
|
|
Title: title,
|
|
Description: fmt.Sprintf("%s", p.Release.Note),
|
|
URL: url,
|
|
Color: color,
|
|
Author: DiscordEmbedAuthor{
|
|
Name: p.Sender.UserName,
|
|
URL: setting.AppURL + p.Sender.UserName,
|
|
IconURL: p.Sender.AvatarURL,
|
|
},
|
|
},
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
// GetDiscordPayload converts a discord webhook into a DiscordPayload
|
|
func GetDiscordPayload(p api.Payloader, event HookEventType, meta string) (*DiscordPayload, error) {
|
|
s := new(DiscordPayload)
|
|
|
|
discord := &DiscordMeta{}
|
|
if err := json.Unmarshal([]byte(meta), &discord); err != nil {
|
|
return s, errors.New("GetDiscordPayload meta json:" + err.Error())
|
|
}
|
|
|
|
switch event {
|
|
case HookEventCreate:
|
|
return getDiscordCreatePayload(p.(*api.CreatePayload), discord)
|
|
case HookEventDelete:
|
|
return getDiscordDeletePayload(p.(*api.DeletePayload), discord)
|
|
case HookEventFork:
|
|
return getDiscordForkPayload(p.(*api.ForkPayload), discord)
|
|
case HookEventIssues:
|
|
return getDiscordIssuesPayload(p.(*api.IssuePayload), discord)
|
|
case HookEventIssueComment:
|
|
return getDiscordIssueCommentPayload(p.(*api.IssueCommentPayload), discord)
|
|
case HookEventPush:
|
|
return getDiscordPushPayload(p.(*api.PushPayload), discord)
|
|
case HookEventPullRequest:
|
|
return getDiscordPullRequestPayload(p.(*api.PullRequestPayload), discord)
|
|
case HookEventRepository:
|
|
return getDiscordRepositoryPayload(p.(*api.RepositoryPayload), discord)
|
|
case HookEventRelease:
|
|
return getDiscordReleasePayload(p.(*api.ReleasePayload), discord)
|
|
}
|
|
|
|
return s, nil
|
|
}
|