/
cmd.go
101 lines (85 loc) · 1.82 KB
/
cmd.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
package yago
import (
"fmt"
"log"
"os"
"strings"
"github.com/spf13/cobra"
)
type Cmd struct {
*cobra.Command
}
func NewCmd() *Cmd {
cmd := &Cmd{&cobra.Command{
// PersistentPreRun: func(cmd *cobra.Command, args []string) {
// conf, _ := cmd.Flags().GetString("c")
// Config = NewAppConfig(conf)
// },
Run: func(cmd *cobra.Command, args []string) {
NewApp().Run()
},
}}
cmd.PersistentFlags().StringP("c", "c", defaultCfgPath(), "config file path")
return cmd
}
func (c *Cmd) LoadCmdRouter() {
if len(CmdRouterMap) == 0 {
return
}
baseCmdMap := make(map[string]*cobra.Command)
for use, router := range CmdRouterMap {
useSlice := strings.Split(use, "/")
length := len(useSlice)
if length == 0 {
log.Fatalf("add cmd router failed: %s", use)
}
var baseCmdSlice []string
var cmd string
if length == 1 {
baseCmdSlice = useSlice
cmd = ""
}
if length >= 2 {
baseCmdSlice = useSlice[:(length - 1)]
cmd = useSlice[length-1]
}
baseCmdStr := strings.Join(baseCmdSlice, "/")
var baseCmd *cobra.Command
if _, ok := baseCmdMap[baseCmdStr]; !ok {
baseCmd = &cobra.Command{
Use: baseCmdStr,
Short: fmt.Sprintf("Help about %s command", baseCmdStr),
}
baseCmdMap[baseCmdStr] = baseCmd
c.AddCommand(baseCmd)
} else {
baseCmd = baseCmdMap[baseCmdStr]
}
var rootCmd *cobra.Command
if cmd == "" {
baseCmd.Short = router.Short
baseCmd.Run = router.Action
rootCmd = baseCmd
} else {
subCmd := &cobra.Command{
Use: cmd,
Short: router.Short,
Run: router.Action,
}
baseCmd.AddCommand(subCmd)
rootCmd = subCmd
}
if len(router.Args) > 0 {
for _, arg := range router.Args {
arg.SetFlag(rootCmd)
}
}
}
}
func (c *Cmd) RunCmd() {
c.LoadCmdRouter()
if err := c.Execute(); err != nil {
log.Println(err)
os.Exit(1)
}
}