psu/common/configs.go

package common

import (
	"fmt"
	"os"

	"github.com/mitchellh/go-homedir"
	"github.com/spf13/viper"
)

func LoadCofig() (v *viper.Viper, err error) {
	// Set config file name
	var configFile string
	if viper.ConfigFileUsed() != "" {
		// Use config file from viper
		configFile = viper.ConfigFileUsed()
	} else {
		// Find home directory
		var home string
		home, err = homedir.Dir()
		if err != nil {
			return
		}

		// Use $HOME/.psu.yaml
		configFile = fmt.Sprintf("%s%s.psu.yaml", home, string(os.PathSeparator))
	}
	v = viper.New()
	v.SetConfigFile(configFile)

	// Read config from file
	err = v.ReadInConfig()

	return
}

func CheckConfigKeyExists(key string) (keyExists bool) {
	for _, k := range viper.AllKeys() {
		if k == key {
			keyExists = true
			break
		}
	}
	return
}