You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
3.2 KiB
Go
116 lines
3.2 KiB
Go
package notifications
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/containrrr/shoutrrr/pkg/format"
|
|
"github.com/spf13/cobra"
|
|
|
|
shoutrrrSmtp "github.com/containrrr/shoutrrr/pkg/services/smtp"
|
|
t "github.com/containrrr/watchtower/pkg/types"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
const (
|
|
emailType = "email"
|
|
)
|
|
|
|
type emailTypeNotifier struct {
|
|
url string
|
|
From, To string
|
|
Server, User, Password, SubjectTag string
|
|
Port int
|
|
tlsSkipVerify bool
|
|
entries []*log.Entry
|
|
logLevels []log.Level
|
|
delay time.Duration
|
|
}
|
|
|
|
// NewEmailNotifier is a factory method creating a new email notifier instance
|
|
func NewEmailNotifier(c *cobra.Command, acceptedLogLevels []log.Level) t.ConvertableNotifier {
|
|
return newEmailNotifier(c, acceptedLogLevels)
|
|
}
|
|
|
|
func newEmailNotifier(c *cobra.Command, acceptedLogLevels []log.Level) t.ConvertableNotifier {
|
|
flags := c.PersistentFlags()
|
|
|
|
from, _ := flags.GetString("notification-email-from")
|
|
to, _ := flags.GetString("notification-email-to")
|
|
server, _ := flags.GetString("notification-email-server")
|
|
user, _ := flags.GetString("notification-email-server-user")
|
|
password, _ := flags.GetString("notification-email-server-password")
|
|
port, _ := flags.GetInt("notification-email-server-port")
|
|
tlsSkipVerify, _ := flags.GetBool("notification-email-server-tls-skip-verify")
|
|
delay, _ := flags.GetInt("notification-email-delay")
|
|
subjecttag, _ := flags.GetString("notification-email-subjecttag")
|
|
|
|
n := &emailTypeNotifier{
|
|
entries: []*log.Entry{},
|
|
From: from,
|
|
To: to,
|
|
Server: server,
|
|
User: user,
|
|
Password: password,
|
|
Port: port,
|
|
tlsSkipVerify: tlsSkipVerify,
|
|
logLevels: acceptedLogLevels,
|
|
delay: time.Duration(delay) * time.Second,
|
|
SubjectTag: subjecttag,
|
|
}
|
|
|
|
return n
|
|
}
|
|
|
|
func (e *emailTypeNotifier) GetURL() string {
|
|
conf := &shoutrrrSmtp.Config{
|
|
FromAddress: e.From,
|
|
FromName: "Watchtower",
|
|
ToAddresses: []string{e.To},
|
|
Port: uint16(e.Port),
|
|
Host: e.Server,
|
|
Subject: e.getSubject(),
|
|
Username: e.User,
|
|
Password: e.Password,
|
|
UseStartTLS: true,
|
|
UseHTML: false,
|
|
}
|
|
|
|
pkr := format.NewPropKeyResolver(conf)
|
|
var err error
|
|
if len(e.User) > 0 {
|
|
err = pkr.Set("auth", "Plain")
|
|
} else {
|
|
err = pkr.Set("auth", "None")
|
|
}
|
|
|
|
if err != nil {
|
|
fmt.Printf("Could not set auth type for email notifier: %v", err)
|
|
}
|
|
|
|
return conf.GetURL().String()
|
|
}
|
|
|
|
func (e *emailTypeNotifier) getSubject() string {
|
|
var emailSubject string
|
|
|
|
if e.SubjectTag == "" {
|
|
emailSubject = "Watchtower updates"
|
|
} else {
|
|
emailSubject = e.SubjectTag + " Watchtower updates"
|
|
}
|
|
if hostname, err := os.Hostname(); err == nil {
|
|
emailSubject += " on " + hostname
|
|
}
|
|
return emailSubject
|
|
}
|
|
|
|
// TODO: Delete these once all notifiers have been converted to shoutrrr
|
|
func (e *emailTypeNotifier) StartNotification() {}
|
|
func (e *emailTypeNotifier) SendNotification() {}
|
|
func (e *emailTypeNotifier) Levels() []log.Level { return nil }
|
|
func (e *emailTypeNotifier) Fire(entry *log.Entry) error { return nil }
|
|
|
|
func (e *emailTypeNotifier) Close() {}
|