|
|
package email
|
|
|
|
|
|
import (
|
|
|
"github.com/HFO4/cloudreve/pkg/util"
|
|
|
"github.com/go-mail/mail"
|
|
|
"time"
|
|
|
)
|
|
|
|
|
|
// SMTP SMTP协议发送邮件
|
|
|
type SMTP struct {
|
|
|
Config SMTPConfig
|
|
|
ch chan *mail.Message
|
|
|
chOpen bool
|
|
|
}
|
|
|
|
|
|
// SMTPConfig SMTP发送配置
|
|
|
type SMTPConfig struct {
|
|
|
Name string // 发送者名
|
|
|
Address string // 发送者地址
|
|
|
ReplyTo string // 回复地址
|
|
|
Host string // 服务器主机名
|
|
|
Port int // 服务器端口
|
|
|
User string // 用户名
|
|
|
Password string // 密码
|
|
|
Encryption string // 是否启用加密
|
|
|
Keepalive int // SMTP 连接保留时长
|
|
|
}
|
|
|
|
|
|
// NewSMTPClient 新建SMTP发送队列
|
|
|
func NewSMTPClient(config SMTPConfig) *SMTP {
|
|
|
client := &SMTP{
|
|
|
Config: config,
|
|
|
ch: make(chan *mail.Message, 30),
|
|
|
chOpen: false,
|
|
|
}
|
|
|
|
|
|
client.Init()
|
|
|
|
|
|
return client
|
|
|
}
|
|
|
|
|
|
// Send 发送邮件
|
|
|
func (client *SMTP) Send(to, title, body string) error {
|
|
|
if !client.chOpen {
|
|
|
return ErrChanNotOpen
|
|
|
}
|
|
|
m := mail.NewMessage()
|
|
|
m.SetHeader("From", client.Config.Address)
|
|
|
m.SetHeader("To", to)
|
|
|
m.SetHeader("Subject", title)
|
|
|
m.SetBody("text/html", body)
|
|
|
client.ch <- m
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
// Close 关闭发送队列
|
|
|
func (client *SMTP) Close() {
|
|
|
if client.ch != nil {
|
|
|
close(client.ch)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// Init 初始化发送队列
|
|
|
func (client *SMTP) Init() {
|
|
|
go func() {
|
|
|
defer func() {
|
|
|
if err := recover(); err != nil {
|
|
|
client.chOpen = false
|
|
|
util.Log().Error("邮件发送队列出现异常, %s ,30 秒后重新连接", err)
|
|
|
time.Sleep(time.Duration(30) * time.Second)
|
|
|
client.Init()
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
d := mail.NewDialer(client.Config.Host, client.Config.Port, client.Config.User, client.Config.Password)
|
|
|
d.Timeout = time.Duration(client.Config.Keepalive+5) * time.Second
|
|
|
client.chOpen = true
|
|
|
|
|
|
var s mail.SendCloser
|
|
|
var err error
|
|
|
open := false
|
|
|
for {
|
|
|
select {
|
|
|
case m, ok := <-client.ch:
|
|
|
if !ok {
|
|
|
client.chOpen = false
|
|
|
return
|
|
|
}
|
|
|
if !open {
|
|
|
if s, err = d.Dial(); err != nil {
|
|
|
panic(err)
|
|
|
}
|
|
|
open = true
|
|
|
}
|
|
|
if err := mail.Send(s, m); err != nil {
|
|
|
util.Log().Warning("邮件发送失败, %s", err)
|
|
|
} else {
|
|
|
util.Log().Debug("邮件已发送")
|
|
|
}
|
|
|
// 长时间没有新邮件，则关闭SMTP连接
|
|
|
case <-time.After(time.Duration(client.Config.Keepalive) * time.Second):
|
|
|
if open {
|
|
|
if err := s.Close(); err != nil {
|
|
|
util.Log().Warning("无法关闭 SMTP 连接 %s", err)
|
|
|
}
|
|
|
open = false
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}()
|
|
|
}
|
