1package termui
2
3import (
4 "bytes"
5 "fmt"
6 "strings"
7 "time"
8
9 "github.com/MichaelMure/go-term-text"
10 "github.com/awesome-gocui/gocui"
11 "github.com/dustin/go-humanize"
12
13 "github.com/MichaelMure/git-bug/cache"
14 "github.com/MichaelMure/git-bug/entity"
15 "github.com/MichaelMure/git-bug/query"
16 "github.com/MichaelMure/git-bug/query/ast"
17 "github.com/MichaelMure/git-bug/util/colors"
18)
19
20const bugTableView = "bugTableView"
21const bugTableHeaderView = "bugTableHeaderView"
22const bugTableFooterView = "bugTableFooterView"
23const bugTableInstructionView = "bugTableInstructionView"
24
25const defaultRemote = "origin"
26const defaultQuery = "status:open"
27
28type bugTable struct {
29 repo *cache.RepoCache
30 queryStr string
31 query *ast.Query
32 allIds []entity.Id
33 excerpts []*cache.BugExcerpt
34 pageCursor int
35 selectCursor int
36}
37
38func newBugTable(c *cache.RepoCache) *bugTable {
39 q, err := query.Parse(defaultQuery)
40 if err != nil {
41 panic(err)
42 }
43
44 return &bugTable{
45 repo: c,
46 query: q,
47 queryStr: defaultQuery,
48 pageCursor: 0,
49 selectCursor: 0,
50 }
51}
52
53func (bt *bugTable) layout(g *gocui.Gui) error {
54 maxX, maxY := g.Size()
55
56 if maxY < 4 {
57 // window too small !
58 return nil
59 }
60
61 v, err := g.SetView(bugTableHeaderView, -1, -1, maxX, 3, 0)
62
63 if err != nil {
64 if !gocui.IsUnknownView(err) {
65 return err
66 }
67
68 v.Frame = false
69 }
70
71 v.Clear()
72 bt.renderHeader(v, maxX)
73
74 v, err = g.SetView(bugTableView, -1, 1, maxX, maxY-3, 0)
75
76 if err != nil {
77 if !gocui.IsUnknownView(err) {
78 return err
79 }
80
81 v.Frame = false
82 v.SelBgColor = gocui.ColorWhite
83 v.SelFgColor = gocui.ColorBlack
84 }
85
86 _, viewHeight := v.Size()
87 err = bt.paginate(viewHeight)
88 if err != nil {
89 return err
90 }
91
92 err = bt.cursorClamp(v)
93 if err != nil {
94 return err
95 }
96
97 v.Clear()
98 bt.render(v, maxX)
99
100 v, err = g.SetView(bugTableFooterView, -1, maxY-4, maxX, maxY, 0)
101
102 if err != nil {
103 if !gocui.IsUnknownView(err) {
104 return err
105 }
106
107 v.Frame = false
108 }
109
110 v.Clear()
111 bt.renderFooter(v, maxX)
112
113 v, err = g.SetView(bugTableInstructionView, -1, maxY-2, maxX, maxY, 0)
114
115 if err != nil {
116 if !gocui.IsUnknownView(err) {
117 return err
118 }
119
120 v.Frame = false
121 v.BgColor = gocui.ColorBlue
122
123 _, _ = fmt.Fprintf(v, "[q] Quit [s] Search [←↓↑→,hjkl] Navigation [↵] Open bug [n] New bug [i] Pull [o] Push")
124 }
125
126 _, err = g.SetCurrentView(bugTableView)
127 return err
128}
129
130func (bt *bugTable) keybindings(g *gocui.Gui) error {
131 // Quit
132 if err := g.SetKeybinding(bugTableView, 'q', gocui.ModNone, quit); err != nil {
133 return err
134 }
135
136 // Down
137 if err := g.SetKeybinding(bugTableView, 'j', gocui.ModNone,
138 bt.cursorDown); err != nil {
139 return err
140 }
141 if err := g.SetKeybinding(bugTableView, gocui.KeyArrowDown, gocui.ModNone,
142 bt.cursorDown); err != nil {
143 return err
144 }
145 // Up
146 if err := g.SetKeybinding(bugTableView, 'k', gocui.ModNone,
147 bt.cursorUp); err != nil {
148 return err
149 }
150 if err := g.SetKeybinding(bugTableView, gocui.KeyArrowUp, gocui.ModNone,
151 bt.cursorUp); err != nil {
152 return err
153 }
154
155 // Previous page
156 if err := g.SetKeybinding(bugTableView, 'h', gocui.ModNone,
157 bt.previousPage); err != nil {
158 return err
159 }
160 if err := g.SetKeybinding(bugTableView, gocui.KeyArrowLeft, gocui.ModNone,
161 bt.previousPage); err != nil {
162 return err
163 }
164 if err := g.SetKeybinding(bugTableView, gocui.KeyPgup, gocui.ModNone,
165 bt.previousPage); err != nil {
166 return err
167 }
168 // Next page
169 if err := g.SetKeybinding(bugTableView, 'l', gocui.ModNone,
170 bt.nextPage); err != nil {
171 return err
172 }
173 if err := g.SetKeybinding(bugTableView, gocui.KeyArrowRight, gocui.ModNone,
174 bt.nextPage); err != nil {
175 return err
176 }
177 if err := g.SetKeybinding(bugTableView, gocui.KeyPgdn, gocui.ModNone,
178 bt.nextPage); err != nil {
179 return err
180 }
181
182 // New bug
183 if err := g.SetKeybinding(bugTableView, 'n', gocui.ModNone,
184 bt.newBug); err != nil {
185 return err
186 }
187
188 // Open bug
189 if err := g.SetKeybinding(bugTableView, gocui.KeyEnter, gocui.ModNone,
190 bt.openBug); err != nil {
191 return err
192 }
193
194 // Pull
195 if err := g.SetKeybinding(bugTableView, 'i', gocui.ModNone,
196 bt.pull); err != nil {
197 return err
198 }
199
200 // Push
201 if err := g.SetKeybinding(bugTableView, 'o', gocui.ModNone,
202 bt.push); err != nil {
203 return err
204 }
205
206 // Query
207 if err := g.SetKeybinding(bugTableView, 's', gocui.ModNone,
208 bt.changeQuery); err != nil {
209 return err
210 }
211
212 return nil
213}
214
215func (bt *bugTable) disable(g *gocui.Gui) error {
216 if err := g.DeleteView(bugTableView); err != nil && !gocui.IsUnknownView(err) {
217 return err
218 }
219 if err := g.DeleteView(bugTableHeaderView); err != nil && !gocui.IsUnknownView(err) {
220 return err
221 }
222 if err := g.DeleteView(bugTableFooterView); err != nil && !gocui.IsUnknownView(err) {
223 return err
224 }
225 if err := g.DeleteView(bugTableInstructionView); err != nil && !gocui.IsUnknownView(err) {
226 return err
227 }
228 return nil
229}
230
231func (bt *bugTable) paginate(max int) error {
232 bt.allIds = bt.repo.QueryBugs(bt.query)
233
234 return bt.doPaginate(max)
235}
236
237func (bt *bugTable) doPaginate(max int) error {
238 // clamp the cursor
239 bt.pageCursor = maxInt(bt.pageCursor, 0)
240 bt.pageCursor = minInt(bt.pageCursor, len(bt.allIds))
241
242 nb := minInt(len(bt.allIds)-bt.pageCursor, max)
243
244 if nb < 0 {
245 bt.excerpts = []*cache.BugExcerpt{}
246 return nil
247 }
248
249 // slice the data
250 ids := bt.allIds[bt.pageCursor : bt.pageCursor+nb]
251
252 bt.excerpts = make([]*cache.BugExcerpt, len(ids))
253
254 for i, id := range ids {
255 excerpt, err := bt.repo.ResolveBugExcerpt(id)
256 if err != nil {
257 return err
258 }
259
260 bt.excerpts[i] = excerpt
261 }
262
263 return nil
264}
265
266func (bt *bugTable) getTableLength() int {
267 return len(bt.excerpts)
268}
269
270func (bt *bugTable) getColumnWidths(maxX int) map[string]int {
271 m := make(map[string]int)
272 m["id"] = 9
273 m["status"] = 7
274
275 left := maxX - 5 - m["id"] - m["status"]
276
277 m["comments"] = 10
278 left -= m["comments"]
279 m["lastEdit"] = 19
280 left -= m["lastEdit"]
281
282 m["author"] = minInt(maxInt(left/3, 15), 10+left/8)
283 m["title"] = maxInt(left-m["author"], 10)
284
285 return m
286}
287
288func (bt *bugTable) render(v *gocui.View, maxX int) {
289 columnWidths := bt.getColumnWidths(maxX)
290
291 for _, excerpt := range bt.excerpts {
292 summaryTxt := fmt.Sprintf("%4d 💬", excerpt.LenComments)
293 if excerpt.LenComments <= 0 {
294 summaryTxt = ""
295 }
296 if excerpt.LenComments > 9999 {
297 summaryTxt = " ∞ 💬"
298 }
299
300 var labelsTxt strings.Builder
301 for _, l := range excerpt.Labels {
302 lc256 := l.Color().Term256()
303 labelsTxt.WriteString(lc256.Escape())
304 labelsTxt.WriteString(" ◼")
305 labelsTxt.WriteString(lc256.Unescape())
306 }
307
308 var authorDisplayName string
309 if excerpt.AuthorId != "" {
310 author, err := bt.repo.ResolveIdentityExcerpt(excerpt.AuthorId)
311 if err != nil {
312 panic(err)
313 }
314 authorDisplayName = author.DisplayName()
315 } else {
316 authorDisplayName = excerpt.LegacyAuthor.DisplayName()
317 }
318
319 lastEditTime := time.Unix(excerpt.EditUnixTime, 0)
320
321 id := text.LeftPadMaxLine(excerpt.Id.Human(), columnWidths["id"], 1)
322 status := text.LeftPadMaxLine(excerpt.Status.String(), columnWidths["status"], 1)
323 labels := text.TruncateMax(labelsTxt.String(), minInt(columnWidths["title"]-2, 10))
324 title := text.LeftPadMaxLine(excerpt.Title, columnWidths["title"]-text.Len(labels), 1)
325 author := text.LeftPadMaxLine(authorDisplayName, columnWidths["author"], 1)
326 comments := text.LeftPadMaxLine(summaryTxt, columnWidths["comments"], 1)
327 lastEdit := text.LeftPadMaxLine(humanize.Time(lastEditTime), columnWidths["lastEdit"], 1)
328
329 _, _ = fmt.Fprintf(v, "%s %s %s%s %s %s %s\n",
330 colors.Cyan(id),
331 colors.Yellow(status),
332 title,
333 labels,
334 colors.Magenta(author),
335 comments,
336 lastEdit,
337 )
338 }
339
340 _ = v.SetHighlight(bt.selectCursor, true)
341}
342
343func (bt *bugTable) renderHeader(v *gocui.View, maxX int) {
344 columnWidths := bt.getColumnWidths(maxX)
345
346 id := text.LeftPadMaxLine("ID", columnWidths["id"], 1)
347 status := text.LeftPadMaxLine("STATUS", columnWidths["status"], 1)
348 title := text.LeftPadMaxLine("TITLE", columnWidths["title"], 1)
349 author := text.LeftPadMaxLine("AUTHOR", columnWidths["author"], 1)
350 comments := text.LeftPadMaxLine("COMMENTS", columnWidths["comments"], 1)
351 lastEdit := text.LeftPadMaxLine("LAST EDIT", columnWidths["lastEdit"], 1)
352
353 _, _ = fmt.Fprintf(v, "\n")
354 _, _ = fmt.Fprintf(v, "%s %s %s %s %s %s\n", id, status, title, author, comments, lastEdit)
355}
356
357func (bt *bugTable) renderFooter(v *gocui.View, maxX int) {
358 _, _ = fmt.Fprintf(v, " \nShowing %d of %d bugs", len(bt.excerpts), len(bt.allIds))
359}
360
361func (bt *bugTable) cursorDown(g *gocui.Gui, v *gocui.View) error {
362 // If we are at the bottom of the page, switch to the next one.
363 if bt.selectCursor+1 > bt.getTableLength()-1 {
364 _, max := v.Size()
365
366 if bt.pageCursor+max >= len(bt.allIds) {
367 return nil
368 }
369
370 bt.pageCursor += max
371 bt.selectCursor = 0
372
373 return bt.doPaginate(max)
374 }
375
376 bt.selectCursor = minInt(bt.selectCursor+1, bt.getTableLength()-1)
377
378 return nil
379}
380
381func (bt *bugTable) cursorUp(g *gocui.Gui, v *gocui.View) error {
382 // If we are at the top of the page, switch to the previous one.
383 if bt.selectCursor-1 < 0 {
384 _, max := v.Size()
385
386 if bt.pageCursor == 0 {
387 return nil
388 }
389
390 bt.pageCursor = maxInt(0, bt.pageCursor-max)
391 bt.selectCursor = max - 1
392
393 return bt.doPaginate(max)
394 }
395
396 bt.selectCursor = maxInt(bt.selectCursor-1, 0)
397
398 return nil
399}
400
401func (bt *bugTable) cursorClamp(v *gocui.View) error {
402 y := bt.selectCursor
403
404 y = minInt(y, bt.getTableLength()-1)
405 y = maxInt(y, 0)
406
407 bt.selectCursor = y
408
409 return nil
410}
411
412func (bt *bugTable) nextPage(g *gocui.Gui, v *gocui.View) error {
413 _, max := v.Size()
414
415 if bt.pageCursor+max >= len(bt.allIds) {
416 return nil
417 }
418
419 bt.pageCursor += max
420
421 return bt.doPaginate(max)
422}
423
424func (bt *bugTable) previousPage(g *gocui.Gui, v *gocui.View) error {
425 _, max := v.Size()
426
427 if bt.pageCursor == 0 {
428 return nil
429 }
430
431 bt.pageCursor = maxInt(0, bt.pageCursor-max)
432
433 return bt.doPaginate(max)
434}
435
436func (bt *bugTable) newBug(g *gocui.Gui, v *gocui.View) error {
437 return newBugWithEditor(bt.repo)
438}
439
440func (bt *bugTable) openBug(g *gocui.Gui, v *gocui.View) error {
441 id := bt.excerpts[bt.selectCursor].Id
442 b, err := bt.repo.ResolveBug(id)
443 if err != nil {
444 return err
445 }
446 ui.showBug.SetBug(b)
447 return ui.activateWindow(ui.showBug)
448}
449
450func (bt *bugTable) pull(g *gocui.Gui, v *gocui.View) error {
451 ui.msgPopup.Activate("Pull from remote "+defaultRemote, "...")
452
453 go func() {
454 stdout, err := bt.repo.Fetch(defaultRemote)
455
456 if err != nil {
457 g.Update(func(gui *gocui.Gui) error {
458 ui.msgPopup.Activate(msgPopupErrorTitle, err.Error())
459 return nil
460 })
461 } else {
462 g.Update(func(gui *gocui.Gui) error {
463 ui.msgPopup.UpdateMessage(stdout)
464 return nil
465 })
466 }
467
468 var buffer bytes.Buffer
469 beginLine := ""
470
471 for result := range bt.repo.MergeAll(defaultRemote) {
472 if result.Status == entity.MergeStatusNothing {
473 continue
474 }
475
476 if result.Err != nil {
477 g.Update(func(gui *gocui.Gui) error {
478 ui.msgPopup.Activate(msgPopupErrorTitle, err.Error())
479 return nil
480 })
481 } else {
482 _, _ = fmt.Fprintf(&buffer, "%s%s: %s",
483 beginLine, colors.Cyan(result.Entity.Id().Human()), result,
484 )
485
486 beginLine = "\n"
487
488 g.Update(func(gui *gocui.Gui) error {
489 ui.msgPopup.UpdateMessage(buffer.String())
490 return nil
491 })
492 }
493 }
494
495 _, _ = fmt.Fprintf(&buffer, "%sdone", beginLine)
496
497 g.Update(func(gui *gocui.Gui) error {
498 ui.msgPopup.UpdateMessage(buffer.String())
499 return nil
500 })
501
502 }()
503
504 return nil
505}
506
507func (bt *bugTable) push(g *gocui.Gui, v *gocui.View) error {
508 ui.msgPopup.Activate("Push to remote "+defaultRemote, "...")
509
510 go func() {
511 // TODO: make the remote configurable
512 stdout, err := bt.repo.Push(defaultRemote)
513
514 if err != nil {
515 g.Update(func(gui *gocui.Gui) error {
516 ui.msgPopup.Activate(msgPopupErrorTitle, err.Error())
517 return nil
518 })
519 } else {
520 g.Update(func(gui *gocui.Gui) error {
521 ui.msgPopup.UpdateMessage(stdout)
522 return nil
523 })
524 }
525 }()
526
527 return nil
528}
529
530func (bt *bugTable) changeQuery(g *gocui.Gui, v *gocui.View) error {
531 return editQueryWithEditor(bt)
532}