1
0
mirror of https://github.com/astaxie/beego.git synced 2024-11-30 01:21:28 +00:00
Beego/logs/smtp.go

161 lines
3.8 KiB
Go
Raw Normal View History

2014-08-18 08:41:43 +00:00
// Copyright 2014 beego Author. All Rights Reserved.
2014-07-03 15:40:21 +00:00
//
2014-08-18 08:41:43 +00:00
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
2014-07-03 15:40:21 +00:00
//
2014-08-18 08:41:43 +00:00
// http://www.apache.org/licenses/LICENSE-2.0
2014-07-03 15:40:21 +00:00
//
2014-08-18 08:41:43 +00:00
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
2013-08-27 15:48:58 +00:00
package logs
import (
2014-08-26 07:43:18 +00:00
"crypto/tls"
2013-08-27 15:48:58 +00:00
"encoding/json"
"fmt"
2014-08-26 07:43:18 +00:00
"net"
2013-08-27 15:48:58 +00:00
"net/smtp"
"strings"
"time"
2013-08-27 15:48:58 +00:00
)
2015-09-11 15:08:24 +00:00
// SMTPWriter implements LoggerInterface and is used to send emails via given SMTP-server.
type SMTPWriter struct {
2015-09-04 14:20:55 +00:00
Username string `json:"username"`
2013-12-19 10:16:06 +00:00
Password string `json:"password"`
2015-09-04 14:20:55 +00:00
Host string `json:"host"`
2013-12-19 10:16:06 +00:00
Subject string `json:"subject"`
2014-08-27 07:53:08 +00:00
FromAddress string `json:"fromAddress"`
2013-12-19 10:16:06 +00:00
RecipientAddresses []string `json:"sendTos"`
Level int `json:"level"`
2013-08-27 15:48:58 +00:00
}
2015-09-11 15:08:24 +00:00
// NewSMTPWriter create smtp writer.
func newSMTPWriter() Logger {
return &SMTPWriter{Level: LevelTrace}
2013-08-27 15:48:58 +00:00
}
2015-09-11 15:08:24 +00:00
// Init smtp writer with json config.
2013-12-30 15:32:57 +00:00
// config like:
// {
2016-01-04 14:22:42 +00:00
// "username":"example@gmail.com",
2013-12-30 15:32:57 +00:00
// "password:"password",
// "host":"smtp.gmail.com:465",
// "subject":"email title",
2014-08-26 10:47:05 +00:00
// "fromAddress":"from@example.com",
2013-12-30 15:32:57 +00:00
// "sendTos":["email1","email2"],
// "level":LevelError
// }
2015-09-11 15:08:24 +00:00
func (s *SMTPWriter) Init(jsonconfig string) error {
2013-12-19 10:16:06 +00:00
err := json.Unmarshal([]byte(jsonconfig), s)
2013-08-27 15:48:58 +00:00
if err != nil {
return err
}
return nil
}
2015-09-11 15:08:24 +00:00
func (s *SMTPWriter) getSMTPAuth(host string) smtp.Auth {
if len(strings.Trim(s.Username, " ")) == 0 && len(strings.Trim(s.Password, " ")) == 0 {
return nil
}
return smtp.PlainAuth(
"",
s.Username,
s.Password,
host,
)
}
2015-09-11 15:08:24 +00:00
func (s *SMTPWriter) sendMail(hostAddressWithPort string, auth smtp.Auth, fromAddress string, recipients []string, msgContent []byte) error {
client, err := smtp.Dial(hostAddressWithPort)
if err != nil {
return err
}
host, _, _ := net.SplitHostPort(hostAddressWithPort)
tlsConn := &tls.Config{
InsecureSkipVerify: true,
ServerName: host,
}
if err = client.StartTLS(tlsConn); err != nil {
return err
}
if auth != nil {
if err = client.Auth(auth); err != nil {
return err
}
}
if err = client.Mail(fromAddress); err != nil {
return err
}
for _, rec := range recipients {
if err = client.Rcpt(rec); err != nil {
return err
}
}
w, err := client.Data()
if err != nil {
return err
}
_, err = w.Write([]byte(msgContent))
if err != nil {
return err
}
err = w.Close()
if err != nil {
return err
}
err = client.Quit()
if err != nil {
return err
}
return nil
}
2015-09-11 15:08:24 +00:00
// WriteMsg write message in smtp writer.
2013-12-30 15:32:57 +00:00
// it will send an email with subject and only this message.
2016-01-23 08:24:58 +00:00
func (s *SMTPWriter) WriteMsg(when time.Time, msg string, level int) error {
if level > s.Level {
2013-08-27 15:48:58 +00:00
return nil
}
2013-12-19 10:16:06 +00:00
hp := strings.Split(s.Host, ":")
2013-08-27 15:48:58 +00:00
// Set up authentication information.
2015-09-11 15:08:24 +00:00
auth := s.getSMTPAuth(hp[0])
2014-08-18 08:41:43 +00:00
2013-08-27 15:48:58 +00:00
// Connect to the server, authenticate, set the sender and recipient,
// and send the email all in one step.
2015-09-11 15:08:24 +00:00
contentType := "Content-Type: text/plain" + "; charset=UTF-8"
2014-08-26 10:47:05 +00:00
mailmsg := []byte("To: " + strings.Join(s.RecipientAddresses, ";") + "\r\nFrom: " + s.FromAddress + "<" + s.FromAddress +
2016-01-23 08:24:58 +00:00
">\r\nSubject: " + s.Subject + "\r\n" + contentType + "\r\n\r\n" + fmt.Sprintf(".%s", when.Format("2006-01-02 15:04:05")) + msg)
2015-02-23 03:42:46 +00:00
return s.sendMail(s.Host, auth, s.FromAddress, s.RecipientAddresses, mailmsg)
2013-08-27 15:48:58 +00:00
}
2015-09-11 15:08:24 +00:00
// Flush implementing method. empty.
func (s *SMTPWriter) Flush() {
2013-11-27 09:50:10 +00:00
return
}
2013-12-30 15:32:57 +00:00
2015-09-11 15:08:24 +00:00
// Destroy implementing method. empty.
func (s *SMTPWriter) Destroy() {
2013-08-27 15:48:58 +00:00
return
}
func init() {
2015-09-11 15:08:24 +00:00
Register("smtp", newSMTPWriter)
2013-08-27 15:48:58 +00:00
}