mirror of https://github.com/gogits/gogs.git
Unknown
11 years ago
10 changed files with 204 additions and 28 deletions
@ -0,0 +1,112 @@ |
|||||||
|
// Copyright 2014 The Gogs Authors. All rights reserved.
|
||||||
|
// Use of this source code is governed by a MIT-style
|
||||||
|
// license that can be found in the LICENSE file.
|
||||||
|
|
||||||
|
package mailer |
||||||
|
|
||||||
|
import ( |
||||||
|
"fmt" |
||||||
|
"net/smtp" |
||||||
|
"strings" |
||||||
|
|
||||||
|
"github.com/gogits/gogs/modules/base" |
||||||
|
"github.com/gogits/gogs/modules/log" |
||||||
|
) |
||||||
|
|
||||||
|
type Message struct { |
||||||
|
To []string |
||||||
|
From string |
||||||
|
Subject string |
||||||
|
Body string |
||||||
|
User string |
||||||
|
Type string |
||||||
|
Massive bool |
||||||
|
Info string |
||||||
|
} |
||||||
|
|
||||||
|
// create mail content
|
||||||
|
func (m Message) Content() string { |
||||||
|
// set mail type
|
||||||
|
contentType := "text/plain; charset=UTF-8" |
||||||
|
if m.Type == "html" { |
||||||
|
contentType = "text/html; charset=UTF-8" |
||||||
|
} |
||||||
|
|
||||||
|
// create mail content
|
||||||
|
content := "From: " + m.User + "<" + m.From + |
||||||
|
">\r\nSubject: " + m.Subject + "\r\nContent-Type: " + contentType + "\r\n\r\n" + m.Body |
||||||
|
return content |
||||||
|
} |
||||||
|
|
||||||
|
// Direct Send mail message
|
||||||
|
func Send(msg Message) (int, error) { |
||||||
|
log.Trace("Sending mails to: %s", strings.Join(msg.To, "; ")) |
||||||
|
host := strings.Split(base.MailService.Host, ":") |
||||||
|
|
||||||
|
// get message body
|
||||||
|
content := msg.Content() |
||||||
|
|
||||||
|
auth := smtp.PlainAuth("", base.MailService.User, base.MailService.Passwd, host[0]) |
||||||
|
|
||||||
|
if len(msg.To) == 0 { |
||||||
|
return 0, fmt.Errorf("empty receive emails") |
||||||
|
} |
||||||
|
|
||||||
|
if len(msg.Body) == 0 { |
||||||
|
return 0, fmt.Errorf("empty email body") |
||||||
|
} |
||||||
|
|
||||||
|
if msg.Massive { |
||||||
|
// send mail to multiple emails one by one
|
||||||
|
num := 0 |
||||||
|
for _, to := range msg.To { |
||||||
|
body := []byte("To: " + to + "\r\n" + content) |
||||||
|
err := smtp.SendMail(base.MailService.Host, auth, msg.From, []string{to}, body) |
||||||
|
if err != nil { |
||||||
|
return num, err |
||||||
|
} |
||||||
|
num++ |
||||||
|
} |
||||||
|
return num, nil |
||||||
|
} else { |
||||||
|
body := []byte("To: " + strings.Join(msg.To, ";") + "\r\n" + content) |
||||||
|
|
||||||
|
// send to multiple emails in one message
|
||||||
|
err := smtp.SendMail(base.MailService.Host, auth, msg.From, msg.To, body) |
||||||
|
if err != nil { |
||||||
|
return 0, err |
||||||
|
} else { |
||||||
|
return 1, nil |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
// Async Send mail message
|
||||||
|
func SendAsync(msg Message) { |
||||||
|
// TODO may be need pools limit concurrent nums
|
||||||
|
go func() { |
||||||
|
num, err := Send(msg) |
||||||
|
tos := strings.Join(msg.To, "; ") |
||||||
|
info := "" |
||||||
|
if err != nil { |
||||||
|
if len(msg.Info) > 0 { |
||||||
|
info = ", info: " + msg.Info |
||||||
|
} |
||||||
|
// log failed
|
||||||
|
log.Error(fmt.Sprintf("Async sent email %d succeed, not send emails: %s%s err: %s", num, tos, info, err)) |
||||||
|
return |
||||||
|
} |
||||||
|
log.Trace(fmt.Sprintf("Async sent email %d succeed, sent emails: %s%s", num, tos, info)) |
||||||
|
}() |
||||||
|
} |
||||||
|
|
||||||
|
// Create html mail message
|
||||||
|
func NewHtmlMessage(To []string, From, Subject, Body string) Message { |
||||||
|
return Message{ |
||||||
|
To: To, |
||||||
|
From: From, |
||||||
|
Subject: Subject, |
||||||
|
Body: Body, |
||||||
|
Type: "html", |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue