hswro-alarm-bot/filters.go

package main

import (
	"log"

	"git.sr.ht/~michalr/go-satel"
)

func FilterByType(ev <-chan satel.Event, allowedTypes []satel.ChangeType) <-chan satel.Event {
	returnChan := make(chan satel.Event)

	if len(allowedTypes) == 0 {
		// no allowed types == all types are allowed
		go func() {
			for e := range ev {
				returnChan <- e
			}
			close(returnChan)
		}()
	} else {
		go func() {
			for e := range ev {
				retEv := satel.Event{BasicEvents: make([]satel.BasicEventElement, 0)}
				for _, basicEventElement := range e.BasicEvents {
					for _, allowedType := range allowedTypes {
						if allowedType == basicEventElement.Type {
							retEv.BasicEvents = append(retEv.BasicEvents, basicEventElement)
							continue
						}
					}
				}
				if len(retEv.BasicEvents) != 0 {
					returnChan <- retEv
				}
			}
			close(returnChan)
		}()
	}

	return returnChan
}

func FilterByIndex(ev <-chan satel.Event, allowedIndexes []int) <-chan satel.Event {
	returnChan := make(chan satel.Event)

	if len(allowedIndexes) == 0 {
		// no allowed indexes == all indexes are allowed
		go func() {
			for e := range ev {
				returnChan <- e
			}
			close(returnChan)
		}()
	} else {
		go func() {
			for e := range ev {
				retEv := satel.Event{BasicEvents: make([]satel.BasicEventElement, 0)}
				for _, basicEventElement := range e.BasicEvents {
					for _, allowedIndex := range allowedIndexes {
						if allowedIndex == basicEventElement.Index {
							retEv.BasicEvents = append(retEv.BasicEvents, basicEventElement)
							continue
						}
					}
				}
				if len(retEv.BasicEvents) != 0 {
					returnChan <- retEv
				}
			}
			close(returnChan)
		}()
	}

	return returnChan
}

func FilterByLastSeen(ev <-chan satel.Event, dataStore *DataStore, logger *log.Logger) <-chan satel.Event {
	returnChan := make(chan satel.Event)

	go func() {
		for e := range ev {
			retEv := satel.Event{BasicEvents: make([]satel.BasicEventElement, 0)}
			for _, basicEventElement := range e.BasicEvents {
				lastSeen := dataStore.GetSystemState()
				val, ok := lastSeen[EventKey{basicEventElement.Type, basicEventElement.Index}]
				if !ok || val.Value != basicEventElement.Value {
					retEv.BasicEvents = append(retEv.BasicEvents, basicEventElement)
					// TODO: flush to disk only after the loop finishes
					dataStore.SetSystemState(EventKey{basicEventElement.Type, basicEventElement.Index}, EventValue{basicEventElement.Value})
				}
			}
			if len(retEv.BasicEvents) != 0 {
				returnChan <- retEv
			}
		}
		close(returnChan)
	}()

	return returnChan
}

func CallWhenClosed(ev <-chan satel.Event, cbk func()) <-chan satel.Event {
	returnChan := make(chan satel.Event)

	go func() {
		for e := range ev {
			returnChan <- e
		}
		cbk()
		close(returnChan)
	}()

	return returnChan
}