main.go

package main

import (
	"embed"
	"flag"
	"os"
	"os/signal"
	"strings"
	"time"
	_ "time/tzdata"

	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"

	"github.com/0x2142/frigate-notify/config"
	frigate "github.com/0x2142/frigate-notify/events"
	"github.com/0x2142/frigate-notify/notifier"
	"github.com/0x2142/frigate-notify/util"
)

var APP_VER = "v0.3.5"
var debug, debugenv bool
var jsonlog, jsonlogenv bool
var nocolor, nocolorenv bool
var configFile string
var logLevel string

//go:embed templates/*
var NotifTemplates embed.FS

func main() {
	// Parse flags
	flag.StringVar(&configFile, "c", "", "Configuration file location (default \"./config.yml\")")
	flag.BoolVar(&debug, "debug", false, "Enable debug logging (Overrides loglevel, if also set)")
	flag.StringVar(&logLevel, "loglevel", "info", "Set logging level")
	flag.BoolVar(&jsonlog, "jsonlog", false, "Enable JSON logging")
	flag.BoolVar(&nocolor, "nocolor", false, "Disable color on console logging")
	flag.Parse()

	// Set up logging
	_, jsonlogenv = os.LookupEnv("FN_JSONLOG")
	if jsonlog || jsonlogenv {
		zerolog.TimeFieldFormat = "2006/01/02 15:04:05 -0700"
	} else {
		_, nocolorenv = os.LookupEnv("FN_NOCOLOR")
		log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stdout, TimeFormat: "2006/01/02 15:04:05 -0700", NoColor: nocolorenv || nocolor})
	}

	// Apply custom log level, if set
	switch logLevel, _ = os.LookupEnv("FN_LOGLEVEL"); strings.ToLower(logLevel) {
	case "panic":
		zerolog.SetGlobalLevel(zerolog.PanicLevel)
	case "fatal":
		zerolog.SetGlobalLevel(zerolog.FatalLevel)
	case "error":
		zerolog.SetGlobalLevel(zerolog.ErrorLevel)
	case "warn":
		zerolog.SetGlobalLevel(zerolog.WarnLevel)
	case "debug":
		log.Debug().Msg("Debug logging enabled")
		zerolog.SetGlobalLevel(zerolog.DebugLevel)
	case "trace":
		log.Trace().Msg("Trace logging enabled")
		zerolog.SetGlobalLevel(zerolog.TraceLevel)
	default:
		zerolog.SetGlobalLevel(zerolog.InfoLevel)
	}

	// Enable debug logging if set
	_, debugenv = os.LookupEnv("FN_DEBUG")
	if debug || debugenv {
		zerolog.SetGlobalLevel(zerolog.DebugLevel)
		// Add calling module to debug logs
		log.Logger = log.With().Caller().Logger()
		log.Debug().Msg("Debug logging enabled")
	}

	log.Info().Msgf("Frigate Notify - %v", APP_VER)
	log.Info().Msg("Starting...")

	// Load & validate config
	config.LoadConfig(configFile)

	notifier.TemplateFiles = NotifTemplates

	// Set up monitor
	if config.ConfigData.Monitor.Enabled {
		log.Debug().Msg("App monitoring enabled.")
		go func() {
			for {
				_, err := util.HTTPGet(config.ConfigData.Monitor.URL, config.ConfigData.Monitor.Insecure, "")
				if err != nil {
					log.Warn().
						Err(err).
						Msg("Unable to reach polling monitoring URL")
				}
				log.Debug().Msg("Completed monitoring check-in.")
				time.Sleep(time.Duration(config.ConfigData.Monitor.Interval) * time.Second)
			}
		}()
	}

	// Set up event cache
	frigate.InitZoneCache()

	// Loop & watch for events
	if config.ConfigData.Frigate.WebAPI.Enabled {
		log.Info().Msg("App ready!")
		for {
			frigate.CheckForEvents()
			time.Sleep(time.Duration(config.ConfigData.Frigate.WebAPI.Interval) * time.Second)
		}
	}
	// Connect MQTT
	if config.ConfigData.Frigate.MQTT.Enabled {
		defer frigate.CloseZoneCache()

		log.Debug().Msg("Connecting to MQTT Server...")
		frigate.SubscribeMQTT()
		log.Info().Msg("App ready!")
		sig := make(chan os.Signal, 1)
		signal.Notify(sig, os.Interrupt)
		<-sig
	}

}