parent
0b1afe6283
commit
cf1ccb5dcf
@ -0,0 +1,59 @@
|
|||||||
|
package message
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
|
||||||
|
"luxshare-daily-report/global"
|
||||||
|
"net/http"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Bark struct {
|
||||||
|
url string
|
||||||
|
key string
|
||||||
|
}
|
||||||
|
|
||||||
|
var b = Bark{}
|
||||||
|
|
||||||
|
func initBark() {
|
||||||
|
b.url = global.GLO_CONFIG.BarkUrl
|
||||||
|
b.key = global.GLO_CONFIG.BarkKey
|
||||||
|
Register("bark", b)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m Bark) Send(message Body) {
|
||||||
|
log.Println("[bark] Sending by bark...")
|
||||||
|
var reqBody = Request{
|
||||||
|
DeviceKey: b.key,
|
||||||
|
Title: message.Title,
|
||||||
|
Body: message.Content,
|
||||||
|
Icon: "https://m.luxshare-ict.com/favicon.ico",
|
||||||
|
//Url: "https://github.com/zggsong",
|
||||||
|
}
|
||||||
|
req, _ := json.Marshal(reqBody)
|
||||||
|
resp, err := http.Post(m.url, "application/json; charset=utf-8", bytes.NewReader(req))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("[bark] http post failed: %v\n", err)
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
log.Printf("[bark] Send successful")
|
||||||
|
}
|
||||||
|
|
||||||
|
type Request struct {
|
||||||
|
Body string `json:"body"`
|
||||||
|
DeviceKey string `json:"device_key"`
|
||||||
|
Title string `json:"title"`
|
||||||
|
Badge int `json:"badge"`
|
||||||
|
Category string `json:"category"`
|
||||||
|
Sound string `json:"sound"`
|
||||||
|
Icon string `json:"icon"`
|
||||||
|
Group string `json:"group"`
|
||||||
|
Url string `json:"url"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type Response struct {
|
||||||
|
Code int `json:"code"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
Timestamp int `json:"timestamp"`
|
||||||
|
}
|
||||||
@ -0,0 +1,48 @@
|
|||||||
|
package message
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/jordan-wright/email"
|
||||||
|
log "github.com/sirupsen/logrus"
|
||||||
|
"luxshare-daily-report/global"
|
||||||
|
"net/smtp"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Mail struct {
|
||||||
|
Host string `json:"host" yaml:"host"`
|
||||||
|
Protocol string `json:"protocol" yaml:"protocol"`
|
||||||
|
Port int `json:"port" yaml:"port"`
|
||||||
|
Username string `json:"username" yaml:"username"`
|
||||||
|
Password string `json:"password" yaml:"password"`
|
||||||
|
FromName string `json:"from_name" yaml:"from_name"`
|
||||||
|
To []string `json:"to" yaml:"to"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var m Mail
|
||||||
|
|
||||||
|
func initMail() {
|
||||||
|
m = Mail{
|
||||||
|
Host: global.GLO_CONFIG.MailHost,
|
||||||
|
Port: global.GLO_CONFIG.MailPort,
|
||||||
|
Username: global.GLO_CONFIG.MailUser,
|
||||||
|
Password: global.GLO_CONFIG.MailPwd,
|
||||||
|
FromName: global.GLO_CONFIG.MailFromName,
|
||||||
|
To: global.GLO_CONFIG.MailTo,
|
||||||
|
}
|
||||||
|
Register("mail", m)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m Mail) Send(message Body) {
|
||||||
|
log.Println("[mail] Sending by mail...")
|
||||||
|
e := email.NewEmail()
|
||||||
|
e.From = m.FromName
|
||||||
|
e.To = m.To
|
||||||
|
e.Subject = message.Title
|
||||||
|
e.Text = []byte(message.Content)
|
||||||
|
addr := fmt.Sprintf("%v:%v", m.Host, m.Port)
|
||||||
|
err := e.Send(addr, smtp.PlainAuth("", m.Username, m.Password, m.Host))
|
||||||
|
if err != nil {
|
||||||
|
log.Fatalf("[mail] Send failed: %v\n", err)
|
||||||
|
}
|
||||||
|
log.Printf("[mail] Send successful")
|
||||||
|
}
|
||||||
@ -1 +1,33 @@
|
|||||||
package message
|
package message
|
||||||
|
|
||||||
|
import "luxshare-daily-report/global"
|
||||||
|
|
||||||
|
var Messages = make(map[string]Message, 0)
|
||||||
|
|
||||||
|
type Message interface {
|
||||||
|
Send(body Body)
|
||||||
|
}
|
||||||
|
|
||||||
|
type Body struct {
|
||||||
|
Title string
|
||||||
|
Content string
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetSupport 获取支持
|
||||||
|
func GetSupport() Message {
|
||||||
|
initBark()
|
||||||
|
initMail()
|
||||||
|
//key := "bark"
|
||||||
|
key := global.GLO_CONFIG.MsgType
|
||||||
|
return Messages[key]
|
||||||
|
}
|
||||||
|
|
||||||
|
// Enabled 是否启用
|
||||||
|
func Enabled() bool {
|
||||||
|
return global.GLO_CONFIG.MsgEnabled
|
||||||
|
}
|
||||||
|
|
||||||
|
// Register 注册
|
||||||
|
func Register(name string, message Message) {
|
||||||
|
Messages[name] = message
|
||||||
|
}
|
||||||
|
|||||||
@ -0,0 +1,35 @@
|
|||||||
|
package support
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"luxshare-daily-report/message"
|
||||||
|
)
|
||||||
|
|
||||||
|
// SendMessageError 发送错误信息
|
||||||
|
func SendMessageError(err error) {
|
||||||
|
var body = message.Body{}
|
||||||
|
body.Title = "LuxShareReportServer"
|
||||||
|
body.Content = fmt.Sprintf("Error occurred: %s", err.Error())
|
||||||
|
send(body)
|
||||||
|
}
|
||||||
|
|
||||||
|
// SendSuccess 发送成功信息
|
||||||
|
func SendSuccess(content string) {
|
||||||
|
body := message.Body{
|
||||||
|
Title: "LuxShareReportServer",
|
||||||
|
Content: content,
|
||||||
|
}
|
||||||
|
send(body)
|
||||||
|
}
|
||||||
|
|
||||||
|
// send 发送
|
||||||
|
func send(body message.Body) {
|
||||||
|
m := message.GetSupport()
|
||||||
|
if m == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
enabled := message.Enabled()
|
||||||
|
if enabled {
|
||||||
|
m.Send(body)
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue