feat: message notify

server/internal/msg/msgx/sender/feishu_bot.go

@@ -0,0 +1,100 @@
+package sender
+
+import (
+	"crypto/hmac"
+	"crypto/sha256"
+	"encoding/base64"
+	"errors"
+	"fmt"
+	"mayfly-go/internal/msg/msgx"
+	"mayfly-go/pkg/httpx"
+	"mayfly-go/pkg/utils/collx"
+	"mayfly-go/pkg/utils/stringx"
+	"strings"
+
+	"time"
+
+	"github.com/may-fly/cast"
+)
+
+type feishuBotMsgReq struct {
+	MsgType string `json:"msg_type"`
+	Content struct {
+		Text string `json:"text"`
+	} `json:"content"`
+	Timestamp string `json:"timestamp"`
+	Sign      string `json:"sign"`
+}
+
+type feishuBotMsgResp struct {
+	Code    int    `json:"code"`
+	Message string `json:"msg"`
+}
+
+// FeishuBotSender 发送飞书机器人消息
+type FeishuBotSender struct{}
+
+func (f FeishuBotSender) Send(channel *msgx.Channel, msg *msgx.Msg) error {
+	// https://open.feishu.cn/document/client-docs/bot-v3/add-custom-bot
+	msgReq := feishuBotMsgReq{
+		MsgType: "text",
+	}
+
+	params := msg.Params
+	receiver := `<at user_id="all"></at>`
+	// 使用receiver参数替换消息内容中可能存在的接收人信息
+	if len(msg.Receivers) > 0 {
+		if to := collx.ArrayMapFilter(msg.Receivers, func(a msgx.Receiver) (string, bool) {
+			if uid := a.GetExtraString("feishuUserId"); uid != "" {
+				// 使用<at user_id="userId"></at>
+				return fmt.Sprintf(`<at user_id="%s"></at>`, uid), true
+			}
+			return "", false
+		}); len(to) > 0 {
+			receiver = strings.Join(to, "")
+		}
+	}
+	params[msgx.ReceiverKey] = receiver
+	content, err := stringx.TemplateResolve(msg.Content, params)
+	if err != nil {
+		return err
+	}
+
+	msgReq.Content.Text = content
+
+	if secret := channel.GetExtraString("secret"); secret != "" {
+		timestamp := time.Now().Unix()
+		if sign, err := f.sign(secret, timestamp); err != nil {
+			return err
+		} else {
+			msgReq.Sign = sign
+		}
+		msgReq.Timestamp = cast.ToString(timestamp)
+	}
+
+	var res feishuBotMsgResp
+	err = httpx.NewReq(channel.URL).
+		PostObj(msgReq).
+		BodyTo(&res)
+	if err != nil {
+		return err
+	}
+
+	if res.Code != 0 {
+		return errors.New(res.Message)
+	}
+	return nil
+}
+
+func (e FeishuBotSender) sign(secret string, timestamp int64) (string, error) {
+	//timestamp + key 做sha256, 再进行base64 encode
+	stringToSign := fmt.Sprintf("%v", timestamp) + "\n" + secret
+	var data []byte
+	h := hmac.New(sha256.New, []byte(stringToSign))
+	_, err := h.Write(data)
+	if err != nil {
+		return "", err
+	}
+	signature := base64.StdEncoding.EncodeToString(h.Sum(nil))
+	return signature, nil
+}