fleg
/
hswro-alarm-bot
mirror of https://git.sr.ht/~michalr/hswro-alarm-bot
1
0
Fork 0
Code

Add ability to specify more than one tg_chat_id

This commit is contained in:
Michał Rudowicz 2024-02-08 20:57:16 +01:00
parent c0a6540500
commit e625bcfcf5
1 changed files with 22 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
main.go
View File

@ -5,6 +5,8 @@ import (
"fmt" "fmt"
"net" "net"
"flag" "flag"
"strings"
"strconv"
"github.com/probakowski/go-satel" "github.com/probakowski/go-satel"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5" tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
@ -13,13 +15,23 @@ import (
func main() { func main() {
satel_api_addr := flag.String("satel_addr", "", "Address that should be used to connect to the SATEL device") 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") 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() 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.") fmt.Println("Use --satel_addr=ADDR, --satel_port=PORT and --tg_chat_id=CHAT_ID command line flags to continue.")
os.Exit(1) 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_addr := fmt.Sprintf("%s:%s", *satel_api_addr, *satel_api_port)
satel_conn, err := net.Dial("tcp", satel_addr) satel_conn, err := net.Dial("tcp", satel_addr)
@ -34,7 +46,8 @@ func main() {
for e, ok := <-s.Events; ok; e, ok = <-s.Events { for e, ok := <-s.Events; ok; e, ok = <-s.Events {
fmt.Println("Change from SATEL: ", "type", e.Type, "index", e.Index, "value", e.Value) 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", 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)) e.Index, e.Type, e.Value))
if _, err := bot.Send(msg); err != nil { if _, err := bot.Send(msg); err != nil {
@ -42,4 +55,5 @@ func main() {
panic(err) panic(err)
} }
} }
}
} }