无参数匹配的 Golang Cobra 子命令

Question

我想使用 Cobra 创建 Golang CLI。目前，它运行良好。我创建了类似

foobar create --username johndoe

的东西。但我需要像

foobar create user --username johndoe

这样的子命令。有

user

子命令。

我已经创建了参数匹配的，

var applicationCmd = &cobra.Command{
   Use:   "application",
   Short: "Create/Read/Update application",
   Long: `You can create any user or partner.`,
   Run: (cmd *cobra.Command, args []string) {
        if len(args) < 1 {
            fmt.Printf("%sPlease choose any object that you want to change e.g (user, partner)%s\n", chalk.Red, chalk.Reset)
        }
        if stringUtil.Contains(args, "user") {
            createUser()
        }
        if stringUtil.Contains(args, "partner") {
            createPartner()
        }
   },
}

但我认为有更好的方法使用 cobra 来做到这一点，而不是使用像这样的参数匹配。有什么建议吗？

Answer 1

我有点太过分了；但是，是的，appCmd.Addcommand(newCmd) 就是您正在寻找的。

（在你发布答案之前我已经开始研究这个问题了，但我不想废弃代码）

package main

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

var rootCmd = &cobra.Command{
    Use:   "app",
    Run: func (cmd *cobra.Command, args []string) {
    },
}

var makeU = &cobra.Command{
    Use:   "makeU",
    Short: "makes a user",
    Long: "Prints stuff about the user. You could also use the flags in your addPartner() function",
    Run: func (cmd *cobra.Command, args []string) {
        fmt.Println("User's name: " + uName)
        fmt.Println("User's number: " + uNumber)
        fmt.Println("User's other stuff: " + uOtherStuff)
    },
}

var makeP = &cobra.Command{
    Use:   "makeP",
    Short: "makes a partner",
    Long: "Prints stuff about the partner. You could also use the flags in your addPartner() function",
    Run: func (cmd *cobra.Command, args []string) {
        fmt.Println("Partner's name: " + uName)
        fmt.Println("Partner's number: " + uNumber)
        fmt.Println("Partner's other stuff: " + uOtherStuff)
    },
}

// adds commands
func addCommands() {
    rootCmd.AddCommand(makeU)
    rootCmd.AddCommand(makeP)
}

// adds flags
var pName, pNumber, pOtherStuff string
var uName, uNumber, uOtherStuff string
func init() {
    makeU.Flags().StringVar(&uName, "username", "default", "--username=<user's name>")
    makeU.Flags().StringVar(&uNumber, "usernumber", "default", "--usernumber=<user's number>")
    makeU.Flags().StringVar(&uOtherStuff, "userother", "default", "--userother=<user's other stuff>")
    makeP.Flags().StringVar(&pName, "partnername", "default", "--partnername=<partner's name>")
    makeP.Flags().StringVar(&pNumber, "partnernumber", "default", "--partnernumber=<partner's number>")
    makeP.Flags().StringVar(&pOtherStuff, "partnerother", "default", "--partnerother=<partner's other stuff>")
}


func main() {
    addCommands()
    rootCmd.Execute()
}

Answer 2

我已经解决了这个问题。只需将新的眼镜蛇命令添加到

applicationCmd

，就像这样

applicationCmd.AddCommand(yourNewCommand)