149 lines
4.0 KiB
Go
149 lines
4.0 KiB
Go
//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,
|
||
})
|
||
}
|