| 
									
										
										
										
											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, | 
					
						
							| 
									
										
										
										
											2019-12-15 22:17:01 +05:00
										 |  |  | // 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 ( | 
					
						
							| 
									
										
										
										
											2019-12-15 22:17:01 +05:00
										 |  |  | 	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...") | 
					
						
							| 
									
										
										
										
											2019-12-15 22:17:01 +05:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-06-29 13:16:43 +05:00
										 |  |  | 		// nolint:exhaustruct | 
					
						
							| 
									
										
										
										
											2017-10-31 22:50:14 +05:00
										 |  |  | 		conn := TelegramConnection{} | 
					
						
							|  |  |  | 		connections[name] = &conn | 
					
						
							| 
									
										
										
										
											2019-12-15 22:17:01 +05:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-03-05 18:32:34 +05:00
										 |  |  | 		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") | 
					
						
							| 
									
										
										
										
											2021-11-21 16:16:10 +05:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2017-10-31 22:50:14 +05:00
										 |  |  | 		return | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2019-12-15 22:17:01 +05:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											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() | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | } |