forked from toolshed/abra
refactor: urfave v3
This commit is contained in:
68
vendor/github.com/urfave/cli/v3/completion.go
generated
vendored
Normal file
68
vendor/github.com/urfave/cli/v3/completion.go
generated
vendored
Normal file
@ -0,0 +1,68 @@
|
||||
package cli
|
||||
|
||||
import (
|
||||
"context"
|
||||
"embed"
|
||||
"fmt"
|
||||
"sort"
|
||||
)
|
||||
|
||||
const (
|
||||
completionCommandName = "generate-completion"
|
||||
)
|
||||
|
||||
var (
|
||||
//go:embed autocomplete
|
||||
autoCompleteFS embed.FS
|
||||
|
||||
shellCompletions = map[string]renderCompletion{
|
||||
"bash": getCompletion("autocomplete/bash_autocomplete"),
|
||||
"ps": getCompletion("autocomplete/powershell_autocomplete.ps1"),
|
||||
"zsh": getCompletion("autocomplete/zsh_autocomplete"),
|
||||
"fish": func(c *Command) (string, error) {
|
||||
return c.ToFishCompletion()
|
||||
},
|
||||
}
|
||||
)
|
||||
|
||||
type renderCompletion func(*Command) (string, error)
|
||||
|
||||
func getCompletion(s string) renderCompletion {
|
||||
return func(c *Command) (string, error) {
|
||||
b, err := autoCompleteFS.ReadFile(s)
|
||||
return string(b), err
|
||||
}
|
||||
}
|
||||
|
||||
func buildCompletionCommand() *Command {
|
||||
return &Command{
|
||||
Name: completionCommandName,
|
||||
Hidden: true,
|
||||
Action: completionCommandAction,
|
||||
}
|
||||
}
|
||||
|
||||
func completionCommandAction(ctx context.Context, cmd *Command) error {
|
||||
var shells []string
|
||||
for k := range shellCompletions {
|
||||
shells = append(shells, k)
|
||||
}
|
||||
|
||||
sort.Strings(shells)
|
||||
|
||||
if cmd.Args().Len() == 0 {
|
||||
return Exit(fmt.Sprintf("no shell provided for completion command. available shells are %+v", shells), 1)
|
||||
}
|
||||
s := cmd.Args().First()
|
||||
|
||||
if rc, ok := shellCompletions[s]; !ok {
|
||||
return Exit(fmt.Sprintf("unknown shell %s, available shells are %+v", s, shells), 1)
|
||||
} else if c, err := rc(cmd); err != nil {
|
||||
return Exit(err, 1)
|
||||
} else {
|
||||
if _, err = cmd.Writer.Write([]byte(c)); err != nil {
|
||||
return Exit(err, 1)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
Reference in New Issue
Block a user