mirror of
https://github.com/grafana/grafana.git
synced 2024-11-30 12:44:10 -06:00
87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
package notifiers
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/grafana/grafana/pkg/bus"
|
|
"github.com/grafana/grafana/pkg/components/simplejson"
|
|
"github.com/grafana/grafana/pkg/log"
|
|
"github.com/grafana/grafana/pkg/metrics"
|
|
m "github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/alerting"
|
|
)
|
|
|
|
var (
|
|
telegeramApiUrl string = "https://api.telegram.org/bot%s/%s"
|
|
)
|
|
|
|
func init() {
|
|
alerting.RegisterNotifier("telegram", NewTelegramNotifier)
|
|
}
|
|
|
|
type TelegramNotifier struct {
|
|
NotifierBase
|
|
BotToken string
|
|
ChatID string
|
|
log log.Logger
|
|
}
|
|
|
|
func NewTelegramNotifier(model *m.AlertNotification) (alerting.Notifier, error) {
|
|
if model.Settings == nil {
|
|
return nil, alerting.ValidationError{Reason: "No Settings Supplied"}
|
|
}
|
|
|
|
botToken := model.Settings.Get("bottoken").MustString()
|
|
chatId := model.Settings.Get("chatid").MustString()
|
|
|
|
if botToken == "" {
|
|
return nil, alerting.ValidationError{Reason: "Could not find Bot Token in settings"}
|
|
}
|
|
|
|
if chatId == "" {
|
|
return nil, alerting.ValidationError{Reason: "Could not find Chat Id in settings"}
|
|
}
|
|
|
|
return &TelegramNotifier{
|
|
NotifierBase: NewNotifierBase(model.Id, model.IsDefault, model.Name, model.Type, model.Settings),
|
|
BotToken: botToken,
|
|
ChatID: chatId,
|
|
log: log.New("alerting.notifier.telegram"),
|
|
}, nil
|
|
}
|
|
|
|
func (this *TelegramNotifier) Notify(evalContext *alerting.EvalContext) error {
|
|
this.log.Info("Sending alert notification to", "bot_token", this.BotToken)
|
|
this.log.Info("Sending alert notification to", "chat_id", this.ChatID)
|
|
metrics.M_Alerting_Notification_Sent_Telegram.Inc(1)
|
|
|
|
bodyJSON := simplejson.New()
|
|
|
|
bodyJSON.Set("chat_id", this.ChatID)
|
|
bodyJSON.Set("parse_mode", "html")
|
|
|
|
message := fmt.Sprintf("%s\nState: %s\nMessage: %s\n", evalContext.GetNotificationTitle(), evalContext.Rule.Name, evalContext.Rule.Message)
|
|
|
|
ruleUrl, err := evalContext.GetRuleUrl()
|
|
if err == nil {
|
|
message = message + fmt.Sprintf("URL: %s\n", ruleUrl)
|
|
}
|
|
bodyJSON.Set("text", message)
|
|
|
|
url := fmt.Sprintf(telegeramApiUrl, this.BotToken, "sendMessage")
|
|
body, _ := bodyJSON.MarshalJSON()
|
|
|
|
cmd := &m.SendWebhookSync{
|
|
Url: url,
|
|
Body: string(body),
|
|
HttpMethod: "POST",
|
|
}
|
|
|
|
if err := bus.DispatchCtx(evalContext.Ctx, cmd); err != nil {
|
|
this.log.Error("Failed to send webhook", "error", err, "webhook", this.Name)
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|