forked from evergreen-ci/evergreen
/
mailer.go
116 lines (101 loc) · 2.58 KB
/
mailer.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
/**
* mailer.go
*
* Created on: October 23 2013
* Author: Valeri Karpov <valeri.karpov@mongodb.com>
*
* Defines a functional abstraction for sending an email and a few concrete
* functions matching that abstraction.
*
*/
package notify
import (
"bytes"
"crypto/tls"
"encoding/base64"
"fmt"
"net/mail"
"net/smtp"
"strings"
"github.com/10gen-labs/slogger/v1"
"github.com/evergreen-ci/evergreen"
)
type Mailer interface {
SendMail([]string, string, string) error
}
type SmtpMailer struct {
From string
Server string
Port int
UseSSL bool
Username string
Password string
}
/* Connects an SMTP server (usually localhost:25 in prod) and uses that to
send an email with the body encoded in base64. */
func (self SmtpMailer) SendMail(recipients []string, subject, body string) error {
// 'recipients' is expected a comma-separated list of emails in either of
// these formats:
// - bob@example.com
// - Bob Smith <bob@example.com>
var c *smtp.Client
var err error
if self.UseSSL {
tlsCon, err := tls.Dial("tcp", fmt.Sprintf("%v:%v", self.Server, self.Port), &tls.Config{})
if err != nil {
return err
}
c, err = smtp.NewClient(tlsCon, self.Server)
} else {
c, err = smtp.Dial(fmt.Sprintf("%v:%v", self.Server, self.Port))
}
if err != nil {
return err
}
if self.Username != "" {
err = c.Auth(smtp.PlainAuth("", self.Username, self.Password, self.Server))
if err != nil {
return err
}
}
// Set the sender
from := mail.Address{"Evergreen Alerts", self.From}
err = c.Mail(self.From)
if err != nil {
evergreen.Logger.Errorf(slogger.ERROR, "Error establishing mail sender (%v): %v", self.From, err)
return err
}
// Set the recipient
for _, recipient := range recipients {
err = c.Rcpt(recipient)
if err != nil {
evergreen.Logger.Errorf(slogger.ERROR, "Error establishing mail recipient (%v): %v", recipient, err)
return err
}
}
// Send the email body.
wc, err := c.Data()
if err != nil {
return err
}
defer wc.Close()
// set header information
header := make(map[string]string)
header["From"] = from.String()
header["To"] = strings.Join(recipients, ", ")
header["Subject"] = encodeRFC2047(subject)
header["MIME-Version"] = "1.0"
header["Content-Type"] = "text/html; charset=\"utf-8\""
header["Content-Transfer-Encoding"] = "base64"
message := ""
for k, v := range header {
message += fmt.Sprintf("%s: %s\r\n", k, v)
}
message += "\r\n" + base64.StdEncoding.EncodeToString([]byte(body))
// write the body
buf := bytes.NewBufferString(message)
if _, err = buf.WriteTo(wc); err != nil {
return err
}
return nil
}