|
|
@ -20,7 +20,7 @@ type Dog struct { |
|
|
|
Keywords string |
|
|
|
Recover string |
|
|
|
Webhooks []string |
|
|
|
AllWebhooks map[string]ConfigWebhook |
|
|
|
AllWebhooks map[string]Webhook |
|
|
|
Logger chan string |
|
|
|
error error |
|
|
|
failures int |
|
|
@ -172,94 +172,98 @@ func (d *Dog) notify(hardFail bool) { |
|
|
|
continue |
|
|
|
} |
|
|
|
|
|
|
|
// TODO do this in main on config init
|
|
|
|
if "" == h.Method { |
|
|
|
h.Method = "POST" |
|
|
|
} |
|
|
|
d.notifyOne(h, hardFail) |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
var body *strings.Reader |
|
|
|
var err error |
|
|
|
// TODO real templates
|
|
|
|
if 0 != len(h.Form) { |
|
|
|
form := url.Values{} |
|
|
|
for k := range h.Form { |
|
|
|
v := h.Form[k] |
|
|
|
// because `{{` gets urlencoded
|
|
|
|
//k = strings.Replace(k, "{{ .Name }}", d.Name, -1)
|
|
|
|
v = strings.Replace(v, "{{ .Name }}", d.Name, -1) |
|
|
|
d.Logger <- fmt.Sprintf("[HEADER] %s: %s", k, v) |
|
|
|
form.Set(k, v) |
|
|
|
} |
|
|
|
body = strings.NewReader(form.Encode()) |
|
|
|
} else if 0 != len(h.JSON) { |
|
|
|
bodyBuf, err := json.Marshal(h.JSON) |
|
|
|
if nil != err { |
|
|
|
d.Logger <- fmt.Sprintf("[Notify] JSON Marshal Error for '%s': %s", h.Name, err) |
|
|
|
continue |
|
|
|
} |
|
|
|
// `{{` should be left alone
|
|
|
|
body = strings.NewReader(strings.Replace(string(bodyBuf), "{{ .Name }}", d.Name, -1)) |
|
|
|
} |
|
|
|
func (d *Dog) notifyOne(h Webhook, hardFail bool) { |
|
|
|
// TODO do this in main on config init
|
|
|
|
if "" == h.Method { |
|
|
|
h.Method = "POST" |
|
|
|
} |
|
|
|
|
|
|
|
client := NewHTTPClient() |
|
|
|
req, err := http.NewRequest(h.Method, h.URL, body) |
|
|
|
var body *strings.Reader |
|
|
|
var err error |
|
|
|
// TODO real templates
|
|
|
|
if 0 != len(h.Form) { |
|
|
|
form := url.Values{} |
|
|
|
for k := range h.Form { |
|
|
|
v := h.Form[k] |
|
|
|
// because `{{` gets urlencoded
|
|
|
|
//k = strings.Replace(k, "{{ .Name }}", d.Name, -1)
|
|
|
|
v = strings.Replace(v, "{{ .Name }}", d.Name, -1) |
|
|
|
d.Logger <- fmt.Sprintf("[HEADER] %s: %s", k, v) |
|
|
|
form.Set(k, v) |
|
|
|
} |
|
|
|
body = strings.NewReader(form.Encode()) |
|
|
|
} else if 0 != len(h.JSON) { |
|
|
|
bodyBuf, err := json.Marshal(h.JSON) |
|
|
|
if nil != err { |
|
|
|
d.Logger <- fmt.Sprintf("[Notify] HTTP Client Network Error for '%s': %s", h.Name, err) |
|
|
|
continue |
|
|
|
d.Logger <- fmt.Sprintf("[Notify] JSON Marshal Error for '%s': %s", h.Name, err) |
|
|
|
return |
|
|
|
} |
|
|
|
// `{{` should be left alone
|
|
|
|
body = strings.NewReader(strings.Replace(string(bodyBuf), "{{ .Name }}", d.Name, -1)) |
|
|
|
} |
|
|
|
|
|
|
|
if 0 != len(h.Form) { |
|
|
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded") |
|
|
|
} else if 0 != len(h.JSON) { |
|
|
|
req.Header.Set("Content-Type", "application/json") |
|
|
|
} |
|
|
|
client := NewHTTPClient() |
|
|
|
req, err := http.NewRequest(h.Method, h.URL, body) |
|
|
|
if nil != err { |
|
|
|
d.Logger <- fmt.Sprintf("[Notify] HTTP Client Network Error for '%s': %s", h.Name, err) |
|
|
|
return |
|
|
|
} |
|
|
|
|
|
|
|
if 0 != len(h.Auth) { |
|
|
|
user := h.Auth["user"] |
|
|
|
if "" == user { |
|
|
|
user = h.Auth["username"] |
|
|
|
} |
|
|
|
pass := h.Auth["pass"] |
|
|
|
if "" == user { |
|
|
|
pass = h.Auth["password"] |
|
|
|
} |
|
|
|
req.SetBasicAuth(user, pass) |
|
|
|
} |
|
|
|
if 0 != len(h.Form) { |
|
|
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded") |
|
|
|
} else if 0 != len(h.JSON) { |
|
|
|
req.Header.Set("Content-Type", "application/json") |
|
|
|
} |
|
|
|
|
|
|
|
req.Header.Set("User-Agent", "Watchdog/1.0") |
|
|
|
for k := range h.Headers { |
|
|
|
req.Header.Set(k, h.Headers[k]) |
|
|
|
if 0 != len(h.Auth) { |
|
|
|
user := h.Auth["user"] |
|
|
|
if "" == user { |
|
|
|
user = h.Auth["username"] |
|
|
|
} |
|
|
|
|
|
|
|
resp, err := client.Do(req) |
|
|
|
if nil != err { |
|
|
|
d.Logger <- fmt.Sprintf("[Notify] HTTP Client Error for '%s': %s", h.Name, err) |
|
|
|
continue |
|
|
|
pass := h.Auth["pass"] |
|
|
|
if "" == user { |
|
|
|
pass = h.Auth["password"] |
|
|
|
} |
|
|
|
req.SetBasicAuth(user, pass) |
|
|
|
} |
|
|
|
|
|
|
|
if !(resp.StatusCode >= 200 && resp.StatusCode < 300) { |
|
|
|
d.Logger <- fmt.Sprintf("[Notify] Response Error for '%s': %s", h.Name, resp.Status) |
|
|
|
continue |
|
|
|
} |
|
|
|
req.Header.Set("User-Agent", "Watchdog/1.0") |
|
|
|
for k := range h.Headers { |
|
|
|
req.Header.Set(k, h.Headers[k]) |
|
|
|
} |
|
|
|
|
|
|
|
// TODO json vs xml vs txt
|
|
|
|
var data map[string]interface{} |
|
|
|
req.Header.Add("Accept", "application/json") |
|
|
|
decoder := json.NewDecoder(resp.Body) |
|
|
|
err = decoder.Decode(&data) |
|
|
|
if err != nil { |
|
|
|
d.Logger <- fmt.Sprintf("[Notify] Response Body Error for '%s': %s", h.Name, resp.Status) |
|
|
|
continue |
|
|
|
} |
|
|
|
resp, err := client.Do(req) |
|
|
|
if nil != err { |
|
|
|
d.Logger <- fmt.Sprintf("[Notify] HTTP Client Error for '%s': %s", h.Name, err) |
|
|
|
return |
|
|
|
} |
|
|
|
|
|
|
|
if !(resp.StatusCode >= 200 && resp.StatusCode < 300) { |
|
|
|
d.Logger <- fmt.Sprintf("[Notify] Response Error for '%s': %s", h.Name, resp.Status) |
|
|
|
return |
|
|
|
} |
|
|
|
|
|
|
|
// TODO some sort of way to determine if data is successful (keywords)
|
|
|
|
d.Logger <- fmt.Sprintf("[Notify] Success? %#v", data) |
|
|
|
// TODO json vs xml vs txt
|
|
|
|
var data map[string]interface{} |
|
|
|
req.Header.Add("Accept", "application/json") |
|
|
|
decoder := json.NewDecoder(resp.Body) |
|
|
|
err = decoder.Decode(&data) |
|
|
|
if err != nil { |
|
|
|
d.Logger <- fmt.Sprintf("[Notify] Response Body Error for '%s': %s", h.Name, resp.Status) |
|
|
|
return |
|
|
|
} |
|
|
|
|
|
|
|
// TODO some sort of way to determine if data is successful (keywords)
|
|
|
|
d.Logger <- fmt.Sprintf("[Notify] Success? %#v", data) |
|
|
|
} |
|
|
|
|
|
|
|
type Config struct { |
|
|
|
Watches []ConfigWatch `json:"watches"` |
|
|
|
Webhooks []ConfigWebhook `json:"webhooks"` |
|
|
|
Watches []ConfigWatch `json:"watches"` |
|
|
|
Webhooks []Webhook `json:"webhooks"` |
|
|
|
} |
|
|
|
|
|
|
|
type ConfigWatch struct { |
|
|
@ -270,7 +274,7 @@ type ConfigWatch struct { |
|
|
|
RecoverScript string `json:"recover_script"` |
|
|
|
} |
|
|
|
|
|
|
|
type ConfigWebhook struct { |
|
|
|
type Webhook struct { |
|
|
|
Name string `json:"name"` |
|
|
|
Method string `json:"method"` |
|
|
|
URL string `json:"url"` |
|
|
|