opensaps/pushers/telegram/telegramconnection.go

116 lines
3.2 KiB
Go
Raw Normal View History

2017-10-31 22:50:14 +05:00
// OpenSAPS - Open Slack API server for everyone.
//
// Copyright (c) 2017, Stanislav N. aka pztrn.
// All rights reserved.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
package telegrampusher
import (
"fmt"
"net/http"
"net/url"
"strings"
configstruct "go.dev.pztrn.name/opensaps/config/struct"
slackmessage "go.dev.pztrn.name/opensaps/slack/message"
2017-10-31 22:50:14 +05:00
)
type TelegramConnection struct {
config configstruct.ConfigTelegram
2017-10-31 22:50:14 +05:00
}
func (tc *TelegramConnection) Initialize(connName string, cfg configstruct.ConfigTelegram) {
tc.config = cfg
2017-10-31 22:50:14 +05:00
}
func (tc *TelegramConnection) ProcessMessage(message slackmessage.SlackMessage) {
// Prepare message body.
2022-06-29 13:16:43 +05:00
messageData := ctx.SendToParser(message.Username, message)
messageToSend, _ := messageData["message"].(string)
// We'll use HTML, so reformat links accordingly (if any).
linksRaw, linksFound := messageData["links"]
if linksFound {
links, _ := linksRaw.([][]string)
for _, link := range links {
messageToSend = strings.ReplaceAll(messageToSend, link[0], `<a href="`+link[1]+`">`+link[2]+`</a>`)
2017-10-31 22:50:14 +05:00
}
}
2022-06-29 13:16:43 +05:00
ctx.Log.Debug().Msgf("Crafted message: %s", messageToSend)
2017-10-31 22:50:14 +05:00
// Send message.
tc.SendMessage(messageToSend)
2017-10-31 22:50:14 +05:00
}
func (tc *TelegramConnection) SendMessage(message string) {
msgdata := url.Values{}
msgdata.Set("chat_id", tc.config.ChatID)
2017-10-31 22:50:14 +05:00
msgdata.Set("text", message)
msgdata.Set("parse_mode", "HTML")
// Are we should use proxy?
2022-06-29 13:16:43 +05:00
// nolint:exhaustruct
httpTransport := &http.Transport{}
// nolint:nestif
if tc.config.Proxy.Enabled {
// Compose proxy URL.
proxyURL := "http://"
if tc.config.Proxy.User != "" {
proxyURL += tc.config.Proxy.User
if tc.config.Proxy.Password != "" {
proxyURL += ":" + tc.config.Proxy.Password
}
proxyURL += "@"
}
proxyURL += tc.config.Proxy.Address
proxyURLParsed, err := url.Parse(proxyURL)
if err != nil {
2022-06-29 13:16:43 +05:00
ctx.Log.Error().Err(err).Msg("Error while constructing/parsing proxy URL")
} else {
httpTransport.Proxy = http.ProxyURL(proxyURLParsed)
}
}
2022-06-29 13:16:43 +05:00
// nolint:exhaustruct
client := &http.Client{Transport: httpTransport}
botURL := fmt.Sprintf("https://api.telegram.org/bot%s/sendMessage", tc.config.BotID)
2022-06-29 13:16:43 +05:00
ctx.Log.Debug().Msgf("Bot URL: %s", botURL)
// ToDo: fix it.
// nolint
response, err := client.PostForm(botURL, msgdata)
if err != nil {
2022-06-29 13:16:43 +05:00
ctx.Log.Error().Err(err).Msg("Error occurred while sending data to Telegram")
} else {
2022-06-29 13:16:43 +05:00
ctx.Log.Debug().Msgf("Status: %s", response.Status)
if response.StatusCode != http.StatusOK {
body := []byte{}
_, _ = response.Body.Read(body)
response.Body.Close()
2022-06-29 13:16:43 +05:00
ctx.Log.Debug().Msg(string(body))
}
}
2017-10-31 22:50:14 +05:00
}
func (tc *TelegramConnection) Shutdown() {
// There is nothing we can do actually.
}