path: root/termui/label_select.go

package termui
import (
	"fmt"
	"strings"
	"github.com/jroimartin/gocui"
	"github.com/MichaelMure/git-bug/bug"
	"github.com/MichaelMure/git-bug/cache"
)
const labelSelectView = "labelSelectView"
const labelSelectInstructionsView = "labelSelectInstructionsView"

type labelSelect struct {
	cache       *cache.RepoCache
	bug         *cache.BugCache
	labels      []bug.Label
	labelSelect []bool
	selected    int
	scroll      int
	childViews  []string
}

func newLabelSelect() *labelSelect {
	return &labelSelect{}
}

func (ls *labelSelect) SetBug(cache *cache.RepoCache, bug *cache.BugCache) {
	ls.cache = cache
	ls.bug = bug
	ls.labels = cache.ValidLabels()

	// Find which labels are currently applied to the bug
	bugLabels := bug.Snapshot().Labels
	labelSelect := make([]bool, len(ls.labels))
	for i, label := range ls.labels {
		for _, bugLabel := range bugLabels {
			if label == bugLabel {
				labelSelect[i] = true
				break
			}
		}
	}

	ls.labelSelect = labelSelect
	ls.selected = 0
	ls.scroll = 0
}

func (ls *labelSelect) keybindings(g *gocui.Gui) error {
	// Abort
	if err := g.SetKeybinding(labelSelectView, gocui.KeyEsc, gocui.ModNone, ls.abort); err != nil {
		return err
	}
	// Save and return
	if err := g.SetKeybinding(labelSelectView, 'q', gocui.ModNone, ls.saveAndReturn); err != nil {
		return err
	}
	// Up
	if err := g.SetKeybinding(labelSelectView, gocui.KeyArrowUp, gocui.ModNone, ls.selectPrevious); err != nil {
		return err
	}
	if err := g.SetKeybinding(labelSelectView, 'k', gocui.ModNone, ls.selectPrevious); err != nil {
		return err
	}
	// Down
	if err := g.SetKeybinding(labelSelectView, gocui.KeyArrowDown, gocui.ModNone, ls.selectNext); err != nil {
		return err
	}
	if err := g.SetKeybinding(labelSelectView, 'j', gocui.ModNone, ls.selectNext); err != nil {
		return err
	}
	// Select
	if err := g.SetKeybinding(labelSelectView, gocui.KeySpace, gocui.ModNone, ls.selectItem); err != nil {
		return err
	}
	if err := g.SetKeybinding(labelSelectView, 'x', gocui.ModNone, ls.selectItem); err != nil {
		return err
	}
	if err := g.SetKeybinding(labelSelectView, gocui.KeyEnter, gocui.ModNone, ls.selectItem); err != nil {
		return err
	}
	// Add
	if err := g.SetKeybinding(labelSelectView, 'a', gocui.ModNone, ls.addItem); err != nil {
		return err
	}
	return nil
}

func (ls *labelSelect) layout(g *gocui.Gui) error {
	maxX, maxY := g.Size()
	ls.childViews = nil

	// TODO: Make width adaptive
	width := 30
	x0 := 1

	v, err := g.SetView(labelSelectView, x0, 0, x0+width, maxY-2)
	if err != nil {
		if err != gocui.ErrUnknownView {
			return err
		}

		v.Frame = false
	}
	v.Clear()
	fmt.Fprintln(v, ls.scroll)
	y0 := 1 - ls.scroll

	for i, label := range ls.labels {
		viewname := fmt.Sprintf("view%d", i)
		v, err := g.SetView(viewname, x0+2, y0, x0+width-2, y0+2)
		if err != nil && err != gocui.ErrUnknownView {
			return err
		}
		ls.childViews = append(ls.childViews, viewname)
		v.Frame = i == ls.selected
		v.Clear()
		selectBox := " [ ] "
		if ls.labelSelect[i] {
			selectBox = " [x] "
		}
		fmt.Fprint(v, selectBox, label)
		y0 += 2
	}

	v, err = g.SetView(labelSelectInstructionsView, -1, maxY-2, maxX, maxY)
	ls.childViews = append(ls.childViews, labelSelectInstructionsView)
	if err != nil {
		if err != gocui.ErrUnknownView{
			return err
		}
		v.Frame = false
		v.BgColor = gocui.ColorBlue
	}
	v.Clear()
	fmt.Fprint(v, "[q] Save and close [↓↑,jk] Nav [a] Add item")
	if _, err = g.SetViewOnTop(labelSelectInstructionsView); err != nil {
		return err
	}
	if _, err := g.SetCurrentView(labelSelectView); err != nil {
		return err
	}
	return nil
}

func (ls *labelSelect) disable(g *gocui.Gui) error {
	for _, view := range ls.childViews {
		if err := g.DeleteView(view); err != nil && err != gocui.ErrUnknownView {
			return err
		}
	}
	return nil
}

func (ls *labelSelect) focusView(g *gocui.Gui) error {
	mainView, err := g.View(labelSelectView)
	if err != nil {
		return err
	}

	_, maxY := mainView.Size()

	_, vy0, _, _, err := g.ViewPosition(fmt.Sprintf("view%d", ls.selected))
	if err != nil {
		return err
	}

	v, err := g.View(fmt.Sprintf("view%d", ls.selected))
	if err != nil {
		return err
	}

	_, vMaxY := v.Size()

	vy1 := vy0 + vMaxY

	if vy0 < 0 {
		ls.scroll += vy0
		return nil
	}

	if vy1 > maxY {
		ls.scroll -= maxY - vy1
	}

	return nil
}

func (ls *labelSelect) selectPrevious(g *gocui.Gui, v *gocui.View) error {
	ls.selected = maxInt(0, ls.selected-1)
	return ls.focusView(g)
}

func (ls *labelSelect) selectNext(g *gocui.Gui, v *gocui.View) error {
	ls.selected = minInt(len(ls.labels)-1, ls.selected+1)
	return ls.focusView(g)
}

func(ls *labelSelect) selectItem(g *gocui.Gui, v *gocui.View) error {
	ls.labelSelect[ls.selected] = !ls.labelSelect[ls.selected]
	return nil
}

func (ls *labelSelect) addItem(g *gocui.Gui, v *gocui.View) error {
	c := ui.inputPopup.Activate("Add a new label")

	go func() error {
		input := <-c
		
		// Standardize label format
		input = strings.TrimSuffix(input, "\n")
		input = strings.Replace(input, " ", "-", -1)

		// Check if label already exists
		for i, label := range ls.labels {
			if input == label.String() {
				ls.labelSelect[i] = true
				ls.selected = i

				g.Update(func(gui *gocui.Gui) error {
					return nil
				})
				return ls.focusView(g)
			}
		}

		// Add new label, make it selected, and move frame
		ls.labels = append(ls.labels, bug.Label(input))
		ls.labelSelect = append(ls.labelSelect, true)
		ls.selected = len(ls.labels) - 1

		ls.layout(g)

		if err := ls.focusView(g); err != nil {
			panic(err)
		}

		g.Update(func(gui *gocui.Gui) error {
			return nil
		})

		return nil
	}()

	return nil
}

func (ls *labelSelect) abort(g *gocui.Gui, v *gocui.View) error {
	return ui.activateWindow(ui.showBug)
}

func (ls *labelSelect) saveAndReturn(g *gocui.Gui, v *gocui.View) error {
	bugLabels := ls.bug.Snapshot().Labels
	selectedLabels := []bug.Label{}
	for i, label := range ls.labels {
		if ls.labelSelect[i] {
			selectedLabels = append(selectedLabels, label)
		}
	}

	// Find the new and removed labels. This makes use of the fact that the first elements
	// of selectedLabels are the not-removed labels in bugLabels
	newLabels := []string{}
	rmLabels := []string{}
	i := 0	// Index for bugLabels
	j := 0	// Index for selectedLabels
	for {
		if j == len(selectedLabels) {
			// No more labels to consider
			break
		} else if i == len(bugLabels) {
			// Remaining labels are all new
			newLabels = append(newLabels, selectedLabels[j].String())
			j += 1
		} else if bugLabels[i] == selectedLabels[j] {
			// Labels match. Move to next pair
			i += 1
			j += 1
		} else {
			// Labels don't match. Prelabel must have been removed
			rmLabels = append(rmLabels, bugLabels[i].String())
			i += 1
		}
	}

	if _, err := ls.bug.ChangeLabels(newLabels, rmLabels); err != nil {
		ui.msgPopup.Activate(msgPopupErrorTitle, err.Error())
	}

	return ui.activateWindow(ui.showBug)
}