-
Notifications
You must be signed in to change notification settings - Fork 2
/
client.go
101 lines (88 loc) · 2.47 KB
/
client.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
package smtp
import (
"crypto/tls"
"net/smtp"
"strconv"
)
type Client interface {
initClient() error
closeClient() error
SendContent(content Content) error
}
type tlsClient struct {
secret string
server string
port int
sender string
signature string
enableTLS bool
client *smtp.Client
}
func (c *tlsClient) initClient() error {
if c.client != nil {
return nil
}
serverAddr := c.server + ":" + strconv.Itoa(c.port)
if conn, dialErr := tls.Dial("tcp", serverAddr, &tls.Config{
ServerName: c.server,
InsecureSkipVerify: c.enableTLS, // nolint:gosec
MinVersion: tls.VersionTLS12,
}); dialErr != nil {
return NewDialSmtpServerError(dialErr)
} else if client, dialClientErr := smtp.NewClient(conn, c.server); dialClientErr != nil {
return NewDialSmtpServerError(dialClientErr)
} else if authErr := client.Auth(smtp.PlainAuth("", c.sender, c.secret, c.server)); authErr != nil {
return NewDialSmtpServerError(authErr)
} else {
c.client = client
return nil
}
}
func (c *tlsClient) closeClient() error {
if c.client == nil {
return nil
}
if closeErr := c.client.Quit(); closeErr != nil {
return NewDialSmtpServerError(closeErr)
}
c.client = nil
return nil
}
func (c *tlsClient) SendContent(content Content) error {
// 检查是否已经连接
if c.client == nil {
if initClientErr := c.initClient(); initClientErr != nil {
return initClientErr
}
}
// 发送邮件
content.setSender(c.signature)
payload := content.ExportToMailText()
if mailToErr := c.client.Mail(c.sender); mailToErr != nil {
return NewInitMailContentError(payload, mailToErr)
} else if rcptErr := c.client.Rcpt(content.getReceiver()); rcptErr != nil {
return NewInitMailContentError(payload, rcptErr)
} else if writer, initWriterErr := c.client.Data(); initWriterErr != nil {
return NewInitMailContentError(payload, initWriterErr)
} else if _, writeContentErr := writer.Write(payload); writeContentErr != nil {
return NewInitMailContentError(payload, writeContentErr)
} else if closeErr := writer.Close(); closeErr != nil {
return NewDialSmtpServerError(closeErr)
} else {
return nil
}
}
func NewTLSClient(cfg Config) (client Client, err error) {
client = &tlsClient{
secret: cfg.Secret,
server: cfg.Server,
port: cfg.Port,
sender: cfg.Sender,
signature: cfg.Signature,
enableTLS: cfg.EnableTLS,
}
if initClientErr := client.initClient(); initClientErr != nil {
return nil, initClientErr
}
return client, nil
}