55 lines
1.1 KiB
Go
55 lines
1.1 KiB
Go
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
|
|
}
|