You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
2.0 KiB
84 lines
2.0 KiB
package cms |
|
|
|
import ( |
|
"bytes" |
|
"context" |
|
"crypto/sha1" |
|
"encoding/hex" |
|
"encoding/json" |
|
"net/http" |
|
"sort" |
|
"strings" |
|
|
|
"go-common/library/log" |
|
|
|
"github.com/pkg/errors" |
|
) |
|
|
|
// MerakNotify send notify |
|
func (d *Dao) MerakNotify(ctx context.Context, title, content string) (err error) { |
|
var ( |
|
cfg = d.conf.Cfg.Merak |
|
sign string |
|
req *http.Request |
|
body []byte |
|
) |
|
params := map[string]string{ |
|
"Action": "CreateWechatMessage", |
|
"PublicKey": cfg.Key, |
|
"UserName": strings.Join(cfg.Names, ","), |
|
"Title": title, |
|
"Content": content, |
|
"TreeId": "", |
|
} |
|
if sign, err = MerakSign(params, cfg.Secret); err != nil { |
|
log.Error("MerakNotify Failed to sign params: %+v: %+v", params, err) |
|
return err |
|
} |
|
params["Signature"] = sign |
|
if body, err = json.Marshal(params); err != nil { |
|
log.Error("MerakNotify Json %v, Err %v", params, err) |
|
return |
|
} |
|
if req, err = http.NewRequest(http.MethodPost, cfg.Host, bytes.NewReader(body)); err != nil { |
|
log.Error("MerakNotify NewRequest Err %v, Host %v", err, cfg.Host) |
|
return |
|
} |
|
req.Header.Set("Content-Type", "application/json; charset=utf-8") |
|
res := struct { |
|
Action string `json:"Action"` |
|
RetCode int `json:"RetCode"` |
|
Data []string `json:"Data"` |
|
}{} |
|
if err = d.client.Do(ctx, req, &res); err != nil { |
|
return |
|
} |
|
if res.RetCode != 0 { |
|
err = errors.Errorf("Merak error: %d", res.RetCode) |
|
log.Error("Failed to send notify by merak with params: %+v: %+v", string(body), err) |
|
return |
|
} |
|
return |
|
} |
|
|
|
// MerakSign is used to sign for merak wechat msg |
|
func MerakSign(params map[string]string, secret string) (string, error) { |
|
keys := make([]string, 0, len(params)) |
|
for k := range params { |
|
keys = append(keys, k) |
|
} |
|
sort.Strings(keys) |
|
buf := bytes.Buffer{} |
|
for _, k := range keys { |
|
buf.WriteString(k + params[k]) |
|
} |
|
h := sha1.New() |
|
if _, err := h.Write(buf.Bytes()); err != nil { |
|
return "", errors.WithStack(err) |
|
} |
|
if _, err := h.Write([]byte(secret)); err != nil { |
|
return "", errors.WithStack(err) |
|
} |
|
sum := h.Sum(nil) |
|
return hex.EncodeToString(sum), nil |
|
}
|
|
|