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) {
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}
81
82func (i *tableCmp) BindingKeys() []key.Binding {
83 return layout.KeyMapToSlice(i.table.KeyMap)
84}
85
86func (i *tableCmp) setRows() {
87 rows := []table.Row{}
88
89 logs := logging.List()
90 slices.SortFunc(logs, func(a, b logging.LogMessage) int {
91 if a.Time.Before(b.Time) {
92 return 1
93 }
94 if a.Time.After(b.Time) {
95 return -1
96 }
97 return 0
98 })
99
100 for _, log := range logs {
101 bm, _ := json.Marshal(log.Attributes)
102
103 row := table.Row{
104 log.ID,
105 log.Time.Format("15:04:05"),
106 log.Level,
107 log.Message,
108 string(bm),
109 }
110 rows = append(rows, row)
111 }
112 i.table.SetRows(rows)
113}
114
115func NewLogsTable() TableComponent {
116 columns := []table.Column{
117 {Title: "ID", Width: 4},
118 {Title: "Time", Width: 4},
119 {Title: "Level", Width: 10},
120 {Title: "Message", Width: 10},
121 {Title: "Attributes", Width: 10},
122 }
123 defaultStyles := table.DefaultStyles()
124 defaultStyles.Selected = defaultStyles.Selected.Foreground(styles.Primary)
125 tableModel := table.New(
126 table.WithColumns(columns),
127 table.WithStyles(defaultStyles),
128 )
129 tableModel.Focus()
130 return &tableCmp{
131 table: tableModel,
132 }
133}