1.4.5.2

EdgeAPI/internal/senders/mediasenders/media_web_hook.go

@@ -0,0 +1,148 @@
+//go:build plus
+
+package mediasenders
+
+import (
+	"errors"
+	teaconst "github.com/TeaOSLab/EdgeAPI/internal/const"
+	"github.com/TeaOSLab/EdgeAPI/internal/utils"
+	"github.com/iwind/TeaGo/types"
+	"github.com/iwind/TeaGo/utils/string"
+	"io"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+)
+
+// WebHookMedia WebHook媒介
+type WebHookMedia struct {
+	URL         string      `yaml:"url" json:"url"` // URL中可以使用${MessageSubject}, ${MessageBody}两个变量
+	Method      string      `yaml:"method" json:"method"`
+	ContentType string      `yaml:"contentType" json:"contentType"` // 内容类型：params|body
+	Headers     []*Variable `yaml:"headers" json:"headers"`
+	Params      []*Variable `yaml:"params" json:"params"`
+	Body        string      `yaml:"body" json:"body"`
+}
+
+// NewWebHookMedia 获取新对象
+func NewWebHookMedia() *WebHookMedia {
+	return &WebHookMedia{}
+}
+
+// Send 发送
+func (this *WebHookMedia) Send(user string, subject string, body string, productName string, datetime string) (resp []byte, err error) {
+	if len(this.URL) == 0 {
+		return nil, errors.New("'url' should be specified")
+	}
+
+	var timeout = 10 * time.Second
+
+	if len(this.Method) == 0 {
+		this.Method = http.MethodGet
+	}
+
+	this.URL = strings.Replace(this.URL, "${MessageUser}", url.QueryEscape(user), -1)
+	this.URL = strings.Replace(this.URL, "${MessageSubject}", url.QueryEscape(subject), -1)
+	this.URL = strings.Replace(this.URL, "${MessageBody}", url.QueryEscape(body), -1)
+
+	var req *http.Request
+	if this.Method == http.MethodGet {
+		req, err = http.NewRequest(this.Method, this.URL, nil)
+	} else {
+		var params = url.Values{
+			"MessageUser":    []string{user},
+			"MessageSubject": []string{subject},
+			"MessageBody":    []string{body},
+		}
+
+		var postBody = ""
+		if this.ContentType == "params" {
+			for _, param := range this.Params {
+				param.Value = strings.Replace(param.Value, "${MessageUser}", user, -1)
+				param.Value = strings.Replace(param.Value, "${MessageSubject}", subject, -1)
+				param.Value = strings.Replace(param.Value, "$MessageBody}", body, -1)
+				params.Add(param.Name, param.Value)
+			}
+			postBody = params.Encode()
+		} else if this.ContentType == "body" {
+			userJSON := stringutil.JSONEncode(user)
+			subjectJSON := stringutil.JSONEncode(subject)
+			bodyJSON := stringutil.JSONEncode(body)
+			if len(userJSON) > 0 {
+				userJSON = userJSON[1 : len(userJSON)-1]
+			}
+			if len(subjectJSON) > 0 {
+				subjectJSON = subjectJSON[1 : len(subjectJSON)-1]
+			}
+			if len(bodyJSON) > 0 {
+				bodyJSON = bodyJSON[1 : len(bodyJSON)-1]
+			}
+			postBody = strings.Replace(this.Body, "${MessageUser}", userJSON, -1)
+			postBody = strings.Replace(postBody, "${MessageSubject}", subjectJSON, -1)
+			postBody = strings.Replace(postBody, "${MessageBody}", bodyJSON, -1)
+		} else {
+			postBody = params.Encode()
+		}
+
+		req, err = http.NewRequest(this.Method, this.URL, strings.NewReader(postBody))
+	}
+	if err != nil {
+		return nil, err
+	}
+	req.Header.Set("User-Agent", teaconst.ProductName+"/"+teaconst.Version)
+
+	if this.Method == http.MethodPost {
+		if this.ContentType == "params" {
+			req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+		}
+	}
+
+	if len(this.Headers) > 0 {
+		for _, h := range this.Headers {
+			req.Header.Set(h.Name, h.Value)
+		}
+	}
+
+	var client = utils.SharedHttpClient(timeout)
+	response, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer func() {
+		_ = response.Body.Close()
+	}()
+
+	data, err := io.ReadAll(response.Body)
+	if err != nil {
+		return data, err
+	}
+
+	// check response code
+	if response.StatusCode >= 400 {
+		err = errors.New("invalid response code '" + types.String(response.StatusCode) + "'")
+	}
+
+	return data, err
+}
+
+// RequireUser 是否需要用户标识
+func (this *WebHookMedia) RequireUser() bool {
+	return false
+}
+
+// AddHeader 添加Header
+func (this *WebHookMedia) AddHeader(name string, value string) {
+	this.Headers = append(this.Headers, &Variable{
+		Name:  name,
+		Value: value,
+	})
+}
+
+// AddParam 添加参数
+func (this *WebHookMedia) AddParam(name string, value string) {
+	this.Params = append(this.Params, &Variable{
+		Name:  name,
+		Value: value,
+	})
+}