/
sendmail.go
115 lines (103 loc) · 2.48 KB
/
sendmail.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
package util
import (
"fmt"
"github.com/chanyipiaomiao/hltool"
"github.com/levigross/grequests"
"github.com/tidwall/gjson"
"strings"
)
type SelfHostMail struct {
Username string
Password string
Port int
Host string
}
type SendCloudMail struct {
APIUser string
APIKey string
APIUrl string
Template string
}
// EmailClient 发送客户端
type EmailClient struct {
Type string
Message *Message
SelfHostMail
SendCloudMail
}
func newEmailClient(subject, content string) *EmailClient {
message := &Message{
Subject: subject,
Content: content,
ContentType: "text/html",
To: []string{to},
SelfHostFrom: selfHostFrom,
SendCloudFrom: sendCloudFrom,
FromName: fromName,
}
client := &EmailClient{Message: message, Type: emailType}
if emailType == "sendcloud" {
client.APIUser = apiUser
client.APIKey = apiKey
client.APIUrl = apiUrl
} else {
client.Host = host
client.Username = username
client.Port = port
client.Password = password
}
return client
}
type SendMail struct {
Client *EmailClient
}
func NewSendMail(subject, content string) *SendMail {
return &SendMail{
Client: newEmailClient(subject, content),
}
}
func (s *SendMail) SendMessage() error {
var err error
switch s.Client.Type {
case "sendcloud":
err = bySendCloudMail(s.Client)
case "selfhost":
err = bySelfHostMail(s.Client)
default:
err = fmt.Errorf("no support %s", s.Client.Type)
}
return err
}
func bySendCloudMail(email *EmailClient) error {
o := &grequests.RequestOptions{
Data: map[string]string{
"apiUser": email.APIUser,
"apiKey": email.APIKey,
"from": email.Message.SendCloudFrom,
"fromName": email.Message.FromName,
"to": strings.Join(email.Message.To, ";"),
"subject": email.Message.Subject,
"html": email.Message.Content,
},
}
resp, err := grequests.Post(email.APIUrl, o)
if err != nil {
return err
}
respS := resp.String()
r := gjson.Get(respS, "statusCode")
if r.String() != "200" {
return fmt.Errorf("call SendCloud error: %s", gjson.Get(respS, "message").String())
}
return nil
}
func bySelfHostMail(email *EmailClient) error {
message := hltool.NewEmailMessage(email.Message.SelfHostFrom, email.Message.Subject, email.Message.ContentType,
email.Message.Content, "", email.Message.To, []string{})
client := hltool.NewEmailClient(email.Host, email.Username, email.Password, email.Port, message)
_, err := hltool.SendMessage(client)
if err != nil {
return err
}
return nil
}