feat(prompt): 尝试增加 select 功能

尝试增加 select 功能

Signed-off-by: mritd <[email protected]>

Author: mritd

pkg/prompt/prompt.go

@@ -20,10 +20,10 @@ const (
 )
 
 type Prompt struct {
-	Ask       string
-	Prompt    string
-	PromptTpl *Tpl
-	FuncMap   template.FuncMap
+	Config
+	Ask     string
+	Prompt  string
+	FuncMap template.FuncMap
 
 	isFirstRun bool
 
@@ -34,7 +34,7 @@ type Prompt struct {
 	errorMsg *template.Template
 }
 
-type Tpl struct {
+type Config struct {
 	AskTpl        string
 	PromptTpl     string
 	ValidTpl      string
@@ -43,8 +43,8 @@ type Tpl struct {
 	CheckListener func(line []rune) error
 }
 
-func NewDefaultTpl(check func(line []rune) error) *Tpl {
-	return &Tpl{
+func NewDefaultConfig(check func(line []rune) error) Config {
+	return Config{
 		AskTpl:        DefaultAskTpl,
 		PromptTpl:     DefaultPromptTpl,
 		InvalidTpl:    DefaultInvalidTpl,
@@ -54,27 +54,27 @@ func NewDefaultTpl(check func(line []rune) error) *Tpl {
 	}
 }
 
-func NewDefaultPrompt(check func(line []rune) error, ask string) *Prompt {
-	return &Prompt{
-		Ask:       ask,
-		Prompt:    DefaultPrompt,
-		PromptTpl: NewDefaultTpl(check),
-		FuncMap:   FuncMap,
+func NewDefaultPrompt(check func(line []rune) error, ask string) Prompt {
+	return Prompt{
+		Ask:     ask,
+		Prompt:  DefaultPrompt,
+		FuncMap: FuncMap,
+		Config:  NewDefaultConfig(check),
 	}
 }
 
 func (p *Prompt) prepareTemplates() {
 
 	var err error
-	p.ask, err = template.New("").Funcs(FuncMap).Parse(p.PromptTpl.AskTpl)
+	p.ask, err = template.New("").Funcs(FuncMap).Parse(p.AskTpl)
 	util.CheckAndExit(err)
-	p.prompt, err = template.New("").Funcs(FuncMap).Parse(p.PromptTpl.PromptTpl)
+	p.prompt, err = template.New("").Funcs(FuncMap).Parse(p.PromptTpl)
 	util.CheckAndExit(err)
-	p.valid, err = template.New("").Funcs(FuncMap).Parse(p.PromptTpl.ValidTpl)
+	p.valid, err = template.New("").Funcs(FuncMap).Parse(p.ValidTpl)
 	util.CheckAndExit(err)
-	p.invalid, err = template.New("").Funcs(FuncMap).Parse(p.PromptTpl.InvalidTpl)
+	p.invalid, err = template.New("").Funcs(FuncMap).Parse(p.InvalidTpl)
 	util.CheckAndExit(err)
-	p.errorMsg, err = template.New("").Funcs(FuncMap).Parse(p.PromptTpl.ErrorMsgTpl)
+	p.errorMsg, err = template.New("").Funcs(FuncMap).Parse(p.ErrorMsgTpl)
 	util.CheckAndExit(err)
 
 }
@@ -121,7 +121,7 @@ func (p *Prompt) Run() string {
 
 	l.Config.SetListener(func(line []rune, pos int, key rune) (newLine []rune, newPos int, ok bool) {
 		// Real-time verification
-		if err = p.PromptTpl.CheckListener(line); err != nil {
+		if err = p.CheckListener(line); err != nil {
 			l.SetPrompt(string(invalidPrompt))
 			l.Refresh()
 		} else {
@@ -139,7 +139,7 @@ func (p *Prompt) Run() string {
 		}
 		s, err := l.Readline()
 		util.CheckAndExit(err)
-		if err = p.PromptTpl.CheckListener([]rune(s)); err != nil {
+		if err = p.CheckListener([]rune(s)); err != nil {
 			fmt.Print(clearLine)
 			fmt.Println(string(render(p.errorMsg, DefaultErrorMsgPrefix+err.Error())))
 			p.isFirstRun = false

pkg/prompt/select.go

@@ -0,0 +1,99 @@
+package prompt
+
+import (
+	"bytes"
+	"text/template"
+
+	"github.com/mritd/idgen/util"
+)
+
+const (
+	DefaultActiveTpl       = "{{ . | cyan }}"
+	DefaultInactiveTpl     = "{{ . | white }}"
+	DefaultDetailsTpl      = "{{ . | white }}"
+	DefaultSelectHeaderTpl = `{{ "Use the arrow keys to navigate: ↓ ↑ → ←" | faint }}`
+	DefaultSelectPromptTpl = `{{ "Select" . | faint }}:`
+	DefaultSelectAsk       = "Please select"
+	NextLine               = "\n"
+)
+
+type Select struct {
+	SelectConfig
+	Items        interface{}
+	SelectPrompt string
+
+	selectPrompt *template.Template
+	selectHeader *template.Template
+	active       *template.Template
+	inactive     *template.Template
+	details      *template.Template
+}
+
+type SelectConfig struct {
+	Prompt
+	SelectHeaderTpl string
+	SelectPromptTpl string
+	ActiveTpl       string
+	InactiveTpl     string
+	DetailsTpl      string
+}
+
+func NewDefaultSelectConfig(check func(line []rune) error, selectAsk string) SelectConfig {
+	return SelectConfig{
+		Prompt:          NewDefaultPrompt(check, selectAsk),
+		SelectHeaderTpl: DefaultSelectHeaderTpl,
+		SelectPromptTpl: DefaultSelectPromptTpl,
+		ActiveTpl:       DefaultActiveTpl,
+		InactiveTpl:     DefaultInactiveTpl,
+		DetailsTpl:      DefaultDetailsTpl,
+	}
+}
+
+func NewDefaultSelect(check func(line []rune) error, items interface{}) *Select {
+	return &Select{
+		SelectConfig: NewDefaultSelectConfig(check, DefaultSelectAsk),
+		Items:        items,
+	}
+}
+
+func (s *Select) prepareTemplates() {
+
+	var err error
+
+	// Prompt prepare
+	s.ask, err = template.New("").Funcs(FuncMap).Parse(s.AskTpl)
+	util.CheckAndExit(err)
+	s.prompt, err = template.New("").Funcs(FuncMap).Parse(s.PromptTpl)
+	util.CheckAndExit(err)
+	s.valid, err = template.New("").Funcs(FuncMap).Parse(s.ValidTpl)
+	util.CheckAndExit(err)
+	s.invalid, err = template.New("").Funcs(FuncMap).Parse(s.InvalidTpl)
+	util.CheckAndExit(err)
+	s.errorMsg, err = template.New("").Funcs(FuncMap).Parse(s.ErrorMsgTpl)
+	util.CheckAndExit(err)
+
+	// Select prepare
+	s.selectHeader, err = template.New("").Funcs(FuncMap).Parse(s.SelectHeaderTpl)
+	util.CheckAndExit(err)
+	s.selectPrompt, err = template.New("").Funcs(FuncMap).Parse(s.SelectPromptTpl)
+	util.CheckAndExit(err)
+	s.active, err = template.New("").Funcs(FuncMap).Parse(s.ActiveTpl)
+	util.CheckAndExit(err)
+	s.inactive, err = template.New("").Funcs(FuncMap).Parse(s.InactiveTpl)
+	util.CheckAndExit(err)
+	s.details, err = template.New("").Funcs(FuncMap).Parse(s.DetailsTpl)
+	util.CheckAndExit(err)
+
+}
+
+func (s *Select) prepareDisplayData(sindex int) {
+	var data bytes.Buffer
+
+	// select header
+	data.Write(render(s.selectHeader, NextLine))
+	// select prompt
+	data.Write(render(s.selectPrompt, s.SelectPrompt+NextLine))
+	//for it := range s.Items {
+	//
+	//}
+}