Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -160,7 +160,7 @@ data | дополнительные данные

Также могут быть **дополнительные** данные (пишутся в поле "data") в зависимости от используемого метода.

Для метода **vk**:
Для метода **vk** и **tg**:

Поле | Описание
------------ | -------------
Expand Down
2 changes: 2 additions & 0 deletions pkg/chat/chat_factory.go
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,8 @@ func NewChat(config *config.ChatConfig) (Chat, error) {
switch method {
case "vk":
cht = &Vk{}
case "tg":
cht = &Tg{}
default:
return nil, errUnknownMethod(method)
}
Expand Down
109 changes: 109 additions & 0 deletions pkg/chat/tg.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,109 @@
package chat

import (
"fmt"
"github.com/NolikTop/watcher/pkg/server"
"io"
"net/http"
"strconv"
)

type Tg struct {
name string

client *http.Client

chatId int
accessToken string
}

func (v *Tg) Init(name string, data map[string]interface{}) error {
v.name = name

if chatId, ok := data["chat_id"]; ok {
v.chatId = int(chatId.(float64)) // сразу в int не дает кастить =(((
} else {
return errNoFieldInData("chat_id")
}

if accessToken, ok := data["access_token"]; ok {
v.accessToken = accessToken.(string)
} else {
return errNoFieldInData("access_token")
}

v.client = &http.Client{}

return nil
}

func (v *Tg) GetName() string {
return v.name
}

func (v *Tg) NotifyServerWentDown(server server.Server, err error) error {
message := fmt.Sprintf(
`Сервер %s упал.
Причина: %s
Призываю %s`,
server.GetFormattedName(), err.Error(), server.GetMentionsText(),
)

return v.sendMessage(message)
}

func (v *Tg) NotifyServerStillIsDown(server server.Server) error {
message := fmt.Sprintf(
`Сервер %s все еще лежит. Прошло уже %d сек.
Призываю %s`,
server.GetFormattedName(), server.GetOffTime(), server.GetMentionsText(),
)

return v.sendMessage(message)
}

func (v *Tg) NotifyServerIsUp(server server.Server) error {
message := fmt.Sprintf(
`Сервер %s встал.
Призываю %s`,
server.GetFormattedName(), server.GetMentionsText(),
)

return v.sendMessage(message)
}

func (v *Tg) sendMessage(message string) error {
request, err := http.NewRequest("POST", fmt.Sprintf("https://api.telegram.org/bot%s/sendMessage", v.accessToken), nil)
if err != nil {
return err
}

query := request.URL.Query()
query.Add("chat_id", strconv.Itoa(v.chatId))
query.Add("text", message)
request.URL.RawQuery = query.Encode()

response, err := v.client.Do(request)
if err != nil {
return err
}

defer response.Body.Close()

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

err = v.getErrorFromResponse(body)
if err != nil {
return err
}

return nil
}

func (v *Tg) getErrorFromResponse(responseBody []byte) error {
//todo
Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

стоит сделать обработку ошибки, если оно упадет как понять что именно не так

return nil
}
8 changes: 4 additions & 4 deletions pkg/chat/vk.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ package chat
import (
"fmt"
"github.com/NolikTop/watcher/pkg/server"
"io/ioutil"
"io"
"net/http"
"strconv"
)
Expand Down Expand Up @@ -93,20 +93,20 @@ func (v *Vk) sendMessage(message string) error {

defer response.Body.Close()

body, err := ioutil.ReadAll(response.Body)
body, err := io.ReadAll(response.Body)
if err != nil {
return err
}

err = getErrorFromResponse(body)
err = v.getErrorFromResponse(body)
if err != nil {
return err
}

return nil
}

func getErrorFromResponse(responseBody []byte) error {
func (v *Vk) getErrorFromResponse(responseBody []byte) error {
//todo
return nil
}