kopia lustrzana https://git.sr.ht/~michalr/hswro-alarm-bot
109 wiersze
2.9 KiB
Go
109 wiersze
2.9 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
|
|
"github.com/probakowski/go-satel"
|
|
)
|
|
|
|
const (
|
|
MessageNotMoreOftenThanSeconds = 15
|
|
)
|
|
|
|
type TgSender struct {
|
|
bot *tgbotapi.BotAPI
|
|
}
|
|
|
|
func (self TgSender) Send(msg GenericMessage) error {
|
|
toSend := tgbotapi.NewMessage(msg.chatId, msg.msg)
|
|
_, err := self.bot.Send(toSend)
|
|
return err
|
|
}
|
|
|
|
func sendTgMessage(tgEvents chan GenericMessage, msg string, chatIds []int64) {
|
|
for _, chatId := range chatIds {
|
|
tgEvents <- GenericMessage{chatId, msg}
|
|
}
|
|
}
|
|
|
|
type RealSleeper struct {
|
|
duration time.Duration
|
|
}
|
|
|
|
func (self RealSleeper) Sleep(ch chan<- interface{}) {
|
|
go func() {
|
|
time.Sleep(self.duration)
|
|
ch <- nil
|
|
}()
|
|
}
|
|
|
|
func getCmdLineParams(logger *log.Logger) (string, []int64) {
|
|
satelApiAddr := flag.String("satel-addr", "", "Address that should be used to connect to the SATEL device")
|
|
satelApiPort := flag.String("satel-port", "7094", "Port that should be used to connect to the SATEL device")
|
|
chatIdRaw := flag.String("tg-chat-id", "", "Telegram Chat ID where to send updates. Use \",\" to specify multiple IDs.")
|
|
flag.Parse()
|
|
|
|
if len(*satelApiAddr) == 0 || len(*satelApiPort) == 0 || len(*chatIdRaw) == 0 {
|
|
logger.Fatal("Use --satel-addr=ADDR, --satel-port=PORT and --tg-chat-id=CHAT_ID command line flags to continue.")
|
|
}
|
|
chatIdsStrings := strings.Split(*chatIdRaw, ",")
|
|
var chatIds []int64
|
|
for _, chatIdStr := range chatIdsStrings {
|
|
chatId, err := strconv.ParseInt(chatIdStr, 10, 64)
|
|
if err != nil {
|
|
logger.Fatalf("Tried to use a non-int value for one of tg_chatIds: %s. That's bad.", chatIdStr)
|
|
}
|
|
chatIds = append(chatIds, chatId)
|
|
}
|
|
|
|
satelAddr := fmt.Sprintf("%s:%s", *satelApiAddr, *satelApiPort)
|
|
return satelAddr, chatIds
|
|
}
|
|
|
|
func makeSatel(satelAddr string) *satel.Satel {
|
|
satelConn, err := net.Dial("tcp", satelAddr)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return satel.NewConfig(satelConn, satel.Config{EventsQueueSize: 10})
|
|
}
|
|
|
|
func main() {
|
|
var (
|
|
wg sync.WaitGroup
|
|
tgEvents = make(chan GenericMessage, 100)
|
|
sleeper = RealSleeper{time.Second * MessageNotMoreOftenThanSeconds}
|
|
logger = log.New(os.Stderr, "Main", log.Lmicroseconds)
|
|
)
|
|
|
|
satelAddr, chatIds := getCmdLineParams(logger)
|
|
|
|
s := makeSatel(satelAddr)
|
|
logger.Printf("Connected to Satel: %s", satelAddr)
|
|
|
|
bot, err := tgbotapi.NewBotAPI(os.Getenv("TELEGRAM_APITOKEN"))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
logger.Print("Created Telegram Bot API client")
|
|
|
|
tgSender := TgSender{bot}
|
|
go tgSenderWorker(tgEvents, tgSender, &wg, sleeper, log.New(os.Stderr, "TgSender", log.Lmicroseconds))
|
|
for e := range FilterByLastSeen(s.Events, "hs_wro_last_seen.bin", log.New(os.Stderr, "FilterByLastSeen", log.Lmicroseconds)) {
|
|
logger.Print("Received change from SATEL: ", e)
|
|
sendTgMessage(tgEvents, fmt.Sprint("Change from SATEL: type:", e.Type, ", index:", e.Index, ", value:", e.Value), chatIds)
|
|
}
|
|
|
|
close(tgEvents)
|
|
wg.Wait()
|
|
}
|