glenux-archive/spf13--cobra
1
0
Fork 0
mirror of https://github.com/spf13/cobra synced 2024-11-25 07:07:15 +00:00
Code Issues Projects Releases Packages Wiki Activity

Conform more to the style

Browse source
This commit is contained in:
Brandon Roehl 2018-06-27 12:46:06 -05:00
parent 009ebb14c9
commit a66d0ae17a
1 changed files with 22 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
zsh_completions.go
View file

@ -21,14 +21,23 @@ func (c *Command) GenZshCompletionFile(filename string) error {
return c.GenZshCompletion(outFile) return c.GenZshCompletion(outFile)
} }
func argName(cmd *Command) string {
for cmd.HasParent() {
cmd = cmd.Parent()
}
name := fmt.Sprintf("%s_cmd_args", cmd.Name())
return strings.Replace(name, "-", "_",-1)
}
// GenZshCompletion generates a zsh completion file and writes to the passed writer. // GenZshCompletion generates a zsh completion file and writes to the passed writer.
func (c *Command) GenZshCompletion(w io.Writer) error { func (c *Command) GenZshCompletion(w io.Writer) error {
buf := new(bytes.Buffer) buf := new(bytes.Buffer)
writeHeader(buf, c) writeHeader(buf, c)
maxDepth := maxDepth(c) maxDepth := maxDepth(c)
writeLevelMapping(buf, maxDepth) fmt.Fprintf(buf, "_%s() {\n", c.Name())
writeLevelMapping(buf, maxDepth, c)
writeLevelCases(buf, maxDepth, c) writeLevelCases(buf, maxDepth, c)
fmt.Fprintf(buf, "}\n_%s \"$@\"\n", c.Name())
_, err := buf.WriteTo(w) _, err := buf.WriteTo(w)
return err return err
@ -52,16 +61,15 @@ func maxDepth(c *Command) int {
return 1 + maxDepthSub return 1 + maxDepthSub
} }
func writeLevelMapping(w io.Writer, numLevels int) { func writeLevelMapping(w io.Writer, numLevels int, root *Command) {
fmt.Fprintln(w, "local -a cmd_options") fmt.Fprintln(w, `local context curcontext="$curcontext" state line`)
fmt.Fprintln(w, `typeset -A opt_args`)
fmt.Fprintln(w, `_arguments -C \`) fmt.Fprintln(w, `_arguments -C \`)
fmt.Fprintln(w, ` $jamf_pro_options \`)
for i := 1; i <= numLevels; i++ { for i := 1; i <= numLevels; i++ {
fmt.Fprintf(w, ` '%d: :->level%d' \`, i, i) fmt.Fprintf(w, " '%d: :->level%d' \\\n", i, i)
fmt.Fprintln(w)
} }
fmt.Fprintf(w, ` '%d: :%s'`, numLevels+1, "_files") fmt.Fprintf(w, " $%s \\\n", argName(root))
fmt.Fprintln(w) fmt.Fprintln(w, ` '*: :_files'`)
} }
func writeLevelCases(w io.Writer, maxDepth int, root *Command) { func writeLevelCases(w io.Writer, maxDepth int, root *Command) {
@ -75,10 +83,10 @@ func writeLevelCases(w io.Writer, maxDepth int, root *Command) {
fmt.Fprintln(w, "esac") fmt.Fprintln(w, "esac")
} }
func writeLevel(w io.Writer, root *Command, level int) { func writeLevel(w io.Writer, root *Command, l int) {
fmt.Fprintf(w, " level%d)\n", level) fmt.Fprintf(w, " level%d)\n", l)
fmt.Fprintf(w, " case $words[%d] in\n", level) fmt.Fprintf(w, " case $words[%d] in\n", l)
for _, c := range filterByLevel(root, level) { for _, c := range filterByLevel(root, l) {
writeCommandArgsBlock(w, c) writeCommandArgsBlock(w, c)
} }
fmt.Fprintln(w, " *)") fmt.Fprintln(w, " *)")
@ -96,11 +104,11 @@ func writeCommandArgsBlock(w io.Writer, c *Command) {
defer fmt.Fprintln(w, " ;;") defer fmt.Fprintln(w, " ;;")
} }
if len(flags) > 0 { if len(flags) > 0 {
fmt.Fprintln(w, " cmd_options=(") fmt.Fprintf(w, " %s=(\n", argName(c))
for _, flag := range flags { for _, flag := range flags {
fmt.Fprintf(w, " %s\n", flag) fmt.Fprintf(w, " %s\n", flag)
} }
fmt.Fprintln(w, "\n )") fmt.Fprintln(w, " )")
} }
if len(names) > 0 { if len(names) > 0 {
fmt.Fprintf(w, " _values 'command' '%s'\n", strings.Join(names, "' '")) fmt.Fprintf(w, " _values 'command' '%s'\n", strings.Join(names, "' '"))