From 59f810fb2b614ed4ed8f67e0fabd6d7e9d88c65a Mon Sep 17 00:00:00 2001
From: cpendery <cpendery@vt.edu>
Date: Tue, 12 Sep 2023 22:07:11 -0400
Subject: [PATCH] feat: add argument description to UI when suggestions are
 available

Signed-off-by: cpendery <cpendery@vt.edu>
---
 ui/suggestions/borders.go     | 25 +++++++++++++++++++++++++
 ui/suggestions/suggestions.go | 18 +++++++++++++++++-
 2 files changed, 42 insertions(+), 1 deletion(-)
 create mode 100644 ui/suggestions/borders.go

diff --git a/ui/suggestions/borders.go b/ui/suggestions/borders.go
new file mode 100644
index 0000000..5c465af
--- /dev/null
+++ b/ui/suggestions/borders.go
@@ -0,0 +1,25 @@
+package suggestions
+
+import "github.com/charmbracelet/lipgloss"
+
+var SuggestionWithDescriptionBorder = lipgloss.Border{
+	Top:         "─",
+	Bottom:      "─",
+	Left:        "│",
+	Right:       "│",
+	TopLeft:     "┌",
+	TopRight:    "┐",
+	BottomLeft:  "├",
+	BottomRight: "┤",
+}
+
+var ArgumentDescriptionBorder = lipgloss.Border{
+	Top:         "",
+	Bottom:      "─",
+	Left:        "│",
+	Right:       "│",
+	TopLeft:     "",
+	TopRight:    "",
+	BottomLeft:  "└",
+	BottomRight: "┘",
+}
diff --git a/ui/suggestions/suggestions.go b/ui/suggestions/suggestions.go
index a22c91b..60983cf 100644
--- a/ui/suggestions/suggestions.go
+++ b/ui/suggestions/suggestions.go
@@ -132,10 +132,15 @@ func pageSuggestions(suggestions []autocomplete.Suggestion, cursorLocation int)
 }
 
 func (m Model) renderSuggestions(width int) string {
+	suggestionBorder := lipgloss.NormalBorder()
+	if m.argDescription != "" {
+		suggestionBorder = SuggestionWithDescriptionBorder
+	}
+
 	var style = lipgloss.NewStyle().
 		Bold(true).
 		Width(width).
-		BorderStyle(lipgloss.NormalBorder()).
+		BorderStyle(suggestionBorder).
 		BorderForeground(lipgloss.Color("63"))
 
 	m.suggestions = pageSuggestions(m.suggestions, m.cursor)
@@ -145,6 +150,17 @@ func (m Model) renderSuggestions(width int) string {
 		r[idx] = m.renderSuggestion(suggestion.Name, idx, m.cursor, width)
 	}
 
+	if m.argDescription != "" {
+		descriptionStyle := lipgloss.NewStyle().
+			Width(width).
+			Border(ArgumentDescriptionBorder, false, true, true, true).
+			BorderForeground(lipgloss.Color("63"))
+
+		return lipgloss.JoinVertical(
+			lipgloss.Left,
+			style.Render(strings.Join(r, "\n")),
+			descriptionStyle.Render(m.argDescription))
+	}
 	return style.Render(strings.Join(r, "\n"))
 }