forked from fluxcd/flux
-
Notifications
You must be signed in to change notification settings - Fork 0
/
slack.go
90 lines (77 loc) · 2.26 KB
/
slack.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package notifications
import (
"bytes"
"encoding/json"
"fmt"
"io"
"io/ioutil"
"net/http"
"strings"
"text/template"
"time"
"github.com/pkg/errors"
"github.com/weaveworks/flux"
)
const (
defaultReleaseTemplate = `Release {{with .Cause}}({{if .User}}{{.User}}{{else}}unknown{{end}}{{if .Message}}: {{.Message}}{{end}}){{end}} {{trim (print .Spec.ImageSpec) "<>"}} to {{with .Spec.ServiceSpecs}}{{range $index, $spec := .}}{{if not (eq $index 0)}}, {{if last $index $.Spec.ServiceSpecs}}and {{end}}{{end}}{{trim (print .) "<>"}}{{end}}{{end}}. {{with .Error}}{{.}}. failed{{else}}done{{end}}`
)
var (
httpClient = &http.Client{Timeout: 5 * time.Second}
)
func slackNotifyRelease(config flux.NotifierConfig, release flux.Release, releaseError error) error {
if release.Spec.Kind == flux.ReleaseKindPlan {
return nil
}
template := defaultReleaseTemplate
if config.ReleaseTemplate != "" {
template = config.ReleaseTemplate
}
errorMessage := ""
if releaseError != nil {
errorMessage = releaseError.Error()
}
text, err := instantiateTemplate("release", template, struct {
flux.Release
Error string
}{
Release: release,
Error: errorMessage,
})
if err != nil {
return err
}
return notify(config, text)
}
func notify(config flux.NotifierConfig, text string) error {
buf := &bytes.Buffer{}
if err := json.NewEncoder(buf).Encode(map[string]string{
"username": config.Username,
"text": text,
}); err != nil {
return errors.Wrap(err, "encoding Slack POST request")
}
req, err := http.NewRequest("POST", config.HookURL, buf)
if err != nil {
return errors.Wrap(err, "constructing Slack HTTP request")
}
resp, err := httpClient.Do(req)
if err != nil {
return errors.Wrap(err, "executing HTTP POST to Slack")
}
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
body, _ := ioutil.ReadAll(io.LimitReader(resp.Body, 1024*1024))
return fmt.Errorf("%s from Slack (%s)", resp.Status, strings.TrimSpace(string(body)))
}
return nil
}
func instantiateTemplate(tmplName, tmplStr string, args interface{}) (string, error) {
tmpl, err := template.New(tmplName).Funcs(templateFuncs).Parse(tmplStr)
if err != nil {
return "", err
}
var buf bytes.Buffer
if err := tmpl.Execute(&buf, args); err != nil {
return "", err
}
return buf.String(), nil
}