不知大家有没有在使用Git命令、Linux的yum命令、Go命令、Maven命令的时候感觉到非常的酷,比如你刚刚拿到一个Go的开源项目,初始化时只需要输入go mod tidy
进行对依赖的下载,或者是git clone xxx
之后拉下来一个GitHub上的项目,mvn package
或者go build
就可以将整合项目打包成一个可执行的文件,当然这些操作现在大多数都能通过可视化的UI界面进行,但是就少了一些韵味,还是使用原生的CLI工具最能体现程序执行的过程,并且真是的生产环境中大多数也只能通过这种命令的方式来进行,因此我们就有必要搞一下这个大蟒蛇工具——Cobra。
1 简介
GitHub:https://github.com/spf13/cobra
Cobra是一个用于创建强大的现代CLI应用程序的库。在很多围棋项目中使用,比如Kubernetes, Hugo和GitHub CLI等等。
那么什么是CLI呢,简单来讲就是命令行工具界面,Command Line Interface for batch scripting的简称,在我们开发中广泛的使用,比如你打开Git Bash,输入git clone XXX,这个就是操作Git的命令行工具界面。
除此之外,Cobra还提供:
- 简单的基于子命令的cli:应用服务器,应用获取,等等。
- 完全兼容posix的标志(包括短版本和长版本)
- 嵌套的子命令
- 全局、本地和级联标志
- 命令和标志的自动帮助生成
- 自动识别-h、——help等帮助标志。
- 为您的应用程序自动生成shell自动完成(bash, zsh, fish, powershell)
- 为您的应用程序自动生成的手册页
- 命令别名,以便您可以更改而不破坏它们
- 定义自己的帮助、用法等的灵活性。
什么是命令&参数&标识符?
Cobra 是构建在命令、参数和标识符之上的:
Commands
表示执行动作Args
就是执行参数Flags
是这些动作的标识符
基本的执行命令如下所示:
$ APPNAME Command Args --Flags # 或者 $ APPNAME Command --Flags Args
2 安装
2.1 安装Cobra-cli脚手架工具
go install github.com/spf13/cobra-cli@latest
下载完成之后会在我们的$GOPATH目录的bin目录下生成一个二进制的文件cobra-cli.exe(Windows下)或cobra-cli(LInux下),如果出现问题,如:
我们可以直接找到$GO_PATH\pkg\mod\github.com\spf13\cobra-cli@v1.3.0目录,直接go build自己编译一个二进制文件。
之后一步,为了方便使用我们可以将生成的二进制文件放在环境变量里,让他在任何地方都能使用,步骤过于简单就忽略了哈。
2.2 在项目中下载Cobra依赖
go get -u github.com/spf13/cobra@latest
3 使用方式
其实在使用方式这一块有两种方法,一种是自己直接写代码,另一种就是使用刚刚我们安装的cobra-cli来生成脚手架代码,我们就在Hello World里面将cobra-cli生成代码试一遍。
3.1 Hello World
首先在项目根目录下打开终端(必须是Go Module项目),然后输入命令cobra-cli init
就会在项目中生成如下代码:
我们看下具体的代码:
main.go
package main import "go-cobra/cmd" func main() { cmd.Execute() }
root.go
package cmd import ( "os" "github.com/spf13/cobra" ) // rootCmd represents the base command when called without any subcommands var rootCmd = &cobra.Command{ Use: "go-cobra", Short: "A brief description of your application", Long: `A longer description that spans multiple lines and likely contains examples and usage of using your application. For example: Cobra is a CLI library for Go that empowers applications. This application is a tool to generate the needed files to quickly create a Cobra application.`, // Uncomment the following line if your bare application // has an action associated with it: // Run: func(cmd *cobra.Command, args []string) { }, } // Execute adds all child commands to the root command and sets flags appropriately. // This is called by main.main(). It only needs to happen once to the rootCmd. func Execute() { err := rootCmd.Execute() if err != nil { os.Exit(1) } } func init() { // Here you will define your flags and configuration settings. // Cobra supports persistent flags, which, if defined here, // will be global for your application. // rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.go-cobra.yaml)") // Cobra also supports local flags, which will only run // when this action is called directly. rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle") }
我们将root.go里面的代码进行部分修改:
package cmd import ( "fmt" "github.com/spf13/cobra" "os" ) var name string var rootCmd = &cobra.Command{ Use: "hello", Run: func(cmd *cobra.Command, args []string) { fmt.Println("hello ", name) }, } func Execute() { err := rootCmd.Execute() if err != nil { os.Exit(1) } } func init() { rootCmd.Flags().StringVarP(&name, "name", "", "world", "") }
测试:
$ go build $ .\go-cobra.exe --name zs hello zs $ .\go-cobra.exe hello world
3.2 开发自己的Cli命令
使用cobra-cli脚手架,或者也可以自己手写如下代码
cobra-cli add hi
这个时候我们就会发现多了一个文件,当然也可以自己手写代码:
在hi.go文件中代码:
package cmd import ( "fmt" "github.com/spf13/cobra" ) var ( age int like []string address string ) var hiCmd = &cobra.Command{ Use: "hi", Short: "", Long: "", Run: func(cmd *cobra.Command, args []string) { Print() }, } func init() { rootCmd.AddCommand(hiCmd) hiCmd.Flags().StringVarP(&name, "name", "", "world", "") hiCmd.Flags().IntVarP(&age, "age", "", 1, "") hiCmd.Flags().StringSliceVarP(&like, "like", "", []string{}, "") hiCmd.Flags().StringVarP(&address, "address", "", "Beijing", "") } func Print() { fmt.Println("name:", name) fmt.Println("age:", age) fmt.Println("like:", like) fmt.Println("address:", address) }
测试:
.\go-cobra.exe hi --name zs --age 100 --like Coding,Running --address ShangHai name: zs age: 100 like: [Coding Running] address: ShangHai
3.3 规则和扩展使用
3.3.1 我们不难看出规则
- 新增命令就是自定义&cobra.Command
- 新增标识符就是命令.Flags().xxx
- 新增的命令都需要在rootCmd中添加,新增的标识符都要在init中添加
3.3.2 小扩展
一些常用到的方法:
- NoArgs:如果有任何参数,命令行将会报错
- ArbitraryArgs: 命令行将会接收任何参数
- OnlyValidArgs: 如果有如何参数不属于 Command 的 ValidArgs 字段,命令行将会报错
- MinimumNArgs(int): 如果参数个数少于 N 个,命令行将会报错
- MaximumNArgs(int): 如果参数个数多于 N 个,命令行将会报错
- ExactArgs(int): 如果参数个数不等于 N 个,命令行将会报错
- RangeArgs(min, max): 如果参数个数不在 min 和 max 之间, 命令行将会报错
3.3.3 自定义帮助命令
... var hiCmd = &cobra.Command{ Use: "hi", Short: "sss", Long: "lll", Run: func(cmd *cobra.Command, args []string) { if len(args) == 0 { cmd.Help() return } Print() }, } ...
测试:
$ .\go-cobra.exe hi Usage: hello hi [flags] Usage: hello hi [flags] Flags: --address string (default "Beijing") --age int (default 1) -h, --help help for hi --like strings --name string (default "world")
4 小总结
本文介绍了Cobra的最基本也是最常用的使用部分,但是Cobra仍然有很多优秀的操作值得我们学习。拜拜喽~