如何实现Cobra命令中标志必须作为首个输入参数？

完全可以用Cobra实现这个需求，推荐两种方案，其中第一种更贴合Cobra的设计理念：

方案一：使用子命令（推荐） #

Cobra的核心设计就是用子命令区分不同动作，这种方式不需要额外的参数验证逻辑，完全依赖Cobra自身的解析机制：

1. 创建config主命令，再给它添加set和unset两个子命令
2. 每个子命令定义需要的位置参数（配置项名称和值）

示例命令变为：

prog_name config set config_var var_value
prog_name config unset config_var var_value

代码示例片段：

var setCmd = &cobra.Command{
    Use:   "set <config_var> <var_value>",
    Short: "Set a configuration variable",
    Args:  cobra.ExactArgs(2), // 强制要求两个位置参数
    RunE: func(cmd *cobra.Command, args []string) error {
        configVar := args[0]
        varValue := args[1]
        // 执行设置配置的逻辑
        return nil
    },
}

var unsetCmd = &cobra.Command{
    Use:   "unset <config_var> <var_value>",
    Short: "Unset a configuration variable",
    Args:  cobra.ExactArgs(2),
    RunE: func(cmd *cobra.Command, args []string) error {
        configVar := args[0]
        varValue := args[1]
        // 执行移除配置的逻辑
        return nil
    },
}

var configCmd = &cobra.Command{
    Use:   "config",
    Short: "Manage configuration variables",
}

func init() {
    rootCmd.AddCommand(configCmd)
    configCmd.AddCommand(setCmd, unsetCmd)
}

这种方式下，Cobra会自动处理参数验证、帮助文档生成，还能避免标志和位置参数的顺序问题。

方案二：使用互斥标志（贴合你提出的命令格式） #

如果坚持要用--set/--unset标志的格式，可以利用Cobra的互斥标志组和参数验证功能，不需要手动解析os.Args：

1. 给config命令添加两个布尔类型的互斥标志--set和--unset
2. 在命令的RunE函数中检查哪个标志被激活，并验证位置参数数量

代码示例片段：

import "fmt"
import "github.com/spf13/cobra"

var configCmd = &cobra.Command{
    Use:   "config",
    Short: "Manage configuration variables",
    RunE: func(cmd *cobra.Command, args []string) error {
        setEnabled, _ := cmd.Flags().GetBool("set")
        unsetEnabled, _ := cmd.Flags().GetBool("unset")

        // 检查位置参数数量是否符合要求
        if len(args) != 2 {
            return fmt.Errorf("must provide exactly two arguments: config_var and var_value")
        }

        configVar := args[0]
        varValue := args[1]

        if setEnabled {
            // 执行设置配置的逻辑
            fmt.Printf("Setting %s to %s\n", configVar, varValue)
        } else if unsetEnabled {
            // 执行移除配置的逻辑
            fmt.Printf("Unsetting %s with value %s\n", configVar, varValue)
        }
        return nil
    },
}

func init() {
    rootCmd.AddCommand(configCmd)
    // 添加标志并设置互斥
    configCmd.Flags().Bool("set", false, "Set a configuration variable")
    configCmd.Flags().Bool("unset", false, "Unset a configuration variable")
    configCmd.MarkFlagsMutuallyExclusive("set", "unset")
}

这里用MarkFlagsMutuallyExclusive让两个标志互斥，Cobra会自动阻止同时设置两个标志的情况，不需要手动验证。另外，Cobra的解析规则允许标志出现在命令后的任何位置，但只要确保激活其中一个标志且提供两个位置参数，就能正确执行动作。

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