1package logs
2
3import (
4 "encoding/json"
5 "slices"
6
7 "github.com/charmbracelet/bubbles/key"
8 "github.com/charmbracelet/bubbles/table"
9 tea "github.com/charmbracelet/bubbletea"
10 "github.com/kujtimiihoxha/opencode/internal/logging"
11 "github.com/kujtimiihoxha/opencode/internal/pubsub"
12 "github.com/kujtimiihoxha/opencode/internal/tui/layout"
13 "github.com/kujtimiihoxha/opencode/internal/tui/styles"
14 "github.com/kujtimiihoxha/opencode/internal/tui/util"
15)
16
17type TableComponent interface {
18 tea.Model
19 layout.Sizeable
20 layout.Bindings
21}
22
23type tableCmp struct {
24 table table.Model
25}
26
27type selectedLogMsg logging.LogMessage
28
29func (i *tableCmp) Init() tea.Cmd {
30 i.setRows()
31 return nil
32}
33
34func (i *tableCmp) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
35 var cmds []tea.Cmd
36 switch msg.(type) {
37 case pubsub.Event[logging.LogMessage]:
38 i.setRows()
39 return i, nil
40 }
41 prevSelectedRow := i.table.SelectedRow()
42 t, cmd := i.table.Update(msg)
43 cmds = append(cmds, cmd)
44 i.table = t
45 selectedRow := i.table.SelectedRow()
46 if selectedRow != nil {
47 if prevSelectedRow == nil || selectedRow[0] == prevSelectedRow[0] {
48 var log logging.LogMessage
49 for _, row := range logging.List() {
50 if row.ID == selectedRow[0] {
51 log = row
52 break
53 }
54 }
55 if log.ID != "" {
56 cmds = append(cmds, util.CmdHandler(selectedLogMsg(log)))
57 }
58 }
59 }
60 return i, tea.Batch(cmds...)
61}
62
63func (i *tableCmp) View() string {
64 return styles.ForceReplaceBackgroundWithLipgloss(i.table.View(), styles.Background)
65}
66
67func (i *tableCmp) GetSize() (int, int) {
68 return i.table.Width(), i.table.Height()
69}
70
71func (i *tableCmp) SetSize(width int, height int) tea.Cmd {
72 i.table.SetWidth(width)
73 i.table.SetHeight(height)
74 cloumns := i.table.Columns()
75 for i, col := range cloumns {
76 col.Width = (width / len(cloumns)) - 2
77 cloumns[i] = col
78 }
79 i.table.SetColumns(cloumns)
80 return nil
81}
82
83func (i *tableCmp) BindingKeys() []key.Binding {
84 return layout.KeyMapToSlice(i.table.KeyMap)
85}
86
87func (i *tableCmp) setRows() {
88 rows := []table.Row{}
89
90 logs := logging.List()
91 slices.SortFunc(logs, func(a, b logging.LogMessage) int {
92 if a.Time.Before(b.Time) {
93 return 1
94 }
95 if a.Time.After(b.Time) {
96 return -1
97 }
98 return 0
99 })
100
101 for _, log := range logs {
102 bm, _ := json.Marshal(log.Attributes)
103
104 row := table.Row{
105 log.ID,
106 log.Time.Format("15:04:05"),
107 log.Level,
108 log.Message,
109 string(bm),
110 }
111 rows = append(rows, row)
112 }
113 i.table.SetRows(rows)
114}
115
116func NewLogsTable() TableComponent {
117 columns := []table.Column{
118 {Title: "ID", Width: 4},
119 {Title: "Time", Width: 4},
120 {Title: "Level", Width: 10},
121 {Title: "Message", Width: 10},
122 {Title: "Attributes", Width: 10},
123 }
124 defaultStyles := table.DefaultStyles()
125 defaultStyles.Selected = defaultStyles.Selected.Foreground(styles.Primary)
126 tableModel := table.New(
127 table.WithColumns(columns),
128 table.WithStyles(defaultStyles),
129 )
130 tableModel.Focus()
131 return &tableCmp{
132 table: tableModel,
133 }
134}