1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
package messagebox
import (
"slices"
"strings"
"time"
"github.com/charmbracelet/bubbles/viewport"
tea "github.com/charmbracelet/bubbletea"
"github.com/charmbracelet/lipgloss"
"github.com/kyren223/eko/internal/client/api"
"github.com/kyren223/eko/internal/data"
"github.com/kyren223/eko/pkg/snowflake"
)
type user struct {
name string
isFetching bool
}
type Model struct {
Viewport viewport.Model
messages []data.Message
senderStyle lipgloss.Style
timestampStyle lipgloss.Style
users map[snowflake.ID]user
}
func New(width, height int) Model {
return Model{
Viewport: viewport.New(width, height),
messages: nil,
senderStyle: lipgloss.NewStyle().Foreground(lipgloss.Color("5")).Bold(true),
timestampStyle: lipgloss.NewStyle().Foreground(lipgloss.Color("#585c62")).Italic(true),
users: make(map[snowflake.ID]user),
}
}
func (m Model) Init() tea.Cmd {
return api.GetMessages
}
func (m Model) View() string {
return m.Viewport.View()
}
func (m Model) Update(msg tea.Msg) (Model, tea.Cmd) {
var cmds []tea.Cmd
for id, user := range m.users {
if !user.isFetching && user.name == "" {
cmds = append(cmds, api.GetUserById(id))
user.isFetching = true
m.users[id] = user
}
}
var cmd tea.Cmd
m.Viewport, cmd = m.Viewport.Update(msg)
cmds = append(cmds, cmd)
switch msg := msg.(type) {
case api.UserProfileUpdate:
m.users[msg.ID] = user{msg.Name, false}
m.UpdateMessages()
}
return m, tea.Batch(cmds...)
}
func (m *Model) AppendMessage(message data.Message) {
m.messages = append(m.messages, message)
m.UpdateMessages()
}
func (m *Model) SetMessages(messages []data.Message) {
m.messages = messages
m.UpdateMessages()
}
func (m *Model) UpdateMessages() {
for _, message := range m.messages {
if _, exists := m.users[message.SenderID]; !exists {
m.users[message.SenderID] = user{name: "", isFetching: false}
}
}
m.sortMessages()
m.updateContent()
m.Viewport.GotoBottom()
}
func (m *Model) sortMessages() {
slices.SortFunc(m.messages, func(a, b data.Message) int {
if a.ID-b.ID < 0 {
return -1
} else {
return 1
}
})
}
func (m *Model) updateContent() {
if len(m.messages) == 0 {
return
}
var builder strings.Builder
for _, message := range m.messages {
sender := m.users[message.SenderID].name
builder.WriteString(m.senderStyle.Render(sender))
builder.WriteByte(' ')
timestamp := timestampFromID(message.ID)
builder.WriteString(m.timestampStyle.Render(timestamp))
builder.WriteByte('\n')
builder.WriteString(message.Content)
builder.WriteByte('\n')
}
m.Viewport.SetContent(builder.String()[:builder.Len()-1])
}
func timestampFromID(id snowflake.ID) string {
localTime := time.UnixMilli(id.Time())
return localTime.Format("02/01/2006 3:04 PM")
}
|
