|
@@ -0,0 +1,86 @@
|
|
|
|
|
+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
|
|
|
|
|
+}
|