34 lines
1.1 KiB
Go
34 lines
1.1 KiB
Go
package command
|
|
|
|
import (
|
|
"github.com/posener/complete"
|
|
)
|
|
|
|
// This file contains some re-usable predictors for auto-complete. The
|
|
// command-specific autocomplete configurations live within each command's
|
|
// own source file, as AutocompleteArgs and AutocompleteFlags methods on each
|
|
// Command implementation.
|
|
|
|
// For completing the value of boolean flags like -foo false
|
|
var completePredictBoolean = complete.PredictSet("true", "false")
|
|
|
|
// We don't currently have a real predictor for module sources, but
|
|
// we'll probably add one later.
|
|
var completePredictModuleSource = complete.PredictAnything
|
|
|
|
type completePredictSequence []complete.Predictor
|
|
|
|
func (s completePredictSequence) Predict(a complete.Args) []string {
|
|
// Only one level of command is stripped off the prefix of a.Completed
|
|
// here, so nested subcommands like "workspace new" will need to provide
|
|
// dummy entries (e.g. complete.PredictNothing) as placeholders for
|
|
// all but the first subcommand. For example, "workspace new" needs
|
|
// one placeholder for the argument "new".
|
|
idx := len(a.Completed)
|
|
if idx >= len(s) {
|
|
return nil
|
|
}
|
|
|
|
return s[idx].Predict(a)
|
|
}
|