使用Go语言Cobra与Viper配置默认目录 实现自定义文件存储路径设置

你给出的示例命令app --config path "~/Documents/"可以通过两种方式实现，以下是完整实现步骤：

标准实现（推荐） #

推荐调整命令格式为更符合CLI规范的app --config "~/Documents/"，直接通过flag传值，实现逻辑如下：

1. 定义全局变量存储路径
   在root.go文件中定义接收参数的变量：

package cmd

import (
	"log"
	"os"
	"path/filepath"
	"strings"

	"github.com/spf13/cobra"
)

var cfgPath string

2. 绑定PersistentFlag
   在root命令的init()函数中添加flag绑定，所有子命令可继承该参数：

func init() {
	cobra.OnInitialize(initConfig)
	// 三个参数分别为：接收变量、flag名、默认值、参数说明
	rootCmd.PersistentFlags().StringVar(&cfgPath, "config", "", "指定文件存储路径（默认值为 $HOME/.你的应用名）")
}

3. 实现路径解析逻辑
   在initConfig函数中处理路径解析，自动展开~开头的用户目录，路径不存在时自动创建：

func initConfig() {
	var targetPath string
	if cfgPath != "" {
		// 展开~开头的用户目录路径
		if strings.HasPrefix(cfgPath, "~") {
			homeDir, err := os.UserHomeDir()
			if err != nil {
				log.Fatalf("获取用户目录失败: %v", err)
			}
			targetPath = filepath.Join(homeDir, cfgPath[1:])
		} else {
			// 解析相对/绝对路径
			var err error
			targetPath, err = filepath.Abs(cfgPath)
			if err != nil {
				log.Fatalf("路径解析失败: %v", err)
			}
		}
	// 未传--config参数时使用默认路径
	} else {
		homeDir, err := os.UserHomeDir()
		if err != nil {
			log.Fatalf("获取用户目录失败: %v", err)
		}
		targetPath = filepath.Join(homeDir, ".你的应用名")
	}
	// 路径不存在则自动创建
	if _, err := os.Stat(targetPath); os.IsNotExist(err) {
		if err := os.MkdirAll(targetPath, 0755); err != nil {
			log.Fatalf("创建存储路径失败: %v", err)
		}
	}
	// 此处可将targetPath存入全局配置对象，供后续业务逻辑使用
	log.Println("当前使用存储路径：", targetPath)
}

适配你给出的命令格式（可选） #

如果要严格保留app --config path "~/Documents/"的格式，可将--config设为布尔flag，通过位置参数读取路径：

1. 绑定布尔类型flag

func init() {
	rootCmd.PersistentFlags().Bool("config", false, "指定文件存储路径，示例：app --config path \"~/Documents/\"")
}

2. 在root命令的Run逻辑中读取参数

var rootCmd = &cobra.Command{
	Use:   "app",
	Short: "你的应用说明",
	Run: func(cmd *cobra.Command, args []string) {
		isSetConfig, _ := cmd.Flags().GetBool("config")
		if isSetConfig {
			if len(args) < 2 || args[0] != "path" {
				log.Fatal("参数格式错误，正确示例：app --config path \"~/Documents/\"")
			}
			cfgPath = args[1]
			// 后续走上述initConfig中的路径解析逻辑即可
		}
	},
}

内容的提问来源于stack exchange，提问作者scottdomini