feat: support Discord now

This commit is contained in:
JustSong
2023-01-10 18:32:04 +08:00
parent 54d1edaf1b
commit cfdc4ce357
6 changed files with 88 additions and 2 deletions
+54
View File
@@ -0,0 +1,54 @@
package channel
import (
"bytes"
"encoding/json"
"errors"
"message-pusher/model"
"net/http"
)
type discordMessageRequest struct {
Content string `json:"content"`
}
type discordMessageResponse struct {
Code int `json:"code"`
Message string `json:"message"`
}
func SendDiscordMessage(message *model.Message, user *model.User) error {
if user.DiscordWebhookURL == "" {
return errors.New("未配置 Discord 群机器人消息推送方式")
}
if message.Content == "" {
message.Content = message.Description
}
messageRequest := discordMessageRequest{
Content: message.Content,
}
jsonData, err := json.Marshal(messageRequest)
if err != nil {
return err
}
resp, err := http.Post(user.DiscordWebhookURL, "application/json", bytes.NewBuffer(jsonData))
if err != nil {
return err
}
if resp.StatusCode == http.StatusNoContent {
return nil
}
var res discordMessageResponse
err = json.NewDecoder(resp.Body).Decode(&res)
if err != nil {
return err
}
if res.Code != 0 {
return errors.New(res.Message)
}
if resp.StatusCode == http.StatusBadRequest {
return errors.New(resp.Status)
}
return nil
}