package app

import (
	"fmt"
	"github.com/marmotedu/component-base/pkg/util/homedir"
	"github.com/spf13/cobra"
	"github.com/spf13/pflag"
	"github.com/spf13/viper"
	"os"
	"path/filepath"
	"strings"
)

const configFlagName = "config"

var cfgFile string

func init() {
	// 通过命令行得到配置文件
	pflag.StringVarP(&cfgFile, "config", "c", cfgFile, "Read configuration from specified `FILE`, "+
		"support JSON, TOML, YAML, HCL, or Java properties formats.")
}

//
func addConfigFlag(basename string, fs *pflag.FlagSet) {
	//  将config 参数到 global组中
	fs.AddFlag(pflag.Lookup(configFlagName)) // 将config参数写到默认组中

	viper.AutomaticEnv() // 读取环境变量
	//设置环境变量前缀为basename(大写)
	viper.SetEnvPrefix(strings.Replace(strings.ToUpper(basename), "-", "_", -1))
	// 将.和-替换为_
	viper.SetEnvKeyReplacer(strings.NewReplacer(".", "_", "-", "_"))
	//判断是否使用文件配置 若无则使用配置文件
	cobra.OnInitialize(func() { //  执行运行命令之前，需要做的初始化工作： 得到配置文件
		if cfgFile != "" { // 使用配置文件路径进行查找
			viper.SetConfigFile(cfgFile)
		} else { // 在本地进行查找
			viper.AddConfigPath(".")
			if names := strings.Split(basename, "-"); len(names) > 1 {
				viper.AddConfigPath(filepath.Join(homedir.HomeDir(), "."+names[0]))
			}
			viper.SetConfigName(basename) // 找到那个文件名的配置
		}
		// 判断是否找到配置文件的位置
		if err := viper.ReadInConfig(); err != nil {
			_, _ = fmt.Fprintf(os.Stderr, "Error: failed to read configuration file(%s): %v\n", cfgFile, err)
			os.Exit(1)
		}
	})
}