129 lines
2.8 KiB
Go
129 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
const dateLayout = "2006-01-02 15:04:05"
|
|
|
|
type silence struct {
|
|
Matchers []matcher `json:"matchers"`
|
|
StartsAt string `json:"startsAt"`
|
|
EndsAt string `json:"endsAt"`
|
|
CreatedBy string `json:"createdBy"`
|
|
Comment string `json:"comment"`
|
|
}
|
|
|
|
type matcher struct {
|
|
Name string `json:"name"`
|
|
Value string `json:"value"`
|
|
IsRegex bool `json:"isRegex"`
|
|
IsEqual bool `json:"isEqual"`
|
|
}
|
|
|
|
type silenceBody struct {
|
|
AlertManagerURL string `json:"alertmanagerURL"`
|
|
Labels map[string]string `json:"labels"`
|
|
}
|
|
|
|
func (rcv *receiver) handleSilences(w http.ResponseWriter, r *http.Request) {
|
|
defer r.Body.Close()
|
|
|
|
if r.Method != http.MethodPost {
|
|
http.Error(w, "Only POST allowed", http.StatusMethodNotAllowed)
|
|
rcv.logger.Debugf("silences: illegal HTTP method: expected %q, got %q", "POST", r.Method)
|
|
return
|
|
}
|
|
|
|
b, err := io.ReadAll(r.Body)
|
|
if err != nil {
|
|
rcv.logger.Debugf("silences: %v", err)
|
|
return
|
|
}
|
|
|
|
b, err = base64.StdEncoding.DecodeString(string(b))
|
|
if err != nil {
|
|
rcv.logger.Debugf("silences: %v", err)
|
|
return
|
|
}
|
|
|
|
var sb silenceBody
|
|
err = json.Unmarshal(b, &sb)
|
|
if err != nil {
|
|
rcv.logger.Debugf("silences: %v", err)
|
|
return
|
|
}
|
|
|
|
var matchers []matcher
|
|
for key, value := range sb.Labels {
|
|
m := matcher{
|
|
Name: key,
|
|
Value: value,
|
|
IsRegex: false,
|
|
IsEqual: true,
|
|
}
|
|
|
|
matchers = append(matchers, m)
|
|
}
|
|
|
|
silence := &silence{
|
|
StartsAt: time.Now().UTC().Format(dateLayout),
|
|
EndsAt: time.Now().Add(rcv.cfg.am.SilenceDuration).UTC().Format(dateLayout),
|
|
CreatedBy: "ntfy-alertmanager",
|
|
Comment: "",
|
|
Matchers: matchers,
|
|
}
|
|
|
|
b, err = json.Marshal(silence)
|
|
if err != nil {
|
|
rcv.logger.Debugf("silences: %v", err)
|
|
return
|
|
}
|
|
|
|
client := &http.Client{Timeout: time.Second * 3}
|
|
|
|
url := sb.AlertManagerURL
|
|
if rcv.cfg.am.URL != "" {
|
|
url = rcv.cfg.am.URL
|
|
}
|
|
url += "/api/v2/silences"
|
|
|
|
req, err := http.NewRequest(http.MethodPost, url, bytes.NewBuffer(b))
|
|
if err != nil {
|
|
rcv.logger.Debugf("silences: %v", err)
|
|
return
|
|
}
|
|
|
|
// Basic auth
|
|
if rcv.cfg.am.User != "" && rcv.cfg.am.Password != "" {
|
|
auth := base64.StdEncoding.EncodeToString([]byte(fmt.Sprintf("%s:%s", rcv.cfg.am.User, rcv.cfg.am.Password)))
|
|
req.Header.Set("Authorization", fmt.Sprintf("Basic %s", auth))
|
|
}
|
|
|
|
req.Header.Add("Content-Type", "application/json")
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
rcv.logger.Debugf("silences: %v", err)
|
|
return
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
b, err = io.ReadAll(resp.Body)
|
|
if err != nil {
|
|
rcv.logger.Debugf("silences: %v", err)
|
|
return
|
|
}
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
rcv.logger.Debugf("silences: received status code %d", resp.StatusCode)
|
|
return
|
|
}
|
|
|
|
rcv.logger.Debugf("silences: created new silence %s", string(b))
|
|
}
|