feat: add tg bot send way

This commit is contained in:
engigu
2026-01-14 20:51:58 +08:00
parent 1d13bc69ab
commit 53a60b54b2
10 changed files with 321 additions and 2 deletions
+4
View File
@@ -119,6 +119,10 @@ func (sw *SendTaskInsService) ValidateDiffIns(ins models.SendTasksIns) (string,
_, Msg := app.CommonPlaygroundValid(Config)
return Msg, Config
}
if ins.WayType == constant.MessageTypeTelegram {
var Config models.InsTelegramConfig
return "", Config
}
return "未知的渠道的config校验", empty
}
@@ -26,6 +26,7 @@ const (
MessageTypeWeChatOFAccount = channels.MessageTypeWeChatOFAccount
MessageTypeMessageNest = channels.MessageTypeMessageNest
MessageTypeAliyunSMS = channels.MessageTypeAliyunSMS
MessageTypeTelegram = channels.MessageTypeTelegram
)
// ChannelRegistry 渠道注册表
@@ -99,6 +100,7 @@ func GetGlobalChannelRegistry() *ChannelRegistry {
globalChannelRegistry.Register(channels.NewWeChatOFAccountChannel())
globalChannelRegistry.Register(channels.NewMessageNestChannel())
globalChannelRegistry.Register(channels.NewAliyunSMSChannel())
globalChannelRegistry.Register(channels.NewTelegramChannel())
})
return globalChannelRegistry
}
@@ -0,0 +1,69 @@
package channels
import (
"fmt"
"message-nest/models"
"message-nest/pkg/message"
"message-nest/service/send_ins_service"
"message-nest/service/send_way_service"
)
type TelegramChannel struct{ *BaseChannel }
func NewTelegramChannel() *TelegramChannel {
return &TelegramChannel{BaseChannel: NewBaseChannel(MessageTypeTelegram, []string{FormatTypeMarkdown, FormatTypeHTML, FormatTypeText})}
}
func (c *TelegramChannel) SendUnified(msgObj interface{}, ins models.SendTasksIns, content *UnifiedMessageContent) (string, string) {
auth, ok := msgObj.(send_way_service.WayDetailTelegram)
if !ok {
return "", "类型转换失败"
}
insService := send_ins_service.SendTaskInsService{}
errStr, configInterface := insService.ValidateDiffIns(ins)
if errStr != "" {
return errStr, ""
}
_, ok = configInterface.(models.InsTelegramConfig)
if !ok {
return "Telegram config校验失败", ""
}
contentType, formattedContent, err := c.FormatContent(content)
if err != nil {
return "", err.Error()
}
cli := message.Telegram{
BotToken: auth.BotToken,
ChatID: auth.ChatID,
ApiHost: auth.ApiHost,
ProxyURL: auth.ProxyURL,
}
var res []byte
var errMsg string
if contentType == FormatTypeText {
res, err = cli.SendMessageText(formattedContent)
if err != nil {
errMsg = fmt.Sprintf("发送失败:%s", err.Error())
}
} else if contentType == FormatTypeMarkdown {
res, err = cli.SendMessageMarkdown(formattedContent)
if err != nil {
errMsg = fmt.Sprintf("发送失败:%s", err.Error())
}
} else if contentType == FormatTypeHTML {
res, err = cli.SendMessageHTML(formattedContent)
if err != nil {
errMsg = fmt.Sprintf("发送失败:%s", err.Error())
}
} else {
errMsg = fmt.Sprintf("未知的Telegram发送内容类型:%s", contentType)
}
return string(res), errMsg
}
@@ -18,6 +18,7 @@ const (
MessageTypeWeChatOFAccount = constant.MessageTypeWeChatOFAccount
MessageTypeMessageNest = constant.MessageTypeMessageNest
MessageTypeAliyunSMS = constant.MessageTypeAliyunSMS
MessageTypeTelegram = constant.MessageTypeTelegram
)
// UnifiedMessageContent 统一的消息内容结构
+35
View File
@@ -45,6 +45,7 @@ var (
constant.MessageTypeWeChatOFAccount: func() WayValidator { return &WeChatOFAccount{} },
constant.MessageTypeMessageNest: func() WayValidator { return &MessageNest{} },
constant.MessageTypeAliyunSMS: func() WayValidator { return &WayDetailAliyunSMS{} },
constant.MessageTypeTelegram: func() WayValidator { return &WayDetailTelegram{} },
}
testerRegistry = map[string]func(interface{}) WayTester{
constant.MessageTypeEmail: func(m interface{}) WayTester { return m.(*WayDetailEmail) },
@@ -55,6 +56,7 @@ var (
constant.MessageTypeWeChatOFAccount: func(m interface{}) WayTester { return m.(*WeChatOFAccount) },
constant.MessageTypeMessageNest: func(m interface{}) WayTester { return m.(*MessageNest) },
constant.MessageTypeAliyunSMS: func(m interface{}) WayTester { return m.(*WayDetailAliyunSMS) },
constant.MessageTypeTelegram: func(m interface{}) WayTester { return m.(*WayDetailTelegram) },
}
)
@@ -251,6 +253,39 @@ func (w *MessageNest) Test() (string, string) {
return "自托管消息不用测试运行,请直接添加", ""
}
// WayDetailTelegram Telegram机器人渠道明细字段
type WayDetailTelegram struct {
BotToken string `json:"bot_token" validate:"required,max=100" label:"Telegram Bot Token"`
ChatID string `json:"chat_id" validate:"required,max=50" label:"Chat ID"`
ApiHost string `json:"api_host" validate:"max=200" label:"自定义API地址"`
ProxyURL string `json:"proxy_url" validate:"max=200" label:"代理地址"`
}
func (w *WayDetailTelegram) Validate(authJson string) (string, interface{}) {
var empty interface{}
err := json.Unmarshal([]byte(authJson), w)
if err != nil {
return "Telegram参数反序列化失败!", empty
}
_, msg := app.CommonPlaygroundValid(*w)
return msg, w
}
func (w *WayDetailTelegram) Test() (string, string) {
testMsg := "This is a test message from message-nest."
var cli = message.Telegram{
BotToken: w.BotToken,
ChatID: w.ChatID,
ApiHost: w.ApiHost,
ProxyURL: w.ProxyURL,
}
res, err := cli.SendMessageText(testMsg)
if err != nil {
return fmt.Sprintf("发送失败:%s", err), string(res)
}
return "", string(res)
}
func (sw *SendWay) GetByID() (interface{}, error) {
return models.GetWayByID(sw.ID)
}