git-bug/termui/bug_table.go
2022-02-15 21:52:36 +01:00

546 lines
12 KiB
Go

package termui
import (
"bytes"
"errors"
"fmt"
"strings"
text "github.com/MichaelMure/go-term-text"
"github.com/awesome-gocui/gocui"
"github.com/dustin/go-humanize"
"github.com/MichaelMure/git-bug/cache"
"github.com/MichaelMure/git-bug/entity"
"github.com/MichaelMure/git-bug/query"
"github.com/MichaelMure/git-bug/util/colors"
)
const bugTableView = "bugTableView"
const bugTableHeaderView = "bugTableHeaderView"
const bugTableFooterView = "bugTableFooterView"
const bugTableInstructionView = "bugTableInstructionView"
const defaultRemote = "origin"
const defaultQuery = "status:open"
var bugTableHelp = helpBar{
{"q", "Quit"},
{"s", "Search"},
{"←↓↑→,hjkl", "Navigation"},
{"↵", "Open bug"},
{"n", "New bug"},
{"i", "Pull"},
{"o", "Push"},
}
type bugTable struct {
repo *cache.RepoCache
queryStr string
query *query.Query
allIds []entity.Id
excerpts []*cache.BugExcerpt
pageCursor int
selectCursor int
}
func newBugTable(c *cache.RepoCache) *bugTable {
q, err := query.Parse(defaultQuery)
if err != nil {
panic(err)
}
return &bugTable{
repo: c,
query: q,
queryStr: defaultQuery,
pageCursor: 0,
selectCursor: 0,
}
}
func (bt *bugTable) layout(g *gocui.Gui) error {
maxX, maxY := g.Size()
if maxY < 4 {
// window too small !
return nil
}
v, err := g.SetView(bugTableHeaderView, -1, -1, maxX, 1, 0)
if err != nil {
if !errors.Is(err, gocui.ErrUnknownView) {
return err
}
v.Frame = false
}
v.Clear()
bt.renderHeader(v, maxX)
v, err = g.SetView(bugTableView, -1, 0, maxX, maxY-3, 0)
if err != nil {
if !errors.Is(err, gocui.ErrUnknownView) {
return err
}
v.Frame = false
v.SelBgColor = gocui.ColorWhite
v.SelFgColor = gocui.ColorBlack
}
viewWidth, viewHeight := v.Size()
err = bt.paginate(viewHeight)
if err != nil {
return err
}
err = bt.cursorClamp(v)
if err != nil {
return err
}
v.Clear()
bt.render(v, viewWidth)
v, err = g.SetView(bugTableFooterView, -1, maxY-4, maxX, maxY, 0)
if err != nil {
if !errors.Is(err, gocui.ErrUnknownView) {
return err
}
v.Frame = false
}
v.Clear()
bt.renderFooter(v, maxX)
v, err = g.SetView(bugTableInstructionView, -1, maxY-2, maxX, maxY, 0)
if err != nil {
if !errors.Is(err, gocui.ErrUnknownView) {
return err
}
v.Frame = false
v.FgColor = gocui.ColorWhite
}
v.Clear()
bt.renderHelp(v, maxX)
_, err = g.SetCurrentView(bugTableView)
return err
}
func (bt *bugTable) keybindings(g *gocui.Gui) error {
// Quit
if err := g.SetKeybinding(bugTableView, 'q', gocui.ModNone, quit); err != nil {
return err
}
// Down
if err := g.SetKeybinding(bugTableView, 'j', gocui.ModNone,
bt.cursorDown); err != nil {
return err
}
if err := g.SetKeybinding(bugTableView, gocui.KeyArrowDown, gocui.ModNone,
bt.cursorDown); err != nil {
return err
}
// Up
if err := g.SetKeybinding(bugTableView, 'k', gocui.ModNone,
bt.cursorUp); err != nil {
return err
}
if err := g.SetKeybinding(bugTableView, gocui.KeyArrowUp, gocui.ModNone,
bt.cursorUp); err != nil {
return err
}
// Previous page
if err := g.SetKeybinding(bugTableView, 'h', gocui.ModNone,
bt.previousPage); err != nil {
return err
}
if err := g.SetKeybinding(bugTableView, gocui.KeyArrowLeft, gocui.ModNone,
bt.previousPage); err != nil {
return err
}
if err := g.SetKeybinding(bugTableView, gocui.KeyPgup, gocui.ModNone,
bt.previousPage); err != nil {
return err
}
// Next page
if err := g.SetKeybinding(bugTableView, 'l', gocui.ModNone,
bt.nextPage); err != nil {
return err
}
if err := g.SetKeybinding(bugTableView, gocui.KeyArrowRight, gocui.ModNone,
bt.nextPage); err != nil {
return err
}
if err := g.SetKeybinding(bugTableView, gocui.KeyPgdn, gocui.ModNone,
bt.nextPage); err != nil {
return err
}
// New bug
if err := g.SetKeybinding(bugTableView, 'n', gocui.ModNone,
bt.newBug); err != nil {
return err
}
// Open bug
if err := g.SetKeybinding(bugTableView, gocui.KeyEnter, gocui.ModNone,
bt.openBug); err != nil {
return err
}
// Pull
if err := g.SetKeybinding(bugTableView, 'i', gocui.ModNone,
bt.pull); err != nil {
return err
}
// Push
if err := g.SetKeybinding(bugTableView, 'o', gocui.ModNone,
bt.push); err != nil {
return err
}
// Query
if err := g.SetKeybinding(bugTableView, 's', gocui.ModNone,
bt.changeQuery); err != nil {
return err
}
return nil
}
func (bt *bugTable) disable(g *gocui.Gui) error {
if err := g.DeleteView(bugTableView); err != nil && !errors.Is(err, gocui.ErrUnknownView) {
return err
}
if err := g.DeleteView(bugTableHeaderView); err != nil && !errors.Is(err, gocui.ErrUnknownView) {
return err
}
if err := g.DeleteView(bugTableFooterView); err != nil && !errors.Is(err, gocui.ErrUnknownView) {
return err
}
if err := g.DeleteView(bugTableInstructionView); err != nil && !errors.Is(err, gocui.ErrUnknownView) {
return err
}
return nil
}
func (bt *bugTable) paginate(max int) error {
var err error
bt.allIds, err = bt.repo.QueryBugs(bt.query)
if err != nil {
return err
}
return bt.doPaginate(max)
}
func (bt *bugTable) doPaginate(max int) error {
// clamp the cursor
bt.pageCursor = maxInt(bt.pageCursor, 0)
bt.pageCursor = minInt(bt.pageCursor, len(bt.allIds))
nb := minInt(len(bt.allIds)-bt.pageCursor, max)
if nb < 0 {
bt.excerpts = []*cache.BugExcerpt{}
return nil
}
// slice the data
ids := bt.allIds[bt.pageCursor : bt.pageCursor+nb]
bt.excerpts = make([]*cache.BugExcerpt, len(ids))
for i, id := range ids {
excerpt, err := bt.repo.ResolveBugExcerpt(id)
if err != nil {
return err
}
bt.excerpts[i] = excerpt
}
return nil
}
func (bt *bugTable) getTableLength() int {
return len(bt.excerpts)
}
func (bt *bugTable) getColumnWidths(maxX int) map[string]int {
m := make(map[string]int)
m["id"] = 7
m["status"] = 6
left := maxX - 5 - m["id"] - m["status"]
m["comments"] = 3
left -= m["comments"]
m["lastEdit"] = 14
left -= m["lastEdit"]
m["author"] = minInt(maxInt(left/3, 15), 10+left/8)
m["title"] = maxInt(left-m["author"], 10)
return m
}
func (bt *bugTable) render(v *gocui.View, maxX int) {
columnWidths := bt.getColumnWidths(maxX)
for _, excerpt := range bt.excerpts {
summaryTxt := fmt.Sprintf("%3d", excerpt.LenComments-1)
if excerpt.LenComments-1 <= 0 {
summaryTxt = ""
}
if excerpt.LenComments-1 > 999 {
summaryTxt = " ∞"
}
var labelsTxt strings.Builder
for _, l := range excerpt.Labels {
labelsTxt.WriteString(" ")
lc256 := l.Color().Term256()
labelsTxt.WriteString(lc256.Escape())
labelsTxt.WriteString("◼")
labelsTxt.WriteString(lc256.Unescape())
}
author, err := bt.repo.ResolveIdentityExcerpt(excerpt.AuthorId)
if err != nil {
panic(err)
}
id := text.LeftPadMaxLine(excerpt.Id.Human(), columnWidths["id"], 0)
status := text.LeftPadMaxLine(excerpt.Status.String(), columnWidths["status"], 0)
labels := text.TruncateMax(labelsTxt.String(), minInt(columnWidths["title"]-2, 10))
title := text.LeftPadMaxLine(strings.TrimSpace(excerpt.Title), columnWidths["title"]-text.Len(labels), 0)
authorTxt := text.LeftPadMaxLine(author.DisplayName(), columnWidths["author"], 0)
comments := text.LeftPadMaxLine(summaryTxt, columnWidths["comments"], 0)
lastEdit := text.LeftPadMaxLine(humanize.Time(excerpt.EditTime()), columnWidths["lastEdit"], 1)
_, _ = fmt.Fprintf(v, "%s %s %s%s %s %s %s\n",
colors.Cyan(id),
colors.Yellow(status),
title,
labels,
colors.Magenta(authorTxt),
comments,
lastEdit,
)
}
_ = v.SetHighlight(bt.selectCursor, true)
}
func (bt *bugTable) renderHeader(v *gocui.View, maxX int) {
columnWidths := bt.getColumnWidths(maxX)
id := text.LeftPadMaxLine("ID", columnWidths["id"], 0)
status := text.LeftPadMaxLine("STATUS", columnWidths["status"], 0)
title := text.LeftPadMaxLine("TITLE", columnWidths["title"], 0)
author := text.LeftPadMaxLine("AUTHOR", columnWidths["author"], 0)
comments := text.LeftPadMaxLine("CMT", columnWidths["comments"], 0)
lastEdit := text.LeftPadMaxLine("LAST EDIT", columnWidths["lastEdit"], 1)
_, _ = fmt.Fprintf(v, "%s %s %s %s %s %s\n", id, status, title, author, comments, lastEdit)
}
func (bt *bugTable) renderFooter(v *gocui.View, maxX int) {
_, _ = fmt.Fprintf(v, " \nShowing %d of %d bugs", len(bt.excerpts), len(bt.allIds))
}
func (bt *bugTable) renderHelp(v *gocui.View, maxX int) {
_, _ = fmt.Fprint(v, bugTableHelp.Render(maxX))
}
func (bt *bugTable) cursorDown(g *gocui.Gui, v *gocui.View) error {
// If we are at the bottom of the page, switch to the next one.
if bt.selectCursor+1 > bt.getTableLength()-1 {
_, max := v.Size()
if bt.pageCursor+max >= len(bt.allIds) {
return nil
}
bt.pageCursor += max
bt.selectCursor = 0
return bt.doPaginate(max)
}
bt.selectCursor = minInt(bt.selectCursor+1, bt.getTableLength()-1)
return nil
}
func (bt *bugTable) cursorUp(g *gocui.Gui, v *gocui.View) error {
// If we are at the top of the page, switch to the previous one.
if bt.selectCursor-1 < 0 {
_, max := v.Size()
if bt.pageCursor == 0 {
return nil
}
bt.pageCursor = maxInt(0, bt.pageCursor-max)
bt.selectCursor = max - 1
return bt.doPaginate(max)
}
bt.selectCursor = maxInt(bt.selectCursor-1, 0)
return nil
}
func (bt *bugTable) cursorClamp(v *gocui.View) error {
y := bt.selectCursor
y = minInt(y, bt.getTableLength()-1)
y = maxInt(y, 0)
bt.selectCursor = y
return nil
}
func (bt *bugTable) nextPage(g *gocui.Gui, v *gocui.View) error {
_, max := v.Size()
if bt.pageCursor+max >= len(bt.allIds) {
return nil
}
bt.pageCursor += max
return bt.doPaginate(max)
}
func (bt *bugTable) previousPage(g *gocui.Gui, v *gocui.View) error {
_, max := v.Size()
if bt.pageCursor == 0 {
return nil
}
bt.pageCursor = maxInt(0, bt.pageCursor-max)
return bt.doPaginate(max)
}
func (bt *bugTable) newBug(g *gocui.Gui, v *gocui.View) error {
return newBugWithEditor(bt.repo)
}
func (bt *bugTable) openBug(g *gocui.Gui, v *gocui.View) error {
if len(bt.excerpts) == 0 {
// There are no open bugs, just do nothing
return nil
}
id := bt.excerpts[bt.selectCursor].Id
b, err := bt.repo.ResolveBug(id)
if err != nil {
return err
}
ui.showBug.SetBug(b)
return ui.activateWindow(ui.showBug)
}
func (bt *bugTable) pull(g *gocui.Gui, v *gocui.View) error {
ui.msgPopup.Activate("Pull from remote "+defaultRemote, "...")
go func() {
stdout, err := bt.repo.Fetch(defaultRemote)
if err != nil {
g.Update(func(gui *gocui.Gui) error {
ui.msgPopup.Activate(msgPopupErrorTitle, err.Error())
return nil
})
} else {
g.Update(func(gui *gocui.Gui) error {
ui.msgPopup.UpdateMessage(stdout)
return nil
})
}
var buffer bytes.Buffer
beginLine := ""
for result := range bt.repo.MergeAll(defaultRemote) {
if result.Status == entity.MergeStatusNothing {
continue
}
if result.Err != nil {
g.Update(func(gui *gocui.Gui) error {
ui.msgPopup.Activate(msgPopupErrorTitle, err.Error())
return nil
})
} else {
_, _ = fmt.Fprintf(&buffer, "%s%s: %s",
beginLine, colors.Cyan(result.Entity.Id().Human()), result,
)
beginLine = "\n"
g.Update(func(gui *gocui.Gui) error {
ui.msgPopup.UpdateMessage(buffer.String())
return nil
})
}
}
_, _ = fmt.Fprintf(&buffer, "%sdone", beginLine)
g.Update(func(gui *gocui.Gui) error {
ui.msgPopup.UpdateMessage(buffer.String())
return nil
})
}()
return nil
}
func (bt *bugTable) push(g *gocui.Gui, v *gocui.View) error {
ui.msgPopup.Activate("Push to remote "+defaultRemote, "...")
go func() {
// TODO: make the remote configurable
stdout, err := bt.repo.Push(defaultRemote)
if err != nil {
g.Update(func(gui *gocui.Gui) error {
ui.msgPopup.Activate(msgPopupErrorTitle, err.Error())
return nil
})
} else {
g.Update(func(gui *gocui.Gui) error {
ui.msgPopup.UpdateMessage(stdout)
return nil
})
}
}()
return nil
}
func (bt *bugTable) changeQuery(g *gocui.Gui, v *gocui.View) error {
return editQueryWithEditor(bt)
}