123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246 |
- // Copyright 2014 The Gogs 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"
- "strings"
-
- "code.gitea.io/git"
- api "code.gitea.io/sdk/gitea"
-
- "code.gitea.io/gitea/modules/setting"
- )
-
- // SlackMeta contains the slack metadata
- type SlackMeta struct {
- Channel string `json:"channel"`
- Username string `json:"username"`
- IconURL string `json:"icon_url"`
- Color string `json:"color"`
- }
-
- // SlackPayload contains the information about the slack channel
- type SlackPayload struct {
- Channel string `json:"channel"`
- Text string `json:"text"`
- Username string `json:"username"`
- IconURL string `json:"icon_url"`
- UnfurlLinks int `json:"unfurl_links"`
- LinkNames int `json:"link_names"`
- Attachments []SlackAttachment `json:"attachments"`
- }
-
- // SlackAttachment contains the slack message
- type SlackAttachment struct {
- Fallback string `json:"fallback"`
- Color string `json:"color"`
- Title string `json:"title"`
- Text string `json:"text"`
- }
-
- // SetSecret sets the slack secret
- func (p *SlackPayload) SetSecret(_ string) {}
-
- // JSONPayload Marshals the SlackPayload to json
- func (p *SlackPayload) JSONPayload() ([]byte, error) {
- data, err := json.MarshalIndent(p, "", " ")
- if err != nil {
- return []byte{}, err
- }
- return data, nil
- }
-
- // SlackTextFormatter replaces &, <, > with HTML characters
- // see: https://api.slack.com/docs/formatting
- func SlackTextFormatter(s string) string {
- // replace & < >
- s = strings.Replace(s, "&", "&", -1)
- s = strings.Replace(s, "<", "<", -1)
- s = strings.Replace(s, ">", ">", -1)
- return s
- }
-
- // SlackShortTextFormatter replaces &, <, > with HTML characters
- func SlackShortTextFormatter(s string) string {
- s = strings.Split(s, "\n")[0]
- // replace & < >
- s = strings.Replace(s, "&", "&", -1)
- s = strings.Replace(s, "<", "<", -1)
- s = strings.Replace(s, ">", ">", -1)
- return s
- }
-
- // SlackLinkFormatter creates a link compatible with slack
- func SlackLinkFormatter(url string, text string) string {
- return fmt.Sprintf("<%s|%s>", url, SlackTextFormatter(text))
- }
-
- // SlackLinkToRef slack-formatter link to a repo ref
- func SlackLinkToRef(repoURL, ref string) string {
- refName := git.RefEndName(ref)
- switch {
- case strings.HasPrefix(ref, git.BranchPrefix):
- return SlackLinkFormatter(repoURL+"/src/branch/"+refName, refName)
- case strings.HasPrefix(ref, git.TagPrefix):
- return SlackLinkFormatter(repoURL+"/src/tag/"+refName, refName)
- default:
- return SlackLinkFormatter(repoURL+"/src/commit/"+refName, refName)
- }
- }
-
- func getSlackCreatePayload(p *api.CreatePayload, slack *SlackMeta) (*SlackPayload, error) {
- repoLink := SlackLinkFormatter(p.Repo.HTMLURL, p.Repo.Name)
- refLink := SlackLinkToRef(p.Repo.HTMLURL, p.Ref)
- text := fmt.Sprintf("[%s:%s] %s created by %s", repoLink, refLink, p.RefType, p.Sender.UserName)
-
- return &SlackPayload{
- Channel: slack.Channel,
- Text: text,
- Username: slack.Username,
- IconURL: slack.IconURL,
- }, nil
- }
-
- func getSlackPushPayload(p *api.PushPayload, slack *SlackMeta) (*SlackPayload, error) {
- // n new commits
- var (
- commitDesc string
- commitString string
- )
-
- if len(p.Commits) == 1 {
- commitDesc = "1 new commit"
- } else {
- commitDesc = fmt.Sprintf("%d new commits", len(p.Commits))
- }
- if len(p.CompareURL) > 0 {
- commitString = SlackLinkFormatter(p.CompareURL, commitDesc)
- } else {
- commitString = commitDesc
- }
-
- repoLink := SlackLinkFormatter(p.Repo.HTMLURL, p.Repo.Name)
- branchLink := SlackLinkToRef(p.Repo.HTMLURL, p.Ref)
- text := fmt.Sprintf("[%s:%s] %s pushed by %s", repoLink, branchLink, commitString, p.Pusher.UserName)
-
- var attachmentText string
- // for each commit, generate attachment text
- for i, commit := range p.Commits {
- attachmentText += fmt.Sprintf("%s: %s - %s", SlackLinkFormatter(commit.URL, commit.ID[:7]), SlackShortTextFormatter(commit.Message), SlackTextFormatter(commit.Author.Name))
- // add linebreak to each commit but the last
- if i < len(p.Commits)-1 {
- attachmentText += "\n"
- }
- }
-
- return &SlackPayload{
- Channel: slack.Channel,
- Text: text,
- Username: slack.Username,
- IconURL: slack.IconURL,
- Attachments: []SlackAttachment{{
- Color: slack.Color,
- Text: attachmentText,
- }},
- }, nil
- }
-
- func getSlackPullRequestPayload(p *api.PullRequestPayload, slack *SlackMeta) (*SlackPayload, error) {
- senderLink := SlackLinkFormatter(setting.AppURL+p.Sender.UserName, p.Sender.UserName)
- titleLink := SlackLinkFormatter(fmt.Sprintf("%s/pulls/%d", p.Repository.HTMLURL, p.Index),
- fmt.Sprintf("#%d %s", p.Index, p.PullRequest.Title))
- var text, title, attachmentText string
- switch p.Action {
- case api.HookIssueOpened:
- text = fmt.Sprintf("[%s] Pull request submitted by %s", p.Repository.FullName, senderLink)
- title = titleLink
- attachmentText = SlackTextFormatter(p.PullRequest.Body)
- case api.HookIssueClosed:
- if p.PullRequest.HasMerged {
- text = fmt.Sprintf("[%s] Pull request merged: %s by %s", p.Repository.FullName, titleLink, senderLink)
- } else {
- text = fmt.Sprintf("[%s] Pull request closed: %s by %s", p.Repository.FullName, titleLink, senderLink)
- }
- case api.HookIssueReOpened:
- text = fmt.Sprintf("[%s] Pull request re-opened: %s by %s", p.Repository.FullName, titleLink, senderLink)
- case api.HookIssueEdited:
- text = fmt.Sprintf("[%s] Pull request edited: %s by %s", p.Repository.FullName, titleLink, senderLink)
- attachmentText = SlackTextFormatter(p.PullRequest.Body)
- case api.HookIssueAssigned:
- text = fmt.Sprintf("[%s] Pull request assigned to %s: %s by %s", p.Repository.FullName,
- SlackLinkFormatter(setting.AppURL+p.PullRequest.Assignee.UserName, p.PullRequest.Assignee.UserName),
- titleLink, senderLink)
- case api.HookIssueUnassigned:
- text = fmt.Sprintf("[%s] Pull request unassigned: %s by %s", p.Repository.FullName, titleLink, senderLink)
- case api.HookIssueLabelUpdated:
- text = fmt.Sprintf("[%s] Pull request labels updated: %s by %s", p.Repository.FullName, titleLink, senderLink)
- case api.HookIssueLabelCleared:
- text = fmt.Sprintf("[%s] Pull request labels cleared: %s by %s", p.Repository.FullName, titleLink, senderLink)
- case api.HookIssueSynchronized:
- text = fmt.Sprintf("[%s] Pull request synchronized: %s by %s", p.Repository.FullName, titleLink, senderLink)
- }
-
- return &SlackPayload{
- Channel: slack.Channel,
- Text: text,
- Username: slack.Username,
- IconURL: slack.IconURL,
- Attachments: []SlackAttachment{{
- Color: slack.Color,
- Title: title,
- Text: attachmentText,
- }},
- }, nil
- }
-
- func getSlackRepositoryPayload(p *api.RepositoryPayload, slack *SlackMeta) (*SlackPayload, error) {
- senderLink := SlackLinkFormatter(setting.AppURL+p.Sender.UserName, p.Sender.UserName)
- var text, title, attachmentText string
- switch p.Action {
- case api.HookRepoCreated:
- text = fmt.Sprintf("[%s] Repository created by %s", p.Repository.FullName, senderLink)
- title = p.Repository.HTMLURL
- case api.HookRepoDeleted:
- text = fmt.Sprintf("[%s] Repository deleted by %s", p.Repository.FullName, senderLink)
- }
-
- return &SlackPayload{
- Channel: slack.Channel,
- Text: text,
- Username: slack.Username,
- IconURL: slack.IconURL,
- Attachments: []SlackAttachment{{
- Color: slack.Color,
- Title: title,
- Text: attachmentText,
- }},
- }, nil
- }
-
- // GetSlackPayload converts a slack webhook into a SlackPayload
- func GetSlackPayload(p api.Payloader, event HookEventType, meta string) (*SlackPayload, error) {
- s := new(SlackPayload)
-
- slack := &SlackMeta{}
- if err := json.Unmarshal([]byte(meta), &slack); err != nil {
- return s, errors.New("GetSlackPayload meta json:" + err.Error())
- }
-
- switch event {
- case HookEventCreate:
- return getSlackCreatePayload(p.(*api.CreatePayload), slack)
- case HookEventPush:
- return getSlackPushPayload(p.(*api.PushPayload), slack)
- case HookEventPullRequest:
- return getSlackPullRequestPayload(p.(*api.PullRequestPayload), slack)
- case HookEventRepository:
- return getSlackRepositoryPayload(p.(*api.RepositoryPayload), slack)
- }
-
- return s, nil
- }
|