2023-11-03 23:21:52 +01:00
|
|
|
package gotify
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"encoding/json"
|
2024-12-17 02:32:13 +01:00
|
|
|
"errors"
|
2023-11-03 23:21:52 +01:00
|
|
|
"fmt"
|
|
|
|
"io"
|
|
|
|
"net/http"
|
|
|
|
|
|
|
|
"github.com/TwiN/gatus/v5/alerting/alert"
|
|
|
|
"github.com/TwiN/gatus/v5/client"
|
2024-05-10 04:56:16 +02:00
|
|
|
"github.com/TwiN/gatus/v5/config/endpoint"
|
2024-12-17 02:32:13 +01:00
|
|
|
"gopkg.in/yaml.v3"
|
2023-11-03 23:21:52 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
const DefaultPriority = 5
|
|
|
|
|
2024-12-17 02:32:13 +01:00
|
|
|
var (
|
|
|
|
ErrServerURLNotSet = errors.New("server URL not set")
|
|
|
|
ErrTokenNotSet = errors.New("token not set")
|
|
|
|
)
|
2023-11-03 23:21:52 +01:00
|
|
|
|
2024-12-17 02:32:13 +01:00
|
|
|
type Config struct {
|
|
|
|
ServerURL string `yaml:"server-url"` // URL of the Gotify server
|
|
|
|
Token string `yaml:"token"` // Token to use when sending a message to the Gotify server
|
|
|
|
Priority int `yaml:"priority,omitempty"` // Priority of the message. Defaults to DefaultPriority.
|
|
|
|
Title string `yaml:"title,omitempty"` // Title of the message that will be sent
|
|
|
|
}
|
2023-11-03 23:21:52 +01:00
|
|
|
|
2024-12-17 02:32:13 +01:00
|
|
|
func (cfg *Config) Validate() error {
|
|
|
|
if cfg.Priority == 0 {
|
|
|
|
cfg.Priority = DefaultPriority
|
|
|
|
}
|
|
|
|
if len(cfg.ServerURL) == 0 {
|
|
|
|
return ErrServerURLNotSet
|
|
|
|
}
|
|
|
|
if len(cfg.Token) == 0 {
|
|
|
|
return ErrTokenNotSet
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (cfg *Config) Merge(override *Config) {
|
|
|
|
if len(override.ServerURL) > 0 {
|
|
|
|
cfg.ServerURL = override.ServerURL
|
|
|
|
}
|
|
|
|
if len(override.Token) > 0 {
|
|
|
|
cfg.Token = override.Token
|
|
|
|
}
|
|
|
|
if override.Priority != 0 {
|
|
|
|
cfg.Priority = override.Priority
|
|
|
|
}
|
|
|
|
if len(override.Title) > 0 {
|
|
|
|
cfg.Title = override.Title
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// AlertProvider is the configuration necessary for sending an alert using Gotify
|
|
|
|
type AlertProvider struct {
|
|
|
|
DefaultConfig Config `yaml:",inline"`
|
2023-11-03 23:21:52 +01:00
|
|
|
|
|
|
|
// DefaultAlert is the default alert configuration to use for endpoints with an alert of the appropriate type
|
|
|
|
DefaultAlert *alert.Alert `yaml:"default-alert,omitempty"`
|
|
|
|
}
|
|
|
|
|
2024-12-17 02:32:13 +01:00
|
|
|
// Validate the provider's configuration
|
|
|
|
func (provider *AlertProvider) Validate() error {
|
|
|
|
return provider.DefaultConfig.Validate()
|
2023-11-03 23:21:52 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// Send an alert using the provider
|
2024-05-10 04:56:16 +02:00
|
|
|
func (provider *AlertProvider) Send(ep *endpoint.Endpoint, alert *alert.Alert, result *endpoint.Result, resolved bool) error {
|
2024-12-17 02:32:13 +01:00
|
|
|
cfg, err := provider.GetConfig(ep.Group, alert)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
buffer := bytes.NewBuffer(provider.buildRequestBody(cfg, ep, alert, result, resolved))
|
|
|
|
request, err := http.NewRequest(http.MethodPost, cfg.ServerURL+"/message?token="+cfg.Token, buffer)
|
2023-11-03 23:21:52 +01:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
request.Header.Set("Content-Type", "application/json")
|
|
|
|
response, err := client.GetHTTPClient(nil).Do(request)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
defer response.Body.Close()
|
|
|
|
if response.StatusCode > 399 {
|
|
|
|
body, _ := io.ReadAll(response.Body)
|
|
|
|
return fmt.Errorf("failed to send alert to Gotify: %s", string(body))
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
type Body struct {
|
|
|
|
Message string `json:"message"`
|
|
|
|
Title string `json:"title"`
|
|
|
|
Priority int `json:"priority"`
|
|
|
|
}
|
|
|
|
|
|
|
|
// buildRequestBody builds the request body for the provider
|
2024-12-17 02:32:13 +01:00
|
|
|
func (provider *AlertProvider) buildRequestBody(cfg *Config, ep *endpoint.Endpoint, alert *alert.Alert, result *endpoint.Result, resolved bool) []byte {
|
2024-04-11 02:46:17 +02:00
|
|
|
var message string
|
2023-11-03 23:21:52 +01:00
|
|
|
if resolved {
|
2024-05-10 04:56:16 +02:00
|
|
|
message = fmt.Sprintf("An alert for `%s` has been resolved after passing successfully %d time(s) in a row", ep.DisplayName(), alert.SuccessThreshold)
|
2023-11-03 23:21:52 +01:00
|
|
|
} else {
|
2024-05-10 04:56:16 +02:00
|
|
|
message = fmt.Sprintf("An alert for `%s` has been triggered due to having failed %d time(s) in a row", ep.DisplayName(), alert.FailureThreshold)
|
2023-11-03 23:21:52 +01:00
|
|
|
}
|
2024-04-11 02:46:17 +02:00
|
|
|
var formattedConditionResults string
|
2023-11-03 23:21:52 +01:00
|
|
|
for _, conditionResult := range result.ConditionResults {
|
|
|
|
var prefix string
|
|
|
|
if conditionResult.Success {
|
|
|
|
prefix = "✓"
|
|
|
|
} else {
|
|
|
|
prefix = "✕"
|
|
|
|
}
|
2024-04-11 02:46:17 +02:00
|
|
|
formattedConditionResults += fmt.Sprintf("\n%s - %s", prefix, conditionResult.Condition)
|
2023-11-03 23:21:52 +01:00
|
|
|
}
|
|
|
|
if len(alert.GetDescription()) > 0 {
|
|
|
|
message += " with the following description: " + alert.GetDescription()
|
|
|
|
}
|
2024-04-11 02:46:17 +02:00
|
|
|
message += formattedConditionResults
|
2024-05-10 04:56:16 +02:00
|
|
|
title := "Gatus: " + ep.DisplayName()
|
2024-12-17 02:32:13 +01:00
|
|
|
if cfg.Title != "" {
|
|
|
|
title = cfg.Title
|
2023-11-03 23:21:52 +01:00
|
|
|
}
|
2024-04-11 02:46:17 +02:00
|
|
|
bodyAsJSON, _ := json.Marshal(Body{
|
2023-11-03 23:21:52 +01:00
|
|
|
Message: message,
|
|
|
|
Title: title,
|
2024-12-17 02:32:13 +01:00
|
|
|
Priority: cfg.Priority,
|
2023-11-03 23:21:52 +01:00
|
|
|
})
|
2024-04-11 02:46:17 +02:00
|
|
|
return bodyAsJSON
|
2023-11-03 23:21:52 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
// GetDefaultAlert returns the provider's default alert configuration
|
2024-02-08 02:09:45 +01:00
|
|
|
func (provider *AlertProvider) GetDefaultAlert() *alert.Alert {
|
2023-11-03 23:21:52 +01:00
|
|
|
return provider.DefaultAlert
|
|
|
|
}
|
2024-12-17 02:32:13 +01:00
|
|
|
|
|
|
|
// GetConfig returns the configuration for the provider with the overrides applied
|
|
|
|
func (provider *AlertProvider) GetConfig(group string, alert *alert.Alert) (*Config, error) {
|
|
|
|
cfg := provider.DefaultConfig
|
|
|
|
// Handle alert overrides
|
|
|
|
if len(alert.ProviderOverride) != 0 {
|
|
|
|
overrideConfig := Config{}
|
|
|
|
if err := yaml.Unmarshal(alert.ProviderOverrideAsBytes(), &overrideConfig); err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
cfg.Merge(&overrideConfig)
|
|
|
|
}
|
|
|
|
// Validate the configuration
|
|
|
|
err := cfg.Validate()
|
|
|
|
return &cfg, err
|
|
|
|
}
|
|
|
|
|
|
|
|
// ValidateOverrides validates the alert's provider override and, if present, the group override
|
|
|
|
func (provider *AlertProvider) ValidateOverrides(group string, alert *alert.Alert) error {
|
|
|
|
_, err := provider.GetConfig(group, alert)
|
|
|
|
return err
|
|
|
|
}
|