Add ability to specify more than one tg_chat_id

command_handler
Michał Rudowicz 2024-02-08 20:57:16 +01:00
rodzic c0a6540500
commit e625bcfcf5
1 zmienionych plików z 22 dodań i 8 usunięć

30
main.go
Wyświetl plik

@ -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)
}
}
}
}