opensaps/pushers/telegram/telegrampusher.go

61 lines
1.8 KiB
Go
Raw Permalink 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
2017-10-31 22:50:14 +05:00
// 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 (
slackmessage "go.dev.pztrn.name/opensaps/slack/message"
2017-10-31 22:50:14 +05:00
)
type TelegramPusher struct{}
func (tp TelegramPusher) Initialize() {
2022-06-29 13:16:43 +05:00
ctx.Log.Info().Msg("Initializing Telegram protocol pusher...")
2017-10-31 22:50:14 +05:00
// Get configuration for pushers and initialize every connection.
2022-06-29 13:16:43 +05:00
cfg := ctx.Config.GetConfig()
2017-10-31 22:50:14 +05:00
for name, config := range cfg.Telegram {
2022-06-29 13:16:43 +05:00
ctx.Log.Info().Str("conn", name).Msg("Initializing connection...")
2022-06-29 13:16:43 +05:00
// nolint:exhaustruct
2017-10-31 22:50:14 +05:00
conn := TelegramConnection{}
connections[name] = &conn
go conn.Initialize(name, config)
2017-10-31 22:50:14 +05:00
}
}
func (tp TelegramPusher) Push(connection string, data slackmessage.SlackMessage) {
conn, found := connections[connection]
if !found {
2022-06-29 13:16:43 +05:00
ctx.Log.Error().Str("conn", connection).Msg("Connection not found")
2017-10-31 22:50:14 +05:00
return
}
2022-06-29 13:16:43 +05:00
ctx.Log.Debug().Str("conn", connection).Msg("Pushing data")
2017-10-31 22:50:14 +05:00
conn.ProcessMessage(data)
}
func (tp TelegramPusher) Shutdown() {
2022-06-29 13:16:43 +05:00
ctx.Log.Info().Msg("Shutting down Telegram pusher...")
2017-10-31 22:50:14 +05:00
for _, conn := range connections {
conn.Shutdown()
}
}