|
@@ -18,7 +18,7 @@ const (
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
var (
|
|
|
- telegramApiUrl = "https://api.telegram.org/bot%s/%s"
|
|
|
|
|
|
|
+ telegramAPIURL = "https://api.telegram.org/bot%s/%s"
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
func init() {
|
|
func init() {
|
|
@@ -52,6 +52,8 @@ func init() {
|
|
|
|
|
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
+// TelegramNotifier is responsible for sending
|
|
|
|
|
+// alert notifications to Telegram.
|
|
|
type TelegramNotifier struct {
|
|
type TelegramNotifier struct {
|
|
|
NotifierBase
|
|
NotifierBase
|
|
|
BotToken string
|
|
BotToken string
|
|
@@ -60,50 +62,51 @@ type TelegramNotifier struct {
|
|
|
log log.Logger
|
|
log log.Logger
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
+// NewTelegramNotifier is the constructor for the Telegram notifier
|
|
|
func NewTelegramNotifier(model *models.AlertNotification) (alerting.Notifier, error) {
|
|
func NewTelegramNotifier(model *models.AlertNotification) (alerting.Notifier, error) {
|
|
|
if model.Settings == nil {
|
|
if model.Settings == nil {
|
|
|
return nil, alerting.ValidationError{Reason: "No Settings Supplied"}
|
|
return nil, alerting.ValidationError{Reason: "No Settings Supplied"}
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
botToken := model.Settings.Get("bottoken").MustString()
|
|
botToken := model.Settings.Get("bottoken").MustString()
|
|
|
- chatId := model.Settings.Get("chatid").MustString()
|
|
|
|
|
|
|
+ chatID := model.Settings.Get("chatid").MustString()
|
|
|
uploadImage := model.Settings.Get("uploadImage").MustBool()
|
|
uploadImage := model.Settings.Get("uploadImage").MustBool()
|
|
|
|
|
|
|
|
if botToken == "" {
|
|
if botToken == "" {
|
|
|
return nil, alerting.ValidationError{Reason: "Could not find Bot Token in settings"}
|
|
return nil, alerting.ValidationError{Reason: "Could not find Bot Token in settings"}
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- if chatId == "" {
|
|
|
|
|
|
|
+ if chatID == "" {
|
|
|
return nil, alerting.ValidationError{Reason: "Could not find Chat Id in settings"}
|
|
return nil, alerting.ValidationError{Reason: "Could not find Chat Id in settings"}
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
return &TelegramNotifier{
|
|
return &TelegramNotifier{
|
|
|
NotifierBase: NewNotifierBase(model),
|
|
NotifierBase: NewNotifierBase(model),
|
|
|
BotToken: botToken,
|
|
BotToken: botToken,
|
|
|
- ChatID: chatId,
|
|
|
|
|
|
|
+ ChatID: chatID,
|
|
|
UploadImage: uploadImage,
|
|
UploadImage: uploadImage,
|
|
|
log: log.New("alerting.notifier.telegram"),
|
|
log: log.New("alerting.notifier.telegram"),
|
|
|
}, nil
|
|
}, nil
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func (this *TelegramNotifier) buildMessage(evalContext *alerting.EvalContext, sendImageInline bool) *models.SendWebhookSync {
|
|
|
|
|
|
|
+func (tn *TelegramNotifier) buildMessage(evalContext *alerting.EvalContext, sendImageInline bool) *models.SendWebhookSync {
|
|
|
if sendImageInline {
|
|
if sendImageInline {
|
|
|
- cmd, err := this.buildMessageInlineImage(evalContext)
|
|
|
|
|
|
|
+ cmd, err := tn.buildMessageInlineImage(evalContext)
|
|
|
if err == nil {
|
|
if err == nil {
|
|
|
return cmd
|
|
return cmd
|
|
|
}
|
|
}
|
|
|
- this.log.Error("Could not generate Telegram message with inline image.", "err", err)
|
|
|
|
|
|
|
+ tn.log.Error("Could not generate Telegram message with inline image.", "err", err)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- return this.buildMessageLinkedImage(evalContext)
|
|
|
|
|
|
|
+ return tn.buildMessageLinkedImage(evalContext)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func (this *TelegramNotifier) buildMessageLinkedImage(evalContext *alerting.EvalContext) *models.SendWebhookSync {
|
|
|
|
|
|
|
+func (tn *TelegramNotifier) buildMessageLinkedImage(evalContext *alerting.EvalContext) *models.SendWebhookSync {
|
|
|
message := fmt.Sprintf("<b>%s</b>\nState: %s\nMessage: %s\n", evalContext.GetNotificationTitle(), evalContext.Rule.Name, evalContext.Rule.Message)
|
|
message := fmt.Sprintf("<b>%s</b>\nState: %s\nMessage: %s\n", evalContext.GetNotificationTitle(), evalContext.Rule.Name, evalContext.Rule.Message)
|
|
|
|
|
|
|
|
- ruleUrl, err := evalContext.GetRuleUrl()
|
|
|
|
|
|
|
+ ruleURL, err := evalContext.GetRuleUrl()
|
|
|
if err == nil {
|
|
if err == nil {
|
|
|
- message = message + fmt.Sprintf("URL: %s\n", ruleUrl)
|
|
|
|
|
|
|
+ message = message + fmt.Sprintf("URL: %s\n", ruleURL)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
if evalContext.ImagePublicUrl != "" {
|
|
if evalContext.ImagePublicUrl != "" {
|
|
@@ -115,14 +118,14 @@ func (this *TelegramNotifier) buildMessageLinkedImage(evalContext *alerting.Eval
|
|
|
message = message + fmt.Sprintf("\n<i>Metrics:</i>%s", metrics)
|
|
message = message + fmt.Sprintf("\n<i>Metrics:</i>%s", metrics)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- cmd := this.generateTelegramCmd(message, "text", "sendMessage", func(w *multipart.Writer) {
|
|
|
|
|
|
|
+ cmd := tn.generateTelegramCmd(message, "text", "sendMessage", func(w *multipart.Writer) {
|
|
|
fw, _ := w.CreateFormField("parse_mode")
|
|
fw, _ := w.CreateFormField("parse_mode")
|
|
|
fw.Write([]byte("html"))
|
|
fw.Write([]byte("html"))
|
|
|
})
|
|
})
|
|
|
return cmd
|
|
return cmd
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func (this *TelegramNotifier) buildMessageInlineImage(evalContext *alerting.EvalContext) (*models.SendWebhookSync, error) {
|
|
|
|
|
|
|
+func (tn *TelegramNotifier) buildMessageInlineImage(evalContext *alerting.EvalContext) (*models.SendWebhookSync, error) {
|
|
|
var imageFile *os.File
|
|
var imageFile *os.File
|
|
|
var err error
|
|
var err error
|
|
|
|
|
|
|
@@ -130,7 +133,7 @@ func (this *TelegramNotifier) buildMessageInlineImage(evalContext *alerting.Eval
|
|
|
defer func() {
|
|
defer func() {
|
|
|
err := imageFile.Close()
|
|
err := imageFile.Close()
|
|
|
if err != nil {
|
|
if err != nil {
|
|
|
- this.log.Error("Could not close Telegram inline image.", "err", err)
|
|
|
|
|
|
|
+ tn.log.Error("Could not close Telegram inline image.", "err", err)
|
|
|
}
|
|
}
|
|
|
}()
|
|
}()
|
|
|
|
|
|
|
@@ -138,27 +141,27 @@ func (this *TelegramNotifier) buildMessageInlineImage(evalContext *alerting.Eval
|
|
|
return nil, err
|
|
return nil, err
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- ruleUrl, err := evalContext.GetRuleUrl()
|
|
|
|
|
|
|
+ ruleURL, err := evalContext.GetRuleUrl()
|
|
|
if err != nil {
|
|
if err != nil {
|
|
|
return nil, err
|
|
return nil, err
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
metrics := generateMetricsMessage(evalContext)
|
|
metrics := generateMetricsMessage(evalContext)
|
|
|
- message := generateImageCaption(evalContext, ruleUrl, metrics)
|
|
|
|
|
|
|
+ message := generateImageCaption(evalContext, ruleURL, metrics)
|
|
|
|
|
|
|
|
- cmd := this.generateTelegramCmd(message, "caption", "sendPhoto", func(w *multipart.Writer) {
|
|
|
|
|
|
|
+ cmd := tn.generateTelegramCmd(message, "caption", "sendPhoto", func(w *multipart.Writer) {
|
|
|
fw, _ := w.CreateFormFile("photo", evalContext.ImageOnDiskPath)
|
|
fw, _ := w.CreateFormFile("photo", evalContext.ImageOnDiskPath)
|
|
|
io.Copy(fw, imageFile)
|
|
io.Copy(fw, imageFile)
|
|
|
})
|
|
})
|
|
|
return cmd, nil
|
|
return cmd, nil
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func (this *TelegramNotifier) generateTelegramCmd(message string, messageField string, apiAction string, extraConf func(writer *multipart.Writer)) *models.SendWebhookSync {
|
|
|
|
|
|
|
+func (tn *TelegramNotifier) generateTelegramCmd(message string, messageField string, apiAction string, extraConf func(writer *multipart.Writer)) *models.SendWebhookSync {
|
|
|
var body bytes.Buffer
|
|
var body bytes.Buffer
|
|
|
w := multipart.NewWriter(&body)
|
|
w := multipart.NewWriter(&body)
|
|
|
|
|
|
|
|
fw, _ := w.CreateFormField("chat_id")
|
|
fw, _ := w.CreateFormField("chat_id")
|
|
|
- fw.Write([]byte(this.ChatID))
|
|
|
|
|
|
|
+ fw.Write([]byte(tn.ChatID))
|
|
|
|
|
|
|
|
fw, _ = w.CreateFormField(messageField)
|
|
fw, _ = w.CreateFormField(messageField)
|
|
|
fw.Write([]byte(message))
|
|
fw.Write([]byte(message))
|
|
@@ -167,8 +170,8 @@ func (this *TelegramNotifier) generateTelegramCmd(message string, messageField s
|
|
|
|
|
|
|
|
w.Close()
|
|
w.Close()
|
|
|
|
|
|
|
|
- this.log.Info("Sending telegram notification", "chat_id", this.ChatID, "bot_token", this.BotToken, "apiAction", apiAction)
|
|
|
|
|
- url := fmt.Sprintf(telegramApiUrl, this.BotToken, apiAction)
|
|
|
|
|
|
|
+ tn.log.Info("Sending telegram notification", "chat_id", tn.ChatID, "bot_token", tn.BotToken, "apiAction", apiAction)
|
|
|
|
|
+ url := fmt.Sprintf(telegramAPIURL, tn.BotToken, apiAction)
|
|
|
|
|
|
|
|
cmd := &models.SendWebhookSync{
|
|
cmd := &models.SendWebhookSync{
|
|
|
Url: url,
|
|
Url: url,
|
|
@@ -193,7 +196,7 @@ func generateMetricsMessage(evalContext *alerting.EvalContext) string {
|
|
|
return metrics
|
|
return metrics
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func generateImageCaption(evalContext *alerting.EvalContext, ruleUrl string, metrics string) string {
|
|
|
|
|
|
|
+func generateImageCaption(evalContext *alerting.EvalContext, ruleURL string, metrics string) string {
|
|
|
message := evalContext.GetNotificationTitle()
|
|
message := evalContext.GetNotificationTitle()
|
|
|
|
|
|
|
|
if len(evalContext.Rule.Message) > 0 {
|
|
if len(evalContext.Rule.Message) > 0 {
|
|
@@ -205,8 +208,8 @@ func generateImageCaption(evalContext *alerting.EvalContext, ruleUrl string, met
|
|
|
|
|
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- if len(ruleUrl) > 0 {
|
|
|
|
|
- urlLine := fmt.Sprintf("\nURL: %s", ruleUrl)
|
|
|
|
|
|
|
+ if len(ruleURL) > 0 {
|
|
|
|
|
+ urlLine := fmt.Sprintf("\nURL: %s", ruleURL)
|
|
|
message = appendIfPossible(message, urlLine, captionLengthLimit)
|
|
message = appendIfPossible(message, urlLine, captionLengthLimit)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
@@ -226,16 +229,17 @@ func appendIfPossible(message string, extra string, sizeLimit int) string {
|
|
|
return message
|
|
return message
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-func (this *TelegramNotifier) Notify(evalContext *alerting.EvalContext) error {
|
|
|
|
|
|
|
+// Notify send an alert notification to Telegram.
|
|
|
|
|
+func (tn *TelegramNotifier) Notify(evalContext *alerting.EvalContext) error {
|
|
|
var cmd *models.SendWebhookSync
|
|
var cmd *models.SendWebhookSync
|
|
|
- if evalContext.ImagePublicUrl == "" && this.UploadImage {
|
|
|
|
|
- cmd = this.buildMessage(evalContext, true)
|
|
|
|
|
|
|
+ if evalContext.ImagePublicUrl == "" && tn.UploadImage {
|
|
|
|
|
+ cmd = tn.buildMessage(evalContext, true)
|
|
|
} else {
|
|
} else {
|
|
|
- cmd = this.buildMessage(evalContext, false)
|
|
|
|
|
|
|
+ cmd = tn.buildMessage(evalContext, false)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
if err := bus.DispatchCtx(evalContext.Ctx, cmd); err != nil {
|
|
if err := bus.DispatchCtx(evalContext.Ctx, cmd); err != nil {
|
|
|
- this.log.Error("Failed to send webhook", "error", err, "webhook", this.Name)
|
|
|
|
|
|
|
+ tn.log.Error("Failed to send webhook", "error", err, "webhook", tn.Name)
|
|
|
return err
|
|
return err
|
|
|
}
|
|
}
|
|
|
|
|
|