😏 Notice object support SREFlow AI

Author: Cairry

pkg/medium/entry.go

@@ -106,6 +106,8 @@ func senderFactory(noticeType string) (SendInter, error) {
 		return getSMSSender()
 	case "Phone":
 		return getPhoneSender()
+	case "SREFlow":
+		return NewSREFlowSender(), nil
 	default:
 		return nil, fmt.Errorf("无效的通知类型: %s", noticeType)
 	}

pkg/medium/sreflow.go

@@ -0,0 +1,45 @@
+package medium
+
+import (
+	"bytes"
+	"errors"
+	"fmt"
+	"io"
+	"watchAlert/pkg/tools"
+)
+
+type (
+	// SREFlowSender
+	SREFlowSender struct{}
+)
+
+var SREFlowTestContent = fmt.Sprintf(`{
+  "text": "%s"
+}`, RobotTestContent)
+
+func NewSREFlowSender() SendInter { return &SREFlowSender{} }
+
+func (w *SREFlowSender) Send(params SendParams) error {
+	return w.post(params.Hook, params.Headers, params.Content)
+}
+
+func (w *SREFlowSender) Test(params SendParams) error {
+	return w.post(params.Hook, params.Headers, SREFlowTestContent)
+}
+
+func (w *SREFlowSender) post(hook string, headers map[string]string, content string) error {
+	res, err := tools.Post(headers, hook, bytes.NewReader([]byte(content)), 10)
+	if err != nil {
+		return err
+	}
+
+	if res.StatusCode != 200 {
+		bodyByte, err := io.ReadAll(res.Body)
+		if err != nil {
+			return fmt.Errorf("读取 Body 失败, err: %s", err.Error())
+		}
+		return errors.New(string(bodyByte))
+	}
+
+	return nil
+}

pkg/medium/webhook.go

@@ -15,7 +15,6 @@ type (
 
 var WebhookTestContent = fmt.Sprintf(`{
   "text": "%s"
-  }
 }`, RobotTestContent)
 
 func NewWebHookSender() SendInter { return &WebHookSender{} }