Merge pull request #19 from ordovicia/config-path-from-env-var

Set config file path via PFTQCONFIG env var

Author: everpeace

README.md

@@ -710,9 +710,11 @@ $ pftaskqueue print-default-config > ~/.pftaskqueue.yaml
 $ PFTQ_REDIS_ADDR=... pftaskqueue ...
 ```
 
-### Config files
+### Config file
 
-`pftaskqueue` automatically reads `${HOME}/.pftaskqueue.yaml` if exists.  Or, you can also set any configuration path with `--config=${CONFIG_FILE_PATH}` flag or both.
+`pftaskqueue` automatically reads `${HOME}/.pftaskqueue.yaml` if exists.
+Or, you can also set any configuration path with `--config=${CONFIG_FILE_PATH}` flag or `PFTQCONFIG` environment variable.
+`--config` flag is prioritized over `PFTQCONFIG` environment variable.
 
 To generate config file with default values, please run `print-default-config` command.
 

cmd/root.go

@@ -144,7 +144,7 @@ func init() {
 
 	rootCmd.SetVersionTemplate(VersionString())
 	flag := rootCmd.PersistentFlags()
-	flag.StringVar(&cfgFile, "config", "", "config file path. [default = $HOME/.pftaskqueue.yaml]")
+	flag.StringVar(&cfgFile, "config", "", "config file path. [default = $PFTQCONFIG or $HOME/.pftaskqueue.yaml]")
 	flag.BoolVar(&displayOpts, "display-options", false, "display loaded config values at startup")
 
 	// Log setting
@@ -239,6 +239,9 @@ func initConfig() {
 	if cfgFile != "" {
 		// Use config file from the flag.
 		viper.SetConfigFile(cfgFile)
+	} else if env, ok := os.LookupEnv("PFTQCONFIG"); ok {
+		// Use config file from env var.
+		viper.SetConfigFile(env)
 	} else {
 		// Find home directory.
 		home, err := homedir.Dir()