1.4.5.2

EdgeAPI/internal/senders/mediasenders/media_qy_weixin_robot.go

@@ -0,0 +1,82 @@
+//go:build plus
+
+package mediasenders
+
+import (
+	"bytes"
+	"errors"
+	"github.com/TeaOSLab/EdgeAPI/internal/utils"
+	"github.com/iwind/TeaGo/maps"
+	"github.com/iwind/TeaGo/utils/string"
+	"io"
+	"net/http"
+	"strings"
+	"time"
+)
+
+// QyWeixinRobotMedia 企业微信群机器人媒介
+type QyWeixinRobotMedia struct {
+	WebHookURL string     `yaml:"webHookURL" json:"webHookURL"`
+	TextFormat TextFormat `yaml:"textFormat" json:"textFormat"`
+}
+
+// NewQyWeixinRobotMedia 获取新对象
+func NewQyWeixinRobotMedia() *QyWeixinRobotMedia {
+	return &QyWeixinRobotMedia{}
+}
+
+func (this *QyWeixinRobotMedia) Send(user string, subject string, body string, productName string, datetime string) (resp []byte, err error) {
+	if len(this.WebHookURL) == 0 {
+		return nil, errors.New("webHook url should not be empty")
+	}
+
+	mobiles := []string{}
+	if len(user) > 0 {
+		for _, u := range strings.Split(user, ",") {
+			u = strings.TrimSpace(u)
+			if len(u) > 0 {
+				mobiles = append(mobiles, u)
+			}
+		}
+	}
+
+	var content maps.Map
+	if this.TextFormat == FormatMarkdown { // markdown
+		content = maps.Map{
+			"msgtype": "markdown",
+			"markdown": maps.Map{
+				"content":               subject + "\n" + body,
+				"mentioned_mobile_list": mobiles,
+			},
+		}
+	} else {
+		content = maps.Map{
+			"msgtype": "text",
+			"text": maps.Map{
+				"content":               subject + "\n" + body,
+				"mentioned_mobile_list": mobiles,
+			},
+		}
+	}
+
+	reader := bytes.NewBufferString(stringutil.JSONEncode(content))
+	req, err := http.NewRequest(http.MethodPost, this.WebHookURL, reader)
+	if err != nil {
+		return nil, err
+	}
+	req.Header.Set("Content-Type", "application/json")
+	client := utils.SharedHttpClient(5 * time.Second)
+	response, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer response.Body.Close()
+
+	resp, err = io.ReadAll(response.Body)
+	return
+}
+
+// RequireUser 是否需要用户标识
+func (this *QyWeixinRobotMedia) RequireUser() bool {
+	return false
+}