feat: add message push function call

This commit is contained in:
engigu
2026-03-04 21:45:43 +08:00
parent d4663cb492
commit 81b8d2a93d
53 changed files with 4161 additions and 33 deletions
@@ -0,0 +1,40 @@
package channels
type AliyunSMSChannel struct{ *BaseChannel }
func NewAliyunSMSChannel() Channel {
return &AliyunSMSChannel{NewBaseChannel(ChannelAliyunSMS, []string{FormatTypeText})}
}
func (c *AliyunSMSChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
accessKeyId := config.GetString("access_key_id")
accessKeySecret := config.GetString("access_key_secret")
signName := config.GetString("sign_name")
regionId := config.GetString("region_id")
phoneNumber := config.GetString("phone_number")
templateCode := config.GetString("template_code")
if accessKeyId == "" || accessKeySecret == "" || signName == "" {
return SendError("aliyun sms config missing: access_key_id, access_key_secret, sign_name are required"), nil
}
if phoneNumber == "" || templateCode == "" {
return SendError("aliyun sms config missing: phone_number, template_code are required"), nil
}
_, formattedContent := c.FormatContent(msg)
if regionId == "" {
regionId = "cn-hangzhou"
}
client, err := createAliyunSMSClient(accessKeyId, accessKeySecret, regionId)
if err != nil {
return SendError("创建阿里云短信客户端失败: %s", err.Error()), nil
}
result, err := sendAliyunSMS(client, phoneNumber, signName, templateCode, formattedContent, msg.Extra)
if err != nil {
return ErrorResult("", err), nil
}
return SuccessResult(result), nil
}
+34
View File
@@ -0,0 +1,34 @@
package channels
import "github.com/engigu/baihu-panel/internal/sdk/message"
type BarkChannel struct{ *BaseChannel }
func NewBarkChannel() Channel {
return &BarkChannel{NewBaseChannel(ChannelBark, []string{FormatTypeText})}
}
func (c *BarkChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
pushKey := config.GetString("push_key")
if pushKey == "" {
return SendError("bark config missing: push_key is required"), nil
}
cli := message.Bark{
PushKey: pushKey,
Archive: config.GetString("archive"),
Group: config.GetString("group"),
Sound: config.GetString("sound"),
Icon: config.GetString("icon"),
Level: config.GetString("level"),
URL: config.GetString("url"),
Key: config.GetString("key"),
IV: config.GetString("iv"),
}
res, err := cli.Request(msg.Title, msg.Text)
if err != nil {
return ErrorResult(string(res), err), nil
}
return SuccessResult(string(res)), nil
}
@@ -0,0 +1,74 @@
package channels
import "fmt"
// Channel 渠道接口 - SDK 版本,零业务依赖
type Channel interface {
// GetType 返回渠道类型标识
GetType() string
// GetSupportedFormats 返回支持的消息格式
GetSupportedFormats() []string
// Send 发送消息
Send(config ChannelConfig, msg *Message) (*Result, error)
}
// BaseChannel 渠道基础实现
type BaseChannel struct {
channelType string
supportedFormats []string
}
func NewBaseChannel(channelType string, supportedFormats []string) *BaseChannel {
return &BaseChannel{channelType: channelType, supportedFormats: supportedFormats}
}
func (c *BaseChannel) GetType() string { return c.channelType }
func (c *BaseChannel) GetSupportedFormats() []string { return c.supportedFormats }
// FormatContent 根据渠道支持的格式选择最佳内容
func (c *BaseChannel) FormatContent(msg *Message) (formatType string, content string) {
for _, ft := range c.supportedFormats {
switch ft {
case FormatTypeMarkdown:
if msg.HasMarkdown() {
return FormatTypeMarkdown, msg.Markdown
}
case FormatTypeHTML:
if msg.HasHTML() {
return FormatTypeHTML, msg.HTML
}
case FormatTypeText:
if msg.HasText() {
return FormatTypeText, msg.Text
}
}
}
if msg.HasText() {
return FormatTypeText, msg.Text
}
return FormatTypeText, ""
}
// SuccessResult 创建成功结果
func SuccessResult(response string) *Result {
return &Result{Success: true, Response: response}
}
// ErrorResult 创建失败结果
func ErrorResult(response string, err error) *Result {
errMsg := ""
if err != nil {
errMsg = err.Error()
}
return &Result{Success: false, Response: response, Error: errMsg}
}
// ErrorResultStr 创建失败结果(字符串错误)
func ErrorResultStr(response string, errMsg string) *Result {
return &Result{Success: false, Response: response, Error: errMsg}
}
// SendError 发送失败时的格式化错误
func SendError(format string, args ...any) *Result {
return &Result{Success: false, Error: fmt.Sprintf(format, args...)}
}
+35
View File
@@ -0,0 +1,35 @@
package channels
import "github.com/engigu/baihu-panel/internal/sdk/message"
type CustomChannel struct{ *BaseChannel }
func NewCustomChannel() Channel {
return &CustomChannel{NewBaseChannel(ChannelCustom, []string{FormatTypeText})}
}
func (c *CustomChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
webhook := config.GetString("webhook")
body := config.GetString("body")
if webhook == "" {
return SendError("custom config missing: webhook is required"), nil
}
_, formattedContent := c.FormatContent(msg)
cli := message.CustomWebhook{}
// 替换 body 模板中的 TEXT 占位符
bodyStr := body
if bodyStr != "" {
bodyStr = replaceBodyPlaceholder(bodyStr, formattedContent)
} else {
bodyStr = formattedContent
}
res, err := cli.Request(webhook, bodyStr)
if err != nil {
return ErrorResult(string(res), err), nil
}
return SuccessResult(string(res)), nil
}
+41
View File
@@ -0,0 +1,41 @@
package channels
import "github.com/engigu/baihu-panel/internal/sdk/message"
type DtalkChannel struct{ *BaseChannel }
func NewDtalkChannel() Channel {
return &DtalkChannel{NewBaseChannel(ChannelDtalk, []string{FormatTypeMarkdown, FormatTypeText})}
}
func (c *DtalkChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
accessToken := config.GetString("access_token")
secret := config.GetString("secret")
if accessToken == "" {
return SendError("dtalk config missing: access_token is required"), nil
}
contentType, formattedContent := c.FormatContent(msg)
atMobiles := msg.GetAtMobiles()
if msg.AtAll {
atMobiles = append(atMobiles, "all")
}
cli := message.Dtalk{AccessToken: accessToken, Secret: secret}
var res []byte
var err error
if contentType == FormatTypeText {
res, err = cli.SendMessageText(formattedContent, atMobiles...)
} else if contentType == FormatTypeMarkdown {
res, err = cli.SendMessageMarkdown(msg.Title, formattedContent, atMobiles...)
} else {
return SendError("未知的钉钉发送内容类型:%s", contentType), nil
}
if err != nil {
return ErrorResult(string(res), err), nil
}
return SuccessResult(string(res)), nil
}
+49
View File
@@ -0,0 +1,49 @@
package channels
import (
"fmt"
"github.com/engigu/baihu-panel/internal/sdk/message"
"strconv"
)
type EmailChannel struct{ *BaseChannel }
func NewEmailChannel() Channel {
return &EmailChannel{NewBaseChannel(ChannelEmail, []string{FormatTypeHTML, FormatTypeText})}
}
func (c *EmailChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
server := config.GetString("server")
portStr := config.GetString("port")
account := config.GetString("account")
passwd := config.GetString("passwd")
fromName := config.GetString("from_name")
toAccount := config.GetString("to_account")
if server == "" || account == "" || passwd == "" {
return SendError("email config missing: server, account, passwd are required"), nil
}
if toAccount == "" {
return SendError("email config missing: to_account is required"), nil
}
port, _ := strconv.Atoi(portStr)
contentType, formattedContent := c.FormatContent(msg)
var emailer message.EmailMessage
emailer.Init(server, port, account, passwd, fromName)
var errMsg string
if contentType == FormatTypeText {
errMsg = emailer.SendTextMessage(toAccount, msg.Title, formattedContent)
} else if contentType == FormatTypeHTML {
errMsg = emailer.SendHtmlMessage(toAccount, msg.Title, formattedContent)
} else {
errMsg = fmt.Sprintf("未知的邮件发送内容类型:%s", contentType)
}
if errMsg != "" {
return ErrorResultStr("", errMsg), nil
}
return SuccessResult(""), nil
}
+43
View File
@@ -0,0 +1,43 @@
package channels
import "github.com/engigu/baihu-panel/internal/sdk/message"
type FeishuChannel struct{ *BaseChannel }
func NewFeishuChannel() Channel {
return &FeishuChannel{NewBaseChannel(ChannelFeishu, []string{FormatTypeMarkdown, FormatTypeText})}
}
func (c *FeishuChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
accessToken := config.GetString("access_token")
secret := config.GetString("secret")
if accessToken == "" {
return SendError("feishu config missing: access_token is required"), nil
}
contentType, formattedContent := c.FormatContent(msg)
atMobiles := msg.GetAtMobiles()
atUserIds := msg.GetAtUserIds()
atList := append(atMobiles, atUserIds...)
if msg.AtAll {
atList = append(atList, "all")
}
cli := message.Feishu{AccessToken: accessToken, Secret: secret}
var res []byte
var err error
if contentType == FormatTypeText {
res, err = cli.SendMessageText(formattedContent, atList...)
} else if contentType == FormatTypeMarkdown {
res, err = cli.SendMessageMarkdown(msg.Title, formattedContent, atList...)
} else {
return SendError("未知的飞书发送内容类型:%s", contentType), nil
}
if err != nil {
return ErrorResult(string(res), err), nil
}
return SuccessResult(string(res)), nil
}
+34
View File
@@ -0,0 +1,34 @@
package channels
import (
"github.com/engigu/baihu-panel/internal/sdk/message"
"strconv"
)
type GotifyChannel struct{ *BaseChannel }
func NewGotifyChannel() Channel {
return &GotifyChannel{NewBaseChannel(ChannelGotify, []string{FormatTypeText})}
}
func (c *GotifyChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
url := config.GetString("url")
token := config.GetString("token")
if url == "" || token == "" {
return SendError("gotify config missing: url and token are required"), nil
}
priority, _ := strconv.Atoi(config.GetString("priority"))
cli := message.Gotify{
Url: url,
Token: token,
Priority: priority,
}
res, err := cli.Request(msg.Title, msg.Text)
if err != nil {
return ErrorResult(string(res), err), nil
}
return SuccessResult(string(res)), nil
}
@@ -0,0 +1,52 @@
package channels
import (
"encoding/json"
"fmt"
"strings"
"github.com/aliyun/alibaba-cloud-sdk-go/services/dysmsapi"
)
// replaceBodyPlaceholder 替换自定义 webhook body 中的 TEXT 占位符
func replaceBodyPlaceholder(body string, content string) string {
data, _ := json.Marshal(content)
dataStr := strings.Trim(string(data), "\"")
return strings.Replace(body, "TEXT", dataStr, -1)
}
// createAliyunSMSClient 创建阿里云短信客户端
func createAliyunSMSClient(accessKeyId, accessKeySecret, regionId string) (*dysmsapi.Client, error) {
return dysmsapi.NewClientWithAccessKey(regionId, accessKeyId, accessKeySecret)
}
// sendAliyunSMS 发送短信
func sendAliyunSMS(client *dysmsapi.Client, phoneNumber, signName, templateCode, content string, extra map[string]any) (string, error) {
templateParam := map[string]interface{}{
"content": content,
}
if extra != nil {
for k, v := range extra {
templateParam[k] = v
}
}
templateParamJSON, _ := json.Marshal(templateParam)
request := dysmsapi.CreateSendSmsRequest()
request.Scheme = "https"
request.PhoneNumbers = phoneNumber
request.SignName = signName
request.TemplateCode = templateCode
request.TemplateParam = string(templateParamJSON)
response, err := client.SendSms(request)
if err != nil {
return "", fmt.Errorf("发送短信失败: %s", err.Error())
}
if response.Code != "OK" {
return "", fmt.Errorf("发送失败: %s - %s", response.Code, response.Message)
}
return fmt.Sprintf("RequestId: %s, BizId: %s", response.RequestId, response.BizId), nil
}
+33
View File
@@ -0,0 +1,33 @@
package channels
import "github.com/engigu/baihu-panel/internal/sdk/message"
type NtfyChannel struct{ *BaseChannel }
func NewNtfyChannel() Channel {
return &NtfyChannel{NewBaseChannel(ChannelNtfy, []string{FormatTypeText})}
}
func (c *NtfyChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
topic := config.GetString("topic")
if topic == "" {
return SendError("ntfy config missing: topic is required"), nil
}
cli := message.Ntfy{
Url: config.GetString("url"),
Topic: topic,
Priority: config.GetString("priority"),
Icon: config.GetString("icon"),
Token: config.GetString("token"),
Username: config.GetString("username"),
Password: config.GetString("password"),
Actions: config.GetString("actions"),
}
res, err := cli.Request(msg.Title, msg.Text)
if err != nil {
return ErrorResult(string(res), err), nil
}
return SuccessResult(string(res)), nil
}
+29
View File
@@ -0,0 +1,29 @@
package channels
import "github.com/engigu/baihu-panel/internal/sdk/message"
type PushMeChannel struct{ *BaseChannel }
func NewPushMeChannel() Channel {
return &PushMeChannel{NewBaseChannel(ChannelPushMe, []string{FormatTypeText})}
}
func (c *PushMeChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
pushKey := config.GetString("push_key")
if pushKey == "" {
return SendError("pushme config missing: push_key is required"), nil
}
cli := message.PushMe{
PushKey: pushKey,
URL: config.GetString("url"),
Date: config.GetString("date"),
Type: config.GetString("type"),
}
res, err := cli.Request(msg.Title, msg.Text)
if err != nil {
return ErrorResult(res, err), nil
}
return SuccessResult(res), nil
}
@@ -0,0 +1,42 @@
package channels
import "github.com/engigu/baihu-panel/internal/sdk/message"
type QyWeiXinChannel struct{ *BaseChannel }
func NewQyWeiXinChannel() Channel {
return &QyWeiXinChannel{NewBaseChannel(ChannelQyWeiXin, []string{FormatTypeMarkdown, FormatTypeText})}
}
func (c *QyWeiXinChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
accessToken := config.GetString("access_token")
if accessToken == "" {
return SendError("qyweixin config missing: access_token is required"), nil
}
contentType, formattedContent := c.FormatContent(msg)
atList := []string{}
atList = append(atList, msg.GetAtUserIds()...)
atList = append(atList, msg.GetAtMobiles()...)
if msg.AtAll {
atList = append(atList, "@all")
}
cli := message.QyWeiXin{AccessToken: accessToken}
var res []byte
var err error
if contentType == FormatTypeText {
res, err = cli.SendMessageText(formattedContent, atList...)
} else if contentType == FormatTypeMarkdown {
res, err = cli.SendMessageMarkdown(msg.Title, formattedContent, atList...)
} else {
return SendError("未知的企业微信发送内容类型:%s", contentType), nil
}
if err != nil {
return ErrorResult(string(res), err), nil
}
return SuccessResult(string(res)), nil
}
@@ -0,0 +1,47 @@
package channels
import "github.com/engigu/baihu-panel/internal/sdk/message"
type TelegramChannel struct{ *BaseChannel }
func NewTelegramChannel() Channel {
return &TelegramChannel{NewBaseChannel(ChannelTelegram, []string{FormatTypeMarkdown, FormatTypeHTML, FormatTypeText})}
}
func (c *TelegramChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
botToken := config.GetString("bot_token")
chatID := config.GetString("chat_id")
apiHost := config.GetString("api_host")
proxyURL := config.GetString("proxy_url")
if botToken == "" || chatID == "" {
return SendError("telegram config missing: bot_token, chat_id are required"), nil
}
contentType, formattedContent := c.FormatContent(msg)
cli := message.Telegram{
BotToken: botToken,
ChatID: chatID,
ApiHost: apiHost,
ProxyURL: proxyURL,
}
var res []byte
var err error
switch contentType {
case FormatTypeText:
res, err = cli.SendMessageText(formattedContent)
case FormatTypeMarkdown:
res, err = cli.SendMessageMarkdown(formattedContent)
case FormatTypeHTML:
res, err = cli.SendMessageHTML(formattedContent)
default:
return SendError("未知的Telegram发送内容类型:%s", contentType), nil
}
if err != nil {
return ErrorResult(string(res), err), nil
}
return SuccessResult(string(res)), nil
}
+75
View File
@@ -0,0 +1,75 @@
package channels
// Message 统一消息内容
type Message struct {
Title string `json:"title"`
Text string `json:"text"`
HTML string `json:"html"`
Markdown string `json:"markdown"`
URL string `json:"url"`
ImageURL string `json:"image_url"`
Summary string `json:"summary"`
AtMobiles []string `json:"at_mobiles"`
AtUserIds []string `json:"at_user_ids"`
AtAll bool `json:"at_all"`
Extra map[string]any `json:"extra"`
}
func (m *Message) HasText() bool { return m.Text != "" }
func (m *Message) HasHTML() bool { return m.HTML != "" }
func (m *Message) HasMarkdown() bool { return m.Markdown != "" }
func (m *Message) GetAtMobiles() []string {
if m.AtMobiles == nil {
return []string{}
}
return m.AtMobiles
}
func (m *Message) GetAtUserIds() []string {
if m.AtUserIds == nil {
return []string{}
}
return m.AtUserIds
}
// ChannelConfig 渠道认证配置(Key-Value 形式,各渠道自行定义字段)
type ChannelConfig map[string]string
// GetString 安全获取配置值
func (c ChannelConfig) GetString(key string) string {
if v, ok := c[key]; ok {
return v
}
return ""
}
// Result 发送结果
type Result struct {
Success bool `json:"success"`
Response string `json:"response"` // 原始响应
Error string `json:"error"` // 错误信息
}
// 消息格式类型常量
const (
FormatTypeText = "text"
FormatTypeHTML = "html"
FormatTypeMarkdown = "markdown"
)
// 渠道类型常量
const (
ChannelEmail = "Email"
ChannelDtalk = "Dtalk"
ChannelQyWeiXin = "QyWeiXin"
ChannelFeishu = "Feishu"
ChannelCustom = "Custom"
ChannelWeChatOFAccount = "WeChatOFAccount"
ChannelAliyunSMS = "AliyunSMS"
ChannelTelegram = "Telegram"
ChannelBark = "Bark"
ChannelPushMe = "PushMe"
ChannelNtfy = "Ntfy"
ChannelGotify = "Gotify"
)
+38
View File
@@ -0,0 +1,38 @@
package channels
import "github.com/engigu/baihu-panel/internal/sdk/message"
type WeChatOFAccountChannel struct{ *BaseChannel }
func NewWeChatOFAccountChannel() Channel {
return &WeChatOFAccountChannel{NewBaseChannel(ChannelWeChatOFAccount, []string{FormatTypeText})}
}
func (c *WeChatOFAccountChannel) Send(config ChannelConfig, msg *Message) (*Result, error) {
appID := config.GetString("appID")
appSecret := config.GetString("appsecret")
tempID := config.GetString("tempid")
toAccount := config.GetString("to_account")
if appID == "" || appSecret == "" {
return SendError("wechat config missing: appID, appsecret are required"), nil
}
if toAccount == "" {
return SendError("wechat config missing: to_account is required"), nil
}
_, formattedContent := c.FormatContent(msg)
cli := message.WeChatOFAccount{
AppID: appID,
AppSecret: appSecret,
TemplateID: tempID,
ToUser: toAccount,
URL: msg.URL,
}
res, err := cli.Send(msg.Title, formattedContent)
if err != nil {
return ErrorResult(res, err), nil
}
return SuccessResult(res), nil
}