-
Notifications
You must be signed in to change notification settings - Fork 5
/
impl.go
98 lines (91 loc) · 2.4 KB
/
impl.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
package configuredemail
import (
"fmt"
"net"
"net/smtp"
"os"
"github.com/Cloud-Foundations/golib/pkg/awsutil/metadata"
"github.com/Cloud-Foundations/golib/pkg/awsutil/secretsmgr"
"github.com/Cloud-Foundations/golib/pkg/log"
)
type emailManager struct {
awsSecret *secretsmgr.CachedSecret
logger log.DebugLogger
password string
smtpAddress string
smtpServer string
username string
}
func newEmailSender(config EmailConfig,
logger log.DebugLogger) (*emailManager, error) {
m := &emailManager{logger: logger}
if host, _, err := net.SplitHostPort(config.SmtpServer); err != nil {
m.smtpAddress = config.SmtpServer + ":25"
m.smtpServer = config.SmtpServer
} else {
m.smtpAddress = config.SmtpServer
m.smtpServer = host
}
if config.AwsSecretId != "" {
metadataClient, err := metadata.GetMetadataClient()
if err != nil {
return nil, err
}
m.awsSecret, err = secretsmgr.NewCachedSecret(metadataClient,
config.AwsSecretId, config.AwsSecretLifetime, logger)
if err != nil {
return nil, err
}
}
if config.PasswordVariable != "" {
m.password = os.Getenv(config.PasswordVariable)
}
if config.UsernameVariable != "" {
m.username = os.Getenv(config.UsernameVariable)
}
return m, nil
}
func (m *emailManager) SendMail(from string, to []string, msg []byte) error {
var username, password string
if m.awsSecret != nil {
var err error
username, password, err = m.getLoginFromAws()
if err != nil {
return err
}
} else if m.username != "" && m.password != "" {
username = username
password = password
}
var auth smtp.Auth
if username != "" && password != "" {
auth = smtp.PlainAuth("", username, password, m.smtpServer)
}
return m.sendMailWithAuth(auth, from, to, msg)
}
func (m *emailManager) getLoginFromAws() (string, string, error) {
secrets, err := m.awsSecret.GetSecret()
if err != nil {
return "", "", err
}
username, ok := secrets["Username"]
if !ok {
return "", "",
fmt.Errorf("no Username in AWS Secret: %s", m.awsSecret)
}
password, ok := secrets["Password"]
if !ok {
return "", "",
fmt.Errorf("no Password in AWS Secret: %s", m.awsSecret)
}
return username, password, nil
}
func (m *emailManager) sendMailWithAuth(auth smtp.Auth, from string,
to []string, msg []byte) error {
err := smtp.SendMail(m.smtpAddress, auth, from, to, msg)
if err != nil {
return err
}
m.logger.Debugf(0, "sent email from: %s to: %v\n", from, to)
return nil
}