Skip to content

Commit

Permalink
Implement Telegram alert client
Browse files Browse the repository at this point in the history
  • Loading branch information
richardgreg committed Feb 20, 2024
1 parent a692c87 commit 3bb3585
Show file tree
Hide file tree
Showing 2 changed files with 106 additions and 1 deletion.
2 changes: 1 addition & 1 deletion internal/client/slack.go
Expand Up @@ -43,7 +43,7 @@ type slackClient struct {
// NewSlackClient ... Initializer
func NewSlackClient(cfg *SlackConfig, name string) SlackClient {
if cfg.URL == "" {
logging.NoContext().Warn("No Slack webhook URL not provided")
logging.NoContext().Warn("No Slack webhook URL provided")
}

return &slackClient{
Expand Down
105 changes: 105 additions & 0 deletions internal/client/telegram.go
@@ -0,0 +1,105 @@
package client

import (
"bytes"
"context"
"encoding/json"
"fmt"
"io"
"net/http"

"github.com/base-org/pessimism/internal/core"
"github.com/base-org/pessimism/internal/logging"
)

type TelegramClient interface {
AlertClient
}

type TelegramConfig struct {
Token string
ChatID string
}

type telegramClient struct {
token string
chatID string
client *http.Client
}

func NewTelegramClient(cfg *TelegramConfig) TelegramClient {
if cfg.Token == "" {
logging.NoContext().Warn("No Telegram token provided")
}

return &telegramClient{
token: cfg.Token,
chatID: cfg.ChatID,
client: &http.Client{},
}
}

type TelegramPayload struct {
ChatID string `json:"chat_id"`
Text string `json:"text"`
}

type TelegramAPIResponse struct {
Ok bool `json:"ok"`
Result json.RawMessage `json:"result"` // Might not be needed for basic response handling
Error string `json:"description"`
}

func (tr *TelegramAPIResponse) ToAlertResponse() *AlertAPIResponse {
if tr.Ok {
return &AlertAPIResponse{
Status: core.SuccessStatus,
Message: "Message sent successfully",
}
}
return &AlertAPIResponse{
Status: core.FailureStatus,
Message: tr.Error,
}
}

func (tc *telegramClient) PostEvent(ctx context.Context, data *AlertEventTrigger) (*AlertAPIResponse, error) {
payload := TelegramPayload{
ChatID: tc.chatID,
Text: data.Message,
}

payloadBytes, err := json.Marshal(payload)
if err != nil {
return nil, err
}

url := fmt.Sprintf("https://api.telegram.org/bot%s/sendMessage", tc.token)
req, err := http.NewRequestWithContext(ctx, "POST", url, bytes.NewReader(payloadBytes))

Check failure on line 78 in internal/client/telegram.go

View workflow job for this annotation

GitHub Actions / lint

"POST" can be replaced by http.MethodPost (usestdlibvars)
if err != nil {
return nil, err
}
req.Header.Set("Content-Type", "application/json")

resp, err := tc.client.Do(req)
if err != nil {
return nil, err
}
defer resp.Body.Close()

respBytes, err := io.ReadAll(resp.Body)
if err != nil {
return nil, err
}

var apiResp TelegramAPIResponse
if err = json.Unmarshal(respBytes, &apiResp); err != nil {
return nil, fmt.Errorf("could not unmarshal telegram response: %w", err)
}

return apiResp.ToAlertResponse(), nil
}

func (tc *telegramClient) GetName() string {
return "TelegramClient"
}

0 comments on commit 3bb3585

Please sign in to comment.