Initial commit

cmd/app/app.go

@@ -0,0 +1,68 @@
+package app
+
+import (
+	"github.com/rs/zerolog"
+	"gopkg.in/yaml.v3"
+	"io"
+	"os"
+	"runtime"
+)
+
+func Init() {
+	data, _ = os.ReadFile("go2rtc.yaml")
+
+	var cfg struct {
+		Mod map[string]string `yaml:"log"`
+	}
+
+	LoadConfig(&cfg)
+
+	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
+
+	log.Info().Msgf("go2rtc %s/%s", runtime.GOOS, runtime.GOARCH)
+}
+
+func LoadConfig(v interface{}) {
+	if data != nil {
+		_ = yaml.Unmarshal(data, v)
+	}
+}
+
+func GetLogger(module string) zerolog.Logger {
+	lvl, err := zerolog.ParseLevel(modules[module])
+	if err != nil {
+		return log
+	}
+	return log.Level(lvl)
+}
+
+// internal
+
+// data - config content
+var data []byte
+
+// log - main logger
+var log zerolog.Logger
+
+// modules log levels
+var modules map[string]string