jiacrontab/pkg/mailer/mail.go

package mailer

import (
	"crypto/tls"
	"errors"
	"fmt"
	"io"
	"net"
	"net/smtp"
	"os"
	"strings"
	"time"

	"github.com/iwannay/log"

	"gopkg.in/gomail.v2"
)

var (
	MailConfig *Mailer
	mailQueue  chan *Message
)

type Mailer struct {
	QueueLength       int
	SubjectPrefix     string
	Host              string
	From              string
	FromEmail         string
	User, Passwd      string
	DisableHelo       bool
	HeloHostname      string
	SkipVerify        bool
	UseCertificate    bool
	CertFile, KeyFile string
	UsePlainText      bool
	HookMode          bool
}

type Message struct {
	*gomail.Message
	Info        string
	confirmChan chan struct{}
}

func NewMessage(to []string, subject, htmlBody string) *Message {
	return NewMessageFrom(to, MailConfig.From, subject, htmlBody)
}

func NewMessageFrom(to []string, from, subject, htmlBody string) *Message {
	log.Printf("QueueLength (%d) NewMessage (htmlBody) \n%s\n", len(mailQueue), htmlBody)
	msg := gomail.NewMessage()
	msg.SetHeader("From", from)
	msg.SetHeader("To", to...)
	msg.SetHeader("Subject", subject)
	msg.SetDateHeader("Date", time.Now())
	contentType := "text/html"

	msg.SetBody(contentType, htmlBody)
	return &Message{
		Message:     msg,
		confirmChan: make(chan struct{}),
	}
}

type Sender struct {
}

func (s *Sender) Send(from string, to []string, msg io.WriterTo) error {
	host, port, err := net.SplitHostPort(MailConfig.Host)
	if err != nil {
		return err
	}

	tlsConfig := &tls.Config{
		InsecureSkipVerify: MailConfig.SkipVerify,
		ServerName:         host,
	}
	if MailConfig.UseCertificate {
		cert, err := tls.LoadX509KeyPair(MailConfig.CertFile, MailConfig.KeyFile)
		if err != nil {
			return err
		}
		tlsConfig.Certificates = []tls.Certificate{cert}
	}

	conn, err := net.DialTimeout("tcp", net.JoinHostPort(host, port), 3*time.Second)
	if err != nil {
		return err
	}
	defer conn.Close()
	isSecureConn := false
	if strings.HasSuffix(port, "465") {
		conn = tls.Client(conn, tlsConfig)
		isSecureConn = true
	}
	client, err := smtp.NewClient(conn, host)
	if err != nil {
		return fmt.Errorf("NewClient: %v", err)
	}

	if MailConfig.DisableHelo {
		hostname := MailConfig.HeloHostname
		if len(hostname) == 0 {
			hostname, err = os.Hostname()
			if err != nil {
				return err
			}
		}

		if err = client.Hello(hostname); err != nil {
			return fmt.Errorf("Hello:%v", err)
		}
	}

	hasStartTLS, _ := client.Extension("STARTTLS")
	if !isSecureConn && hasStartTLS {
		if err = client.StartTLS(tlsConfig); err != nil {
			return fmt.Errorf("StartTLS:%v", err)
		}
	}

	canAuth, options := client.Extension("AUTH")

	if canAuth && len(MailConfig.User) > 0 {
		var auth smtp.Auth
		if strings.Contains(options, "CRAM-MD5") {
			auth = smtp.CRAMMD5Auth(MailConfig.User, MailConfig.Passwd)
		} else if strings.Contains(options, "PLAIN") {
			auth = smtp.PlainAuth("", MailConfig.User, MailConfig.Passwd, host)
		} else if strings.Contains(options, "LOGIN") {
			// Patch for AUTH LOGIN
			auth = LoginAuth(MailConfig.User, MailConfig.Passwd)
		}
		if auth != nil {
			if err = client.Auth(auth); err != nil {
				return fmt.Errorf("Auth: %v", err)
			}
		}
	}

	if err = client.Mail(from); err != nil {
		return fmt.Errorf("Mail: %v", err)
	}
	for _, rec := range to {
		if err = client.Rcpt(rec); err != nil {
			return fmt.Errorf("Rcpt: %v", err)
		}
	}
	w, err := client.Data()
	if err != nil {
		return fmt.Errorf("Data: %v", err)
	} else if _, err = msg.WriteTo(w); err != nil {
		return fmt.Errorf("WriteTo: %v", err)
	} else if err = w.Close(); err != nil {
		return fmt.Errorf("Close: %v", err)
	}
	return client.Quit()
}

func processMailQueue() {
	sender := &Sender{}
	for {
		select {
		case msg := <-mailQueue:
			if err := gomail.Send(sender, msg.Message); err != nil {
				log.Errorf("Fail to send emails %s: %s - %v\n", msg.GetHeader("To"), msg.Info, err)
			} else {
				log.Infof("E-mails sent %s: %s\n", msg.GetHeader("To"), msg.Info)
			}
			msg.confirmChan <- struct{}{}
		}
	}
}

func InitMailer(m *Mailer) {
	MailConfig = m
	if MailConfig == nil || mailQueue != nil {
		return
	}

	mailQueue = make(chan *Message, MailConfig.QueueLength)
	go processMailQueue()
}

func Send(msg *Message) {
	mailQueue <- msg
	if MailConfig.HookMode {
		<-msg.confirmChan
		return
	}
	go func() {
		<-msg.confirmChan
	}()
}

func SendMail(to []string, subject, content string) error {
	if MailConfig == nil {
		return errors.New("update mail config must restart service")
	}
	msg := NewMessage(to, subject, content)
	Send(msg)
	return nil
}