Add feishu notify method

prometheus · Dec 23, 2024 · b8ae83a · b8ae83a
1 parent 3b61ae8
commit b8ae83a
Show file tree

Hide file tree

Showing 8 changed files with 328 additions and 6 deletions.
diff --git a/asset/assets_vfsdata.go b/asset/assets_vfsdata.go
diff --git a/config/config.go b/config/config.go
@@ -272,6 +272,9 @@ func resolveFilepaths(baseDir string, cfg *Config) {
 		for _, cfg := range receiver.RocketchatConfigs {
 			cfg.HTTPConfig.SetDirectory(baseDir)
 		}
+		for _, cfg := range receiver.FeishuConfigs {
+			cfg.HTTPConfig.SetDirectory(baseDir)
+		}
 	}
 }
 
@@ -606,6 +609,29 @@ func (c *Config) UnmarshalYAML(unmarshal func(interface{}) error) error {
 				rocketchat.TokenFile = c.Global.RocketchatTokenFile
 			}
 		}
+		for _, feishu := range rcv.FeishuConfigs {
+			if feishu.HTTPConfig == nil {
+				feishu.HTTPConfig = c.Global.HTTPConfig
+			}
+			if feishu.APIURL == nil {
+				if c.Global.FeishuAPIURL == nil {
+					return errors.New("no global FeishuAPIURL set ")
+				}
+				feishu.APIURL = c.Global.FeishuAPIURL
+			}
+			if feishu.APPID == "" {
+				if c.Global.FeishuAPPID == "" {
+					return errors.New("no global FeishuAPPID set ")
+				}
+				feishu.APPID = c.Global.FeishuAPPID
+			}
+			if feishu.APPSecret == "" {
+				if c.Global.FeishuAPPSecret == "" {
+					return errors.New("no global FeishuAPPSecret set ")
+				}
+				feishu.APPSecret = c.Global.FeishuAPPSecret
+			}
+		}
 
 		names[rcv.Name] = struct{}{}
 	}
@@ -710,6 +736,7 @@ func DefaultGlobalConfig() GlobalConfig {
 		TelegramAPIUrl:   mustParseURL("https://api.telegram.org"),
 		WebexAPIURL:      mustParseURL("https://webexapis.com/v1/messages"),
 		RocketchatAPIURL: mustParseURL("https://open.rocket.chat/"),
+		FeishuAPIURL:     mustParseURL("https://open.feishu.cn/open-apis"),
 	}
 }
 
@@ -831,6 +858,9 @@ type GlobalConfig struct {
 	WeChatAPIURL          *URL                 `yaml:"wechat_api_url,omitempty" json:"wechat_api_url,omitempty"`
 	WeChatAPISecret       Secret               `yaml:"wechat_api_secret,omitempty" json:"wechat_api_secret,omitempty"`
 	WeChatAPICorpID       string               `yaml:"wechat_api_corp_id,omitempty" json:"wechat_api_corp_id,omitempty"`
+	FeishuAPIURL          *URL                 `yaml:"feishu_api_url,omitempty" json:"feishu_api_url,omitempty"`
+	FeishuAPPID           string               `yaml:"feishu_app_id,omitempty" json:"feishu_app_id,omitempty"`
+	FeishuAPPSecret       Secret               `yaml:"feishu_app_secret,omitempty" json:"feishu_app_secret,omitempty"`
 	VictorOpsAPIURL       *URL                 `yaml:"victorops_api_url,omitempty" json:"victorops_api_url,omitempty"`
 	VictorOpsAPIKey       Secret               `yaml:"victorops_api_key,omitempty" json:"victorops_api_key,omitempty"`
 	VictorOpsAPIKeyFile   string               `yaml:"victorops_api_key_file,omitempty" json:"victorops_api_key_file,omitempty"`
@@ -988,6 +1018,7 @@ type Receiver struct {
 	MSTeamsV2Configs  []*MSTeamsV2Config  `yaml:"msteamsv2_configs,omitempty" json:"msteamsv2_configs,omitempty"`
 	JiraConfigs       []*JiraConfig       `yaml:"jira_configs,omitempty" json:"jira_configs,omitempty"`
 	RocketchatConfigs []*RocketchatConfig `yaml:"rocketchat_configs,omitempty" json:"rocketchat_configs,omitempty"`
+	FeishuConfigs     []*FeishuConfig     `yaml:"feishu_configs,omitempty" json:"feishu_configs,omitempty"`
 }
 
 // UnmarshalYAML implements the yaml.Unmarshaler interface for Receiver.

diff --git a/config/notifiers.go b/config/notifiers.go
@@ -203,6 +203,15 @@ var (
 		Description: `{{ template "jira.default.description" . }}`,
 		Priority:    `{{ template "jira.default.priority" . }}`,
 	}
+
+	DefaultFeishuConfig = FeishuConfig{
+		NotifierConfig: NotifierConfig{
+			VSendResolved: true,
+		},
+		ToUser:  `{{ template "feishu.default.to_user" . }}`,
+		ToChat:  `{{ template "feishu.default.to_chat" . }}`,
+		Message: `{{ template "feishu.default.message" . }}`,
+	}
 )
 
 // NotifierConfig contains base options common across all notifier configurations.
@@ -597,6 +606,29 @@ func (c *WechatConfig) UnmarshalYAML(unmarshal func(interface{}) error) error {
 	return nil
 }
 
+// FeishuConfig configures notifications via Feishu.
+type FeishuConfig struct {
+	NotifierConfig `yaml:",inline" json:",inline"`
+
+	HTTPConfig *commoncfg.HTTPClientConfig `yaml:"http_config,omitempty" json:"http_config,omitempty"`
+
+	APIURL    *URL   `yaml:"api_url,omitempty" json:"api_url,omitempty"`
+	APPID     string `yaml:"app_id,omitempty" json:"app_id,omitempty"`
+	APPSecret Secret `yaml:"app_secret,omitempty" json:"app_secret,omitempty"`
+	ToUser    string `yaml:"to_user,omitempty" json:"to_user,omitempty"`
+	ToChat    string `yaml:"to_chat,omitempty" json:"to_chat,omitempty"`
+	Message   string `yaml:"message,omitempty" json:"message,omitempty"`
+}
+
+func (c *FeishuConfig) UnmarshalYAML(unmarshal func(interface{}) error) error {
+	*c = DefaultFeishuConfig
+	type plain FeishuConfig
+	if err := unmarshal((*plain)(c)); err != nil {
+		return err
+	}
+	return nil
+}
+
 // OpsGenieConfig configures notifications via OpsGenie.
 type OpsGenieConfig struct {
 	NotifierConfig `yaml:",inline" json:",inline"`

diff --git a/config/receiver/receiver.go b/config/receiver/receiver.go
@@ -23,6 +23,7 @@ import (
 	"github.com/prometheus/alertmanager/notify"
 	"github.com/prometheus/alertmanager/notify/discord"
 	"github.com/prometheus/alertmanager/notify/email"
+	"github.com/prometheus/alertmanager/notify/feishu"
 	"github.com/prometheus/alertmanager/notify/jira"
 	"github.com/prometheus/alertmanager/notify/msteams"
 	"github.com/prometheus/alertmanager/notify/msteamsv2"
@@ -109,7 +110,9 @@ func BuildReceiverIntegrations(nc config.Receiver, tmpl *template.Template, logg
 	for i, c := range nc.RocketchatConfigs {
 		add("rocketchat", i, c, func(l *slog.Logger) (notify.Notifier, error) { return rocketchat.New(c, tmpl, l, httpOpts...) })
 	}
-
+	for i, c := range nc.FeishuConfigs {
+		add("feishu", i, c, func(l *slog.Logger) (notify.Notifier, error) { return feishu.New(c, tmpl, l, httpOpts...) })
+	}
 	if errs.Len() > 0 {
 		return nil, &errs
 	}