83 lines
2.0 KiB
Go
83 lines
2.0 KiB
Go
// License: GPLv3 Copyright: 2022, Kovid Goyal, <kovid at kovidgoyal.net>
|
|
|
|
package completion
|
|
|
|
import (
|
|
"strings"
|
|
)
|
|
|
|
func complete_word(word string, completions *Completions, only_args_allowed bool, expecting_arg_for *option) {
|
|
cmd := Completions.current_cmd
|
|
if expecting_arg_for != nil {
|
|
if expecting_arg_for.completion_for_arg != nil {
|
|
expecting_arg_for.completion_for_arg(completions, word)
|
|
}
|
|
return
|
|
}
|
|
if !only_args_allowed && strings.HasPrefix(word, "-") {
|
|
// handle single letter multiple options -abcd
|
|
// handle standalone --long-opt
|
|
// handle long opt ends with =
|
|
// handle long opt containing =
|
|
completions.add_options_group(cmd.options, word)
|
|
return
|
|
}
|
|
if cmd.completion_for_arg != nil {
|
|
cmd.completion_for_arg(completions, word)
|
|
}
|
|
return
|
|
}
|
|
|
|
func parse_args(cmd *command, words []string, completions *Completions) {
|
|
completions.current_cmd = cmd
|
|
if len(words) == 0 {
|
|
complete_word("", completions, false, nil)
|
|
return
|
|
}
|
|
|
|
var expecting_arg_for *option
|
|
only_args_allowed := false
|
|
arg_num := 0
|
|
|
|
for i, word := range words {
|
|
cmd = completions.current_cmd
|
|
is_last_word := i == len(words)-1
|
|
if is_last_word {
|
|
complete_word(word, completions, only_args_allowed, expecting_arg_for)
|
|
} else {
|
|
if expecting_arg_for != nil {
|
|
expecting_arg_for = nil
|
|
continue
|
|
}
|
|
if word == "--" {
|
|
only_args_allowed = true
|
|
continue
|
|
}
|
|
arg_num++
|
|
if !only_args_allowed && strings.HasPrefix(word, "-") {
|
|
// handle single letter multiple options -abcd
|
|
// handle standalone --long-opt
|
|
// handle long opt ends with =
|
|
// handle long opt containing =
|
|
continue
|
|
}
|
|
if len(cmd.subcommands) > 0 && arg_num == 1 {
|
|
sc := cmd.find_subcommand(word)
|
|
if sc == nil {
|
|
only_args_allowed = true
|
|
continue
|
|
}
|
|
completions.current_cmd = sc
|
|
cmd = sc
|
|
arg_num = 0
|
|
only_args_allowed = false
|
|
} else if cmd.stop_processing_at_arg > 0 && arg_num >= cmd.stop_processing_at_arg {
|
|
return
|
|
} else {
|
|
only_args_allowed = true
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
}
|