mirror of
https://github.com/containrrr/watchtower.git
synced 2025-12-14 06:06:38 +01:00
dc12a1ac7f
* feat: allow hostname override for email notifier As it currently stands all notifiers utilise `os.Hostname` to populate their titles/subjects. When utilising Docker with a bridged network if you set the hostname for a container to an external DNS hostname Docker's internal DNS resolver will override said hostname for all containers within the bridged network. This change allows a user to specify what hostname should be represented in the email notifications without having to change the `os.Hostname`. * feat: allow custom hostname for all notifiers * docs: adjust notification hostname flag
78 lines
2.2 KiB
Go
78 lines
2.2 KiB
Go
package notifications
|
|
|
|
import (
|
|
"strings"
|
|
|
|
shoutrrrDisco "github.com/containrrr/shoutrrr/pkg/services/discord"
|
|
shoutrrrSlack "github.com/containrrr/shoutrrr/pkg/services/slack"
|
|
t "github.com/containrrr/watchtower/pkg/types"
|
|
"github.com/johntdyer/slackrus"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
const (
|
|
slackType = "slack"
|
|
)
|
|
|
|
type slackTypeNotifier struct {
|
|
slackrus.SlackrusHook
|
|
}
|
|
|
|
// NewSlackNotifier is a factory function used to generate new instance of the slack notifier type
|
|
func NewSlackNotifier(c *cobra.Command, acceptedLogLevels []log.Level) t.ConvertibleNotifier {
|
|
return newSlackNotifier(c, acceptedLogLevels)
|
|
}
|
|
|
|
func newSlackNotifier(c *cobra.Command, acceptedLogLevels []log.Level) t.ConvertibleNotifier {
|
|
flags := c.PersistentFlags()
|
|
|
|
hookURL, _ := flags.GetString("notification-slack-hook-url")
|
|
userName, _ := flags.GetString("notification-slack-identifier")
|
|
channel, _ := flags.GetString("notification-slack-channel")
|
|
emoji, _ := flags.GetString("notification-slack-icon-emoji")
|
|
iconURL, _ := flags.GetString("notification-slack-icon-url")
|
|
|
|
n := &slackTypeNotifier{
|
|
SlackrusHook: slackrus.SlackrusHook{
|
|
HookURL: hookURL,
|
|
Username: userName,
|
|
Channel: channel,
|
|
IconEmoji: emoji,
|
|
IconURL: iconURL,
|
|
AcceptedLevels: acceptedLogLevels,
|
|
},
|
|
}
|
|
return n
|
|
}
|
|
|
|
func (s *slackTypeNotifier) GetURL(c *cobra.Command) (string, error) {
|
|
trimmedURL := strings.TrimRight(s.HookURL, "/")
|
|
trimmedURL = strings.TrimLeft(trimmedURL, "https://")
|
|
parts := strings.Split(trimmedURL, "/")
|
|
|
|
if parts[0] == "discord.com" || parts[0] == "discordapp.com" {
|
|
log.Debug("Detected a discord slack wrapper URL, using shoutrrr discord service")
|
|
conf := &shoutrrrDisco.Config{
|
|
Channel: parts[len(parts)-3],
|
|
Token: parts[len(parts)-2],
|
|
Color: ColorInt,
|
|
Title: GetTitle(c),
|
|
SplitLines: true,
|
|
Username: s.Username,
|
|
}
|
|
return conf.GetURL().String(), nil
|
|
}
|
|
|
|
rawTokens := strings.Replace(s.HookURL, "https://hooks.slack.com/services/", "", 1)
|
|
tokens := strings.Split(rawTokens, "/")
|
|
|
|
conf := &shoutrrrSlack.Config{
|
|
BotName: s.Username,
|
|
Token: tokens,
|
|
Color: ColorHex,
|
|
Title: GetTitle(c),
|
|
}
|
|
|
|
return conf.GetURL().String(), nil
|
|
}
|