crawlab/core/notification/mail.go

package notification

import (
	"errors"
	"github.com/PuerkitoBio/goquery"
	"github.com/crawlab-team/crawlab/core/models/models"
	"gopkg.in/gomail.v2"
	"net/mail"
	"regexp"
	"strings"
)

func SendMail(s *models.NotificationSetting, ch *models.NotificationChannel, to, cc, bcc []string, title, content string) error {
	// sender identity
	senderIdentity := ch.SMTPUsername

	// sender email
	senderEmail := ch.SMTPUsername
	if s != nil && s.UseCustomSenderEmail {
		if s.SenderName != "" {
			senderIdentity = s.SenderName
		}
		senderEmail = s.SenderEmail
	}

	// config
	smtpConfig := smtpAuthentication{
		Server:         ch.SMTPServer,
		Port:           ch.SMTPPort,
		SenderIdentity: senderIdentity,
		SenderEmail:    senderEmail,
		SMTPUser:       ch.SMTPUsername,
		SMTPPassword:   ch.SMTPPassword,
	}

	options := sendOptions{
		Subject: title,
		To:      to,
		Cc:      cc,
		Bcc:     bcc,
	}

	// convert html to text
	text := content
	if isHtml(text) {
		text = convertHtmlToText(text)
	}

	// apply theme
	if isHtml(content) {
		content = GetTheme() + content
	}

	switch ch.Provider {
	case ChannelMailProviderGmail:
		return sendMailGmail(ch, smtpConfig, options, content, text)
	default:
		return sendMail(smtpConfig, options, content, text)
	}
}

func isHtml(content string) bool {
	regex := regexp.MustCompile(`(?i)<\s*(html|head|body|div|span|p|a|img|table|tr|td|th|tbody|thead|tfoot|ul|ol|li|dl|dt|dd|form|input|textarea|button|select|option|optgroup|fieldset|legend|label|iframe|embed|object|param|video|audio|source|canvas|svg|math|style|link|script|meta|base|title|br|hr|b|strong|i|em|u|s|strike|del|ins|mark|small|sub|sup|big|pre|code|q|blockquote|abbr|address|bdo|cite|dfn|kbd|var|samp|ruby|rt|rp|time|progress|meter|output|area|map)`)
	return regex.MatchString(content)
}

func convertHtmlToText(content string) string {
	doc, err := goquery.NewDocumentFromReader(strings.NewReader(content))
	if err != nil {
		logger.Errorf("failed to convert html to text: %v", err)
		return ""
	}
	return doc.Text()
}

type smtpAuthentication struct {
	Server         string
	Port           int
	SenderEmail    string
	SenderIdentity string
	SMTPUser       string
	SMTPPassword   string
}

// sendOptions are options for sending an email
type sendOptions struct {
	Subject string
	To      []string
	Cc      []string
	Bcc     []string
}

func getMailMessage(smtpConfig smtpAuthentication, options sendOptions, htmlBody string, txtBody string) (m *gomail.Message, err error) {
	if len(options.To) == 0 {
		return nil, errors.New("no receiver emails configured")
	}

	// from
	from := mail.Address{
		Name:    smtpConfig.SenderIdentity,
		Address: smtpConfig.SenderEmail,
	}

	// message
	m = gomail.NewMessage()
	m.SetHeader("From", from.String())
	m.SetHeader("To", options.To...)
	m.SetHeader("Subject", options.Subject)
	if len(options.Cc) > 0 {
		m.SetHeader("Cc", options.Cc...)
	}
	if len(options.Bcc) > 0 {
		m.SetHeader("Bcc", options.Bcc...)
	}
	m.SetBody("text/plain", txtBody)
	m.AddAlternative("text/html", htmlBody)

	return m, nil
}

// send email
func sendMail(smtpConfig smtpAuthentication, options sendOptions, htmlBody string, txtBody string) error {
	if smtpConfig.Server == "" {
		return errors.New("SMTP server config is empty")
	}

	if smtpConfig.Port == 0 {
		return errors.New("SMTP port config is empty")
	}

	if smtpConfig.SMTPUser == "" {
		return errors.New("SMTP user is empty")
	}

	m, err := getMailMessage(smtpConfig, options, htmlBody, txtBody)
	if err != nil {
		return err
	}

	// dialer
	d := gomail.NewDialer(smtpConfig.Server, smtpConfig.Port, smtpConfig.SMTPUser, smtpConfig.SMTPPassword)

	return d.DialAndSend(m)
}