override config flags

1 files changed, 17 insertions, 70 deletions

diff --git a/cmd/rr/cmd/root.go b/cmd/rr/cmd/root.go
index 15a004e0..ceeeb840 100644
--- a/cmd/rr/cmd/root.go
+++ b/cmd/rr/cmd/root.go
@@ -23,16 +23,15 @@ package cmd
 import (
 	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
-	"github.com/spf13/viper"
 	"github.com/spiral/roadrunner/cmd/util"
 	"github.com/spiral/roadrunner/service"
 	"os"
-	"path/filepath"
 )
 
 // Service bus for all the commands.
 var (
-	cfgFile string
+	cfgFile  string
+	override []string
 
 	// Verbose enables verbosity mode (container specific).
 	Verbose bool
@@ -59,26 +58,6 @@ var (
 	}
 )
 
-// ViperWrapper provides interface bridge between v configs and service.Config.
-type ViperWrapper struct {
-	v *viper.Viper
-}
-
-// Get nested config section (sub-map), returns nil if section not found.
-func (w *ViperWrapper) Get(key string) service.Config {
-	sub := w.v.Sub(key)
-	if sub == nil {
-		return nil
-	}
-
-	return &ViperWrapper{sub}
-}
-
-// Unmarshal unmarshal config data into given struct.
-func (w *ViperWrapper) Unmarshal(out interface{}) error {
-	return w.v.Unmarshal(out)
-}
-
 // Execute adds all child commands to the CLI command and sets flags appropriately.
 // This is called by main.main(). It only needs to happen once to the CLI.
 func Execute() {
@@ -92,60 +71,28 @@ func init() {
 	CLI.PersistentFlags().BoolVarP(&Verbose, "verbose", "v", false, "Verbose output")
 	CLI.PersistentFlags().BoolVarP(&Debug, "debug", "d", false, "debug mode")
 	CLI.PersistentFlags().StringVarP(&cfgFile, "config", "c", "", "config file (default is .rr.yaml)")
+	CLI.PersistentFlags().StringArrayVarP(
+		&override,
+		"override",
+		"o",
+		nil,
+		"override config value (dot.notation=value)",
+	)
 
 	cobra.OnInitialize(func() {
 		if Verbose {
 			Logger.SetLevel(logrus.DebugLevel)
 		}
 
-		if cfg := initConfig(cfgFile, []string{"."}, ".rr"); cfg != nil {
-			if err := Container.Init(cfg); err != nil {
-				util.Printf("<red+hb>Error:</reset> <red>%s</reset>\n", err)
-				os.Exit(1)
-			}
+		cfg, err := util.LoadConfig(cfgFile, []string{"."}, ".rr", override)
+		if err != nil {
+			Logger.Warnf("config: %s", err)
+			return
 		}
-	})
-}
-
-func initConfig(cfgFile string, path []string, name string) service.Config {
-	cfg := viper.New()
-
-	if cfgFile != "" {
-		if absPath, err := filepath.Abs(cfgFile); err == nil {
-			cfgFile = absPath
 
-			// force working absPath related to config file
-			if err := os.Chdir(filepath.Dir(absPath)); err != nil {
-				Logger.Error(err)
-			}
+		if err := Container.Init(cfg); err != nil {
+			util.Printf("<red+hb>Error:</reset> <red>%s</reset>\n", err)
+			os.Exit(1)
 		}
-
-		// Use cfg file from the flag.
-		cfg.SetConfigFile(cfgFile)
-
-		if dir, err := filepath.Abs(cfgFile); err == nil {
-			// force working absPath related to config file
-			if err := os.Chdir(filepath.Dir(dir)); err != nil {
-				Logger.Error(err)
-			}
-		}
-	} else {
-		// automatic location
-		for _, p := range path {
-			cfg.AddConfigPath(p)
-		}
-
-		cfg.SetConfigName(name)
-	}
-
-	// read in environment variables that match
-	cfg.AutomaticEnv()
-
-	// If a cfg file is found, read it in.
-	if err := cfg.ReadInConfig(); err != nil {
-		Logger.Warnf("config: %s", err)
-		return nil
-	}
-
-	return &ViperWrapper{cfg}
+	})
 }