diff options
author | Elwardi <elwardifadeli@gmail.com> | 2021-08-05 15:24:17 +0100 |
---|---|---|
committer | Elwardi <elwardifadeli@gmail.com> | 2021-08-05 15:45:18 +0100 |
commit | a8ec044f0e52bff8921592fb4d334a4d6f4c7468 (patch) | |
tree | 0f9dcdf7bb9b041c81f5c34e9fe97f5f9a92afb1 /pkg | |
parent | d626bcac0029267d3f45223198902f5cb78d9dc1 (diff) |
Make menuFromCommand format menu items and their description
Diffstat (limited to 'pkg')
-rw-r--r-- | pkg/config/user_config.go | 3 | ||||
-rw-r--r-- | pkg/gui/custom_commands.go | 37 | ||||
-rw-r--r-- | pkg/gui/gui_test.go | 29 |
3 files changed, 45 insertions, 24 deletions
diff --git a/pkg/config/user_config.go b/pkg/config/user_config.go index 38259f6f2..239a79acf 100644 --- a/pkg/config/user_config.go +++ b/pkg/config/user_config.go @@ -285,7 +285,8 @@ type CustomCommandPrompt struct { // this only applies to menuFromCommand Command string `yaml:"command"` Filter string `yaml:"filter"` - Format string `yaml:"format"` + TFormat string `yaml:"itemFormat"` + DFormat string `yaml:"descriptionFormat"` } type CustomCommandMenuOption struct { diff --git a/pkg/gui/custom_commands.go b/pkg/gui/custom_commands.go index 2fb17049f..080831aa0 100644 --- a/pkg/gui/custom_commands.go +++ b/pkg/gui/custom_commands.go @@ -118,16 +118,22 @@ func (gui *Gui) menuPrompt(prompt config.CustomCommandPrompt, promptResponses [] return gui.createMenu(title, menuItems, createMenuOptions{showCancel: true}) } -func (gui *Gui) GenerateMenuCandidates(commandOutput string, filter string, format string) ([]string, error) { +func (gui *Gui) GenerateMenuCandidates(commandOutput string, filter string, tFormat string, dFormat string) ([]string, []string, error) { candidates := []string{} + descriptions := []string{} reg, err := regexp.Compile(filter) if err != nil { - return candidates, gui.surfaceError(errors.New("unable to parse filter regex, error: " + err.Error())) + return candidates, descriptions, gui.surfaceError(errors.New("unable to parse filter regex, error: " + err.Error())) } - buff := bytes.NewBuffer(nil) - temp, err := template.New("format").Parse(format) + buffTitle := bytes.NewBuffer(nil) + tempTitle, err := template.New("format").Parse(tFormat) if err != nil { - return candidates, gui.surfaceError(errors.New("unable to parse format, error: " + err.Error())) + return candidates, descriptions, gui.surfaceError(errors.New("unable to parse item format, error: " + err.Error())) + } + buffDescr := bytes.NewBuffer(nil) + tempDescr, err := template.New("format").Parse(dFormat) + if err != nil { + return candidates, descriptions, gui.surfaceError(errors.New("unable to parse item description format, error: " + err.Error())) } for _, str := range strings.Split(string(commandOutput), "\n") { if str == "" { @@ -146,15 +152,21 @@ func (gui *Gui) GenerateMenuCandidates(commandOutput string, filter string, form } } } - err = temp.Execute(buff, tmplData) + err = tempTitle.Execute(buffTitle, tmplData) + if err != nil { + return candidates, descriptions, gui.surfaceError(err) + } + err = tempDescr.Execute(buffDescr, tmplData) if err != nil { - return candidates, gui.surfaceError(err) + return candidates, descriptions, gui.surfaceError(err) } - candidates = append(candidates, strings.TrimSpace(buff.String())) - buff.Reset() + candidates = append(candidates, strings.TrimSpace(buffTitle.String())) + descriptions = append(descriptions, strings.TrimSpace(buffDescr.String())) + buffTitle.Reset() + buffDescr.Reset() } - return candidates, err + return candidates, descriptions, err } func (gui *Gui) menuPromptFromCommand(prompt config.CustomCommandPrompt, promptResponses []string, responseIdx int, wrappedF func() error) error { @@ -177,7 +189,7 @@ func (gui *Gui) menuPromptFromCommand(prompt config.CustomCommandPrompt, promptR } // Need to make a menu out of what the cmd has displayed - candidates, err := gui.GenerateMenuCandidates(message, filter, prompt.Format) + candidates, descriptions, err := gui.GenerateMenuCandidates(message, filter, prompt.TFormat, prompt.DFormat) if err != nil { return gui.surfaceError(err) } @@ -185,7 +197,8 @@ func (gui *Gui) menuPromptFromCommand(prompt config.CustomCommandPrompt, promptR menuItems := make([]*menuItem, len(candidates)) for i := range candidates { menuItems[i] = &menuItem{ - displayStrings: []string{candidates[i]}, + // Put in candidate and its description + displayStrings: []string{candidates[i], style.FgYellow.Sprint(descriptions[i])}, onPress: func() error { promptResponses[responseIdx] = candidates[i] return wrappedF() diff --git a/pkg/gui/gui_test.go b/pkg/gui/gui_test.go index ec1279608..f3f798d03 100644 --- a/pkg/gui/gui_test.go +++ b/pkg/gui/gui_test.go @@ -86,29 +86,34 @@ func TestGuiGenerateMenuCandidates(t *testing.T) { testName string cmdOut string filter string - format string - test func([]string, error) + tFormat string + dFormat string + test func([]string, []string, error) } scenarios := []scenario{ { "Extract remote branch name", "upstream/pr-1", - "upstream/(?P<branch>.*)", + "(?P<remote>[a-z_]+)/(?P<branch>.*)", "{{ .branch }}", - func(actual []string, err error) { + "Remote: {{ .remote }}", + func(actualCandidate []string, actualDescr []string, err error) { assert.NoError(t, err) - assert.EqualValues(t, "pr-1", actual[0]) + assert.EqualValues(t, "pr-1", actualCandidate[0]) + assert.EqualValues(t, "Remote: upstream", actualDescr[0]) }, }, { - "Multiple named groups", + "Multiple named groups with empty description", "upstream/pr-1", "(?P<remote>[a-z]*)/(?P<branch>.*)", "{{ .branch }}|{{ .remote }}", - func(actual []string, err error) { + "", + func(actualCandidate []string, actualDescr []string, err error) { assert.NoError(t, err) - assert.EqualValues(t, "pr-1|upstream", actual[0]) + assert.EqualValues(t, "pr-1|upstream", actualCandidate[0]) + assert.EqualValues(t, "", actualDescr[0]) }, }, { @@ -116,16 +121,18 @@ func TestGuiGenerateMenuCandidates(t *testing.T) { "upstream/pr-1", "(?P<remote>[a-z]*)/(?P<branch>.*)", "{{ .group_2 }}|{{ .group_1 }}", - func(actual []string, err error) { + "Remote: {{ .group_1 }}", + func(actualCandidate []string, actualDescr []string, err error) { assert.NoError(t, err) - assert.EqualValues(t, "pr-1|upstream", actual[0]) + assert.EqualValues(t, "pr-1|upstream", actualCandidate[0]) + assert.EqualValues(t, "Remote: upstream", actualDescr[0]) }, }, } for _, s := range scenarios { t.Run(s.testName, func(t *testing.T) { - s.test(NewDummyGui().GenerateMenuCandidates(s.cmdOut, s.filter, s.format)) + s.test(NewDummyGui().GenerateMenuCandidates(s.cmdOut, s.filter, s.tFormat, s.dFormat)) }) } } |