From 704f101af32dd4d52d49b5df28556d2f0a1500a7 Mon Sep 17 00:00:00 2001
From: Carlos Alexandro Becker <caarlos0@users.noreply.github.com>
Date: Thu, 12 Sep 2024 11:17:29 -0300
Subject: [PATCH] feat: formatter

---
 textarea/textarea.go | 51 +++++++++++++++++++++++++++++++++++---------
 1 file changed, 41 insertions(+), 10 deletions(-)

diff --git a/textarea/textarea.go b/textarea/textarea.go
index 51d555e..8df4f8e 100644
--- a/textarea/textarea.go
+++ b/textarea/textarea.go
@@ -243,6 +243,7 @@ type Model struct {
 	MaxWidth int
 
 	SyntaxHighlighter func(string) string
+	Formatter         func(string) string
 
 	// Should the input suggest to complete
 	ShowSuggestions bool
@@ -473,6 +474,7 @@ func (m *Model) insertRunesFromUserInput(runes []rune) {
 	// Finally add the tail at the end of the last line inserted.
 	m.value[m.row] = append(m.value[m.row], tail...)
 
+	m.format()
 	m.SetCursor(m.col)
 }
 
@@ -997,6 +999,7 @@ func (m Model) Update(msg tea.Msg) (Model, tea.Cmd) {
 	if ok && key.Matches(keyMsg, m.KeyMap.AcceptSuggestion) {
 		if m.canAcceptSuggestion() {
 			m.value = m.matchedSuggestions[m.currentSuggestionIndex]
+			m.format()
 			m.row = len(m.value) - 1
 			m.CursorEnd()
 		}
@@ -1078,7 +1081,11 @@ func (m Model) Update(msg tea.Msg) (Model, tea.Cmd) {
 		case key.Matches(msg, m.KeyMap.CharacterForward):
 			m.characterRight()
 		case key.Matches(msg, m.KeyMap.LineNext):
-			m.CursorDown()
+			if m.row == 0 {
+				m.nextSuggestion()
+			} else {
+				m.CursorDown()
+			}
 		case key.Matches(msg, m.KeyMap.WordForward):
 			m.wordRight()
 		case key.Matches(msg, m.KeyMap.Paste):
@@ -1086,7 +1093,11 @@ func (m Model) Update(msg tea.Msg) (Model, tea.Cmd) {
 		case key.Matches(msg, m.KeyMap.CharacterBackward):
 			m.characterLeft(false /* insideLine */)
 		case key.Matches(msg, m.KeyMap.LinePrevious):
-			m.CursorUp()
+			if m.row == 0 {
+				m.previousSuggestion()
+			} else {
+				m.CursorUp()
+			}
 		case key.Matches(msg, m.KeyMap.WordBackward):
 			m.wordLeft()
 		case key.Matches(msg, m.KeyMap.InputBegin):
@@ -1484,6 +1495,7 @@ func (m *Model) splitLine(row, col int) {
 	m.value[row+1] = tail
 
 	m.col = 0
+	m.SetHeight(m.row + 2)
 	m.row++
 }
 
@@ -1493,14 +1505,6 @@ func (m *Model) canAcceptSuggestion() bool {
 	return len(m.matchedSuggestions) > 0
 }
 
-func linesToString(lines [][]rune) string {
-	var result []string
-	for _, line := range lines {
-		result = append(result, string(line))
-	}
-	return strings.Join(result, "\n")
-}
-
 // updateSuggestions refreshes the list of matching suggestions.
 func (m *Model) updateSuggestions() {
 	if !m.ShowSuggestions {
@@ -1544,6 +1548,17 @@ func (m *Model) previousSuggestion() {
 	}
 }
 
+func (m *Model) format() {
+	if m.Formatter == nil {
+		return
+	}
+	m.value = stringToLines(m.Formatter(linesToString(m.value)))
+	m.row = len(m.value) - 1
+	if m.col > len(m.value[m.row]) {
+		m.col = len(m.value[m.row]) - 1
+	}
+}
+
 // Paste is a command for pasting from the clipboard into the text input.
 func Paste() tea.Msg {
 	str, err := clipboard.ReadAll()
@@ -1642,3 +1657,19 @@ func max(a, b int) int {
 	}
 	return b
 }
+
+func stringToLines(s string) [][]rune {
+	var r [][]rune
+	for _, line := range strings.Split(s, "\n") {
+		r = append(r, []rune(line))
+	}
+	return r
+}
+
+func linesToString(lines [][]rune) string {
+	var result []string
+	for _, line := range lines {
+		result = append(result, string(line))
+	}
+	return strings.Join(result, "\n")
+}