mirror of
https://github.com/astaxie/beego.git
synced 2024-11-18 09:20:55 +00:00
105 lines
2.3 KiB
Go
105 lines
2.3 KiB
Go
package logs
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net/smtp"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
subjectPhrase = "Diagnostic message from server"
|
|
)
|
|
|
|
// smtpWriter is used to send emails via given SMTP-server.
|
|
type SmtpWriter struct {
|
|
username string
|
|
password string
|
|
host string
|
|
subject string
|
|
recipientAddresses []string
|
|
level int
|
|
}
|
|
|
|
func NewSmtpWriter() LoggerInterface {
|
|
return &SmtpWriter{level: LevelTrace}
|
|
}
|
|
|
|
func (s *SmtpWriter) Init(jsonconfig string) error {
|
|
var m map[string]interface{}
|
|
err := json.Unmarshal([]byte(jsonconfig), &m)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if username, ok := m["username"]; !ok {
|
|
return errors.New("smtp config must have auth username")
|
|
} else if password, ok := m["password"]; !ok {
|
|
return errors.New("smtp config must have auth password")
|
|
} else if hostname, ok := m["host"]; !ok {
|
|
return errors.New("smtp config must have host like 'mail.example.com:25'")
|
|
} else if sendTos, ok := m["sendTos"]; !ok {
|
|
return errors.New("smtp config must have sendTos")
|
|
} else {
|
|
s.username = username.(string)
|
|
s.password = password.(string)
|
|
s.host = hostname.(string)
|
|
for _, v := range sendTos.([]interface{}) {
|
|
s.recipientAddresses = append(s.recipientAddresses, v.(string))
|
|
}
|
|
}
|
|
|
|
if subject, ok := m["subject"]; ok {
|
|
s.subject = subject.(string)
|
|
} else {
|
|
s.subject = subjectPhrase
|
|
}
|
|
if lv, ok := m["level"]; ok {
|
|
s.level = int(lv.(float64))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *SmtpWriter) WriteMsg(msg string, level int) error {
|
|
if level < s.level {
|
|
return nil
|
|
}
|
|
|
|
hp := strings.Split(s.host, ":")
|
|
|
|
// Set up authentication information.
|
|
auth := smtp.PlainAuth(
|
|
"",
|
|
s.username,
|
|
s.password,
|
|
hp[0],
|
|
)
|
|
// Connect to the server, authenticate, set the sender and recipient,
|
|
// and send the email all in one step.
|
|
content_type := "Content-Type: text/plain" + "; charset=UTF-8"
|
|
mailmsg := []byte("To: " + strings.Join(s.recipientAddresses, ";") + "\r\nFrom: " + s.username + "<" + s.username +
|
|
">\r\nSubject: " + s.subject + "\r\n" + content_type + "\r\n\r\n" + fmt.Sprintf(".%s", time.Now().Format("2006-01-02 15:04:05")) + msg)
|
|
|
|
err := smtp.SendMail(
|
|
s.host,
|
|
auth,
|
|
s.username,
|
|
s.recipientAddresses,
|
|
mailmsg,
|
|
)
|
|
|
|
return err
|
|
}
|
|
|
|
func (s *SmtpWriter) Flush() {
|
|
return
|
|
}
|
|
func (s *SmtpWriter) Destroy() {
|
|
return
|
|
}
|
|
|
|
func init() {
|
|
Register("smtp", NewSmtpWriter)
|
|
}
|