woodpecker/remote/coding/internal/webhook.go

93 lines
2.2 KiB
Go
Raw Normal View History

2017-07-22 09:12:09 +00:00
package internal
import (
"encoding/json"
"fmt"
"net/url"
)
type Webhook struct {
Id int `json:"id"`
HookURL string `json:"hook_url"`
}
func (c *Client) GetWebhooks(globalKey, projectName string) ([]*Webhook, error) {
u := fmt.Sprintf("/user/%s/project/%s/git/hooks", globalKey, projectName)
resp, err := c.Get(u, nil)
if err != nil {
return nil, err
}
webhooks := make([]*Webhook, 0)
err = json.Unmarshal(resp, &webhooks)
if err != nil {
return nil, APIClientErr{"fail to parse webhooks data", u, err}
}
return webhooks, nil
}
func (c *Client) AddWebhook(globalKey, projectName, link string) error {
webhooks, err := c.GetWebhooks(globalKey, projectName)
if err != nil {
return err
}
webhook := matchingHooks(webhooks, link)
if webhook != nil {
u := fmt.Sprintf("/user/%s/project/%s/git/hook/%d", globalKey, projectName, webhook.Id)
params := url.Values{}
params.Set("hook_url", link)
params.Set("type_pust", "true")
params.Set("type_mr_pr", "true")
_, err := c.Do("PUT", u, params)
if err != nil {
return APIClientErr{"fail to edit webhook", u, err}
}
return nil
}
u := fmt.Sprintf("/user/%s/project/%s/git/hook", globalKey, projectName)
params := url.Values{}
params.Set("hook_url", link)
params.Set("type_push", "true")
params.Set("type_mr_pr", "true")
_, err = c.Do("POST", u, params)
if err != nil {
return APIClientErr{"fail to add webhook", u, err}
}
return nil
}
func (c *Client) RemoveWebhook(globalKey, projectName, link string) error {
webhooks, err := c.GetWebhooks(globalKey, projectName)
if err != nil {
return err
}
webhook := matchingHooks(webhooks, link)
if webhook == nil {
return nil
}
u := fmt.Sprintf("/user/%s/project/%s/git/hook/%d", globalKey, projectName, webhook.Id)
_, err = c.Do("DELETE", u, nil)
if err != nil {
return APIClientErr{"fail to remove webhook", u, err}
}
return nil
}
// helper function to return matching hook.
func matchingHooks(hooks []*Webhook, rawurl string) *Webhook {
link, err := url.Parse(rawurl)
if err != nil {
return nil
}
for _, hook := range hooks {
hookurl, err := url.Parse(hook.HookURL)
if err == nil && hookurl.Host == link.Host {
return hook
}
}
return nil
}