support labels for suggestions which are distinct from values

1 files changed, 15 insertions, 3 deletions

diff --git a/pkg/gui/suggestions_panel.go b/pkg/gui/suggestions_panel.go
index eb3b7808a..9f5469496 100644
--- a/pkg/gui/suggestions_panel.go
+++ b/pkg/gui/suggestions_panel.go
@@ -1,15 +1,27 @@
 package gui
 
-func (gui *Gui) getSelectedSuggestion() string {
+import "github.com/jesseduffield/lazygit/pkg/gui/types"
+
+func (gui *Gui) getSelectedSuggestionValue() string {
+	selectedSuggestion := gui.getSelectedSuggestion()
+
+	if selectedSuggestion != nil {
+		return selectedSuggestion.Value
+	}
+
+	return ""
+}
+
+func (gui *Gui) getSelectedSuggestion() *types.Suggestion {
 	selectedLine := gui.State.Panels.Suggestions.SelectedLineIdx
 	if selectedLine == -1 {
-		return ""
+		return nil
 	}
 
 	return gui.State.Suggestions[selectedLine]
 }
 
-func (gui *Gui) setSuggestions(suggestions []string) {
+func (gui *Gui) setSuggestions(suggestions []*types.Suggestion) {
 	view := gui.getSuggestionsView()
 	if view == nil {
 		return