feat: support Discord now

channel/discord.go

@@ -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
+}

channel/main.go

@@ -13,6 +13,7 @@ const (
 	TypeLark              = "lark"
 	TypeDing              = "ding"
 	TypeTelegram          = "telegram"
+	TypeDiscord           = "discord"
 	TypeBark              = "bark"
 	TypeClient            = "client"
 	TypeNone              = "none"
@@ -38,6 +39,8 @@ func SendMessage(message *model.Message, user *model.User) error {
 		return SendClientMessage(message, user)
 	case TypeTelegram:
 		return SendTelegramMessage(message, user)
+	case TypeDiscord:
+		return SendDiscordMessage(message, user)
 	case TypeNone:
 		return nil
 	default: