feat(prompt): 使用自定义 prompt

使用自定义的 prompt 替换原有库，以实现终端移动，编辑更友好

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

Author: mritd

pkg/commit/ci.go

@@ -1,15 +1,14 @@
 package commit
 
 import (
-	"errors"
 	"fmt"
 	"io/ioutil"
 	"os"
-	"regexp"
 	"strings"
 	"text/template"
 
 	"github.com/mritd/gitflow-toolkit/pkg/consts"
+	gitprompt "github.com/mritd/gitflow-toolkit/pkg/prompt"
 	"github.com/mritd/gitflow-toolkit/pkg/util"
 	"github.com/mritd/promptui"
 )
@@ -84,106 +83,55 @@ func SelectCommitType() consts.CommitType {
 // 输入影响范围
 func InputScope() string {
 
-	validate := func(input string) error {
-		reg := regexp.MustCompile("\\s+")
-		if reg.ReplaceAllString(input, "") == "" {
-			return errors.New("scope is blank")
+	p := gitprompt.NewDefaultPrompt(func(line []rune) bool {
+		if strings.TrimSpace(string(line)) == "" {
+			return false
+		} else {
+			return true
 		}
-		return nil
-	}
-
-	templates := &promptui.PromptTemplates{
-		Prompt:  "{{ . }} ",
-		Valid:   "{{ . | green }} ",
-		Invalid: "{{ . | red }} ",
-		Success: "{{ . | bold }} ",
-	}
+	}, "Scope:")
 
-	prompt := promptui.Prompt{
-		Label:     "❯ Scope:",
-		Templates: templates,
-		Validate:  validate,
-	}
-
-	result, err := prompt.Run()
-	util.CheckAndExit(err)
-	return result
+	return p.Run()
 
 }
 
 // 输入提交主题
 func InputSubject() string {
 
-	validate := func(input string) error {
-		if strings.TrimSpace(input) == "" {
-			return errors.New("subject is blank")
+	p := gitprompt.NewDefaultPrompt(func(line []rune) bool {
+		if strings.TrimSpace(string(line)) == "" {
+			return false
+		} else {
+			return true
 		}
-		if r := []rune(input); len(r) > 50 {
-			return errors.New("subject too long")
-		}
-
-		return nil
-	}
+	}, "Subject:")
 
-	templates := &promptui.PromptTemplates{
-		Prompt:  "{{ . }} ",
-		Valid:   "{{ . | green }} ",
-		Invalid: "{{ . | red }} ",
-		Success: "{{ . | bold }} ",
-	}
-
-	prompt := promptui.Prompt{
-		Label:     "❯ Subject:",
-		Templates: templates,
-		Validate:  validate,
-	}
-
-	result, err := prompt.Run()
-	util.CheckAndExit(err)
-	return result
+	return p.Run()
 }
 
 // 输入完整提交信息
 func InputBody() string {
 
-	templates := &promptui.PromptTemplates{
-		Prompt:  "{{ . }} ",
-		Valid:   "{{ . | green }} ",
-		Invalid: "{{ . | red }} ",
-		Success: "{{ . | bold }} ",
-	}
+	p := gitprompt.NewDefaultPrompt(func(line []rune) bool {
+		return true
+	}, "Body:")
 
-	prompt := promptui.Prompt{
-		Label:     "❯ Body:",
-		Templates: templates,
-	}
-
-	result, err := prompt.Run()
-	util.CheckAndExit(err)
-	if result == "big" {
+	body := p.Run()
+	if body == "big" {
 		return util.OSEditInput()
 	}
-	return result
+
+	return body
 }
 
 // 输入提交关联信息
 func InputFooter() string {
 
-	templates := &promptui.PromptTemplates{
-		Prompt:  "{{ . }} ",
-		Valid:   "{{ . | green }} ",
-		Invalid: "{{ . | red }} ",
-		Success: "{{ . | bold }} ",
-	}
-
-	prompt := promptui.Prompt{
-		Label:     "❯ Footer:",
-		Templates: templates,
-	}
+	p := gitprompt.NewDefaultPrompt(func(line []rune) bool {
+		return true
+	}, "Footer:")
 
-	result, err := prompt.Run()
-	util.CheckAndExit(err)
-	return result
+	return p.Run()
 }
 
 // 生成 SOB 签名
@@ -221,5 +169,5 @@ func Commit(cm *Message) {
 	t.Execute(f, cm)
 	util.MustExec("git", "commit", "-F", f.Name())
 
-	fmt.Println("\nAlways code as if the guy who ends up maintaining your code will be a violent psychopath who knows where you live.")
+	fmt.Println("\n✔ Always code as if the guy who ends up maintaining your code will be a violent psychopath who knows where you live.")
 }

pkg/prompt/codes.go

@@ -0,0 +1,108 @@
+package prompt
+
+import (
+	"strconv"
+	"strings"
+	"fmt"
+	"text/template"
+)
+
+const esc = "\033["
+
+type attribute int
+
+// Foreground weight/decoration attributes.
+const (
+	reset attribute = iota
+
+	FGBold
+	FGFaint
+	FGItalic
+	FGUnderline
+)
+
+// Foreground color attributes
+const (
+	FGBlack attribute = iota + 30
+	FGRed
+	FGGreen
+	FGYellow
+	FGBlue
+	FGMagenta
+	FGCyan
+	FGWhite
+)
+
+// Background color attributes
+const (
+	BGBlack attribute = iota + 40
+	BGRed
+	BGGreen
+	BGYellow
+	BGBlue
+	BGMagenta
+	BGCyan
+	BGWhite
+)
+
+// ResetCode is the character code used to reset the terminal formatting
+var ResetCode = fmt.Sprintf("%s%dm", esc, reset)
+
+const (
+	hideCursor = esc + "?25l"
+	showCursor = esc + "?25h"
+	clearLine  = esc + "2K"
+)
+
+// FuncMap defines template helpers for the output. It can be extended as a
+// regular map.
+var FuncMap = template.FuncMap{
+	"black":     Styler(FGBlack),
+	"red":       Styler(FGRed),
+	"green":     Styler(FGGreen),
+	"yellow":    Styler(FGYellow),
+	"blue":      Styler(FGBlue),
+	"magenta":   Styler(FGMagenta),
+	"cyan":      Styler(FGCyan),
+	"white":     Styler(FGWhite),
+	"bgBlack":   Styler(BGBlack),
+	"bgRed":     Styler(BGRed),
+	"bgGreen":   Styler(BGGreen),
+	"bgYellow":  Styler(BGYellow),
+	"bgBlue":    Styler(BGBlue),
+	"bgMagenta": Styler(BGMagenta),
+	"bgCyan":    Styler(BGCyan),
+	"bgWhite":   Styler(BGWhite),
+	"bold":      Styler(FGBold),
+	"faint":     Styler(FGFaint),
+	"italic":    Styler(FGItalic),
+	"underline": Styler(FGUnderline),
+}
+
+func upLine(n uint) string {
+	return movementCode(n, 'A')
+}
+
+func movementCode(n uint, code rune) string {
+	return esc + strconv.FormatUint(uint64(n), 10) + string(code)
+}
+
+// Styler returns a func that applies the attributes given in the Styler call
+// to the provided string.
+func Styler(attrs ...attribute) func(interface{}) string {
+	attrstrs := make([]string, len(attrs))
+	for i, v := range attrs {
+		attrstrs[i] = strconv.Itoa(int(v))
+	}
+
+	seq := strings.Join(attrstrs, ";")
+
+	return func(v interface{}) string {
+		end := ""
+		s, ok := v.(string)
+		if !ok || !strings.HasSuffix(s, ResetCode) {
+			end = ResetCode
+		}
+		return fmt.Sprintf("%s%sm%v%s", esc, seq, v, end)
+	}
+}