210 lines · 7.1 KB · Go
RawDownload
// Copyright 2021 The Gitea Authors. All rights reserved.// SPDX-License-Identifier: MIT package webhook import (	"context"	"fmt"	"html/template"	"net/http"	"strings" 	webhook_model "forgejo.org/models/webhook"	"forgejo.org/modules/git"	api "forgejo.org/modules/structs"	webhook_module "forgejo.org/modules/webhook"	"forgejo.org/services/forms"	"forgejo.org/services/webhook/shared") type wechatworkHandler struct{} func (wechatworkHandler) Type() webhook_module.HookType       { return webhook_module.WECHATWORK }func (wechatworkHandler) Metadata(*webhook_model.Webhook) any { return nil } func (wechatworkHandler) Icon(size int) template.HTML {	return shared.ImgIcon("wechatwork.png", size)} func (wechatworkHandler) UnmarshalForm(bind func(any)) forms.WebhookForm {	var form struct {		forms.WebhookCoreForm		PayloadURL string `binding:"Required;ValidUrl"`	}	bind(&form) 	return forms.WebhookForm{		WebhookCoreForm: form.WebhookCoreForm,		URL:             form.PayloadURL,		ContentType:     webhook_model.ContentTypeJSON,		Secret:          "",		HTTPMethod:      http.MethodPost,		Metadata:        nil,	}} type (	// WechatworkPayload represents	WechatworkPayload struct {		Msgtype string `json:"msgtype"`		Text    struct {			Content             string   `json:"content"`			MentionedList       []string `json:"mentioned_list"`			MentionedMobileList []string `json:"mentioned_mobile_list"`		} `json:"text"`		Markdown struct {			Content string `json:"content"`		} `json:"markdown"`	}) func newWechatworkMarkdownPayload(title string) WechatworkPayload {	return WechatworkPayload{		Msgtype: "markdown",		Markdown: struct {			Content string `json:"content"`		}{			Content: title,		},	}} // Create implements PayloadConvertor Create methodfunc (wc wechatworkConvertor) Create(p *api.CreatePayload) (WechatworkPayload, error) {	// created tag/branch	refName := git.RefName(p.Ref).ShortName()	title := fmt.Sprintf("[%s] %s %s created", p.Repo.FullName, p.RefType, refName) 	return newWechatworkMarkdownPayload(title), nil} // Delete implements PayloadConvertor Delete methodfunc (wc wechatworkConvertor) Delete(p *api.DeletePayload) (WechatworkPayload, error) {	// created tag/branch	refName := git.RefName(p.Ref).ShortName()	title := fmt.Sprintf("[%s] %s %s deleted", p.Repo.FullName, p.RefType, refName) 	return newWechatworkMarkdownPayload(title), nil} // Fork implements PayloadConvertor Fork methodfunc (wc wechatworkConvertor) Fork(p *api.ForkPayload) (WechatworkPayload, error) {	title := fmt.Sprintf("%s is forked to %s", p.Forkee.FullName, p.Repo.FullName) 	return newWechatworkMarkdownPayload(title), nil} // Push implements PayloadConvertor Push methodfunc (wc wechatworkConvertor) Push(p *api.PushPayload) (WechatworkPayload, error) {	var (		branchName = git.RefName(p.Ref).ShortName()		commitDesc string	) 	title := fmt.Sprintf("# %s:%s <font color=\"warning\">  %s  </font>", p.Repo.FullName, branchName, commitDesc) 	var text string	// for each commit, generate attachment text	for i, commit := range p.Commits {		var authorName string		if commit.Author != nil {			authorName = "Author: " + commit.Author.Name		} 		message := strings.ReplaceAll(commit.Message, "\n\n", "\r\n")		text += fmt.Sprintf(" > [%s](%s) \r\n ><font color=\"info\">%s</font> \n ><font color=\"warning\">%s</font>", commit.ID[:7], commit.URL,			message, authorName) 		// add linebreak to each commit but the last		if i < len(p.Commits)-1 {			text += "\n"		}	}	return newWechatworkMarkdownPayload(title + "\r\n\r\n" + text), nil} // Issue implements PayloadConvertor Issue methodfunc (wc wechatworkConvertor) Issue(p *api.IssuePayload) (WechatworkPayload, error) {	text, issueTitle, attachmentText, _ := getIssuesPayloadInfo(p, noneLinkFormatter, true)	var content string	content += fmt.Sprintf(" ><font color=\"info\">%s</font>\n >%s \n ><font color=\"warning\"> %s</font> \n [%s](%s)", text, attachmentText, issueTitle, p.Issue.HTMLURL, p.Issue.HTMLURL) 	return newWechatworkMarkdownPayload(content), nil} // IssueComment implements PayloadConvertor IssueComment methodfunc (wc wechatworkConvertor) IssueComment(p *api.IssueCommentPayload) (WechatworkPayload, error) {	text, issueTitle, _ := getIssueCommentPayloadInfo(p, noneLinkFormatter, true)	var content string	content += fmt.Sprintf(" ><font color=\"info\">%s</font>\n >%s \n ><font color=\"warning\">%s</font> \n [%s](%s)", text, p.Comment.Body, issueTitle, p.Comment.HTMLURL, p.Comment.HTMLURL) 	return newWechatworkMarkdownPayload(content), nil} // PullRequest implements PayloadConvertor PullRequest methodfunc (wc wechatworkConvertor) PullRequest(p *api.PullRequestPayload) (WechatworkPayload, error) {	text, issueTitle, attachmentText, _ := getPullRequestPayloadInfo(p, noneLinkFormatter, true)	pr := fmt.Sprintf("> <font color=\"info\"> %s </font> \r\n > <font color=\"comment\">%s </font> \r\n > <font color=\"comment\">%s </font> \r\n",		text, issueTitle, attachmentText) 	return newWechatworkMarkdownPayload(pr), nil} // Review implements PayloadConvertor Review methodfunc (wc wechatworkConvertor) Review(p *api.PullRequestPayload, event webhook_module.HookEventType) (WechatworkPayload, error) {	var text, title string	if p.Action == api.HookIssueReviewed {		action, err := parseHookPullRequestEventType(event)		if err != nil {			return WechatworkPayload{}, err		}		title = fmt.Sprintf("[%s] Pull request review %s : #%d %s", p.Repository.FullName, action, p.Index, p.PullRequest.Title)		text = p.Review.Content	} 	return newWechatworkMarkdownPayload("# " + title + "\r\n\r\n >" + text), nil} // Repository implements PayloadConvertor Repository methodfunc (wc wechatworkConvertor) Repository(p *api.RepositoryPayload) (WechatworkPayload, error) {	var title string	switch p.Action {	case api.HookRepoCreated:		title = fmt.Sprintf("[%s] Repository created", p.Repository.FullName)		return newWechatworkMarkdownPayload(title), nil	case api.HookRepoDeleted:		title = fmt.Sprintf("[%s] Repository deleted", p.Repository.FullName)		return newWechatworkMarkdownPayload(title), nil	} 	return WechatworkPayload{}, nil} // Wiki implements PayloadConvertor Wiki methodfunc (wc wechatworkConvertor) Wiki(p *api.WikiPayload) (WechatworkPayload, error) {	text, _, _ := getWikiPayloadInfo(p, noneLinkFormatter, true) 	return newWechatworkMarkdownPayload(text), nil} // Release implements PayloadConvertor Release methodfunc (wc wechatworkConvertor) Release(p *api.ReleasePayload) (WechatworkPayload, error) {	text, _ := getReleasePayloadInfo(p, noneLinkFormatter, true) 	return newWechatworkMarkdownPayload(text), nil} func (wc wechatworkConvertor) Package(p *api.PackagePayload) (WechatworkPayload, error) {	text, _ := getPackagePayloadInfo(p, noneLinkFormatter, true) 	return newWechatworkMarkdownPayload(text), nil} type wechatworkConvertor struct{} var _ shared.PayloadConvertor[WechatworkPayload] = wechatworkConvertor{} func (wechatworkHandler) NewRequest(ctx context.Context, w *webhook_model.Webhook, t *webhook_model.HookTask) (*http.Request, []byte, error) {	return shared.NewJSONRequest(wechatworkConvertor{}, w, t, true)}