hswro-alarm-bot/main.go

package main

import (
	"flag"
	"fmt"
	"html/template"
	"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
	PersistenceFilename            = "hs_wro_last_seen.bin"
)

type TgSender struct {
	bot *tgbotapi.BotAPI
}

func (self TgSender) Send(msg GenericMessage, tpl *template.Template) error {
	chatIds := msg.ChatIds.GetTgIds()
	if chatIds == nil {
		return nil
	}
	message := msg.Format(tpl)
	for _, chatId := range *chatIds {
		toSend := tgbotapi.NewMessage(chatId, message)
		toSend.ParseMode = "HTML"
		_, err := self.bot.Send(toSend)
		if err != nil {
			return err
		}
	}
	return nil
}

func sendTgMessage(tgEvents chan GenericMessage, msg MsgContent, chatId int64) {
	tgEvents <- GenericMessage{TgChatId{chatId}, []MsgContent{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, []satel.ChangeType, []int) {
	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.")
	allowedTypesRaw := flag.String("allowed-types", "", "Satel change types that are allowed. All other types will be discarded. By default all are allowed. Use \",\" to specify multiple types.")
	allowedIndexesRaw := flag.String("allowed-indexes", "", "Satel indexes (zones?) that are allowed. All other indexes will be discarded. By default all are allowed. Use \",\" to specify multiple indexes.")
	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)
	}
	allowedTypesStrings := strings.Split(*allowedTypesRaw, ",")
	var allowedTypes []satel.ChangeType
	for _, allowedTypeStr := range allowedTypesStrings {
		if len(allowedTypeStr) == 0 {
			continue
		}
		allowedType, err := StringToSatelChangeType(allowedTypeStr)
		if err != nil {
			logger.Fatalf("Error trying to understand an allowed type: %s.", err)
		}
		allowedTypes = append(allowedTypes, allowedType)
	}
	allowedIndexesStrings := strings.Split(*allowedIndexesRaw, ",")
	var allowedIndexes []int
	for _, allowedIndexStr := range allowedIndexesStrings {
		if len(allowedIndexStr) == 0 {
			continue
		}
		allowedIndex, err := strconv.ParseInt(allowedIndexStr, 10, 0)
		if err != nil {
			logger.Fatalf("Tried to use a non-int value for one of allowed indexes: %s. That's bad.", allowedIndexStr)
		}
		allowedIndexes = append(allowedIndexes, int(allowedIndex))
	}

	satelAddr := fmt.Sprintf("%s:%s", *satelApiAddr, *satelApiPort)
	return satelAddr, chatIds, allowedTypes, allowedIndexes
}

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, allowedTypes, allowedIndexes := 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}

	tpl := template.Must(template.New("TelegramMessage").Parse(TelegramMessageTemplate))

	dataStore := MakeDataStore(log.New(os.Stderr, "DataStore", log.Lmicroseconds), PersistenceFilename)

	Consume(
		SendToTg(
			tgSenderWorker(tgEvents, &wg, sleeper, log.New(os.Stderr, "TgSender", log.Lmicroseconds)),
			tgSender, &wg, log.New(os.Stderr, "SendToTg", log.Lmicroseconds), tpl))

	go CloseSatelOnCtrlC(s)

	for e := range FilterByIndex(FilterByType(
		FilterByLastSeen(
			CallWhenClosed(s.Events, func() { logger.Print("Satel disconnected.") }),
			&dataStore, log.New(os.Stderr, "FilterByLastSeen", log.Lmicroseconds)),
		allowedTypes),
		allowedIndexes) {
		logger.Print("Received change from SATEL: ", e)
		for _, chatId := range chatIds {
			sendTgMessage(tgEvents, MsgContent{e}, chatId)
		}
	}

	close(tgEvents)
	wg.Wait()
}