go2rtc/cmd/app/app.go

package app

import (
	"flag"
	"github.com/rs/zerolog"
	"gopkg.in/yaml.v3"
	"io"
	"os"
	"runtime"
)

func Init() {
	config := flag.String(
		"config",
		"go2rtc.yaml",
		"Path to go2rtc configuration file",
	)

	flag.Parse()

	data, _ = os.ReadFile(*config)

	var cfg struct {
		Mod map[string]string `yaml:"log"`
	}

	if data != nil {
		if err := yaml.Unmarshal(data, &cfg); err != nil {
			println("ERROR: " + err.Error())
		}
	}

	var writer io.Writer = os.Stdout

	// styles
	format := cfg.Mod["format"]
	if format != "json" {
		writer = zerolog.ConsoleWriter{
			Out: writer, TimeFormat: "15:04:05.000",
			NoColor: writer != os.Stdout || format == "text",
		}
	}

	zerolog.TimeFieldFormat = zerolog.TimeFormatUnixMs

	lvl, err := zerolog.ParseLevel(cfg.Mod["level"])
	if err != nil || lvl == zerolog.NoLevel {
		lvl = zerolog.InfoLevel
	}

	log = zerolog.New(writer).With().Timestamp().Logger().Level(lvl)

	modules = cfg.Mod

	path, _ := os.Getwd()
	log.Debug().Str("os", runtime.GOOS).Str("arch", runtime.GOARCH).
		Str("cwd", path).Int("conf_size", len(data)).Msgf("[app]")
}

func LoadConfig(v interface{}) {
	if data != nil {
		if err := yaml.Unmarshal(data, v); err != nil {
			log.Warn().Err(err).Msg("[app] read config")
		}
	}
}

func GetLogger(module string) zerolog.Logger {
	if s, ok := modules[module]; ok {
		lvl, err := zerolog.ParseLevel(s)
		if err != nil {
			log.Warn().Err(err).Msg("[log]")
			return log
		}

		return log.Level(lvl)
	}

	return log
}

// internal

// data - config content
var data []byte

// log - main logger
var log zerolog.Logger

// modules log levels
var modules map[string]string