feat(drivers): 饭碗警告(WIP)
This commit is contained in:
@@ -1,4 +1,5 @@
|
|||||||
# goodBaby v2
|
# goodBaby v2
|
||||||
|
|
||||||

|

|
||||||

|

|
||||||

|

|
||||||
|
|||||||
@@ -9,8 +9,6 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
// DingTalkExecutor 钉钉机器人执行器
|
// DingTalkExecutor 钉钉机器人执行器
|
||||||
//
|
|
||||||
// 之前只注册了校验器却没有执行器,导致钉钉规则可以创建但永远不会被执行。
|
|
||||||
type DingTalkExecutor struct{}
|
type DingTalkExecutor struct{}
|
||||||
|
|
||||||
func (e *DingTalkExecutor) GetType() string {
|
func (e *DingTalkExecutor) GetType() string {
|
||||||
|
|||||||
@@ -0,0 +1,21 @@
|
|||||||
|
package fwalert
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"github.com/go-resty/resty/v2"
|
||||||
|
)
|
||||||
|
|
||||||
|
// send 发送消息
|
||||||
|
func send(config *FwAlertRuleConfig) error {
|
||||||
|
restyClient := resty.New()
|
||||||
|
req := restyClient.R()
|
||||||
|
resp, err := req.Execute("GET", config.WebhookURL+"?message="+config.Msg)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if resp.StatusCode() != 200 {
|
||||||
|
return fmt.Errorf("发送饭碗警告失败: %s", resp.Status())
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,8 @@
|
|||||||
|
package fwalert
|
||||||
|
|
||||||
|
const RuleTypeFwalert = "fwalert"
|
||||||
|
|
||||||
|
type FwAlertRuleConfig struct {
|
||||||
|
WebhookURL string `json:"webhook_url"`
|
||||||
|
Msg string `json:"msg"`
|
||||||
|
}
|
||||||
@@ -0,0 +1,62 @@
|
|||||||
|
package fwalert
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/url"
|
||||||
|
|
||||||
|
"github.com/ssdomei232/goodBaby/internal/meta"
|
||||||
|
)
|
||||||
|
|
||||||
|
type FwalertRuleValidator struct{}
|
||||||
|
|
||||||
|
func (v *FwalertRuleValidator) GetType() string {
|
||||||
|
return RuleTypeFwalert
|
||||||
|
}
|
||||||
|
|
||||||
|
func (v *FwalertRuleValidator) Validate(configJSON string) error {
|
||||||
|
var config FwAlertRuleConfig
|
||||||
|
if err := json.Unmarshal([]byte(configJSON), &config); err != nil {
|
||||||
|
return fmt.Errorf("解析饭碗警告规则配置失败: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
parsed, err := url.Parse(config.WebhookURL)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("饭碗警告规则配置中 webhook_url 格式不正确: %v", err)
|
||||||
|
}
|
||||||
|
if parsed.Scheme != "http" && parsed.Scheme != "https" {
|
||||||
|
return fmt.Errorf("饭碗警告规则配置中 webhook_url 必须是 HTTP 或 HTTPS URL")
|
||||||
|
}
|
||||||
|
|
||||||
|
if config.Msg == "" {
|
||||||
|
return fmt.Errorf("饭碗警告规则配置中 msg 不能为空")
|
||||||
|
}
|
||||||
|
if len([]rune(config.Msg)) > 1000 {
|
||||||
|
return fmt.Errorf("饭碗警告内容过长")
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (v *FwalertRuleValidator) Meta() meta.RuleMeta {
|
||||||
|
return meta.RuleMeta{
|
||||||
|
Type: RuleTypeFwalert,
|
||||||
|
Label: "发送饭碗警告",
|
||||||
|
Description: "触发时发送饭碗警告消息。",
|
||||||
|
Docs: "docs/fwalert-config.md",
|
||||||
|
AccountType: "",
|
||||||
|
Fields: []meta.Field{
|
||||||
|
{Key: "webhook_url", Label: "Webhook URL", Type: meta.FieldString, Required: true, Help: "饭碗警告的触发地址", Placeholder: "https://fwalert.com/xxx-xxx-xxx-xxx-xxx"},
|
||||||
|
{Key: "msg", Label: "消息内容", Type: meta.FieldTextarea, Required: true, Help: "最大长度 1000 个字符"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParseRuleConfig 解析饭碗警告规则配置
|
||||||
|
func ParseRuleConfig(configJSON string) (*FwAlertRuleConfig, error) {
|
||||||
|
var config FwAlertRuleConfig
|
||||||
|
if err := json.Unmarshal([]byte(configJSON), &config); err != nil {
|
||||||
|
return nil, fmt.Errorf("解析饭碗警告规则配置失败: %v", err)
|
||||||
|
}
|
||||||
|
return &config, nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,29 @@
|
|||||||
|
package fwalert
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"log"
|
||||||
|
|
||||||
|
"github.com/ssdomei232/goodBaby/internal/retry"
|
||||||
|
"github.com/ssdomei232/goodBaby/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
// FwalertExecutor 饭碗警告执行器
|
||||||
|
type FwalertExecutor struct{}
|
||||||
|
|
||||||
|
func (e *FwalertExecutor) GetType() string {
|
||||||
|
return RuleTypeFwalert
|
||||||
|
}
|
||||||
|
|
||||||
|
func (e *FwalertExecutor) Execute(ctx context.Context, rule *model.Rule) error {
|
||||||
|
log.Printf("执行饭碗警告规则: %s (ID: %d)", rule.Name, rule.ID)
|
||||||
|
|
||||||
|
config, err := ParseRuleConfig(rule.ConfigJson)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return retry.Do(ctx, func() error {
|
||||||
|
return send(config)
|
||||||
|
})
|
||||||
|
}
|
||||||
@@ -9,6 +9,7 @@ import (
|
|||||||
"github.com/ssdomei232/goodBaby/drivers/bilibili"
|
"github.com/ssdomei232/goodBaby/drivers/bilibili"
|
||||||
"github.com/ssdomei232/goodBaby/drivers/dingtalk"
|
"github.com/ssdomei232/goodBaby/drivers/dingtalk"
|
||||||
"github.com/ssdomei232/goodBaby/drivers/email"
|
"github.com/ssdomei232/goodBaby/drivers/email"
|
||||||
|
"github.com/ssdomei232/goodBaby/drivers/fwalert"
|
||||||
"github.com/ssdomei232/goodBaby/drivers/github"
|
"github.com/ssdomei232/goodBaby/drivers/github"
|
||||||
"github.com/ssdomei232/goodBaby/drivers/onebot"
|
"github.com/ssdomei232/goodBaby/drivers/onebot"
|
||||||
"github.com/ssdomei232/goodBaby/model"
|
"github.com/ssdomei232/goodBaby/model"
|
||||||
@@ -76,6 +77,7 @@ func InitExecutorRegistry() *ExecutorRegistry {
|
|||||||
registry.Register(&github.GithubMakeRepoPublicExecutor{})
|
registry.Register(&github.GithubMakeRepoPublicExecutor{})
|
||||||
registry.Register(&onebot.OneBotExecutor{})
|
registry.Register(&onebot.OneBotExecutor{})
|
||||||
registry.Register(&dingtalk.DingTalkExecutor{})
|
registry.Register(&dingtalk.DingTalkExecutor{})
|
||||||
|
registry.Register(&fwalert.FwalertExecutor{})
|
||||||
// 未来添加新规则类型时,在这里注册即可
|
// 未来添加新规则类型时,在这里注册即可
|
||||||
|
|
||||||
globalExecutorRegistry = registry
|
globalExecutorRegistry = registry
|
||||||
|
|||||||
@@ -6,6 +6,7 @@ import (
|
|||||||
"github.com/ssdomei232/goodBaby/drivers/bilibili"
|
"github.com/ssdomei232/goodBaby/drivers/bilibili"
|
||||||
"github.com/ssdomei232/goodBaby/drivers/dingtalk"
|
"github.com/ssdomei232/goodBaby/drivers/dingtalk"
|
||||||
"github.com/ssdomei232/goodBaby/drivers/email"
|
"github.com/ssdomei232/goodBaby/drivers/email"
|
||||||
|
"github.com/ssdomei232/goodBaby/drivers/fwalert"
|
||||||
"github.com/ssdomei232/goodBaby/drivers/github"
|
"github.com/ssdomei232/goodBaby/drivers/github"
|
||||||
"github.com/ssdomei232/goodBaby/drivers/onebot"
|
"github.com/ssdomei232/goodBaby/drivers/onebot"
|
||||||
)
|
)
|
||||||
@@ -27,6 +28,7 @@ func InitValidatorRegistry() *ValidatorRegistry {
|
|||||||
r.Register(&onebot.OneBotRuleValidator{})
|
r.Register(&onebot.OneBotRuleValidator{})
|
||||||
r.Register(&dingtalk.DingTalkRuleValidator{})
|
r.Register(&dingtalk.DingTalkRuleValidator{})
|
||||||
r.Register(&github.GithubMakeRepositoryPublicRuleValidator{})
|
r.Register(&github.GithubMakeRepositoryPublicRuleValidator{})
|
||||||
|
r.Register(&fwalert.FwalertRuleValidator{})
|
||||||
|
|
||||||
registry = r
|
registry = r
|
||||||
})
|
})
|
||||||
|
|||||||
Reference in New Issue
Block a user