170 lines
3.2 KiB
Go
170 lines
3.2 KiB
Go
package cmd
|
||
|
||
import (
|
||
"agent-wdd/config"
|
||
"fmt"
|
||
"io"
|
||
"os"
|
||
"strings"
|
||
|
||
"github.com/spf13/cobra"
|
||
)
|
||
|
||
var rootCmd = &cobra.Command{
|
||
Use: "wdd",
|
||
Short: "wdd应用程序是wdd封装的NB工具",
|
||
Long: `使用golang的强大特性，加上 WDD 的高超技术，打造一个方便、高效、适用于各种LINUX系统的瑞士军刀工具!
|
||
尽情享用！ 尽情享用！ 尽情享用！`,
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
// Do Stuff Here
|
||
},
|
||
}
|
||
|
||
func init() {
|
||
|
||
cobra.OnInitialize(config.InitConfig)
|
||
}
|
||
|
||
func Execute() {
|
||
|
||
// 1. base命令
|
||
baseCmd := &cobra.Command{
|
||
Use: "base",
|
||
Short: "服务器基础操作",
|
||
}
|
||
addBaseSubcommands(baseCmd)
|
||
|
||
// 2. zsh命令
|
||
zshCmd := &cobra.Command{
|
||
Use: "zsh",
|
||
Short: "zsh相关的内容",
|
||
}
|
||
addZshSubcommands(zshCmd)
|
||
|
||
// 3. proxy命令
|
||
proxyCmd := &cobra.Command{
|
||
Use: "proxy",
|
||
Short: "主机代理相关的内容",
|
||
}
|
||
addProxySubcommands(proxyCmd)
|
||
|
||
// 4. acme命令
|
||
acmeCmd := &cobra.Command{
|
||
Use: "acme",
|
||
Short: "acme相关的内容",
|
||
}
|
||
addAcmeSubcommands(acmeCmd)
|
||
|
||
// 5. wdd命令
|
||
wddCmd := &cobra.Command{
|
||
Use: "wdd",
|
||
Short: "WDD相关操作",
|
||
}
|
||
addWddSubcommands(wddCmd)
|
||
|
||
// 6. security命令
|
||
securityCmd := &cobra.Command{
|
||
Use: "security",
|
||
Short: "安全相关操作",
|
||
}
|
||
|
||
// 7. info命令
|
||
infoCmd := &cobra.Command{
|
||
Use: "info",
|
||
Short: "打印主机详细信息",
|
||
}
|
||
addInfoSubcommands(infoCmd)
|
||
|
||
// 8. version命令
|
||
versionCmd := &cobra.Command{
|
||
Use: "version",
|
||
Short: "打印版本信息",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
// 实现version逻辑
|
||
fmt.Println("版本信息: 1.0.0")
|
||
},
|
||
}
|
||
|
||
// 9. config命令
|
||
configCmd := &cobra.Command{
|
||
Use: "config",
|
||
Short: "配置文件管理",
|
||
}
|
||
addConfigSubcommands(configCmd)
|
||
|
||
// 10. download命令
|
||
downloadCmd := &cobra.Command{
|
||
Use: "download",
|
||
Short: "文件下载，直接下载 [url] [dest_path]",
|
||
}
|
||
|
||
addDownloadSubcommands(downloadCmd)
|
||
|
||
helpCmd := &cobra.Command{
|
||
Use: "help",
|
||
Short: "帮助信息",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
// 在你的命令执行函数中
|
||
// cmd.Root().SetHelpFunc(func(c *cobra.Command, s []string) {
|
||
// // 直接使用默认帮助模板
|
||
// c.Usage()
|
||
// })
|
||
|
||
// 自定义实现 打印全部的命令
|
||
printAllCommands(cmd.Root(), os.Stdout)
|
||
},
|
||
}
|
||
|
||
// 添加所有根命令
|
||
rootCmd.AddCommand(
|
||
baseCmd,
|
||
zshCmd,
|
||
proxyCmd,
|
||
acmeCmd,
|
||
wddCmd,
|
||
securityCmd,
|
||
infoCmd,
|
||
versionCmd,
|
||
configCmd,
|
||
downloadCmd,
|
||
helpCmd,
|
||
)
|
||
|
||
if err := rootCmd.Execute(); err != nil {
|
||
fmt.Println(err)
|
||
}
|
||
|
||
}
|
||
|
||
func printAllCommands(cmd *cobra.Command, writer io.Writer) {
|
||
// 打印命令标题
|
||
fmt.Fprintf(writer, "Available commands:\n")
|
||
|
||
// 递归打印命令树
|
||
var traverse func(*cobra.Command, int)
|
||
traverse = func(c *cobra.Command, depth int) {
|
||
if c.Hidden || c.Deprecated != "" {
|
||
return
|
||
}
|
||
|
||
// 生成命令前缀缩进
|
||
indent := strings.Repeat(" ", depth)
|
||
|
||
// 打印当前命令信息
|
||
fmt.Fprintf(writer, "%s%-20s%s\n",
|
||
indent,
|
||
c.Name(),
|
||
strings.TrimSpace(c.Short))
|
||
|
||
// 递归子命令
|
||
for _, subCmd := range c.Commands() {
|
||
traverse(subCmd, depth+1)
|
||
}
|
||
}
|
||
|
||
// 从根命令开始遍历
|
||
for _, c := range cmd.Commands() {
|
||
traverse(c, 0)
|
||
}
|
||
}
|