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.
58 lines
1.3 KiB
58 lines
1.3 KiB
package dingtalk
|
|
|
|
import (
|
|
"crypto/hmac"
|
|
"crypto/sha256"
|
|
"encoding/base64"
|
|
"encoding/json"
|
|
"fmt"
|
|
"gopkg.in/dtapps/go-library.v3/service/dingtalk/config"
|
|
"gopkg.in/dtapps/go-library.v3/service/dingtalk/message"
|
|
"gopkg.in/dtapps/go-library.v3/utils/gojson"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type DingBot struct {
|
|
Secret string
|
|
AccessToken string
|
|
}
|
|
|
|
type Response struct {
|
|
Errcode int64 `json:"errcode"`
|
|
Errmsg string `json:"errmsg"`
|
|
}
|
|
|
|
func (bot *DingBot) Send(msg message.Message) (Response, error) {
|
|
timestamp := time.Now().UnixNano() / 1e6
|
|
var response Response
|
|
signStr := sign(timestamp, bot.Secret)
|
|
dingUrl := fmt.Sprintf("%s?access_token=%s×tamp=%d&sign=%s", config.Api, bot.AccessToken, timestamp, signStr)
|
|
toString, err := gojson.MarshalToString(msg)
|
|
if err != nil {
|
|
return response, err
|
|
}
|
|
resp, e := http.Post(dingUrl, "application/json", strings.NewReader(toString))
|
|
if e != nil {
|
|
return response, e
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
body, _ := ioutil.ReadAll(resp.Body)
|
|
e = json.Unmarshal(body, &response)
|
|
if e != nil {
|
|
return response, e
|
|
}
|
|
return response, nil
|
|
}
|
|
|
|
func sign(t int64, secret string) string {
|
|
secStr := fmt.Sprintf("%d\n%s", t, secret)
|
|
hmac256 := hmac.New(sha256.New, []byte(secret))
|
|
hmac256.Write([]byte(secStr))
|
|
result := hmac256.Sum(nil)
|
|
return base64.StdEncoding.EncodeToString(result)
|
|
}
|