v2.0

server/internal/library/notify/feishu/client.go

@@ -0,0 +1,83 @@
+package feishu
+
+import (
+	"encoding/json"
+	"fmt"
+	"strconv"
+	"time"
+
+	"github.com/go-resty/resty/v2"
+	"hotgo/internal/library/notify/feishu/internal/security"
+)
+
+const feishuAPI = "https://open.feishu.cn/open-apis/bot/v2/hook/"
+
+// Client feishu client
+type Client struct {
+	AccessToken string
+	Secret      string
+}
+
+// NewClient new client
+func NewClient(accessToken, secret string) *Client {
+	return &Client{
+		AccessToken: accessToken,
+		Secret:      secret,
+	}
+}
+
+// Response response struct
+type Response struct {
+	Code int64       `json:"code"`
+	Msg  string      `json:"msg"`
+	Data interface{} `json:"data"`
+
+	Extra         interface{} `json:"Extra"`
+	StatusCode    int64       `json:"StatusCode"`
+	StatusMessage string      `json:"StatusMessage"`
+}
+
+// Send send message
+func (d *Client) Send(message Message) (string, *Response, error) {
+	res := &Response{}
+
+	if len(d.AccessToken) < 1 {
+		return "", res, fmt.Errorf("accessToken is empty")
+	}
+
+	timestamp := time.Now().Unix()
+	sign, err := security.GenSign(d.Secret, timestamp)
+	if err != nil {
+		return "", res, err
+	}
+
+	body := message.Body()
+	body["timestamp"] = strconv.FormatInt(timestamp, 10)
+	body["sign"] = sign
+
+	reqBytes, err := json.Marshal(body)
+	if err != nil {
+		return "", res, err
+	}
+	reqString := string(reqBytes)
+
+	client := resty.New()
+	URL := fmt.Sprintf("%v%v", feishuAPI, d.AccessToken)
+	resp, err := client.SetRetryCount(3).R().
+		SetBody(body).
+		SetHeader("Accept", "application/json").
+		SetHeader("Content-Type", "application/json").
+		SetResult(&Response{}).
+		ForceContentType("application/json").
+		Post(URL)
+
+	if err != nil {
+		return reqString, nil, err
+	}
+
+	result := resp.Result().(*Response)
+	if result.Code != 0 {
+		return reqString, result, fmt.Errorf("send message to feishu error = %s", result.Msg)
+	}
+	return reqString, result, nil
+}