dnscontrol/pkg/notifications/telegram.go
Amin Vakil 9622e5ba8d
Revert "Remove ParseMode"
This reverts commit 4d39e9b8bb.
2025-08-29 17:41:28 +03:30

74 lines
1.9 KiB
Go

package notifications
import (
"bytes"
"encoding/json"
"fmt"
"net/http"
"strconv"
"io"
)
func init() {
initers = append(initers, func(cfg map[string]string) Notifier {
if botToken, ok := cfg["telegram_bot_token"]; ok {
if chatID, ok := cfg["telegram_chat_id"]; ok {
notifier := &telegramNotifier{
BotToken: botToken,
ChatID: chatID,
}
return notifier
}
}
return nil
})
}
// telegramNotifier sends notifications to Telegram
type telegramNotifier struct {
BotToken string
ChatID string
}
func (s *telegramNotifier) Notify(domain, provider, msg string, err error, preview bool) error {
var payload struct {
ChatID int64 `json:"chat_id"`
Text string `json:"text"`
ParseMode string `json:"parse_mode"`
}
url := fmt.Sprintf("https://api.telegram.org/bot%s/sendMessage", s.BotToken)
payload.ChatID, _ = strconv.ParseInt(s.ChatID, 10, 64)
payload.ParseMode = "MarkdownV2"
if preview {
payload.Text = fmt.Sprintf(`DNSControl preview: %s[%s]:\n%s`, domain, provider, msg)
} else if err != nil {
payload.Text = fmt.Sprintf(`DNSControl ERROR running correction on %s[%s]:\n%s\nError: %s`, domain, provider, msg, err)
} else {
payload.Text = fmt.Sprintf(`DNSControl successfully ran correction for %s[%s]:\n%s`, domain, provider, msg)
}
// Debugging version
marshaledPayload, err := json.Marshal(payload)
if err != nil {
return fmt.Errorf("failed to marshal telegram payload: %w", err)
}
resp, posterr := http.Post(url, "application/json", bytes.NewBuffer(marshaledPayload))
if posterr != nil {
return posterr
}
defer resp.Body.Close()
// Check if Telegram returned an error (anything other than a 200 OK)
if resp.StatusCode != http.StatusOK {
body, _ := io.ReadAll(resp.Body) // You'll need to import "io"
return fmt.Errorf("telegram API error: %s", string(body))
}
return nil
}
func (s *telegramNotifier) Done() {}