mirror of
https://github.com/containrrr/watchtower.git
synced 2025-09-22 05:40:50 +02:00
85 lines
2 KiB
Go
85 lines
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"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
const (
|
|
slackType = "slack"
|
|
)
|
|
|
|
type slackTypeNotifier struct {
|
|
HookURL string
|
|
Username string
|
|
Channel string
|
|
IconEmoji string
|
|
IconURL string
|
|
}
|
|
|
|
func newSlackNotifier(c *cobra.Command) t.ConvertibleNotifier {
|
|
flags := c.Flags()
|
|
|
|
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{
|
|
HookURL: hookURL,
|
|
Username: userName,
|
|
Channel: channel,
|
|
IconEmoji: emoji,
|
|
IconURL: iconURL,
|
|
}
|
|
return n
|
|
}
|
|
|
|
func (s *slackTypeNotifier) GetURL(c *cobra.Command) (string, error) {
|
|
trimmedURL := strings.TrimRight(s.HookURL, "/")
|
|
trimmedURL = strings.TrimPrefix(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{
|
|
WebhookID: parts[len(parts)-3],
|
|
Token: parts[len(parts)-2],
|
|
Color: ColorInt,
|
|
SplitLines: true,
|
|
Username: s.Username,
|
|
}
|
|
|
|
if s.IconURL != "" {
|
|
conf.Avatar = s.IconURL
|
|
}
|
|
|
|
return conf.GetURL().String(), nil
|
|
}
|
|
|
|
webhookToken := strings.Replace(s.HookURL, "https://hooks.slack.com/services/", "", 1)
|
|
|
|
conf := &shoutrrrSlack.Config{
|
|
BotName: s.Username,
|
|
Color: ColorHex,
|
|
Channel: "webhook",
|
|
}
|
|
|
|
if s.IconURL != "" {
|
|
conf.Icon = s.IconURL
|
|
} else if s.IconEmoji != "" {
|
|
conf.Icon = s.IconEmoji
|
|
}
|
|
|
|
if err := conf.Token.SetFromProp(webhookToken); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return conf.GetURL().String(), nil
|
|
}
|