pocketbase/tools/mailer/smtp.go

package mailer

import (
	"fmt"
	"io"
	"net/mail"
	"net/smtp"

	"github.com/domodwyer/mailyak/v3"
)

var _ Mailer = (*SmtpClient)(nil)

// NewSmtpClient creates new `SmtpClient` with the provided configuration.
func NewSmtpClient(
	host string,
	port int,
	username string,
	password string,
	tls bool,
) *SmtpClient {
	return &SmtpClient{
		host:     host,
		port:     port,
		username: username,
		password: password,
		tls:      tls,
	}
}

// SmtpClient defines a SMTP mail client structure that implements
// `mailer.Mailer` interface.
type SmtpClient struct {
	host     string
	port     int
	username string
	password string
	tls      bool
}

// Send implements `mailer.Mailer` interface.
func (m *SmtpClient) Send(
	fromEmail mail.Address,
	toEmail mail.Address,
	subject string,
	htmlContent string,
	attachments map[string]io.Reader,
) error {
	var smtpAuth smtp.Auth
	if m.username != "" || m.password != "" {
		smtpAuth = smtp.PlainAuth("", m.username, m.password, m.host)
	}

	// create mail instance
	var yak *mailyak.MailYak
	if m.tls {
		var tlsErr error
		yak, tlsErr = mailyak.NewWithTLS(fmt.Sprintf("%s:%d", m.host, m.port), smtpAuth, nil)
		if tlsErr != nil {
			return tlsErr
		}
	} else {
		yak = mailyak.New(fmt.Sprintf("%s:%d", m.host, m.port), smtpAuth)
	}

	if fromEmail.Name != "" {
		yak.FromName(fromEmail.Name)
	}
	yak.From(fromEmail.Address)
	yak.To(toEmail.Address)
	yak.Subject(subject)
	yak.HTML().Set(htmlContent)

	// try to generate a plain text version of the HTML
	if plain, err := html2Text(htmlContent); err == nil {
		yak.Plain().Set(plain)
	}

	for name, data := range attachments {
		yak.Attach(name, data)
	}

	return yak.Send()
}