KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Allow generating completions for multiple command lines for more efficient testing

Browse Source
This commit is contained in:
Kovid Goyal 2022-09-09 15:17:34 +05:30
parent 54378de52b
commit d679ea2cdf
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tools/completion/main.go
View File

@ -14,18 +14,18 @@ import (
"kitty/tools/utils" "kitty/tools/utils"
) )
func json_input_parser(data []byte, shell_state map[string]string) ([]string, error) { func json_input_parser(data []byte, shell_state map[string]string) ([][]string, error) {
ans := make([]string, 0, 32) ans := make([][]string, 0, 32)
err := json.Unmarshal(data, &ans) err := json.Unmarshal(data, &ans)
return ans, err return ans, err
} }
func json_output_serializer(completions *Completions, shell_state map[string]string) ([]byte, error) { func json_output_serializer(completions []*Completions, shell_state map[string]string) ([]byte, error) {
return json.Marshal(completions) return json.Marshal(completions)
} }
type parser_func func(data []byte, shell_state map[string]string) ([]string, error) type parser_func func(data []byte, shell_state map[string]string) ([][]string, error)
type serializer_func func(completions *Completions, shell_state map[string]string) ([]byte, error) type serializer_func func(completions []*Completions, shell_state map[string]string) ([]byte, error)
var input_parsers = make(map[string]parser_func, 4) var input_parsers = make(map[string]parser_func, 4)
var output_serializers = make(map[string]serializer_func, 4) var output_serializers = make(map[string]serializer_func, 4)
@ -60,7 +60,7 @@ func main(args []string) error {
if err != nil { if err != nil {
return err return err
} }
argv, err := input_parser(data, shell_state) all_argv, err := input_parser(data, shell_state)
if err != nil { if err != nil {
return err return err
} }
@ -69,8 +69,11 @@ func main(args []string) error {
re(&root) re(&root)
} }
completions := root.GetCompletions(argv) all_completions := make([]*Completions, 0, 1)
output, err := output_serializer(completions, shell_state) for _, argv := range all_argv {
all_completions = append(all_completions, root.GetCompletions(argv))
}
output, err := output_serializer(all_completions, shell_state)
if err == nil { if err == nil {
_, err = os.Stdout.Write(output) _, err = os.Stdout.Write(output)
} }