123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821 |
- // 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 db
- import (
- "crypto/hmac"
- "crypto/sha256"
- "crypto/tls"
- "encoding/hex"
- "fmt"
- "io/ioutil"
- "net/url"
- "strings"
- "time"
- jsoniter "github.com/json-iterator/go"
- gouuid "github.com/satori/go.uuid"
- log "unknwon.dev/clog/v2"
- "xorm.io/xorm"
- api "github.com/gogs/go-gogs-client"
- "gogs.io/gogs/internal/conf"
- "gogs.io/gogs/internal/errutil"
- "gogs.io/gogs/internal/httplib"
- "gogs.io/gogs/internal/netutil"
- "gogs.io/gogs/internal/sync"
- )
- var HookQueue = sync.NewUniqueQueue(1000)
- type HookContentType int
- const (
- JSON HookContentType = iota + 1
- FORM
- )
- var hookContentTypes = map[string]HookContentType{
- "json": JSON,
- "form": FORM,
- }
- // ToHookContentType returns HookContentType by given name.
- func ToHookContentType(name string) HookContentType {
- return hookContentTypes[name]
- }
- func (t HookContentType) Name() string {
- switch t {
- case JSON:
- return "json"
- case FORM:
- return "form"
- }
- return ""
- }
- // IsValidHookContentType returns true if given name is a valid hook content type.
- func IsValidHookContentType(name string) bool {
- _, ok := hookContentTypes[name]
- return ok
- }
- type HookEvents struct {
- Create bool `json:"create"`
- Delete bool `json:"delete"`
- Fork bool `json:"fork"`
- Push bool `json:"push"`
- Issues bool `json:"issues"`
- PullRequest bool `json:"pull_request"`
- IssueComment bool `json:"issue_comment"`
- Release bool `json:"release"`
- }
- // HookEvent represents events that will delivery hook.
- type HookEvent struct {
- PushOnly bool `json:"push_only"`
- SendEverything bool `json:"send_everything"`
- ChooseEvents bool `json:"choose_events"`
- HookEvents `json:"events"`
- }
- type HookStatus int
- const (
- HOOK_STATUS_NONE = iota
- HOOK_STATUS_SUCCEED
- HOOK_STATUS_FAILED
- )
- // Webhook represents a web hook object.
- type Webhook struct {
- ID int64
- RepoID int64
- OrgID int64
- URL string `xorm:"url TEXT"`
- ContentType HookContentType
- Secret string `xorm:"TEXT"`
- Events string `xorm:"TEXT"`
- *HookEvent `xorm:"-"` // LEGACY [1.0]: Cannot ignore JSON (i.e. json:"-") here, it breaks old backup archive
- IsSSL bool `xorm:"is_ssl"`
- IsActive bool
- HookTaskType HookTaskType
- Meta string `xorm:"TEXT"` // store hook-specific attributes
- LastStatus HookStatus // Last delivery status
- Created time.Time `xorm:"-" json:"-"`
- CreatedUnix int64
- Updated time.Time `xorm:"-" json:"-"`
- UpdatedUnix int64
- }
- func (w *Webhook) BeforeInsert() {
- w.CreatedUnix = time.Now().Unix()
- w.UpdatedUnix = w.CreatedUnix
- }
- func (w *Webhook) BeforeUpdate() {
- w.UpdatedUnix = time.Now().Unix()
- }
- func (w *Webhook) AfterSet(colName string, _ xorm.Cell) {
- var err error
- switch colName {
- case "events":
- w.HookEvent = &HookEvent{}
- if err = jsoniter.Unmarshal([]byte(w.Events), w.HookEvent); err != nil {
- log.Error("Unmarshal [%d]: %v", w.ID, err)
- }
- case "created_unix":
- w.Created = time.Unix(w.CreatedUnix, 0).Local()
- case "updated_unix":
- w.Updated = time.Unix(w.UpdatedUnix, 0).Local()
- }
- }
- func (w *Webhook) SlackMeta() *SlackMeta {
- s := &SlackMeta{}
- if err := jsoniter.Unmarshal([]byte(w.Meta), s); err != nil {
- log.Error("Failed to get Slack meta [webhook_id: %d]: %v", w.ID, err)
- }
- return s
- }
- // History returns history of webhook by given conditions.
- func (w *Webhook) History(page int) ([]*HookTask, error) {
- return HookTasks(w.ID, page)
- }
- // UpdateEvent handles conversion from HookEvent to Events.
- func (w *Webhook) UpdateEvent() error {
- data, err := jsoniter.Marshal(w.HookEvent)
- w.Events = string(data)
- return err
- }
- // HasCreateEvent returns true if hook enabled create event.
- func (w *Webhook) HasCreateEvent() bool {
- return w.SendEverything ||
- (w.ChooseEvents && w.HookEvents.Create)
- }
- // HasDeleteEvent returns true if hook enabled delete event.
- func (w *Webhook) HasDeleteEvent() bool {
- return w.SendEverything ||
- (w.ChooseEvents && w.HookEvents.Delete)
- }
- // HasForkEvent returns true if hook enabled fork event.
- func (w *Webhook) HasForkEvent() bool {
- return w.SendEverything ||
- (w.ChooseEvents && w.HookEvents.Fork)
- }
- // HasPushEvent returns true if hook enabled push event.
- func (w *Webhook) HasPushEvent() bool {
- return w.PushOnly || w.SendEverything ||
- (w.ChooseEvents && w.HookEvents.Push)
- }
- // HasIssuesEvent returns true if hook enabled issues event.
- func (w *Webhook) HasIssuesEvent() bool {
- return w.SendEverything ||
- (w.ChooseEvents && w.HookEvents.Issues)
- }
- // HasPullRequestEvent returns true if hook enabled pull request event.
- func (w *Webhook) HasPullRequestEvent() bool {
- return w.SendEverything ||
- (w.ChooseEvents && w.HookEvents.PullRequest)
- }
- // HasIssueCommentEvent returns true if hook enabled issue comment event.
- func (w *Webhook) HasIssueCommentEvent() bool {
- return w.SendEverything ||
- (w.ChooseEvents && w.HookEvents.IssueComment)
- }
- // HasReleaseEvent returns true if hook enabled release event.
- func (w *Webhook) HasReleaseEvent() bool {
- return w.SendEverything ||
- (w.ChooseEvents && w.HookEvents.Release)
- }
- type eventChecker struct {
- checker func() bool
- typ HookEventType
- }
- func (w *Webhook) EventsArray() []string {
- events := make([]string, 0, 8)
- eventCheckers := []eventChecker{
- {w.HasCreateEvent, HOOK_EVENT_CREATE},
- {w.HasDeleteEvent, HOOK_EVENT_DELETE},
- {w.HasForkEvent, HOOK_EVENT_FORK},
- {w.HasPushEvent, HOOK_EVENT_PUSH},
- {w.HasIssuesEvent, HOOK_EVENT_ISSUES},
- {w.HasPullRequestEvent, HOOK_EVENT_PULL_REQUEST},
- {w.HasIssueCommentEvent, HOOK_EVENT_ISSUE_COMMENT},
- {w.HasReleaseEvent, HOOK_EVENT_RELEASE},
- }
- for _, c := range eventCheckers {
- if c.checker() {
- events = append(events, string(c.typ))
- }
- }
- return events
- }
- // CreateWebhook creates a new web hook.
- func CreateWebhook(w *Webhook) error {
- _, err := x.Insert(w)
- return err
- }
- var _ errutil.NotFound = (*ErrWebhookNotExist)(nil)
- type ErrWebhookNotExist struct {
- args map[string]interface{}
- }
- func IsErrWebhookNotExist(err error) bool {
- _, ok := err.(ErrWebhookNotExist)
- return ok
- }
- func (err ErrWebhookNotExist) Error() string {
- return fmt.Sprintf("webhook does not exist: %v", err.args)
- }
- func (ErrWebhookNotExist) NotFound() bool {
- return true
- }
- // getWebhook uses argument bean as query condition,
- // ID must be specified and do not assign unnecessary fields.
- func getWebhook(bean *Webhook) (*Webhook, error) {
- has, err := x.Get(bean)
- if err != nil {
- return nil, err
- } else if !has {
- return nil, ErrWebhookNotExist{args: map[string]interface{}{"webhookID": bean.ID}}
- }
- return bean, nil
- }
- // GetWebhookByID returns webhook by given ID.
- // Use this function with caution of accessing unauthorized webhook,
- // which means should only be used in non-user interactive functions.
- func GetWebhookByID(id int64) (*Webhook, error) {
- return getWebhook(&Webhook{
- ID: id,
- })
- }
- // GetWebhookOfRepoByID returns webhook of repository by given ID.
- func GetWebhookOfRepoByID(repoID, id int64) (*Webhook, error) {
- return getWebhook(&Webhook{
- ID: id,
- RepoID: repoID,
- })
- }
- // GetWebhookByOrgID returns webhook of organization by given ID.
- func GetWebhookByOrgID(orgID, id int64) (*Webhook, error) {
- return getWebhook(&Webhook{
- ID: id,
- OrgID: orgID,
- })
- }
- // getActiveWebhooksByRepoID returns all active webhooks of repository.
- func getActiveWebhooksByRepoID(e Engine, repoID int64) ([]*Webhook, error) {
- webhooks := make([]*Webhook, 0, 5)
- return webhooks, e.Where("repo_id = ?", repoID).And("is_active = ?", true).Find(&webhooks)
- }
- // GetWebhooksByRepoID returns all webhooks of a repository.
- func GetWebhooksByRepoID(repoID int64) ([]*Webhook, error) {
- webhooks := make([]*Webhook, 0, 5)
- return webhooks, x.Find(&webhooks, &Webhook{RepoID: repoID})
- }
- // UpdateWebhook updates information of webhook.
- func UpdateWebhook(w *Webhook) error {
- _, err := x.Id(w.ID).AllCols().Update(w)
- return err
- }
- // deleteWebhook uses argument bean as query condition,
- // ID must be specified and do not assign unnecessary fields.
- func deleteWebhook(bean *Webhook) (err error) {
- sess := x.NewSession()
- defer sess.Close()
- if err = sess.Begin(); err != nil {
- return err
- }
- if _, err = sess.Delete(bean); err != nil {
- return err
- } else if _, err = sess.Delete(&HookTask{HookID: bean.ID}); err != nil {
- return err
- }
- return sess.Commit()
- }
- // DeleteWebhookOfRepoByID deletes webhook of repository by given ID.
- func DeleteWebhookOfRepoByID(repoID, id int64) error {
- return deleteWebhook(&Webhook{
- ID: id,
- RepoID: repoID,
- })
- }
- // DeleteWebhookOfOrgByID deletes webhook of organization by given ID.
- func DeleteWebhookOfOrgByID(orgID, id int64) error {
- return deleteWebhook(&Webhook{
- ID: id,
- OrgID: orgID,
- })
- }
- // GetWebhooksByOrgID returns all webhooks for an organization.
- func GetWebhooksByOrgID(orgID int64) (ws []*Webhook, err error) {
- err = x.Find(&ws, &Webhook{OrgID: orgID})
- return ws, err
- }
- // getActiveWebhooksByOrgID returns all active webhooks for an organization.
- func getActiveWebhooksByOrgID(e Engine, orgID int64) ([]*Webhook, error) {
- ws := make([]*Webhook, 0, 3)
- return ws, e.Where("org_id=?", orgID).And("is_active=?", true).Find(&ws)
- }
- // ___ ___ __ ___________ __
- // / | \ ____ ____ | | _\__ ___/____ _____| | __
- // / ~ \/ _ \ / _ \| |/ / | | \__ \ / ___/ |/ /
- // \ Y ( <_> | <_> ) < | | / __ \_\___ \| <
- // \___|_ / \____/ \____/|__|_ \ |____| (____ /____ >__|_ \
- // \/ \/ \/ \/ \/
- type HookTaskType int
- const (
- GOGS HookTaskType = iota + 1
- SLACK
- DISCORD
- DINGTALK
- )
- var hookTaskTypes = map[string]HookTaskType{
- "gogs": GOGS,
- "slack": SLACK,
- "discord": DISCORD,
- "dingtalk": DINGTALK,
- }
- // ToHookTaskType returns HookTaskType by given name.
- func ToHookTaskType(name string) HookTaskType {
- return hookTaskTypes[name]
- }
- func (t HookTaskType) Name() string {
- switch t {
- case GOGS:
- return "gogs"
- case SLACK:
- return "slack"
- case DISCORD:
- return "discord"
- case DINGTALK:
- return "dingtalk"
- }
- return ""
- }
- // IsValidHookTaskType returns true if given name is a valid hook task type.
- func IsValidHookTaskType(name string) bool {
- _, ok := hookTaskTypes[name]
- return ok
- }
- type HookEventType string
- const (
- HOOK_EVENT_CREATE HookEventType = "create"
- HOOK_EVENT_DELETE HookEventType = "delete"
- HOOK_EVENT_FORK HookEventType = "fork"
- HOOK_EVENT_PUSH HookEventType = "push"
- HOOK_EVENT_ISSUES HookEventType = "issues"
- HOOK_EVENT_PULL_REQUEST HookEventType = "pull_request"
- HOOK_EVENT_ISSUE_COMMENT HookEventType = "issue_comment"
- HOOK_EVENT_RELEASE HookEventType = "release"
- )
- // HookRequest represents hook task request information.
- type HookRequest struct {
- Headers map[string]string `json:"headers"`
- }
- // HookResponse represents hook task response information.
- type HookResponse struct {
- Status int `json:"status"`
- Headers map[string]string `json:"headers"`
- Body string `json:"body"`
- }
- // HookTask represents a hook task.
- type HookTask struct {
- ID int64
- RepoID int64 `xorm:"INDEX"`
- HookID int64
- UUID string
- Type HookTaskType
- URL string `xorm:"TEXT"`
- Signature string `xorm:"TEXT"`
- api.Payloader `xorm:"-" json:"-"`
- PayloadContent string `xorm:"TEXT"`
- ContentType HookContentType
- EventType HookEventType
- IsSSL bool
- IsDelivered bool
- Delivered int64
- DeliveredString string `xorm:"-" json:"-"`
- // History info.
- IsSucceed bool
- RequestContent string `xorm:"TEXT"`
- RequestInfo *HookRequest `xorm:"-" json:"-"`
- ResponseContent string `xorm:"TEXT"`
- ResponseInfo *HookResponse `xorm:"-" json:"-"`
- }
- func (t *HookTask) BeforeUpdate() {
- if t.RequestInfo != nil {
- t.RequestContent = t.ToJSON(t.RequestInfo)
- }
- if t.ResponseInfo != nil {
- t.ResponseContent = t.ToJSON(t.ResponseInfo)
- }
- }
- func (t *HookTask) AfterSet(colName string, _ xorm.Cell) {
- var err error
- switch colName {
- case "delivered":
- t.DeliveredString = time.Unix(0, t.Delivered).Format("2006-01-02 15:04:05 MST")
- case "request_content":
- if len(t.RequestContent) == 0 {
- return
- }
- t.RequestInfo = &HookRequest{}
- if err = jsoniter.Unmarshal([]byte(t.RequestContent), t.RequestInfo); err != nil {
- log.Error("Unmarshal[%d]: %v", t.ID, err)
- }
- case "response_content":
- if len(t.ResponseContent) == 0 {
- return
- }
- t.ResponseInfo = &HookResponse{}
- if err = jsoniter.Unmarshal([]byte(t.ResponseContent), t.ResponseInfo); err != nil {
- log.Error("Unmarshal [%d]: %v", t.ID, err)
- }
- }
- }
- func (t *HookTask) ToJSON(v interface{}) string {
- p, err := jsoniter.Marshal(v)
- if err != nil {
- log.Error("Marshal [%d]: %v", t.ID, err)
- }
- return string(p)
- }
- // HookTasks returns a list of hook tasks by given conditions.
- func HookTasks(hookID int64, page int) ([]*HookTask, error) {
- tasks := make([]*HookTask, 0, conf.Webhook.PagingNum)
- return tasks, x.Limit(conf.Webhook.PagingNum, (page-1)*conf.Webhook.PagingNum).Where("hook_id=?", hookID).Desc("id").Find(&tasks)
- }
- // createHookTask creates a new hook task,
- // it handles conversion from Payload to PayloadContent.
- func createHookTask(e Engine, t *HookTask) error {
- data, err := t.Payloader.JSONPayload()
- if err != nil {
- return err
- }
- t.UUID = gouuid.NewV4().String()
- t.PayloadContent = string(data)
- _, err = e.Insert(t)
- return err
- }
- var _ errutil.NotFound = (*ErrHookTaskNotExist)(nil)
- type ErrHookTaskNotExist struct {
- args map[string]interface{}
- }
- func IsHookTaskNotExist(err error) bool {
- _, ok := err.(ErrHookTaskNotExist)
- return ok
- }
- func (err ErrHookTaskNotExist) Error() string {
- return fmt.Sprintf("hook task does not exist: %v", err.args)
- }
- func (ErrHookTaskNotExist) NotFound() bool {
- return true
- }
- // GetHookTaskOfWebhookByUUID returns hook task of given webhook by UUID.
- func GetHookTaskOfWebhookByUUID(webhookID int64, uuid string) (*HookTask, error) {
- hookTask := &HookTask{
- HookID: webhookID,
- UUID: uuid,
- }
- has, err := x.Get(hookTask)
- if err != nil {
- return nil, err
- } else if !has {
- return nil, ErrHookTaskNotExist{args: map[string]interface{}{"webhookID": webhookID, "uuid": uuid}}
- }
- return hookTask, nil
- }
- // UpdateHookTask updates information of hook task.
- func UpdateHookTask(t *HookTask) error {
- _, err := x.Id(t.ID).AllCols().Update(t)
- return err
- }
- // prepareHookTasks adds list of webhooks to task queue.
- func prepareHookTasks(e Engine, repo *Repository, event HookEventType, p api.Payloader, webhooks []*Webhook) (err error) {
- if len(webhooks) == 0 {
- return nil
- }
- var payloader api.Payloader
- for _, w := range webhooks {
- switch event {
- case HOOK_EVENT_CREATE:
- if !w.HasCreateEvent() {
- continue
- }
- case HOOK_EVENT_DELETE:
- if !w.HasDeleteEvent() {
- continue
- }
- case HOOK_EVENT_FORK:
- if !w.HasForkEvent() {
- continue
- }
- case HOOK_EVENT_PUSH:
- if !w.HasPushEvent() {
- continue
- }
- case HOOK_EVENT_ISSUES:
- if !w.HasIssuesEvent() {
- continue
- }
- case HOOK_EVENT_PULL_REQUEST:
- if !w.HasPullRequestEvent() {
- continue
- }
- case HOOK_EVENT_ISSUE_COMMENT:
- if !w.HasIssueCommentEvent() {
- continue
- }
- case HOOK_EVENT_RELEASE:
- if !w.HasReleaseEvent() {
- continue
- }
- }
- // Use separate objects so modifcations won't be made on payload on non-Gogs type hooks.
- switch w.HookTaskType {
- case SLACK:
- payloader, err = GetSlackPayload(p, event, w.Meta)
- if err != nil {
- return fmt.Errorf("GetSlackPayload: %v", err)
- }
- case DISCORD:
- payloader, err = GetDiscordPayload(p, event, w.Meta)
- if err != nil {
- return fmt.Errorf("GetDiscordPayload: %v", err)
- }
- case DINGTALK:
- payloader, err = GetDingtalkPayload(p, event)
- if err != nil {
- return fmt.Errorf("GetDingtalkPayload: %v", err)
- }
- default:
- payloader = p
- }
- var signature string
- if len(w.Secret) > 0 {
- data, err := payloader.JSONPayload()
- if err != nil {
- log.Error("prepareWebhooks.JSONPayload: %v", err)
- }
- sig := hmac.New(sha256.New, []byte(w.Secret))
- _, _ = sig.Write(data)
- signature = hex.EncodeToString(sig.Sum(nil))
- }
- if err = createHookTask(e, &HookTask{
- RepoID: repo.ID,
- HookID: w.ID,
- Type: w.HookTaskType,
- URL: w.URL,
- Signature: signature,
- Payloader: payloader,
- ContentType: w.ContentType,
- EventType: event,
- IsSSL: w.IsSSL,
- }); err != nil {
- return fmt.Errorf("createHookTask: %v", err)
- }
- }
- // It's safe to fail when the whole function is called during hook execution
- // because resource released after exit. Also, there is no process started to
- // consume this input during hook execution.
- go HookQueue.Add(repo.ID)
- return nil
- }
- func prepareWebhooks(e Engine, repo *Repository, event HookEventType, p api.Payloader) error {
- webhooks, err := getActiveWebhooksByRepoID(e, repo.ID)
- if err != nil {
- return fmt.Errorf("getActiveWebhooksByRepoID [%d]: %v", repo.ID, err)
- }
- // check if repo belongs to org and append additional webhooks
- if repo.mustOwner(e).IsOrganization() {
- // get hooks for org
- orgws, err := getActiveWebhooksByOrgID(e, repo.OwnerID)
- if err != nil {
- return fmt.Errorf("getActiveWebhooksByOrgID [%d]: %v", repo.OwnerID, err)
- }
- webhooks = append(webhooks, orgws...)
- }
- return prepareHookTasks(e, repo, event, p, webhooks)
- }
- // PrepareWebhooks adds all active webhooks to task queue.
- func PrepareWebhooks(repo *Repository, event HookEventType, p api.Payloader) error {
- return prepareWebhooks(x, repo, event, p)
- }
- // TestWebhook adds the test webhook matches the ID to task queue.
- func TestWebhook(repo *Repository, event HookEventType, p api.Payloader, webhookID int64) error {
- webhook, err := GetWebhookOfRepoByID(repo.ID, webhookID)
- if err != nil {
- return fmt.Errorf("GetWebhookOfRepoByID [repo_id: %d, id: %d]: %v", repo.ID, webhookID, err)
- }
- return prepareHookTasks(x, repo, event, p, []*Webhook{webhook})
- }
- func (t *HookTask) deliver() {
- payloadURL, err := url.Parse(t.URL)
- if err != nil {
- t.ResponseContent = fmt.Sprintf(`{"body": "Cannot parse payload URL: %v"}`, err)
- return
- }
- if netutil.IsBlockedLocalHostname(payloadURL.Hostname(), conf.Security.LocalNetworkAllowlist) {
- t.ResponseContent = `{"body": "Payload URL resolved to a local network address that is implicitly blocked."}`
- return
- }
- t.IsDelivered = true
- timeout := time.Duration(conf.Webhook.DeliverTimeout) * time.Second
- req := httplib.Post(t.URL).SetTimeout(timeout, timeout).
- Header("X-Github-Delivery", t.UUID).
- Header("X-Github-Event", string(t.EventType)).
- Header("X-Gogs-Delivery", t.UUID).
- Header("X-Gogs-Signature", t.Signature).
- Header("X-Gogs-Event", string(t.EventType)).
- SetTLSClientConfig(&tls.Config{InsecureSkipVerify: conf.Webhook.SkipTLSVerify})
- switch t.ContentType {
- case JSON:
- req = req.Header("Content-Type", "application/json").Body(t.PayloadContent)
- case FORM:
- req.Param("payload", t.PayloadContent)
- }
- // Record delivery information.
- t.RequestInfo = &HookRequest{
- Headers: map[string]string{},
- }
- for k, vals := range req.Headers() {
- t.RequestInfo.Headers[k] = strings.Join(vals, ",")
- }
- t.ResponseInfo = &HookResponse{
- Headers: map[string]string{},
- }
- defer func() {
- t.Delivered = time.Now().UnixNano()
- if t.IsSucceed {
- log.Trace("Hook delivered: %s", t.UUID)
- } else {
- log.Trace("Hook delivery failed: %s", t.UUID)
- }
- // Update webhook last delivery status.
- w, err := GetWebhookByID(t.HookID)
- if err != nil {
- log.Error("GetWebhookByID: %v", err)
- return
- }
- if t.IsSucceed {
- w.LastStatus = HOOK_STATUS_SUCCEED
- } else {
- w.LastStatus = HOOK_STATUS_FAILED
- }
- if err = UpdateWebhook(w); err != nil {
- log.Error("UpdateWebhook: %v", err)
- return
- }
- }()
- resp, err := req.Response()
- if err != nil {
- t.ResponseInfo.Body = fmt.Sprintf("Delivery: %v", err)
- return
- }
- defer resp.Body.Close()
- // Status code is 20x can be seen as succeed.
- t.IsSucceed = resp.StatusCode/100 == 2
- t.ResponseInfo.Status = resp.StatusCode
- for k, vals := range resp.Header {
- t.ResponseInfo.Headers[k] = strings.Join(vals, ",")
- }
- p, err := ioutil.ReadAll(resp.Body)
- if err != nil {
- t.ResponseInfo.Body = fmt.Sprintf("read body: %s", err)
- return
- }
- t.ResponseInfo.Body = string(p)
- }
- // DeliverHooks checks and delivers undelivered hooks.
- // TODO: shoot more hooks at same time.
- func DeliverHooks() {
- tasks := make([]*HookTask, 0, 10)
- _ = x.Where("is_delivered = ?", false).Iterate(new(HookTask),
- func(idx int, bean interface{}) error {
- t := bean.(*HookTask)
- t.deliver()
- tasks = append(tasks, t)
- return nil
- })
- // Update hook task status.
- for _, t := range tasks {
- if err := UpdateHookTask(t); err != nil {
- log.Error("UpdateHookTask [%d]: %v", t.ID, err)
- }
- }
- // Start listening on new hook requests.
- for repoID := range HookQueue.Queue() {
- log.Trace("DeliverHooks [repo_id: %v]", repoID)
- HookQueue.Remove(repoID)
- tasks = make([]*HookTask, 0, 5)
- if err := x.Where("repo_id = ?", repoID).And("is_delivered = ?", false).Find(&tasks); err != nil {
- log.Error("Get repository [%s] hook tasks: %v", repoID, err)
- continue
- }
- for _, t := range tasks {
- t.deliver()
- if err := UpdateHookTask(t); err != nil {
- log.Error("UpdateHookTask [%d]: %v", t.ID, err)
- continue
- }
- }
- }
- }
- func InitDeliverHooks() {
- go DeliverHooks()
- }
|