1package termui
  2
  3import (
  4	"bytes"
  5	"fmt"
  6	"strings"
  7
  8	"github.com/MichaelMure/git-bug/bug"
  9	"github.com/MichaelMure/git-bug/cache"
 10	"github.com/MichaelMure/git-bug/util/colors"
 11	"github.com/MichaelMure/git-bug/util/git"
 12	"github.com/MichaelMure/git-bug/util/text"
 13	"github.com/MichaelMure/gocui"
 14)
 15
 16const showBugView = "showBugView"
 17const showBugSidebarView = "showBugSidebarView"
 18const showBugInstructionView = "showBugInstructionView"
 19const showBugHeaderView = "showBugHeaderView"
 20
 21const timeLayout = "Jan 2 2006"
 22
 23type showBug struct {
 24	cache              *cache.RepoCache
 25	bug                *cache.BugCache
 26	childViews         []string
 27	mainSelectableView []string
 28	sideSelectableView []string
 29	selected           string
 30	isOnSide           bool
 31	scroll             int
 32}
 33
 34func newShowBug(cache *cache.RepoCache) *showBug {
 35	return &showBug{
 36		cache: cache,
 37	}
 38}
 39
 40func (sb *showBug) SetBug(bug *cache.BugCache) {
 41	sb.bug = bug
 42	sb.scroll = 0
 43	sb.selected = ""
 44	sb.isOnSide = false
 45}
 46
 47func (sb *showBug) layout(g *gocui.Gui) error {
 48	maxX, maxY := g.Size()
 49	sb.childViews = nil
 50
 51	v, err := g.SetView(showBugView, 0, 0, maxX*2/3, maxY-2)
 52
 53	if err != nil {
 54		if err != gocui.ErrUnknownView {
 55			return err
 56		}
 57
 58		sb.childViews = append(sb.childViews, showBugView)
 59		v.Frame = false
 60	}
 61
 62	v.Clear()
 63	err = sb.renderMain(g, v)
 64	if err != nil {
 65		return err
 66	}
 67
 68	v, err = g.SetView(showBugSidebarView, maxX*2/3+1, 0, maxX-1, maxY-2)
 69
 70	if err != nil {
 71		if err != gocui.ErrUnknownView {
 72			return err
 73		}
 74
 75		sb.childViews = append(sb.childViews, showBugSidebarView)
 76		v.Frame = false
 77	}
 78
 79	v.Clear()
 80	err = sb.renderSidebar(g, v)
 81	if err != nil {
 82		return err
 83	}
 84
 85	v, err = g.SetView(showBugInstructionView, -1, maxY-2, maxX, maxY)
 86
 87	if err != nil {
 88		if err != gocui.ErrUnknownView {
 89			return err
 90		}
 91
 92		sb.childViews = append(sb.childViews, showBugInstructionView)
 93		v.Frame = false
 94		v.BgColor = gocui.ColorBlue
 95	}
 96
 97	v.Clear()
 98	_, _ = fmt.Fprintf(v, "[q] Save and return [←↓↑→,hjkl] Navigation [o] Toggle open/close [e] Edit [c] Comment [t] Change title")
 99
100	_, err = g.SetViewOnTop(showBugInstructionView)
101	if err != nil {
102		return err
103	}
104
105	_, err = g.SetCurrentView(showBugView)
106	return err
107}
108
109func (sb *showBug) keybindings(g *gocui.Gui) error {
110	// Return
111	if err := g.SetKeybinding(showBugView, 'q', gocui.ModNone, sb.saveAndBack); err != nil {
112		return err
113	}
114
115	// Scrolling
116	if err := g.SetKeybinding(showBugView, gocui.KeyPgup, gocui.ModNone,
117		sb.scrollUp); err != nil {
118		return err
119	}
120	if err := g.SetKeybinding(showBugView, gocui.KeyPgdn, gocui.ModNone,
121		sb.scrollDown); err != nil {
122		return err
123	}
124
125	// Down
126	if err := g.SetKeybinding(showBugView, 'j', gocui.ModNone,
127		sb.selectNext); err != nil {
128		return err
129	}
130	if err := g.SetKeybinding(showBugView, gocui.KeyArrowDown, gocui.ModNone,
131		sb.selectNext); err != nil {
132		return err
133	}
134	// Up
135	if err := g.SetKeybinding(showBugView, 'k', gocui.ModNone,
136		sb.selectPrevious); err != nil {
137		return err
138	}
139	if err := g.SetKeybinding(showBugView, gocui.KeyArrowUp, gocui.ModNone,
140		sb.selectPrevious); err != nil {
141		return err
142	}
143
144	// Left
145	if err := g.SetKeybinding(showBugView, 'h', gocui.ModNone,
146		sb.left); err != nil {
147		return err
148	}
149	if err := g.SetKeybinding(showBugView, gocui.KeyArrowLeft, gocui.ModNone,
150		sb.left); err != nil {
151		return err
152	}
153	// Right
154	if err := g.SetKeybinding(showBugView, 'l', gocui.ModNone,
155		sb.right); err != nil {
156		return err
157	}
158	if err := g.SetKeybinding(showBugView, gocui.KeyArrowRight, gocui.ModNone,
159		sb.right); err != nil {
160		return err
161	}
162
163	// Comment
164	if err := g.SetKeybinding(showBugView, 'c', gocui.ModNone,
165		sb.comment); err != nil {
166		return err
167	}
168
169	// Open/close
170	if err := g.SetKeybinding(showBugView, 'o', gocui.ModNone,
171		sb.toggleOpenClose); err != nil {
172		return err
173	}
174
175	// Title
176	if err := g.SetKeybinding(showBugView, 't', gocui.ModNone,
177		sb.setTitle); err != nil {
178		return err
179	}
180
181	// Edit
182	if err := g.SetKeybinding(showBugView, 'e', gocui.ModNone,
183		sb.edit); err != nil {
184		return err
185	}
186
187	return nil
188}
189
190func (sb *showBug) disable(g *gocui.Gui) error {
191	for _, view := range sb.childViews {
192		if err := g.DeleteView(view); err != nil && err != gocui.ErrUnknownView {
193			return err
194		}
195	}
196	return nil
197}
198
199func (sb *showBug) renderMain(g *gocui.Gui, mainView *gocui.View) error {
200	maxX, _ := mainView.Size()
201	x0, y0, _, _, _ := g.ViewPosition(mainView.Name())
202
203	y0 -= sb.scroll
204
205	snap := sb.bug.Snapshot()
206
207	sb.mainSelectableView = nil
208
209	createTimelineItem := snap.Timeline[0].(*bug.CreateTimelineItem)
210
211	edited := ""
212	if createTimelineItem.Edited() {
213		edited = " (edited)"
214	}
215
216	bugHeader := fmt.Sprintf("[%s] %s\n\n[%s] %s opened this bug on %s%s",
217		colors.Cyan(snap.HumanId()),
218		colors.Bold(snap.Title),
219		colors.Yellow(snap.Status),
220		colors.Magenta(snap.Author.DisplayName()),
221		snap.CreatedAt.Format(timeLayout),
222		edited,
223	)
224	bugHeader, lines := text.Wrap(bugHeader, maxX)
225
226	v, err := sb.createOpView(g, showBugHeaderView, x0, y0, maxX+1, lines, false)
227	if err != nil {
228		return err
229	}
230
231	_, _ = fmt.Fprint(v, bugHeader)
232	y0 += lines + 1
233
234	for _, op := range snap.Timeline {
235		viewName := op.Hash().String()
236
237		// TODO: me might skip the rendering of blocks that are outside of the view
238		// but to do that we need to rework how sb.mainSelectableView is maintained
239
240		switch op.(type) {
241
242		case *bug.CreateTimelineItem:
243			create := op.(*bug.CreateTimelineItem)
244
245			var content string
246			var lines int
247
248			if create.MessageIsEmpty() {
249				content, lines = text.WrapLeftPadded(emptyMessagePlaceholder(), maxX-1, 4)
250			} else {
251				content, lines = text.WrapLeftPadded(create.Message, maxX-1, 4)
252			}
253
254			v, err := sb.createOpView(g, viewName, x0, y0, maxX+1, lines, true)
255			if err != nil {
256				return err
257			}
258			_, _ = fmt.Fprint(v, content)
259			y0 += lines + 2
260
261		case *bug.AddCommentTimelineItem:
262			comment := op.(*bug.AddCommentTimelineItem)
263
264			edited := ""
265			if comment.Edited() {
266				edited = " (edited)"
267			}
268
269			var message string
270			if comment.MessageIsEmpty() {
271				message, _ = text.WrapLeftPadded(emptyMessagePlaceholder(), maxX-1, 4)
272			} else {
273				message, _ = text.WrapLeftPadded(comment.Message, maxX-1, 4)
274			}
275
276			content := fmt.Sprintf("%s commented on %s%s\n\n%s",
277				colors.Magenta(comment.Author.DisplayName()),
278				comment.CreatedAt.Time().Format(timeLayout),
279				edited,
280				message,
281			)
282			content, lines = text.Wrap(content, maxX)
283
284			v, err := sb.createOpView(g, viewName, x0, y0, maxX+1, lines, true)
285			if err != nil {
286				return err
287			}
288			_, _ = fmt.Fprint(v, content)
289			y0 += lines + 2
290
291		case *bug.SetTitleTimelineItem:
292			setTitle := op.(*bug.SetTitleTimelineItem)
293
294			content := fmt.Sprintf("%s changed the title to %s on %s",
295				colors.Magenta(setTitle.Author.DisplayName()),
296				colors.Bold(setTitle.Title),
297				setTitle.UnixTime.Time().Format(timeLayout),
298			)
299			content, lines := text.Wrap(content, maxX)
300
301			v, err := sb.createOpView(g, viewName, x0, y0, maxX+1, lines, true)
302			if err != nil {
303				return err
304			}
305			_, _ = fmt.Fprint(v, content)
306			y0 += lines + 2
307
308		case *bug.SetStatusTimelineItem:
309			setStatus := op.(*bug.SetStatusTimelineItem)
310
311			content := fmt.Sprintf("%s %s the bug on %s",
312				colors.Magenta(setStatus.Author.DisplayName()),
313				colors.Bold(setStatus.Status.Action()),
314				setStatus.UnixTime.Time().Format(timeLayout),
315			)
316			content, lines := text.Wrap(content, maxX)
317
318			v, err := sb.createOpView(g, viewName, x0, y0, maxX+1, lines, true)
319			if err != nil {
320				return err
321			}
322			_, _ = fmt.Fprint(v, content)
323			y0 += lines + 2
324
325		case *bug.LabelChangeTimelineItem:
326			labelChange := op.(*bug.LabelChangeTimelineItem)
327
328			var added []string
329			for _, label := range labelChange.Added {
330				added = append(added, colors.Bold("\""+label+"\""))
331			}
332
333			var removed []string
334			for _, label := range labelChange.Removed {
335				removed = append(removed, colors.Bold("\""+label+"\""))
336			}
337
338			var action bytes.Buffer
339
340			if len(added) > 0 {
341				action.WriteString("added ")
342				action.WriteString(strings.Join(added, ", "))
343
344				if len(removed) > 0 {
345					action.WriteString(" and ")
346				}
347			}
348
349			if len(removed) > 0 {
350				action.WriteString("removed ")
351				action.WriteString(strings.Join(removed, ", "))
352			}
353
354			if len(added)+len(removed) > 1 {
355				action.WriteString(" labels")
356			} else {
357				action.WriteString(" label")
358			}
359
360			content := fmt.Sprintf("%s %s on %s",
361				colors.Magenta(labelChange.Author.DisplayName()),
362				action.String(),
363				labelChange.UnixTime.Time().Format(timeLayout),
364			)
365			content, lines := text.Wrap(content, maxX)
366
367			v, err := sb.createOpView(g, viewName, x0, y0, maxX+1, lines, true)
368			if err != nil {
369				return err
370			}
371			_, _ = fmt.Fprint(v, content)
372			y0 += lines + 2
373		}
374	}
375
376	return nil
377}
378
379// emptyMessagePlaceholder return a formatted placeholder for an empty message
380func emptyMessagePlaceholder() string {
381	return colors.GreyBold("No description provided.")
382}
383
384func (sb *showBug) createOpView(g *gocui.Gui, name string, x0 int, y0 int, maxX int, height int, selectable bool) (*gocui.View, error) {
385	v, err := g.SetView(name, x0, y0, maxX, y0+height+1)
386
387	if err != nil && err != gocui.ErrUnknownView {
388		return nil, err
389	}
390
391	sb.childViews = append(sb.childViews, name)
392
393	if selectable {
394		sb.mainSelectableView = append(sb.mainSelectableView, name)
395	}
396
397	v.Frame = sb.selected == name
398
399	v.Clear()
400
401	return v, nil
402}
403
404func (sb *showBug) createSideView(g *gocui.Gui, name string, x0 int, y0 int, maxX int, height int) (*gocui.View, error) {
405	v, err := g.SetView(name, x0, y0, maxX, y0+height+1)
406
407	if err != nil && err != gocui.ErrUnknownView {
408		return nil, err
409	}
410
411	sb.childViews = append(sb.childViews, name)
412	sb.sideSelectableView = append(sb.sideSelectableView, name)
413
414	v.Frame = sb.selected == name
415
416	v.Clear()
417
418	return v, nil
419}
420
421func (sb *showBug) renderSidebar(g *gocui.Gui, sideView *gocui.View) error {
422	maxX, _ := sideView.Size()
423	x0, y0, _, _, _ := g.ViewPosition(sideView.Name())
424	maxX += x0
425
426	snap := sb.bug.Snapshot()
427
428	sb.sideSelectableView = nil
429
430	labelStr := make([]string, len(snap.Labels))
431	for i, l := range snap.Labels {
432		labelStr[i] = string(l)
433	}
434
435	labels := strings.Join(labelStr, "\n")
436	labels, lines := text.WrapLeftPadded(labels, maxX, 2)
437
438	content := fmt.Sprintf("%s\n\n%s", colors.Bold("Labels"), labels)
439
440	v, err := sb.createSideView(g, "sideLabels", x0, y0, maxX, lines+2)
441	if err != nil {
442		return err
443	}
444
445	_, _ = fmt.Fprint(v, content)
446
447	return nil
448}
449
450func (sb *showBug) saveAndBack(g *gocui.Gui, v *gocui.View) error {
451	err := sb.bug.CommitAsNeeded()
452	if err != nil {
453		return err
454	}
455	err = ui.activateWindow(ui.bugTable)
456	if err != nil {
457		return err
458	}
459	return nil
460}
461
462func (sb *showBug) scrollUp(g *gocui.Gui, v *gocui.View) error {
463	mainView, err := g.View(showBugView)
464	if err != nil {
465		return err
466	}
467
468	_, maxY := mainView.Size()
469
470	sb.scroll -= maxY / 2
471
472	sb.scroll = maxInt(sb.scroll, 0)
473
474	return nil
475}
476
477func (sb *showBug) scrollDown(g *gocui.Gui, v *gocui.View) error {
478	_, maxY := v.Size()
479
480	lastViewName := sb.mainSelectableView[len(sb.mainSelectableView)-1]
481
482	lastView, err := g.View(lastViewName)
483	if err != nil {
484		return err
485	}
486
487	_, vMaxY := lastView.Size()
488
489	_, vy0, _, _, err := g.ViewPosition(lastViewName)
490	if err != nil {
491		return err
492	}
493
494	maxScroll := vy0 + sb.scroll + vMaxY - maxY
495
496	sb.scroll += maxY / 2
497
498	sb.scroll = minInt(sb.scroll, maxScroll)
499
500	return nil
501}
502
503func (sb *showBug) selectPrevious(g *gocui.Gui, v *gocui.View) error {
504	var selectable []string
505	if sb.isOnSide {
506		selectable = sb.sideSelectableView
507	} else {
508		selectable = sb.mainSelectableView
509	}
510
511	for i, name := range selectable {
512		if name == sb.selected {
513			// special case to scroll up to the top
514			if i == 0 {
515				sb.scroll = 0
516			}
517
518			sb.selected = selectable[maxInt(i-1, 0)]
519			return sb.focusView(g)
520		}
521	}
522
523	if sb.selected == "" && len(selectable) > 0 {
524		sb.selected = selectable[0]
525	}
526
527	return sb.focusView(g)
528}
529
530func (sb *showBug) selectNext(g *gocui.Gui, v *gocui.View) error {
531	var selectable []string
532	if sb.isOnSide {
533		selectable = sb.sideSelectableView
534	} else {
535		selectable = sb.mainSelectableView
536	}
537
538	for i, name := range selectable {
539		if name == sb.selected {
540			sb.selected = selectable[minInt(i+1, len(selectable)-1)]
541			return sb.focusView(g)
542		}
543	}
544
545	if sb.selected == "" && len(selectable) > 0 {
546		sb.selected = selectable[0]
547	}
548
549	return sb.focusView(g)
550}
551
552func (sb *showBug) left(g *gocui.Gui, v *gocui.View) error {
553	if sb.isOnSide {
554		sb.isOnSide = false
555		sb.selected = ""
556		return sb.selectNext(g, v)
557	}
558
559	if sb.selected == "" {
560		return sb.selectNext(g, v)
561	}
562
563	return nil
564}
565
566func (sb *showBug) right(g *gocui.Gui, v *gocui.View) error {
567	if !sb.isOnSide {
568		sb.isOnSide = true
569		sb.selected = ""
570		return sb.selectNext(g, v)
571	}
572
573	if sb.selected == "" {
574		return sb.selectNext(g, v)
575	}
576
577	return nil
578}
579
580func (sb *showBug) focusView(g *gocui.Gui) error {
581	mainView, err := g.View(showBugView)
582	if err != nil {
583		return err
584	}
585
586	_, maxY := mainView.Size()
587
588	_, vy0, _, _, err := g.ViewPosition(sb.selected)
589	if err != nil {
590		return err
591	}
592
593	v, err := g.View(sb.selected)
594	if err != nil {
595		return err
596	}
597
598	_, vMaxY := v.Size()
599
600	vy1 := vy0 + vMaxY
601
602	if vy0 < 0 {
603		sb.scroll += vy0
604		return nil
605	}
606
607	if vy1 > maxY {
608		sb.scroll -= maxY - vy1
609	}
610
611	return nil
612}
613
614func (sb *showBug) comment(g *gocui.Gui, v *gocui.View) error {
615	return addCommentWithEditor(sb.bug)
616}
617
618func (sb *showBug) setTitle(g *gocui.Gui, v *gocui.View) error {
619	return setTitleWithEditor(sb.bug)
620}
621
622func (sb *showBug) toggleOpenClose(g *gocui.Gui, v *gocui.View) error {
623	switch sb.bug.Snapshot().Status {
624	case bug.OpenStatus:
625		_, err := sb.bug.Close()
626		return err
627	case bug.ClosedStatus:
628		_, err := sb.bug.Open()
629		return err
630	default:
631		return nil
632	}
633}
634
635func (sb *showBug) edit(g *gocui.Gui, v *gocui.View) error {
636	snap := sb.bug.Snapshot()
637
638	if sb.isOnSide {
639		return sb.editLabels(g, snap)
640	}
641
642	op, err := snap.SearchTimelineItem(git.Hash(sb.selected))
643	if err != nil {
644		return err
645	}
646
647	switch op.(type) {
648	case *bug.AddCommentTimelineItem:
649		message := op.(*bug.AddCommentTimelineItem).Message
650		return editCommentWithEditor(sb.bug, op.Hash(), message)
651	case *bug.CreateTimelineItem:
652		preMessage := op.(*bug.CreateTimelineItem).Message
653		return editCommentWithEditor(sb.bug, op.Hash(), preMessage)
654	case *bug.LabelChangeTimelineItem:
655		return sb.editLabels(g, snap)
656	}
657
658	ui.msgPopup.Activate(msgPopupErrorTitle, "Selected field is not editable.")
659	return nil
660}
661
662func (sb *showBug) editLabels(g *gocui.Gui, snap *bug.Snapshot) error {
663	ui.labelSelect.SetBug(sb.cache, sb.bug)
664	return ui.activateWindow(ui.labelSelect)
665}