From 8087907b4b942c7d8c9296c3fd0e1f08d5d88694 Mon Sep 17 00:00:00 2001
From: Kyren223 <Kyren223@proton.me>
Date: Thu, 5 Dec 2024 14:11:28 +0200
Subject: Refactored how the core UI looks

---
 internal/client/ui/viminput/linedecorations.go |  4 +--
 internal/client/ui/viminput/viminput.go        | 38 ++++++++++++++++++++++----
 2 files changed, 35 insertions(+), 7 deletions(-)

(limited to 'internal/client/ui/viminput')

diff --git a/internal/client/ui/viminput/linedecorations.go b/internal/client/ui/viminput/linedecorations.go
index 82e51f1..631c0ce 100644
--- a/internal/client/ui/viminput/linedecorations.go
+++ b/internal/client/ui/viminput/linedecorations.go
@@ -7,11 +7,11 @@ import (
 )
 
 func LineNumberDecoration(style lipgloss.Style) LineDecoration {
-	return func(lnum int, line string, cursorLnum int) string {
+	return func(lnum int, m Model) string {
 		return style.Render(strconv.FormatInt(int64(lnum), 10))
 	}
 }
 
-func EmptyLineDecoration(lnum int, line string, cursorLnum int) string {
+func EmptyLineDecoration(lnum int, m Model) string {
 	return ""
 }
diff --git a/internal/client/ui/viminput/viminput.go b/internal/client/ui/viminput/viminput.go
index e2b65a8..94c5413 100644
--- a/internal/client/ui/viminput/viminput.go
+++ b/internal/client/ui/viminput/viminput.go
@@ -1,13 +1,15 @@
 package viminput
 
 import (
+	"strings"
+
 	tea "github.com/charmbracelet/bubbletea"
 	"github.com/charmbracelet/lipgloss"
 )
 
 type (
 	Keymap         struct{}
-	LineDecoration = func(lnum int, line string, cursorLnum int) string
+	LineDecoration = func(lnum int, m Model) string
 )
 
 type Line []byte
@@ -21,7 +23,9 @@ type Model struct {
 	Placeholder    string
 	LineDecoration LineDecoration
 
-	lines []Line
+	lines      []Line
+	CursorLine int
+	CursorColumn int
 
 	width  int
 	height int
@@ -46,11 +50,23 @@ func (m Model) Init() tea.Cmd {
 }
 
 func (m Model) View() string {
-	placeholder := m.PlaceholderStyle.Render(m.Placeholder)
-	lineDecoration := m.LineDecoration(1, "", 1)
+	var lines []Line
+	if len(m.lines) == 0 {
+		placeholder := m.PlaceholderStyle.Render(m.Placeholder)
+		lines = append(lines, Line(placeholder))
+	} else {
+		lines = m.lines
+	}
 
-	result := lipgloss.JoinHorizontal(lipgloss.Top, lineDecoration, placeholder)
+	var builder strings.Builder
+	for i, line := range lines {
+		lineDecoration := m.LineDecoration(i, m)
+		builder.WriteString(lineDecoration)
+		builder.Write(line)
+		builder.WriteByte('\n')
+	}
 
+	result := builder.String()
 	result = lipgloss.NewStyle().Width(m.width).Height(m.height).Render(result)
 
 	return result
@@ -87,3 +103,15 @@ func (m *Model) Focus() {
 func (m *Model) Blur() {
 	m.focus = false
 }
+
+func (m *Model) SetLines(lines ...Line) {
+	m.lines = lines
+}
+
+func (m *Model) SetLine(lnum int, line Line) {
+	m.lines[lnum] = line
+}
+
+func (m *Model) Line(lnum int) Line {
+	return m.lines[lnum]
+}
-- 
cgit v1.3.1