1

我正在使用Cobra为我的应用程序更新一些 cli。我想让这个命令成为必需的，这意味着如果应用程序找不到它正在寻找的参数，它应该会失败。

package commands

import (
    "github.com/spf13/cobra"
    "errors"
    "fmt"
)

var (
    Env string
)

var RootCmd = &cobra.Command{
    Use:   "myapp",
    Short: "tool",
    Long:  `tool`,
    Run: func(cmd *cobra.Command, args []string)  {
        // Root command does nothing
    },
}

func init() {
    RootCmd.AddCommand(Environment)
}

var Environment = &cobra.Command{
    Use:   "env",
    Short: "Specify Environment to run against",
    Long: `Can be dev or prod`,
    Args: func(cmd *cobra.Command, args []string) error {
        if len(args) != 1 {
            return errors.New("requires at least one arg")
        }

        if args[0] == "dev" || args[0] == "prod" {
            return nil
        }else {
            return errors.New("input can only be dev or prod")
        }
        return fmt.Errorf("invalid env specified: %s", args[0])
    },
    PreRunE: func(cmd *cobra.Command, args []string) error {
        if len(args) != 1 {
          return fmt.Errorf("env is required")
        }
      return nil
    },
    Run: func(cmd *cobra.Command, args []string) {
        Env = args[0]
    },
}

主包是

package main

import (
    "fmt"
    "log"
    "os"
    "util"
    "commands"
)

func main() {
    log.Println("Executing")

    if err := commands.RootCmd.Execute(); err != nil {
        fmt.Println(err)
        os.Exit(1)
    }
    log.Println("Executing")
}

现在，如果我在没有任何环境的情况下将其作为 ./myApp 运行，它不会抱怨它。但是，如果我在 myapp 之后使用 env ，那么它会激活该功能并运行所有验证。

4

2 回答 2

1

您可以让函数的主体处理它，也许通过打印帮助并以不成功的方式退出：

Run: func(cmd *cobra.Command, args []string)  {
    // Root command does nothing
    cmd.Help()
    os.Exit(1)
},
于 2017-11-13T19:50:04.770 回答
0

从 中省略Run(and RunE) 字段cobra.Command将要求提供有效的子命令：

var RootCmd = &cobra.Command{
    Use:   "myapp",
    Short: "tool",
    Long:  `tool long help...`,
}

如果命令行上没有给出子命令，Cobra 将打印出命令的Help()文本，其中将包括根命令的Long帮助文本和所有子命令的自动生成的使用帮助。

于 2020-12-03T16:56:32.160 回答