feat: support Discord now
This commit is contained in:
@@ -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
|
||||
}
|
||||
Reference in New Issue
Block a user