From e625bcfcf5bca688854c4489e9a85a4e79fead9a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Rudowicz?= Date: Thu, 8 Feb 2024 20:57:16 +0100 Subject: [PATCH] Add ability to specify more than one tg_chat_id --- main.go | 30 ++++++++++++++++++++++-------- 1 file changed, 22 insertions(+), 8 deletions(-) diff --git a/main.go b/main.go index 898d8b0..c9d28fb 100644 --- a/main.go +++ b/main.go @@ -5,6 +5,8 @@ import ( "fmt" "net" "flag" + "strings" + "strconv" "github.com/probakowski/go-satel" tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5" @@ -13,13 +15,23 @@ import ( func main() { satel_api_addr := flag.String("satel_addr", "", "Address that should be used to connect to the SATEL device") satel_api_port := flag.String("satel_port", "7094", "Port that should be used to connect to the SATEL device") - chat_id := flag.Int64("tg_chat_id", -1, "Telegram Chat ID where to send updates") + chat_id_raw := flag.String("tg_chat_id", "", "Telegram Chat ID where to send updates. Use \",\" to specify multiple IDs.") flag.Parse() - if len(*satel_api_addr) == 0 || len(*satel_api_port) == 0 || *chat_id == -1 { + if len(*satel_api_addr) == 0 || len(*satel_api_port) == 0 || len(*chat_id_raw) == 0 { fmt.Println("Use --satel_addr=ADDR, --satel_port=PORT and --tg_chat_id=CHAT_ID command line flags to continue.") os.Exit(1) } + chat_ids_strings := strings.Split(*chat_id_raw, ",") + var chat_ids []int64 + for _, chat_id_str := range chat_ids_strings { + chat_id, err := strconv.ParseInt(chat_id_str, 10, 64) + if err != nil { + fmt.Printf("Tried to use a non-int value for one of tg_chat_ids: %s. That's bad.", chat_id_str) + os.Exit(1) + } + chat_ids = append(chat_ids, chat_id) + } satel_addr := fmt.Sprintf("%s:%s", *satel_api_addr, *satel_api_port) satel_conn, err := net.Dial("tcp", satel_addr) @@ -34,12 +46,14 @@ func main() { for e, ok := <-s.Events; ok; e, ok = <-s.Events { fmt.Println("Change from SATEL: ", "type", e.Type, "index", e.Index, "value", e.Value) - msg := tgbotapi.NewMessage(*chat_id, fmt.Sprintf("Change from SATEL: Zone: %d Type: %s Value: %t", - e.Index, e.Type, e.Value)) + for _, chat_id := range chat_ids { + msg := tgbotapi.NewMessage(chat_id, fmt.Sprintf("Change from SATEL: Zone: %d Type: %s Value: %t", + e.Index, e.Type, e.Value)) - if _, err := bot.Send(msg); err != nil { - // TODO: retry sending later in case of problems - panic(err) - } + if _, err := bot.Send(msg); err != nil { + // TODO: retry sending later in case of problems + panic(err) + } + } } }