190 lines
4.7 KiB
Go
190 lines
4.7 KiB
Go
package tui
|
|
|
|
import (
|
|
"fmt"
|
|
"slices"
|
|
"strings"
|
|
"time"
|
|
|
|
models "git.mlow.ca/mlow/lmcli/pkg/lmcli/model"
|
|
"git.mlow.ca/mlow/lmcli/pkg/util"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/charmbracelet/lipgloss"
|
|
)
|
|
|
|
type (
|
|
// send when conversation list is loaded
|
|
msgConversationsLoaded []models.Conversation
|
|
)
|
|
|
|
func (m *model) handleConversationListInput(msg tea.KeyMsg) tea.Cmd {
|
|
switch msg.String() {
|
|
case "enter":
|
|
m.state = stateConversation
|
|
m.updateContent()
|
|
// load selected conversation and switch state
|
|
case "n":
|
|
// new conversation
|
|
case "d":
|
|
// show prompt to delete conversation
|
|
case "c":
|
|
// copy/clone conversation
|
|
case "r":
|
|
// show prompt to rename conversation
|
|
case "shift+r":
|
|
// show prompt to generate name for conversation
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m *model) handleConversationListUpdate(msg tea.Msg) []tea.Cmd {
|
|
var cmds []tea.Cmd
|
|
switch msg := msg.(type) {
|
|
case msgConversationsLoaded:
|
|
m.conversations = msg
|
|
m.content.SetContent(m.conversationListView())
|
|
}
|
|
|
|
var cmd tea.Cmd
|
|
m.content, cmd = m.content.Update(msg)
|
|
if cmd != nil {
|
|
cmds = append(cmds, cmd)
|
|
}
|
|
|
|
if m.width > 0 {
|
|
m.views.header = m.headerView()
|
|
m.views.footer = m.footerView()
|
|
m.views.error = m.errorView()
|
|
fixedHeight := height(m.views.header) + height(m.views.error) + height(m.views.footer)
|
|
|
|
m.content.Height = m.height - fixedHeight
|
|
m.views.content = m.content.View()
|
|
}
|
|
return cmds
|
|
}
|
|
|
|
func (m *model) loadConversations() tea.Cmd {
|
|
return func() tea.Msg {
|
|
c, err := m.ctx.Store.Conversations()
|
|
if err != nil {
|
|
return msgError(fmt.Errorf("Could not load conversations: %v", err))
|
|
}
|
|
return msgConversationsLoaded(c)
|
|
}
|
|
}
|
|
|
|
func (m *model) loadConversationLastReplies(conversations []models.Conversation) tea.Cmd {
|
|
return func() tea.Msg {
|
|
//lastMessage, err := m.ctx.Store.LastMessage(&conversation)
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func (m *model) setConversations(conversations []models.Conversation) {
|
|
}
|
|
|
|
func (m *model) conversationListView() string {
|
|
sb := &strings.Builder{}
|
|
type AgeGroup struct {
|
|
name string
|
|
cutoff time.Duration
|
|
}
|
|
|
|
type ConversationLine struct {
|
|
id uint
|
|
short string
|
|
title string
|
|
elapsed string
|
|
lastReplyAge time.Duration
|
|
}
|
|
|
|
now := time.Now()
|
|
|
|
midnight := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location())
|
|
monthStart := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location())
|
|
dayOfWeek := int(now.Weekday())
|
|
categories := []AgeGroup{
|
|
{"Today", now.Sub(midnight)},
|
|
{"Yesterday", now.Sub(midnight.AddDate(0, 0, -1))},
|
|
{"This week", now.Sub(midnight.AddDate(0, 0, -dayOfWeek))},
|
|
{"Last week", now.Sub(midnight.AddDate(0, 0, -(dayOfWeek + 7)))},
|
|
{"This month", now.Sub(monthStart)},
|
|
{"Last month", now.Sub(monthStart.AddDate(0, -1, 0))},
|
|
{"2 Months ago", now.Sub(monthStart.AddDate(0, -2, 0))},
|
|
{"3 Months ago", now.Sub(monthStart.AddDate(0, -3, 0))},
|
|
{"4 Months ago", now.Sub(monthStart.AddDate(0, -4, 0))},
|
|
{"5 Months ago", now.Sub(monthStart.AddDate(0, -5, 0))},
|
|
{"6 Months ago", now.Sub(monthStart.AddDate(0, -6, 0))},
|
|
{"Older", now.Sub(time.Time{})},
|
|
}
|
|
categorized := map[string][]ConversationLine{}
|
|
|
|
for _, conversation := range m.conversations {
|
|
lastMessage, err := m.ctx.Store.LastMessage(&conversation)
|
|
if lastMessage == nil || err != nil {
|
|
continue
|
|
}
|
|
messageAge := now.Sub(lastMessage.CreatedAt)
|
|
var category string
|
|
for _, c := range categories {
|
|
if messageAge < c.cutoff {
|
|
category = c.name
|
|
break
|
|
}
|
|
}
|
|
|
|
categorized[category] = append(
|
|
categorized[category],
|
|
ConversationLine{
|
|
id: conversation.ID,
|
|
title: conversation.Title,
|
|
short: conversation.ShortName.String,
|
|
lastReplyAge: messageAge,
|
|
},
|
|
)
|
|
}
|
|
|
|
// TODO: pick nice color
|
|
categoryStyle := lipgloss.NewStyle().
|
|
MarginBottom(1).
|
|
Foreground(lipgloss.Color("170")).
|
|
PaddingLeft(1).
|
|
Bold(true)
|
|
|
|
conversationHeadingStyle := lipgloss.NewStyle().
|
|
MarginBottom(1).
|
|
PaddingLeft(2)
|
|
|
|
for _, category := range categories {
|
|
conversationLines, ok := categorized[category.name]
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
slices.SortFunc(conversationLines, func(a, b ConversationLine) int {
|
|
return int(a.lastReplyAge - b.lastReplyAge)
|
|
})
|
|
|
|
ageStyle := lipgloss.NewStyle().Faint(true)
|
|
titleStyle := lipgloss.NewStyle().Bold(true)
|
|
untitledStyle := titleStyle.Copy().Italic(true).SetString("(untitled)")
|
|
|
|
fmt.Fprintf(sb, "%s\n", categoryStyle.Render(category.name))
|
|
for _, c := range conversationLines {
|
|
tstyle := titleStyle
|
|
if c.title == "" {
|
|
tstyle = untitledStyle
|
|
}
|
|
heading := fmt.Sprintf(
|
|
"%s\n%s",
|
|
tstyle.Render(c.title),
|
|
ageStyle.Render(util.HumanTimeElapsedSince(c.lastReplyAge)),
|
|
)
|
|
sb.WriteString(conversationHeadingStyle.Render(heading))
|
|
sb.WriteRune('\n')
|
|
}
|
|
}
|
|
|
|
return sb.String()
|
|
}
|