From 0b291f5185026953ce3fdeb9210586216de05248 Mon Sep 17 00:00:00 2001
From: Alexey Khit <alexey.khit@gmail.com>
Date: Sat, 14 Jan 2023 21:12:17 +0300
Subject: [PATCH] Support multiple configs and config in raw yaml form

---
 cmd/app/app.go | 59 ++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 43 insertions(+), 16 deletions(-)

diff --git a/cmd/app/app.go b/cmd/app/app.go
index c9dd1d63..438b504c 100644
--- a/cmd/app/app.go
+++ b/cmd/app/app.go
@@ -9,6 +9,7 @@ import (
 	"io"
 	"os"
 	"runtime"
+	"strings"
 )
 
 var Version = "0.1-rc.8"
@@ -20,28 +21,44 @@ var Info = map[string]interface{}{
 }
 
 func Init() {
-	flag.StringVar(
-		&ConfigPath, "config", "go2rtc.yaml",
-		"Path to go2rtc configuration file",
-	)
+	var confs Config
 
+	flag.Var(&confs, "config", "go2rtc config (path to file or raw text), support multiple")
 	flag.Parse()
 
-	Info["config_path"] = ConfigPath
+	if confs == nil {
+		confs = []string{"go2rtc.yaml"}
+	}
 
-	data, _ = os.ReadFile(ConfigPath)
+	for _, conf := range confs {
+		if conf[0] != '{' {
+			// config as file
+			if ConfigPath == "" {
+				ConfigPath = conf
+			}
+
+			data, _ := os.ReadFile(conf)
+			if data == nil {
+				continue
+			}
+
+			data = []byte(shell.ReplaceEnvVars(string(data)))
+			configs = append(configs, data)
+		} else {
+			// config as raw YAML
+			configs = append(configs, []byte(conf))
+		}
+	}
+
+	if ConfigPath != "" {
+		Info["config_path"] = ConfigPath
+	}
 
 	var cfg struct {
 		Mod map[string]string `yaml:"log"`
 	}
 
-	if data != nil {
-		data = []byte(shell.ReplaceEnvVars(string(data)))
-
-		if err := yaml.Unmarshal(data, &cfg); err != nil {
-			println("ERROR: " + err.Error())
-		}
-	}
+	LoadConfig(&cfg)
 
 	log.Logger = NewLogger(cfg.Mod["format"], cfg.Mod["level"])
 
@@ -74,7 +91,7 @@ func NewLogger(format string, level string) zerolog.Logger {
 }
 
 func LoadConfig(v interface{}) {
-	if data != nil {
+	for _, data := range configs {
 		if err := yaml.Unmarshal(data, v); err != nil {
 			log.Warn().Err(err).Msg("[app] read config")
 		}
@@ -95,8 +112,18 @@ func GetLogger(module string) zerolog.Logger {
 
 // internal
 
-// data - config content
-var data []byte
+type Config []string
+
+func (c *Config) String() string {
+	return strings.Join(*c, " ")
+}
+
+func (c *Config) Set(value string) error {
+	*c = append(*c, value)
+	return nil
+}
+
+var configs [][]byte
 
 // modules log levels
 var modules map[string]string