mirror of
https://github.com/easychen/pushdeer.git
synced 2025-04-30 16:29:51 +08:00
312 lines
6.8 KiB
Go
312 lines
6.8 KiB
Go
package notify
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"fmt"
|
||
"github.com/appleboy/gorush/config"
|
||
"github.com/appleboy/gorush/core"
|
||
"github.com/appleboy/gorush/logx"
|
||
"github.com/appleboy/gorush/status"
|
||
"github.com/geek-go/xmpush"
|
||
"strconv"
|
||
"strings"
|
||
"sync"
|
||
)
|
||
|
||
var (
|
||
mipushError error
|
||
mipushClient *XMPush
|
||
mipushOnce sync.Once
|
||
)
|
||
|
||
// GetMIPushClient use for create HMS Push.
|
||
func GetMIPushClient(conf *XmpushConfig) (*XMPush, error) {
|
||
once.Do(func() {
|
||
client := &XMPush{
|
||
Config: &XmpushConfig{
|
||
AppSecret: conf.AppSecret,
|
||
Package: conf.Package,
|
||
},
|
||
}
|
||
mipushClient = client
|
||
pushError = nil
|
||
})
|
||
|
||
return mipushClient, pushError
|
||
}
|
||
|
||
func InitMIPUSHClient(cfg *config.ConfYaml, appSecret, pkg string) (*XMPush, error) {
|
||
if appSecret == "" {
|
||
return nil, errors.New("Missing MI App Secret")
|
||
}
|
||
|
||
if pkg == "" {
|
||
return nil, errors.New("Missing MI App Package")
|
||
}
|
||
conf := &XmpushConfig{
|
||
AppSecret: appSecret,
|
||
Package: pkg,
|
||
}
|
||
if appSecret != cfg.MI.AppSecret || pkg != cfg.MI.Package {
|
||
return GetMIPushClient(conf)
|
||
}
|
||
if MIPUSHClient == nil {
|
||
return GetMIPushClient(conf)
|
||
}
|
||
return MIPUSHClient, nil
|
||
|
||
}
|
||
|
||
func GetMINotification(req *PushNotification, pkg string) (*xmpush.Message, error) {
|
||
payload := &Payload{
|
||
PushTitle: req.Title,
|
||
PushBody: req.Message,
|
||
IsShowNotify: "1",
|
||
}
|
||
payloadStr, err := json.Marshal(payload)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
//是否透传
|
||
passThrough := 1
|
||
if payload.IsShowNotify == "1" {
|
||
passThrough = 0 //通知栏消息
|
||
}
|
||
|
||
message := &xmpush.Message{
|
||
Payload: string(payloadStr),
|
||
Title: payload.PushTitle,
|
||
Description: payload.PushBody,
|
||
PassThrough: int32(passThrough),
|
||
NotifyType: 1,
|
||
RestrictedPackageName: pkg,
|
||
Extra: map[string]string{
|
||
"notify_foreground": "1",
|
||
},
|
||
}
|
||
if len(req.Tokens) > 0 {
|
||
for i, token := range req.Tokens {
|
||
if i == 0 {
|
||
message.RegistrationId = token
|
||
} else {
|
||
message.RegistrationId = message.RegistrationId + "," + token
|
||
}
|
||
}
|
||
}
|
||
return message, nil
|
||
}
|
||
|
||
func PushToMI(req *PushNotification, cfg *config.ConfYaml) (resp *ResponsePush, err error) {
|
||
logx.LogAccess.Debug("Start push notification for MI")
|
||
|
||
var (
|
||
client *XMPush
|
||
retryCount = 0
|
||
maxRetry = cfg.MI.MaxRetry
|
||
)
|
||
|
||
if req.Retry > 0 && req.Retry < maxRetry {
|
||
maxRetry = req.Retry
|
||
}
|
||
|
||
client, err = InitMIPUSHClient(cfg, cfg.MI.AppSecret, cfg.MI.Package)
|
||
if err != nil {
|
||
// MI server error
|
||
logx.LogError.Error("MI server error: " + err.Error())
|
||
return
|
||
}
|
||
|
||
resp = &ResponsePush{}
|
||
|
||
Retry:
|
||
isError := false
|
||
notification, err1 := GetMINotification(req, client.Config.Package)
|
||
if err1 != nil {
|
||
return nil, err1
|
||
}
|
||
|
||
res, err := client.SendMessage(context.Background(), notification)
|
||
if err != nil {
|
||
// Send Message error
|
||
errLog := logPush(cfg, core.FailedPush, req.To, req, err)
|
||
resp.Logs = append(resp.Logs, errLog)
|
||
logx.LogError.Error("HMS server send message error: " + err.Error())
|
||
return
|
||
}
|
||
// Huawei Push Send API does not support exact results for each token
|
||
if res.Code == 0 {
|
||
status.StatStorage.AddMISuccess(int64(1))
|
||
logx.LogAccess.Debug("MI Send Notification is completed successfully!")
|
||
} else {
|
||
isError = true
|
||
status.StatStorage.AddMIError(int64(1))
|
||
logx.LogAccess.Debug("MI Send Notification is failed! Code: " + strconv.Itoa(int(res.Code)))
|
||
}
|
||
|
||
if isError && retryCount < maxRetry {
|
||
retryCount++
|
||
|
||
// resend all tokens
|
||
goto Retry
|
||
}
|
||
|
||
return nil, nil
|
||
}
|
||
|
||
var Cfg = &XmpushConfig{
|
||
AppSecret: "",
|
||
Package: "",
|
||
}
|
||
|
||
// XmpushConfig ...
|
||
type XmpushConfig struct {
|
||
AppSecret string `toml:"app_secret"`
|
||
Package string `toml:"package"`
|
||
}
|
||
|
||
//Payload 消息payload，根据业务自定义
|
||
type Payload struct {
|
||
PushTitle string `json:"push_title"`
|
||
PushBody string `json:"push_body"`
|
||
IsShowNotify string `json:"is_show_notify"`
|
||
Ext string `json:"ext"`
|
||
}
|
||
|
||
type XMPush struct {
|
||
Config *XmpushConfig
|
||
}
|
||
|
||
// NewXmPush 获取实例
|
||
func NewXmPush(config *XmpushConfig) (*XMPush, error) {
|
||
|
||
if config.Package == "" || config.AppSecret == "" {
|
||
return nil, errors.New("please check config")
|
||
}
|
||
|
||
xm := &XMPush{
|
||
Config: config,
|
||
}
|
||
|
||
return xm, nil
|
||
}
|
||
|
||
func (m *XMPush) SendMessage(ctx context.Context, notification *xmpush.Message) (*xmpush.Result, error) {
|
||
// 通过 regID 推送
|
||
if notification.RegistrationId != "" {
|
||
return xmpush.SendMessageByRegIds(m.Config.AppSecret, notification)
|
||
}
|
||
// 群推
|
||
return xmpush.SendMessageAll(m.Config.AppSecret, notification)
|
||
|
||
}
|
||
|
||
//SendByCid 根据用户cid推送
|
||
func (m *XMPush) SendByCid(cid string, payload *Payload) error {
|
||
return m.SendByCids([]string{cid}, payload)
|
||
}
|
||
|
||
//SendByCids 根据用户cids批量推送
|
||
func (m *XMPush) SendByCids(cids []string, payload *Payload) error {
|
||
|
||
payloadStr, _ := json.Marshal(payload)
|
||
|
||
//是否透传
|
||
passThrough := 1
|
||
if payload.IsShowNotify == "1" {
|
||
passThrough = 0 //通知栏消息
|
||
}
|
||
|
||
message := &xmpush.Message{
|
||
Payload: string(payloadStr),
|
||
Title: payload.PushTitle,
|
||
Description: payload.PushBody,
|
||
PassThrough: int32(passThrough),
|
||
NotifyType: 1,
|
||
RestrictedPackageName: m.Config.Package,
|
||
Extra: map[string]string{
|
||
"notify_foreground": "1",
|
||
},
|
||
}
|
||
|
||
message.RegistrationId = strings.Join(cids, ",")
|
||
|
||
result, err := xmpush.SendMessageByRegIds(m.Config.AppSecret, message)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
fmt.Println(result)
|
||
|
||
return nil
|
||
}
|
||
|
||
// SendByAliases 根据别名推送批量推送
|
||
func (m *XMPush) SendByAliases(aliases []string, payload *Payload) error {
|
||
|
||
payloadStr, _ := json.Marshal(payload)
|
||
|
||
//是否透传
|
||
passThrough := 1
|
||
if payload.IsShowNotify == "1" {
|
||
passThrough = 0 //通知栏消息
|
||
}
|
||
|
||
message := &xmpush.Message{
|
||
Payload: string(payloadStr),
|
||
Title: payload.PushTitle,
|
||
Description: payload.PushBody,
|
||
PassThrough: int32(passThrough),
|
||
NotifyType: 1,
|
||
RestrictedPackageName: m.Config.Package,
|
||
Extra: map[string]string{
|
||
"notify_foreground": "1",
|
||
},
|
||
}
|
||
|
||
message.Alias = strings.Join(aliases, ",")
|
||
|
||
result, err := xmpush.SendMessageByRegAliasIds(m.Config.AppSecret, message)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
fmt.Println(result)
|
||
|
||
return nil
|
||
}
|
||
|
||
// SendAll 推送给所有人
|
||
func (m *XMPush) SendAll(payload *Payload) error {
|
||
|
||
payloadStr, _ := json.Marshal(payload)
|
||
|
||
//是否透传
|
||
passThrough := 1
|
||
if payload.IsShowNotify == "1" {
|
||
passThrough = 0 //通知栏消息
|
||
}
|
||
|
||
message := &xmpush.Message{
|
||
Payload: string(payloadStr),
|
||
Title: payload.PushTitle,
|
||
Description: payload.PushBody,
|
||
PassThrough: int32(passThrough),
|
||
NotifyType: 1,
|
||
RestrictedPackageName: m.Config.Package,
|
||
Extra: map[string]string{
|
||
"notify_foreground": "1",
|
||
},
|
||
}
|
||
|
||
result, err := xmpush.SendMessageAll(m.Config.AppSecret, message)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
fmt.Println(result)
|
||
|
||
return nil
|
||
}
|