mirror of https://github.com/astaxie/beego.git
100 lines
2.4 KiB
Go
100 lines
2.4 KiB
Go
// Beego (http://beego.me/)
|
|
// @description beego is an open-source, high-performance web framework for the Go programming language.
|
|
// @link http://github.com/astaxie/beego for the canonical source repository
|
|
// @license http://github.com/astaxie/beego/blob/master/LICENSE
|
|
// @authors astaxie
|
|
|
|
package logs
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/smtp"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
const (
|
|
subjectPhrase = "Diagnostic message from server"
|
|
)
|
|
|
|
// smtpWriter implements LoggerInterface and is used to send emails via given SMTP-server.
|
|
type SmtpWriter struct {
|
|
Username string `json:"Username"`
|
|
Password string `json:"password"`
|
|
Host string `json:"Host"`
|
|
Subject string `json:"subject"`
|
|
RecipientAddresses []string `json:"sendTos"`
|
|
Level int `json:"level"`
|
|
}
|
|
|
|
// create smtp writer.
|
|
func NewSmtpWriter() LoggerInterface {
|
|
return &SmtpWriter{Level: LevelTrace}
|
|
}
|
|
|
|
// init smtp writer with json config.
|
|
// config like:
|
|
// {
|
|
// "Username":"example@gmail.com",
|
|
// "password:"password",
|
|
// "host":"smtp.gmail.com:465",
|
|
// "subject":"email title",
|
|
// "sendTos":["email1","email2"],
|
|
// "level":LevelError
|
|
// }
|
|
func (s *SmtpWriter) Init(jsonconfig string) error {
|
|
err := json.Unmarshal([]byte(jsonconfig), s)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// write message in smtp writer.
|
|
// it will send an email with subject and only this message.
|
|
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
|
|
}
|
|
|
|
// implementing method. empty.
|
|
func (s *SmtpWriter) Flush() {
|
|
return
|
|
}
|
|
|
|
// implementing method. empty.
|
|
func (s *SmtpWriter) Destroy() {
|
|
return
|
|
}
|
|
|
|
func init() {
|
|
Register("smtp", NewSmtpWriter)
|
|
}
|