authelia/internal/notification/smtp_notifier.go

146 lines
4.0 KiB
Go

package notification
import (
"context"
"crypto/tls"
"crypto/x509"
"fmt"
"net/mail"
"strings"
"github.com/sirupsen/logrus"
gomail "github.com/wneessen/go-mail"
"github.com/authelia/authelia/v4/internal/configuration/schema"
"github.com/authelia/authelia/v4/internal/logging"
"github.com/authelia/authelia/v4/internal/templates"
"github.com/authelia/authelia/v4/internal/utils"
)
// NewSMTPNotifier creates a SMTPNotifier using the notifier configuration.
func NewSMTPNotifier(config *schema.SMTPNotifierConfiguration, certPool *x509.CertPool) *SMTPNotifier {
opts := []gomail.Option{
gomail.WithPort(config.Port),
gomail.WithTLSConfig(utils.NewTLSConfig(config.TLS, certPool)),
gomail.WithHELO(config.Identifier),
}
switch {
case config.DisableStartTLS:
opts = append(opts, gomail.WithTLSPolicy(gomail.NoTLS))
case config.DisableRequireTLS:
opts = append(opts, gomail.WithTLSPolicy(gomail.TLSOpportunistic))
default:
opts = append(opts, gomail.WithTLSPolicy(gomail.TLSMandatory))
}
if config.Port == smtpPortSUBMISSIONS {
opts = append(opts, gomail.WithSSL())
}
var domain string
at := strings.LastIndex(config.Sender.Address, "@")
if at >= 0 {
domain = config.Sender.Address[at:]
}
return &SMTPNotifier{
config: config,
domain: domain,
tls: utils.NewTLSConfig(config.TLS, certPool),
log: logging.Logger(),
opts: opts,
}
}
// SMTPNotifier a notifier to send emails to SMTP servers.
type SMTPNotifier struct {
config *schema.SMTPNotifierConfiguration
domain string
tls *tls.Config
log *logrus.Logger
opts []gomail.Option
}
// StartupCheck implements model.StartupCheck to perform startup check operations.
func (n *SMTPNotifier) StartupCheck() (err error) {
var client *gomail.Client
if client, err = gomail.NewClient(n.config.Host, n.opts...); err != nil {
return fmt.Errorf("failed to establish client: %w", err)
}
ctx := context.Background()
if err = client.DialWithContext(ctx); err != nil {
return fmt.Errorf("failed to dial connection: %w", err)
}
if err = client.Close(); err != nil {
return fmt.Errorf("failed to close connection: %w", err)
}
return nil
}
// Send a notification via the SMTPNotifier.
func (n *SMTPNotifier) Send(ctx context.Context, recipient mail.Address, subject string, et *templates.EmailTemplate, data any) (err error) {
msg := gomail.NewMsg(
gomail.WithMIMEVersion(gomail.Mime10),
gomail.WithBoundary(utils.RandomString(30, utils.CharSetAlphaNumeric, true)),
)
if err = msg.From(n.config.Sender.String()); err != nil {
return fmt.Errorf("notifier: smtp: failed to set from address: %w", err)
}
if err = msg.AddTo(recipient.String()); err != nil {
return fmt.Errorf("notifier: smtp: failed to set to address: %w", err)
}
msg.Subject(strings.ReplaceAll(n.config.Subject, "{title}", subject))
switch {
case n.config.DisableHTMLEmails:
if err = msg.SetBodyTextTemplate(et.Text, data); err != nil {
return fmt.Errorf("notifier: smtp: failed to set body: text template errored: %w", err)
}
default:
if err = msg.AddAlternativeHTMLTemplate(et.HTML, data); err != nil {
return fmt.Errorf("notifier: smtp: failed to set body: html template errored: %w", err)
}
if err = msg.AddAlternativeTextTemplate(et.Text, data); err != nil {
return fmt.Errorf("notifier: smtp: failed to set body: text template errored: %w", err)
}
}
var client *gomail.Client
n.log.Debugf("creating client with %d options: %+v", len(n.opts), n.opts)
if client, err = gomail.NewClient(n.config.Host, n.opts...); err != nil {
return fmt.Errorf("notifier: smtp: failed to establish client: %w", err)
}
if auth := NewOpportunisticSMTPAuth(n.config); auth != nil {
client.SetSMTPAuthCustom(auth)
}
if err = client.DialWithContext(ctx); err != nil {
return fmt.Errorf("notifier: smtp: failed to dial connection: %w", err)
}
if err = client.Send(msg); err != nil {
return fmt.Errorf("notifier: smtp: failed to send message: %w", err)
}
if err = client.Close(); err != nil {
return fmt.Errorf("notifier: smtp: failed to close connection: %w", err)
}
return nil
}