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

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
}