feat: channel corp is done

This commit is contained in:
JustSong
2022-11-25 12:04:36 +08:00
parent 4cf9aad148
commit d894fe32d8
6 changed files with 113 additions and 17 deletions
+60
View File
@@ -0,0 +1,60 @@
package channel
import (
"bytes"
"encoding/json"
"errors"
"fmt"
"message-pusher/model"
"net/http"
)
type corpMessageRequest struct {
MessageType string `json:"msgtype"`
Text struct {
Content string `json:"content"`
} `json:"text"`
Markdown struct {
Content string `json:"content"`
} `json:"markdown"`
}
type corpMessageResponse struct {
Code int `json:"errcode"`
Message string `json:"errmsg"`
}
func SendCorpMessage(message *Message, user *model.User) error {
if user.CorpWebhookURL == "" {
return errors.New("未配置企业微信群机器人消息推送方式")
}
messageRequest := corpMessageRequest{
MessageType: "text",
}
if message.Content == "" {
messageRequest.MessageType = "text"
messageRequest.Text.Content = message.Description
} else {
messageRequest.MessageType = "markdown"
messageRequest.Markdown.Content = message.Content
}
jsonData, err := json.Marshal(messageRequest)
if err != nil {
return err
}
resp, err := http.Post(fmt.Sprintf("%s", user.CorpWebhookURL), "application/json",
bytes.NewBuffer(jsonData))
if err != nil {
return err
}
var res corpMessageResponse
err = json.NewDecoder(resp.Body).Decode(&res)
if err != nil {
return err
}
if res.Code != 0 {
return errors.New(res.Message)
}
return nil
}
+4 -1
View File
@@ -8,7 +8,8 @@ import (
const (
TypeEmail = "email"
TypeWeChatTestAccount = "test"
TypeWeChatCorpAccount = "corp"
TypeWeChatCorpAccount = "corp_app"
TypeCorp = "corp"
TypeLark = "lark"
TypeDing = "ding"
TypeTelegram = "telegram"
@@ -33,6 +34,8 @@ func (message *Message) Send(user *model.User) error {
return SendWeChatTestMessage(message, user)
case TypeWeChatCorpAccount:
return SendWeChatCorpMessage(message, user)
case TypeCorp:
return SendCorpMessage(message, user)
case TypeLark:
return SendLarkMessage(message, user)
case TypeDing: