mlow
/
lmcli
1
0
Fork 0
Code Issues 3 Pull Requests Packages Projects Releases Wiki Activity

tui: renamed uiCache to views, cleanup

This commit is contained in:
Matt Low 2024-03-29 20:43:19 +00:00
parent 0b991800d6
commit 9cd28d28d7
1 changed files with 63 additions and 67 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

128
pkg/tui/tui.go
View File

@ -42,14 +42,6 @@ const (
selectedMessage
)
type uiCache struct {
header string
content string
error string
input string
footer string
}
type model struct {
width int
height int
@ -81,15 +73,50 @@ type model struct {
content viewport.Model
input textarea.Model
spinner spinner.Model
cache *uiCache
views *views
}
type message struct {
role string
// we populate these fields in the main Update() function, and let View()
// be responsible for returning the final composition of elements
type views struct {
header string
content string
error string
input string
footer string
}
// styles
var (
headerStyle = lipgloss.NewStyle().
PaddingLeft(1).
PaddingRight(1).
Background(lipgloss.Color("0"))
messageHeadingStyle = lipgloss.NewStyle().
MarginTop(1).
MarginBottom(1).
PaddingLeft(1).
Bold(true)
userStyle = lipgloss.NewStyle().Faint(true).Foreground(lipgloss.Color("10"))
assistantStyle = lipgloss.NewStyle().Faint(true).Foreground(lipgloss.Color("12"))
messageStyle = lipgloss.NewStyle().
PaddingLeft(2).
PaddingRight(2)
inputFocusedStyle = lipgloss.NewStyle().
Border(lipgloss.RoundedBorder(), true, true, true, false)
inputBlurredStyle = lipgloss.NewStyle().
Faint(true).
Border(lipgloss.RoundedBorder(), true, true, true, false)
footerStyle = lipgloss.NewStyle()
)
// custom tea.Msg types
type (
// sent on each chunk received from LLM
@ -110,24 +137,11 @@ type (
msgError error
)
// styles
var (
headingStyle = lipgloss.NewStyle().
MarginTop(1).
MarginBottom(1).
PaddingLeft(1).
Bold(true)
userStyle = lipgloss.NewStyle().Faint(true).Foreground(lipgloss.Color("10"))
assistantStyle = lipgloss.NewStyle().Faint(true).Foreground(lipgloss.Color("12"))
messageStyle = lipgloss.NewStyle().
PaddingLeft(2).
PaddingRight(2)
headerStyle = lipgloss.NewStyle().
PaddingLeft(1).
PaddingRight(1).
Background(lipgloss.Color("0"))
footerStyle = lipgloss.NewStyle()
)
func wrapError(err error) tea.Cmd {
return func() tea.Msg {
return msgError(err)
}
}
func (m model) Init() tea.Cmd {
return tea.Batch(
@ -139,12 +153,6 @@ func (m model) Init() tea.Cmd {
)
}
func wrapError(err error) tea.Cmd {
return func() tea.Msg {
return msgError(err)
}
}
func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
var cmds []tea.Cmd
@ -293,7 +301,6 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
cmds = append(cmds, cmd)
}
prevInputLineCnt := m.input.LineCount()
inputCaptured := false
m.input, cmd = m.input.Update(msg)
@ -310,19 +317,19 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
}
if m.width > 0 {
m.cache.header = m.headerView()
m.cache.footer = m.footerView()
m.cache.error = m.errorView()
m.views.header = m.headerView()
m.views.footer = m.footerView()
m.views.error = m.errorView()
fixedHeight := height(m.cache.header) + height(m.cache.error) + height(m.cache.footer)
fixedHeight := height(m.views.header) + height(m.views.error) + height(m.views.footer)
// calculate clamped input height to accomodate input text
newHeight := max(4, min((m.height-fixedHeight-1)/2, m.input.LineCount()))
m.input.SetHeight(newHeight)
m.cache.input = m.inputView()
m.views.input = m.input.View()
m.content.Height = m.height - height(m.cache.input) - fixedHeight
m.cache.content = m.contentView()
m.content.Height = m.height - height(m.views.input) - fixedHeight
m.views.content = m.content.View()
}
// this is a pretty nasty hack to ensure the input area viewport doesn't
@ -384,13 +391,13 @@ func (m model) View() string {
}
sections := make([]string, 0, 6)
sections = append(sections, m.cache.header)
sections = append(sections, m.cache.content)
if m.cache.error != "" {
sections = append(sections, m.cache.error)
sections = append(sections, m.views.header)
sections = append(sections, m.views.content)
if m.views.error != "" {
sections = append(sections, m.views.error)
}
sections = append(sections, m.cache.input)
sections = append(sections, m.cache.footer)
sections = append(sections, m.views.input)
sections = append(sections, m.views.footer)
return lipgloss.JoinVertical(
lipgloss.Left,
@ -411,10 +418,6 @@ func (m *model) headerView() string {
return headerStyle.Width(m.width).Render(part)
}
func (m *model) contentView() string {
return m.content.View()
}
func (m *model) errorView() string {
if m.err == nil {
return ""
@ -427,10 +430,6 @@ func (m *model) errorView() string {
Render(fmt.Sprintf("%s", m.err))
}
func (m *model) inputView() string {
return m.input.View()
}
func (m *model) footerView() string {
segmentStyle := lipgloss.NewStyle().PaddingLeft(1).PaddingRight(1).Faint(true)
segmentSeparator := "|"
@ -488,7 +487,7 @@ func initialModel(ctx *lmcli.Context, convShortname string) model {
wrap: true,
selectedMessage: -1,
cache: &uiCache{},
views: &views{},
}
m.content = viewport.New(0, 0)
@ -497,14 +496,11 @@ func initialModel(ctx *lmcli.Context, convShortname string) model {
m.input.CharLimit = 0
m.input.Placeholder = "Enter a message"
m.input.ShowLineNumbers = false
m.input.Focus()
m.input.FocusedStyle.CursorLine = lipgloss.NewStyle()
m.input.FocusedStyle.Base = lipgloss.NewStyle().
Border(lipgloss.RoundedBorder(), true, true, true, false)
m.input.BlurredStyle.Base = lipgloss.NewStyle().
Faint(true).
Border(lipgloss.RoundedBorder(), true, true, true, false)
m.input.FocusedStyle.Base = inputFocusedStyle
m.input.BlurredStyle.Base = inputBlurredStyle
m.input.ShowLineNumbers = false
m.spinner = spinner.New(spinner.WithSpinner(
spinner.Spinner{
@ -826,7 +822,7 @@ func (m *model) renderMessageHeading(i int, message *models.Message) string {
suffix += faint.Render(" (not saved)")
}
return headingStyle.Render(prefix + user + suffix)
return messageHeadingStyle.Render(prefix + user + suffix)
}
func (m *model) renderMessage(msg *models.Message) string {