From 95655dccb87f3316934f88f679190d30060f5e47 Mon Sep 17 00:00:00 2001 From: Jeff Wong Date: Thu, 24 Oct 2024 13:12:05 -0700 Subject: [PATCH] Discord notifier: Add username and avatar_url (#4081) * Feat(discord): Allow for custom username and avatar URLs to be set in discord notifications. Add `username` and `avatar_url` to discord configuration, default empty string. Re-implement #3821 Signed-off-by: Jeff Wong * Test the new fields Signed-off-by: gotjosh * These are not templeatable strings Signed-off-by: gotjosh --------- Signed-off-by: Jeff Wong Signed-off-by: gotjosh Co-authored-by: gotjosh --- config/notifiers.go | 8 +++++--- docs/configuration.md | 6 ++++++ notify/discord/discord.go | 18 +++++++++++++++--- notify/discord/discord_test.go | 4 +++- 4 files changed, 29 insertions(+), 7 deletions(-) diff --git a/config/notifiers.go b/config/notifiers.go index ac437096..fe28ca05 100644 --- a/config/notifiers.go +++ b/config/notifiers.go @@ -251,9 +251,11 @@ type DiscordConfig struct { WebhookURL *SecretURL `yaml:"webhook_url,omitempty" json:"webhook_url,omitempty"` WebhookURLFile string `yaml:"webhook_url_file,omitempty" json:"webhook_url_file,omitempty"` - Content string `yaml:"content,omitempty" json:"content,omitempty"` - Title string `yaml:"title,omitempty" json:"title,omitempty"` - Message string `yaml:"message,omitempty" json:"message,omitempty"` + Content string `yaml:"content,omitempty" json:"content,omitempty"` + Title string `yaml:"title,omitempty" json:"title,omitempty"` + Message string `yaml:"message,omitempty" json:"message,omitempty"` + Username string `yaml:"username,omitempty" json:"username,omitempty"` + AvatarURL string `yaml:"avatar_url,omitempty" json:"avatar_url,omitempty"` } // UnmarshalYAML implements the yaml.Unmarshaler interface. diff --git a/docs/configuration.md b/docs/configuration.md index 69654865..4f20a1ef 100644 --- a/docs/configuration.md +++ b/docs/configuration.md @@ -886,6 +886,12 @@ webhook_url_file: # Message content template. Limited to 2000 characters. [ content: | default = '{{ template "discord.default.content" . }}' ] +# Message username. +[ username: | default = '' ] + +# Message avatar URL. +[ avatar_url: | default = '' ] + # The HTTP client's configuration. [ http_config: | default = global.http_config ] ``` diff --git a/notify/discord/discord.go b/notify/discord/discord.go index 7ab60bb8..a63137c6 100644 --- a/notify/discord/discord.go +++ b/notify/discord/discord.go @@ -19,6 +19,7 @@ import ( "encoding/json" "fmt" "net/http" + netUrl "net/url" "os" "strings" @@ -76,8 +77,10 @@ func New(c *config.DiscordConfig, t *template.Template, l log.Logger, httpOpts . } type webhook struct { - Content string `json:"content"` - Embeds []webhookEmbed `json:"embeds"` + Content string `json:"content"` + Embeds []webhookEmbed `json:"embeds"` + Username string `json:"username,omitempty"` + AvatarURL string `json:"avatar_url,omitempty"` } type webhookEmbed struct { @@ -145,7 +148,8 @@ func (n *Notifier) Notify(ctx context.Context, as ...*types.Alert) (bool, error) } w := webhook{ - Content: content, + Content: content, + Username: n.conf.Username, Embeds: []webhookEmbed{{ Title: title, Description: description, @@ -153,6 +157,14 @@ func (n *Notifier) Notify(ctx context.Context, as ...*types.Alert) (bool, error) }}, } + if len(n.conf.AvatarURL) != 0 { + if _, err := netUrl.Parse(n.conf.AvatarURL); err == nil { + w.AvatarURL = n.conf.AvatarURL + } else { + level.Warn(n.logger).Log("msg", "Bad avatar url", "key", key) + } + } + var payload bytes.Buffer if err = json.NewEncoder(&payload).Encode(w); err != nil { return false, err diff --git a/notify/discord/discord_test.go b/notify/discord/discord_test.go index 70251094..f19c8b62 100644 --- a/notify/discord/discord_test.go +++ b/notify/discord/discord_test.go @@ -201,6 +201,8 @@ func TestDiscord_Notify(t *testing.T) { Title: "Test Title", Message: "Test Message", Content: "Test Content", + Username: "Test Username", + AvatarURL: "http://example.com/avatar.png", } // Create a new Discord notifier @@ -227,5 +229,5 @@ func TestDiscord_Notify(t *testing.T) { require.NoError(t, err) require.False(t, ok) - require.Equal(t, "{\"content\":\"Test Content\",\"embeds\":[{\"title\":\"Test Title\",\"description\":\"Test Message\",\"color\":10038562}]}\n", resp) + require.Equal(t, "{\"content\":\"Test Content\",\"embeds\":[{\"title\":\"Test Title\",\"description\":\"Test Message\",\"color\":10038562}],\"username\":\"Test Username\",\"avatar_url\":\"http://example.com/avatar.png\"}\n", resp) }