mirror of https://github.com/databricks/cli.git
90 lines
1.6 KiB
Go
90 lines
1.6 KiB
Go
package prompt
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/databricks/bricks/project"
|
|
"github.com/manifoldco/promptui"
|
|
)
|
|
|
|
type Results map[string]Answer
|
|
|
|
type Question interface {
|
|
Ask(res Results) (key string, ans Answer, err error)
|
|
}
|
|
|
|
type Questions []Question
|
|
|
|
func (qq Questions) Ask(res Results) error {
|
|
for _, v := range qq {
|
|
key, ans, err := v.Ask(res)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
res[key] = ans
|
|
}
|
|
return nil
|
|
}
|
|
|
|
type Text struct {
|
|
Key string
|
|
Label string
|
|
Default func(res Results) string
|
|
Callback AnswerCallback
|
|
Stdin io.ReadCloser
|
|
}
|
|
|
|
func (t Text) Ask(res Results) (string, Answer, error) {
|
|
def := ""
|
|
if t.Default != nil {
|
|
def = t.Default(res)
|
|
}
|
|
v, err := (&promptui.Prompt{
|
|
Label: t.Label,
|
|
Default: def,
|
|
Stdin: t.Stdin,
|
|
}).Run()
|
|
return t.Key, Answer{
|
|
Value: v,
|
|
Callback: t.Callback,
|
|
}, err
|
|
}
|
|
|
|
type Choice struct {
|
|
Key string
|
|
Label string
|
|
Answers []Answer
|
|
Stdin io.ReadCloser
|
|
}
|
|
|
|
func (q Choice) Ask(res Results) (string, Answer, error) {
|
|
// TODO: validate and re-ask
|
|
prompt := promptui.Select{
|
|
Label: q.Label,
|
|
Items: q.Answers,
|
|
Templates: &promptui.SelectTemplates{
|
|
Label: `{{ .Value }}`,
|
|
Details: `{{ .Details | green }}`,
|
|
Selected: fmt.Sprintf(`{{ "%s" | faint }}: {{ .Value | bold }}`, q.Label),
|
|
},
|
|
Stdin: q.Stdin,
|
|
}
|
|
i, _, err := prompt.Run()
|
|
return q.Key, q.Answers[i], err
|
|
}
|
|
|
|
type Answers []Answer
|
|
|
|
type AnswerCallback func(ans Answer, prj *project.Project, res Results)
|
|
|
|
type Answer struct {
|
|
Value string
|
|
Details string
|
|
Callback AnswerCallback
|
|
}
|
|
|
|
func (a Answer) String() string {
|
|
return a.Value
|
|
}
|