feat: override a path to config file in4it#5

harnyk · Mar 16, 2023 · 779251f · 779251f
1 parent 0c8238c
commit 779251f
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/cmd/root.go b/cmd/root.go
@@ -2,6 +2,7 @@ package cmd
 
 import (
 	"log"
+	"os"
 
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
@@ -55,9 +56,14 @@ func init() {
 }
 
 func LoadConfig(path string) (config Config, err error) {
-	viper.AddConfigPath(path)
-	viper.SetConfigName("config")
-	viper.SetConfigType("yaml")
+	userDefinedConfigFile := os.Getenv("ONELOGIN_AUTH_CLI_CONFIG_FILE")
+	if userDefinedConfigFile != "" {
+		viper.SetConfigFile(userDefinedConfigFile)
+	} else {
+		viper.AddConfigPath(path)
+		viper.SetConfigName("config")
+		viper.SetConfigType("yaml")
+	}
 
 	viper.AutomaticEnv()