474 lines
13 KiB
Go
474 lines
13 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"github.com/leanote/leanote/app/info"
|
|||
|
|
"github.com/leanote/leanote/app/db"
|
|||
|
|
. "github.com/leanote/leanote/app/lea"
|
|||
|
|
"gopkg.in/mgo.v2/bson"
|
|||
|
|
"time"
|
|||
|
|
"strings"
|
|||
|
|
"net/smtp"
|
|||
|
|
"strconv"
|
|||
|
|
"fmt"
|
|||
|
|
"html/template"
|
|||
|
|
"bytes"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 发送邮件
|
|||
|
|
|
|||
|
|
type EmailService struct {
|
|||
|
|
tpls map[string]*template.Template
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func NewEmailService() (*EmailService) {
|
|||
|
|
return &EmailService{tpls: map[string]*template.Template{}}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送邮件
|
|||
|
|
var host = ""
|
|||
|
|
var emailPort = ""
|
|||
|
|
var username = ""
|
|||
|
|
var password = ""
|
|||
|
|
|
|||
|
|
func InitEmailFromDb() {
|
|||
|
|
host = configService.GetGlobalStringConfig("emailHost")
|
|||
|
|
emailPort = configService.GetGlobalStringConfig("emailPort")
|
|||
|
|
username = configService.GetGlobalStringConfig("emailUsername")
|
|||
|
|
password = configService.GetGlobalStringConfig("emailPassword")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *EmailService) SendEmail(to, subject, body string) (ok bool, e string) {
|
|||
|
|
InitEmailFromDb()
|
|||
|
|
|
|||
|
|
if host == "" || emailPort == "" || username == "" || password == "" {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
hp := strings.Split(host, ":")
|
|||
|
|
auth := smtp.PlainAuth("", username, password, hp[0])
|
|||
|
|
|
|||
|
|
var content_type string
|
|||
|
|
|
|||
|
|
mailtype := "html"
|
|||
|
|
if mailtype == "html" {
|
|||
|
|
content_type = "Content-Type: text/"+ mailtype + "; charset=UTF-8"
|
|||
|
|
} else{
|
|||
|
|
content_type = "Content-Type: text/plain" + "; charset=UTF-8"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
msg := []byte("To: " + to + "\r\nFrom: " + username + "<"+ username +">\r\nSubject: " + subject + "\r\n" + content_type + "\r\n\r\n" + body)
|
|||
|
|
send_to := strings.Split(to, ";")
|
|||
|
|
err := smtp.SendMail(host+":"+emailPort, auth, username, send_to, msg)
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
e = fmt.Sprint(err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
ok = true
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AddUser调用
|
|||
|
|
// 可以使用一个goroutine
|
|||
|
|
func (this *EmailService) RegisterSendActiveEmail(userInfo info.User, email string) bool {
|
|||
|
|
token := tokenService.NewToken(userInfo.UserId.Hex(), email, info.TokenActiveEmail)
|
|||
|
|
if token == "" {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
subject := configService.GetGlobalStringConfig("emailTemplateRegisterSubject");
|
|||
|
|
tpl := configService.GetGlobalStringConfig("emailTemplateRegister");
|
|||
|
|
|
|||
|
|
if(tpl == "") {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
tokenUrl := siteUrl + "/user/activeEmail?token=" + token
|
|||
|
|
// {siteUrl} {tokenUrl} {token} {tokenTimeout} {user.id} {user.email} {user.username}
|
|||
|
|
token2Value := map[string]interface{}{"siteUrl": siteUrl, "tokenUrl": tokenUrl, "token": token, "tokenTimeout": strconv.Itoa(int(tokenService.GetOverHours(info.TokenActiveEmail))),
|
|||
|
|
"user": map[string]interface{}{
|
|||
|
|
"userId": userInfo.UserId.Hex(),
|
|||
|
|
"email": userInfo.Email,
|
|||
|
|
"username": userInfo.Username,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var ok bool
|
|||
|
|
ok, _, subject, tpl = this.renderEmail(subject, tpl, token2Value)
|
|||
|
|
if !ok {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送邮件
|
|||
|
|
ok, _ = this.SendEmail(email, subject, tpl)
|
|||
|
|
return ok
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 修改邮箱
|
|||
|
|
func (this *EmailService) UpdateEmailSendActiveEmail(userInfo info.User, email string) (ok bool, msg string) {
|
|||
|
|
// 先验证该email是否被注册了
|
|||
|
|
if userService.IsExistsUser(email) {
|
|||
|
|
ok = false
|
|||
|
|
msg = "该邮箱已注册"
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
token := tokenService.NewToken(userInfo.UserId.Hex(), email, info.TokenUpdateEmail)
|
|||
|
|
|
|||
|
|
if token == "" {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
subject := configService.GetGlobalStringConfig("emailTemplateUpdateEmailSubject");
|
|||
|
|
tpl := configService.GetGlobalStringConfig("emailTemplateUpdateEmail");
|
|||
|
|
|
|||
|
|
// 发送邮件
|
|||
|
|
tokenUrl := siteUrl + "/user/updateEmail?token=" + token
|
|||
|
|
// {siteUrl} {tokenUrl} {token} {tokenTimeout} {user.userId} {user.email} {user.username}
|
|||
|
|
token2Value := map[string]interface{}{"siteUrl": siteUrl, "tokenUrl": tokenUrl, "token": token, "tokenTimeout": strconv.Itoa(int(tokenService.GetOverHours(info.TokenActiveEmail))),
|
|||
|
|
"newEmail": email,
|
|||
|
|
"user": map[string]interface{}{
|
|||
|
|
"userId": userInfo.UserId.Hex(),
|
|||
|
|
"email": userInfo.Email,
|
|||
|
|
"username": userInfo.Username,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ok, msg, subject, tpl = this.renderEmail(subject, tpl, token2Value)
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送邮件
|
|||
|
|
ok, msg = this.SendEmail(email, subject, tpl)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *EmailService) FindPwdSendEmail(token, email string) (ok bool, msg string) {
|
|||
|
|
subject := configService.GetGlobalStringConfig("emailTemplateFindPasswordSubject");
|
|||
|
|
tpl := configService.GetGlobalStringConfig("emailTemplateFindPassword");
|
|||
|
|
|
|||
|
|
// 发送邮件
|
|||
|
|
tokenUrl := siteUrl + "/findPassword/" + token
|
|||
|
|
// {siteUrl} {tokenUrl} {token} {tokenTimeout} {user.id} {user.email} {user.username}
|
|||
|
|
token2Value := map[string]interface{}{"siteUrl": siteUrl, "tokenUrl": tokenUrl,
|
|||
|
|
"token": token, "tokenTimeout": strconv.Itoa(int(tokenService.GetOverHours(info.TokenActiveEmail)))}
|
|||
|
|
|
|||
|
|
ok, msg, subject, tpl = this.renderEmail(subject, tpl, token2Value)
|
|||
|
|
if !ok {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 发送邮件
|
|||
|
|
ok, msg = this.SendEmail(email, subject, tpl)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送邀请链接
|
|||
|
|
func (this *EmailService) SendInviteEmail(userInfo info.User, email, content string) bool {
|
|||
|
|
subject := configService.GetGlobalStringConfig("emailTemplateInviteSubject");
|
|||
|
|
tpl := configService.GetGlobalStringConfig("emailTemplateInvite");
|
|||
|
|
|
|||
|
|
token2Value := map[string]interface{}{"siteUrl": siteUrl,
|
|||
|
|
"registerUrl": siteUrl + "/register?from=" + userInfo.Username,
|
|||
|
|
"content": content,
|
|||
|
|
"user": map[string]interface{}{
|
|||
|
|
"username": userInfo.Username,
|
|||
|
|
"email": userInfo.Email,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
var ok bool
|
|||
|
|
ok, _, subject, tpl = this.renderEmail(subject, tpl, token2Value)
|
|||
|
|
if !ok {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
// 发送邮件
|
|||
|
|
ok, _ = this.SendEmail(email, subject, tpl)
|
|||
|
|
return ok
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送评论
|
|||
|
|
func (this *EmailService) SendCommentEmail(note info.Note, comment info.BlogComment, userId, content string) bool {
|
|||
|
|
subject := configService.GetGlobalStringConfig("emailTemplateCommentSubject");
|
|||
|
|
tpl := configService.GetGlobalStringConfig("emailTemplateComment");
|
|||
|
|
|
|||
|
|
// title := "评论提醒"
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
toUserId := note.UserId.Hex()
|
|||
|
|
// title := "评论提醒"
|
|||
|
|
|
|||
|
|
// 表示回复回复的内容, 那么发送给之前回复的
|
|||
|
|
if comment.CommentId != "" {
|
|||
|
|
toUserId = comment.UserId.Hex()
|
|||
|
|
}
|
|||
|
|
toUserInfo := userService.GetUserInfo(toUserId)
|
|||
|
|
sendUserInfo := userService.GetUserInfo(userId)
|
|||
|
|
|
|||
|
|
subject := note.Title + " 收到 " + sendUserInfo.Username + " 的评论";
|
|||
|
|
if comment.CommentId != "" {
|
|||
|
|
subject = "您在 " + note.Title + " 发表的评论收到 " + sendUserInfo.Username;
|
|||
|
|
if userId == note.UserId.Hex() {
|
|||
|
|
subject += "(作者)";
|
|||
|
|
}
|
|||
|
|
subject += " 的评论";
|
|||
|
|
}
|
|||
|
|
*/
|
|||
|
|
|
|||
|
|
toUserId := note.UserId.Hex()
|
|||
|
|
// 表示回复回复的内容, 那么发送给之前回复的
|
|||
|
|
if comment.CommentId != "" {
|
|||
|
|
toUserId = comment.UserId.Hex()
|
|||
|
|
}
|
|||
|
|
toUserInfo := userService.GetUserInfo(toUserId) // 被评论者
|
|||
|
|
sendUserInfo := userService.GetUserInfo(userId) // 评论者
|
|||
|
|
|
|||
|
|
// {siteUrl} {blogUrl}
|
|||
|
|
// {blog.id} {blog.title} {blog.url}
|
|||
|
|
// {commentUser.userId} {commentUser.username} {commentUser.email}
|
|||
|
|
// {commentedUser.userId} {commentedUser.username} {commentedUser.email}
|
|||
|
|
token2Value := map[string]interface{}{"siteUrl": siteUrl, "blogUrl": configService.GetBlogUrl(),
|
|||
|
|
"blog": map[string]string{
|
|||
|
|
"id": note.NoteId.Hex(),
|
|||
|
|
"title": note.Title,
|
|||
|
|
"url": configService.GetBlogUrl() + "/view/" + note.NoteId.Hex(),
|
|||
|
|
},
|
|||
|
|
"commentContent": content,
|
|||
|
|
// 评论者信息
|
|||
|
|
"commentUser": map[string]interface{}{"userId": sendUserInfo.UserId.Hex(),
|
|||
|
|
"username": sendUserInfo.Username,
|
|||
|
|
"email": sendUserInfo.Email,
|
|||
|
|
"isBlogAuthor": userId == note.UserId.Hex(),
|
|||
|
|
},
|
|||
|
|
// 被评论者信息
|
|||
|
|
"commentedUser": map[string]interface{}{"userId": toUserId,
|
|||
|
|
"username": toUserInfo.Username,
|
|||
|
|
"email": toUserInfo.Email,
|
|||
|
|
"isBlogAuthor": toUserId == note.UserId.Hex(),
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
ok := false
|
|||
|
|
ok, _, subject, tpl = this.renderEmail(subject, tpl, token2Value)
|
|||
|
|
if !ok {
|
|||
|
|
return false
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送邮件
|
|||
|
|
ok, _ = this.SendEmail(toUserInfo.Email, subject, tpl)
|
|||
|
|
return ok
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
|
|||
|
|
// 验证模板是否正确
|
|||
|
|
func (this *EmailService) ValidTpl(str string) (ok bool, msg string){
|
|||
|
|
defer func() {
|
|||
|
|
if err := recover(); err != nil {
|
|||
|
|
ok = false
|
|||
|
|
msg = fmt.Sprint(err)
|
|||
|
|
}
|
|||
|
|
}();
|
|||
|
|
header := configService.GetGlobalStringConfig("emailTemplateHeader");
|
|||
|
|
footer := configService.GetGlobalStringConfig("emailTemplateFooter");
|
|||
|
|
str = strings.Replace(str, "{{header}}", header, -1)
|
|||
|
|
str = strings.Replace(str, "{{footer}}", footer, -1)
|
|||
|
|
_, err := template.New("tpl name").Parse(str)
|
|||
|
|
if err != nil {
|
|||
|
|
msg = fmt.Sprint(err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
ok = true
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// ok, msg, subject, tpl
|
|||
|
|
func (this *EmailService) getTpl(str string) (ok bool, msg string, tpl *template.Template){
|
|||
|
|
defer func() {
|
|||
|
|
if err := recover(); err != nil {
|
|||
|
|
ok = false
|
|||
|
|
msg = fmt.Sprint(err)
|
|||
|
|
}
|
|||
|
|
}();
|
|||
|
|
|
|||
|
|
var err error
|
|||
|
|
var has bool
|
|||
|
|
|
|||
|
|
if tpl, has = this.tpls[str]; !has {
|
|||
|
|
tpl, err = template.New("tpl name").Parse(str)
|
|||
|
|
if err != nil {
|
|||
|
|
msg = fmt.Sprint(err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
this.tpls[str] = tpl
|
|||
|
|
}
|
|||
|
|
ok = true
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 通过subject, body和值得到内容
|
|||
|
|
func (this *EmailService) renderEmail(subject, body string, values map[string]interface{}) (ok bool, msg string, o string, b string) {
|
|||
|
|
ok = false
|
|||
|
|
msg = ""
|
|||
|
|
defer func() { // 必须要先声明defer，否则不能捕获到panic异常
|
|||
|
|
if err := recover(); err != nil {
|
|||
|
|
ok = false
|
|||
|
|
msg = fmt.Sprint(err) // 这里的err其实就是panic传入的内容，
|
|||
|
|
}
|
|||
|
|
}();
|
|||
|
|
|
|||
|
|
var tpl *template.Template
|
|||
|
|
|
|||
|
|
values["siteUrl"] = siteUrl;
|
|||
|
|
|
|||
|
|
// subject
|
|||
|
|
if subject != "" {
|
|||
|
|
ok, msg, tpl = this.getTpl(subject)
|
|||
|
|
if(!ok) {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
var buffer bytes.Buffer
|
|||
|
|
err := tpl.Execute(&buffer, values)
|
|||
|
|
if err != nil {
|
|||
|
|
msg = fmt.Sprint(err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
o = buffer.String()
|
|||
|
|
} else {
|
|||
|
|
o = ""
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// content
|
|||
|
|
header := configService.GetGlobalStringConfig("emailTemplateHeader");
|
|||
|
|
footer := configService.GetGlobalStringConfig("emailTemplateFooter");
|
|||
|
|
body = strings.Replace(body, "{{header}}", header, -1)
|
|||
|
|
body = strings.Replace(body, "{{footer}}", footer, -1)
|
|||
|
|
values["subject"] = o
|
|||
|
|
ok, msg, tpl = this.getTpl(body)
|
|||
|
|
if(!ok) {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
var buffer2 bytes.Buffer
|
|||
|
|
err := tpl.Execute(&buffer2, values)
|
|||
|
|
if err != nil {
|
|||
|
|
msg = fmt.Sprint(err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
b = buffer2.String()
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 发送email给用户
|
|||
|
|
// 需要记录
|
|||
|
|
func (this *EmailService) SendEmailToUsers(users []info.User, subject, body string) (ok bool, msg string) {
|
|||
|
|
if(users == nil || len(users) == 0) {
|
|||
|
|
msg = "no users"
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 尝试renderHtml
|
|||
|
|
ok, msg, _, _ = this.renderEmail(subject, body, map[string]interface{}{})
|
|||
|
|
if(!ok) {
|
|||
|
|
Log(msg)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
go func() {
|
|||
|
|
for _, user := range users {
|
|||
|
|
LogJ(user)
|
|||
|
|
m := map[string]interface{}{}
|
|||
|
|
m["userId"] = user.UserId.Hex()
|
|||
|
|
m["username"] = user.Username
|
|||
|
|
m["email"] = user.Email
|
|||
|
|
ok2, msg2, subject2, body2 := this.renderEmail(subject, body, m)
|
|||
|
|
ok = ok2
|
|||
|
|
msg = msg2
|
|||
|
|
if(ok2) {
|
|||
|
|
sendOk, msg := this.SendEmail(user.Email, subject2, body2);
|
|||
|
|
this.AddEmailLog(user.Email, subject, body, sendOk, msg) // 把模板记录下
|
|||
|
|
// 记录到Email Log
|
|||
|
|
if sendOk {
|
|||
|
|
// Log("ok " + user.Email)
|
|||
|
|
} else {
|
|||
|
|
// Log("no " + user.Email)
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
// Log(msg);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (this *EmailService) SendEmailToEmails(emails []string, subject, body string) (ok bool, msg string) {
|
|||
|
|
if(emails == nil || len(emails) == 0) {
|
|||
|
|
msg = "no emails"
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 尝试renderHtml
|
|||
|
|
ok, msg, _, _ = this.renderEmail(subject, body, map[string]interface{}{})
|
|||
|
|
if(!ok) {
|
|||
|
|
Log(msg)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// go func() {
|
|||
|
|
for _, email := range emails {
|
|||
|
|
if email == "" {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
m := map[string]interface{}{}
|
|||
|
|
m["email"] = email
|
|||
|
|
ok, msg, subject, body = this.renderEmail(subject, body, m)
|
|||
|
|
if(ok) {
|
|||
|
|
sendOk, msg := this.SendEmail(email, subject, body);
|
|||
|
|
this.AddEmailLog(email, subject, body, sendOk, msg)
|
|||
|
|
// 记录到Email Log
|
|||
|
|
if sendOk {
|
|||
|
|
Log("ok " + email)
|
|||
|
|
} else {
|
|||
|
|
Log("no " + email)
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
Log(msg);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
// }()
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 添加邮件日志
|
|||
|
|
func (this *EmailService) AddEmailLog(email, subject, body string, ok bool, msg string) {
|
|||
|
|
log := info.EmailLog{LogId: bson.NewObjectId(), Email: email, Subject: subject, Body: body, Ok: ok, Msg: msg, CreatedTime: time.Now()}
|
|||
|
|
db.Insert(db.EmailLogs, log)
|
|||
|
|
}
|
|||
|
|
// 展示邮件日志
|
|||
|
|
|
|||
|
|
func (this *EmailService) DeleteEmails(ids []string) bool {
|
|||
|
|
idsO := make([]bson.ObjectId, len(ids))
|
|||
|
|
for i, id := range ids {
|
|||
|
|
idsO[i] = bson.ObjectIdHex(id)
|
|||
|
|
}
|
|||
|
|
db.DeleteAll(db.EmailLogs, bson.M{"_id": bson.M{"$in": idsO}})
|
|||
|
|
|
|||
|
|
return true
|
|||
|
|
}
|
|||
|
|
func (this *EmailService) ListEmailLogs(pageNumber, pageSize int, sortField string, isAsc bool, email string) (page info.Page, emailLogs []info.EmailLog) {
|
|||
|
|
emailLogs = []info.EmailLog{}
|
|||
|
|
skipNum, sortFieldR := parsePageAndSort(pageNumber, pageSize, sortField, isAsc)
|
|||
|
|
query := bson.M{}
|
|||
|
|
if email != "" {
|
|||
|
|
query["Email"] = bson.M{"$regex": bson.RegEx{".*?" + email + ".*", "i"}}
|
|||
|
|
}
|
|||
|
|
q := db.EmailLogs.Find(query);
|
|||
|
|
// 总记录数
|
|||
|
|
count, _ := q.Count()
|
|||
|
|
// 列表
|
|||
|
|
q.Sort(sortFieldR).
|
|||
|
|
Skip(skipNum).
|
|||
|
|
Limit(pageSize).
|
|||
|
|
All(&emailLogs)
|
|||
|
|
page = info.NewPage(pageNumber, pageSize, count, nil)
|
|||
|
|
return
|
|||
|
|
}
|