feat(alert): add Telegram, PagerDuty, Pushover, Gotify providers
Expand alert provider count from 5 to 9. All new providers use the shared HTTPProvider with closure-based payload functions. Includes TUI form support and tests for each provider.
This commit is contained in:
@@ -7,6 +7,7 @@ import (
|
||||
"go-upkeep/internal/models"
|
||||
"net/http"
|
||||
"net/smtp"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
@@ -52,6 +53,52 @@ func webhookPayload(title, message string) ([]byte, error) {
|
||||
return json.Marshal(map[string]string{"title": title, "message": message, "status": "alert"})
|
||||
}
|
||||
|
||||
func telegramPayload(chatID string) PayloadFunc {
|
||||
return func(title, message string) ([]byte, error) {
|
||||
return json.Marshal(map[string]string{
|
||||
"chat_id": chatID,
|
||||
"text": fmt.Sprintf("*%s*\n%s", title, message),
|
||||
"parse_mode": "Markdown",
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func pagerdutyPayload(routingKey, severity string) PayloadFunc {
|
||||
return func(title, message string) ([]byte, error) {
|
||||
return json.Marshal(map[string]any{
|
||||
"routing_key": routingKey,
|
||||
"event_action": "trigger",
|
||||
"payload": map[string]string{
|
||||
"summary": fmt.Sprintf("%s: %s", title, message),
|
||||
"source": "go-upkeep",
|
||||
"severity": severity,
|
||||
},
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func pushoverPayload(token, user string) PayloadFunc {
|
||||
return func(title, message string) ([]byte, error) {
|
||||
return json.Marshal(map[string]string{
|
||||
"token": token,
|
||||
"user": user,
|
||||
"title": title,
|
||||
"message": message,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func gotifyPayload(priority string) PayloadFunc {
|
||||
return func(title, message string) ([]byte, error) {
|
||||
pri, _ := strconv.Atoi(priority)
|
||||
return json.Marshal(map[string]any{
|
||||
"title": title,
|
||||
"message": message,
|
||||
"priority": pri,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func GetProvider(cfg models.AlertConfig) Provider {
|
||||
switch cfg.Type {
|
||||
case "discord":
|
||||
@@ -85,6 +132,35 @@ func GetProvider(cfg models.AlertConfig) Provider {
|
||||
Username: cfg.Settings["username"],
|
||||
Password: cfg.Settings["password"],
|
||||
}
|
||||
case "telegram":
|
||||
return &HTTPProvider{
|
||||
URL: fmt.Sprintf("https://api.telegram.org/bot%s/sendMessage", cfg.Settings["token"]),
|
||||
Payload: telegramPayload(cfg.Settings["chat_id"]),
|
||||
}
|
||||
case "pagerduty":
|
||||
severity := "critical"
|
||||
if s, ok := cfg.Settings["severity"]; ok && s != "" {
|
||||
severity = s
|
||||
}
|
||||
return &HTTPProvider{
|
||||
URL: "https://events.pagerduty.com/v2/enqueue",
|
||||
Payload: pagerdutyPayload(cfg.Settings["routing_key"], severity),
|
||||
}
|
||||
case "pushover":
|
||||
return &HTTPProvider{
|
||||
URL: "https://api.pushover.net/1/messages.json",
|
||||
Payload: pushoverPayload(cfg.Settings["token"], cfg.Settings["user"]),
|
||||
}
|
||||
case "gotify":
|
||||
priority := "5"
|
||||
if p, ok := cfg.Settings["priority"]; ok && p != "" {
|
||||
priority = p
|
||||
}
|
||||
serverURL := strings.TrimRight(cfg.Settings["url"], "/")
|
||||
return &HTTPProvider{
|
||||
URL: fmt.Sprintf("%s/message?token=%s", serverURL, cfg.Settings["token"]),
|
||||
Payload: gotifyPayload(priority),
|
||||
}
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
|
||||
@@ -101,6 +101,110 @@ func TestNtfyProvider(t *testing.T) {
|
||||
}
|
||||
}
|
||||
|
||||
func TestHTTPProviderTelegram(t *testing.T) {
|
||||
var received map[string]string
|
||||
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
json.NewDecoder(r.Body).Decode(&received)
|
||||
w.WriteHeader(200)
|
||||
}))
|
||||
defer srv.Close()
|
||||
|
||||
p := &HTTPProvider{URL: srv.URL, Payload: telegramPayload("12345")}
|
||||
if err := p.Send("Alert", "Down"); err != nil {
|
||||
t.Fatalf("Send: %v", err)
|
||||
}
|
||||
if received["chat_id"] != "12345" {
|
||||
t.Errorf("expected chat_id '12345', got '%s'", received["chat_id"])
|
||||
}
|
||||
if received["text"] != "*Alert*\nDown" {
|
||||
t.Errorf("unexpected text: %s", received["text"])
|
||||
}
|
||||
if received["parse_mode"] != "Markdown" {
|
||||
t.Errorf("expected parse_mode 'Markdown', got '%s'", received["parse_mode"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestHTTPProviderPagerDuty(t *testing.T) {
|
||||
var received map[string]any
|
||||
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
json.NewDecoder(r.Body).Decode(&received)
|
||||
w.WriteHeader(200)
|
||||
}))
|
||||
defer srv.Close()
|
||||
|
||||
p := &HTTPProvider{URL: srv.URL, Payload: pagerdutyPayload("test-key", "critical")}
|
||||
if err := p.Send("Alert", "Down"); err != nil {
|
||||
t.Fatalf("Send: %v", err)
|
||||
}
|
||||
if received["routing_key"] != "test-key" {
|
||||
t.Errorf("expected routing_key 'test-key', got '%v'", received["routing_key"])
|
||||
}
|
||||
if received["event_action"] != "trigger" {
|
||||
t.Errorf("expected event_action 'trigger', got '%v'", received["event_action"])
|
||||
}
|
||||
payload := received["payload"].(map[string]any)
|
||||
if payload["summary"] != "Alert: Down" {
|
||||
t.Errorf("unexpected summary: %v", payload["summary"])
|
||||
}
|
||||
if payload["severity"] != "critical" {
|
||||
t.Errorf("expected severity 'critical', got '%v'", payload["severity"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestHTTPProviderPushover(t *testing.T) {
|
||||
var received map[string]string
|
||||
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
json.NewDecoder(r.Body).Decode(&received)
|
||||
w.WriteHeader(200)
|
||||
}))
|
||||
defer srv.Close()
|
||||
|
||||
p := &HTTPProvider{URL: srv.URL, Payload: pushoverPayload("app-tok", "user-key")}
|
||||
if err := p.Send("Alert", "Down"); err != nil {
|
||||
t.Fatalf("Send: %v", err)
|
||||
}
|
||||
if received["token"] != "app-tok" {
|
||||
t.Errorf("expected token 'app-tok', got '%s'", received["token"])
|
||||
}
|
||||
if received["user"] != "user-key" {
|
||||
t.Errorf("expected user 'user-key', got '%s'", received["user"])
|
||||
}
|
||||
if received["title"] != "Alert" || received["message"] != "Down" {
|
||||
t.Errorf("unexpected payload: %v", received)
|
||||
}
|
||||
}
|
||||
|
||||
func TestHTTPProviderGotify(t *testing.T) {
|
||||
var received map[string]any
|
||||
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
json.NewDecoder(r.Body).Decode(&received)
|
||||
w.WriteHeader(200)
|
||||
}))
|
||||
defer srv.Close()
|
||||
|
||||
p := &HTTPProvider{URL: srv.URL, Payload: gotifyPayload("8")}
|
||||
if err := p.Send("Alert", "Down"); err != nil {
|
||||
t.Fatalf("Send: %v", err)
|
||||
}
|
||||
if received["title"] != "Alert" || received["message"] != "Down" {
|
||||
t.Errorf("unexpected payload: %v", received)
|
||||
}
|
||||
if pri, ok := received["priority"].(float64); !ok || pri != 8 {
|
||||
t.Errorf("expected priority 8, got %v", received["priority"])
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetProviderNewTypes(t *testing.T) {
|
||||
for _, typ := range []string{"telegram", "pagerduty", "pushover", "gotify"} {
|
||||
p := GetProvider(models.AlertConfig{Type: typ, Settings: map[string]string{
|
||||
"token": "x", "chat_id": "1", "routing_key": "k", "user": "u", "url": "http://localhost",
|
||||
}})
|
||||
if p == nil {
|
||||
t.Errorf("GetProvider(%q) returned nil", typ)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetProviderUnknown(t *testing.T) {
|
||||
p := GetProvider(models.AlertConfig{Type: "unknown"})
|
||||
if p != nil {
|
||||
|
||||
Reference in New Issue
Block a user