feat: now support Telegram bot

This commit is contained in:
JustSong
2022-12-21 18:33:00 +08:00
parent c67e639f26
commit a36d21d18e
5 changed files with 99 additions and 1 deletions
+2
View File
@@ -46,6 +46,8 @@ func (message *Message) Send(user *model.User) error {
return SendBarkMessage(message, user)
case TypeClient:
return SendClientMessage(message, user)
case TypeTelegram:
return SendTelegramMessage(message, user)
default:
return errors.New("不支持的消息通道:" + message.Channel)
}
+53
View File
@@ -0,0 +1,53 @@
package channel
import (
"bytes"
"encoding/json"
"errors"
"fmt"
"message-pusher/model"
"net/http"
)
type telegramMessageRequest struct {
ChatId string `json:"chat_id"`
Text string `json:"text"`
ParseMode string `json:"parse_mode"`
}
type telegramMessageResponse struct {
Ok bool `json:"ok"`
Description string `json:"description"`
}
func SendTelegramMessage(message *Message, user *model.User) error {
if user.TelegramBotToken == "" || user.TelegramChatId == "" {
return errors.New("未配置 Telegram 机器人消息推送方式")
}
messageRequest := telegramMessageRequest{
ChatId: user.TelegramChatId,
Text: message.Content,
ParseMode: "markdown",
}
if messageRequest.Text == "" {
messageRequest.Text = message.Description
}
jsonData, err := json.Marshal(messageRequest)
if err != nil {
return err
}
resp, err := http.Post(fmt.Sprintf("https://api.telegram.org/bot%s/sendMessage", user.TelegramBotToken), "application/json",
bytes.NewBuffer(jsonData))
if err != nil {
return err
}
var res telegramMessageResponse
err = json.NewDecoder(resp.Body).Decode(&res)
if err != nil {
return err
}
if !res.Ok {
return errors.New(res.Description)
}
return nil
}