feat: add tg bot send way
This commit is contained in:
@@ -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 统一的消息内容结构
|
||||
|
||||
@@ -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)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user