arno/kube-bench
1
0
Fork 0
mirror of https://github.com/aquasecurity/kube-bench.git synced 2024-11-25 17:38:21 +00:00
Code Issues Releases Wiki Activity

Allow config file to be specified on the command line

Browse Source
This commit is contained in:
Liz Rice 2017-06-22 15:34:21 +01:00
parent 96364e3f29
commit 0d6d3a03ef
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cmd/root.go
View File

@ -71,18 +71,18 @@ func init() {
"",
`Run all the checks under this comma-delimited list of groups. Example --group="1.1"`,
)
RootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is ./cfg/config.yaml)")
}
// initConfig reads in config file and ENV variables if set.
func initConfig() {
if cfgFile != "" { // enable ability to specify config file via flag
viper.SetConfigFile(cfgFile)
} else {
viper.SetConfigName("config") // name of config file (without extension)
viper.AddConfigPath(cfgDir) // adding ./cfg as first search path
}
viper.SetConfigName("config") // name of config file (without extension)
viper.AddConfigPath(cfgDir) // adding home directory as first search path
viper.SetEnvPrefix("CISK8S")
viper.AutomaticEnv() // read in environment variables that match