gime/ui/list_timers.go

584 lines
15 KiB
Go

package ui
import (
"errors"
"fmt"
"strings"
"time"
"git.bullercodeworks.com/brian/gime/util"
"git.bullercodeworks.com/brian/go-timertxt"
"git.bullercodeworks.com/brian/wandle"
"git.bullercodeworks.com/brian/widdles"
"github.com/nsf/termbox-go"
)
const (
activeToggleActive = iota
activeToggleInactive
activeToggleAll
activeToggleErr
)
type listTimersScreen struct {
ui *Ui
initialized bool
menu *widdles.TopMenu
scrollbar *widdles.Scrollbar
cursor int
activeToggle int
fullList *timertxt.TimerList
timerList *timertxt.TimerList
doneList *timertxt.TimerList
fullFilterList *timertxt.TimerList
timerFilterList *timertxt.TimerList
doneFilterList *timertxt.TimerList
selected map[int]bool
confirm *widdles.ConfirmDialog
filter string
choiceMenu *widdles.MenuV
tagEditor *PromptForTagWiddle
//partManager *PartManager
msg string
err error
}
type ListTimersMsg ScreenMsg
func NewListTimersMsg(data interface{}, err error) ListTimersMsg {
return ListTimersMsg{
source: ListTimersId,
command: CmdArchiveTimer,
data: data,
err: err,
}
}
func NewListTimersScreen(u *Ui) *listTimersScreen {
s := listTimersScreen{
ui: u,
menu: widdles.NewTopMenu(0, 0, 0),
scrollbar: widdles.NewScrollbar(0, 0, 0, 0),
selected: make(map[int]bool),
confirm: widdles.NewConfirmDialog("", ""),
choiceMenu: widdles.NewMenuV(0, 0, 0, 0),
tagEditor: NewPromptForTagWiddle(0, 0, widdles.AUTO_SIZE, widdles.AUTO_SIZE, "", ""),
}
return &s
}
func (s *listTimersScreen) Init() wandle.Cmd {
if s.initialized {
return nil
}
s.initialized = true
// Set up the top menu
fileMenu := s.menu.NewSubMenu("File")
settingsOption := widdles.NewMenuItem("Settings")
settingsOption.SetCommand(s.ui.GotoScreen(SettingsId))
fileMenu.AddOption(settingsOption)
quitOption := widdles.NewMenuItem("Quit")
quitOption.SetHotkey(widdles.NewHotkey(termbox.KeyCtrlC))
quitOption.SetCommand(func() wandle.Msg { return wandle.Quit() })
fileMenu.AddOption(quitOption)
s.menu.Measure()
// Timer Lists
s.timerList, s.doneList = s.ui.program.TimerList, s.ui.program.DoneList
s.fullList = timertxt.NewTimerList()
s.fullList.AddTimers(s.timerList.GetTimerSlice())
s.fullList.AddTimers(s.doneList.GetTimerSlice())
s.timerFilterList, s.doneFilterList = s.timerList, s.doneList
s.timerFilterList.Sort(timertxt.SortStartDateDesc)
s.doneFilterList.Sort(timertxt.SortStartDateDesc)
s.updateFullFilterList()
w, h := termbox.Size()
s.choiceMenu.SetBorder(wandle.BRD_CSIMPLE)
s.choiceMenu.SetX((w / 2) - 7)
s.choiceMenu.SetY((h / 2) - 7)
s.choiceMenu.SetWidth(widdles.AUTO_SIZE)
s.choiceMenu.SetHeight(widdles.AUTO_SIZE)
s.choiceMenu.SetPadding(0, 1, 0, 1)
s.tagEditor.SetX(w / 4)
s.tagEditor.SetY(h / 4)
s.tagEditor.SetWidth(w / 2)
s.tagEditor.SetHeight(h / 2)
s.confirm.EnableHotkeys()
s.confirm.SetX(w / 4)
s.confirm.SetY(h / 4)
s.confirm.SetWidth(w / 2)
s.confirm.SetHeight(h / 2)
s.updateFullFilterList()
return nil
}
func (s *listTimersScreen) Update(msg wandle.Msg) wandle.Cmd {
switch msg := msg.(type) {
case ScreenMsg:
s.err = msg.err
case termbox.Event:
return s.handleTermboxEvent(msg)
}
return nil
}
func (s *listTimersScreen) View(style wandle.Style) {
_, h := termbox.Size()
y := 2
printedTimers := 0
if s.activeToggle == activeToggleAll || s.activeToggle == activeToggleActive {
wandle.Print(1, y, style.Bold(true), "Active Timers")
y++
for idx, tmr := range s.timerFilterList.GetTimerSlice() {
if y > h-2 {
break
}
st := style
if s.cursor == idx {
st = st.Invert()
}
if s.selected[idx] {
wandle.Print(1, y, st, "[✔] ")
} else {
wandle.Print(1, y, st, "[ ] ")
}
s.ViewTimer(5, y, st, tmr)
y++
}
y++
printedTimers = s.timerFilterList.Size()
}
if s.activeToggle == activeToggleAll || s.activeToggle == activeToggleInactive {
wandle.Print(1, y, style.Bold(true), "Done Timers")
y++
for idx, tmr := range s.doneFilterList.GetTimerSlice() {
if y > h-3 {
break
}
st := style
if s.cursor == printedTimers+idx {
st = st.Invert()
}
if s.selected[printedTimers+idx] {
wandle.Print(1, y, st, "[✔] ")
} else {
wandle.Print(1, y, st, "[ ] ")
}
s.ViewTimer(5, y, st, tmr)
y++
}
}
selectedStatus := fmt.Sprintf("%s", s.getSelectedTimerDuration())
if len(s.selected) > 0 {
selectedStatus = fmt.Sprintf("%s (%d / %d selected)", selectedStatus, len(s.selected), s.fullFilterList.Size())
}
wandle.Print(1, h-2, style, selectedStatus)
var archiveText string
if s.areSelectedInSameList() {
if s.areSelectedInDoneList() {
archiveText = "Un[A]rchive Selected, "
} else {
archiveText = "[A]rchive Selected, "
}
} else {
archiveText = "Not in Same List"
}
help := fmt.Sprintf("[T]oggle Display, [p]roject(+), [c]ontext(@), [t]ags(:), %s[Ctrl+A]: Select All/None, [Ctrl+I]: Invert Selection", archiveText)
wandle.Print(1, h-1, style, help)
s.scrollbar.View(style)
if s.menu.IsActive() {
s.menu.View(style)
}
if s.choiceMenu.IsActive() {
s.choiceMenu.View(style)
}
if s.tagEditor.IsActive() {
s.tagEditor.View(style)
}
if s.confirm.IsActive() {
s.confirm.View(style)
}
wandle.Print(1, h-3, style, s.msg)
if s.err != nil {
wandle.Print(1, h-4, ErrStyle, s.err.Error())
}
}
func (s *listTimersScreen) ViewTimer(x, y int, style wandle.Style, tmr *timertxt.Timer) {
var tags []string
for _, k := range util.SortedTagKeyList(tmr.AdditionalTags) {
tags = append(tags, fmt.Sprintf("%s:%s", k, tmr.AdditionalTags[k]))
}
wandle.Print(x, y, style, fmt.Sprintf("%s %s %s %s %s", tmr.StartDate.Format(time.Stamp), tmr.Duration(), tmr.Contexts, tmr.Projects, strings.Join(tags, "; ")))
}
func (s *listTimersScreen) handleTermboxEvent(msg termbox.Event) wandle.Cmd {
if s.confirm.IsActive() {
return s.confirm.Update(msg)
}
if s.choiceMenu.IsActive() {
if msg.Type == termbox.EventKey && msg.Key == termbox.KeyEsc {
s.choiceMenu.SetActive(false)
return wandle.EmptyCmd
} else {
return s.choiceMenu.Update(msg)
}
}
if s.tagEditor.IsActive() {
return s.tagEditor.Update(msg)
}
if (msg.Type == termbox.EventKey && msg.Key == termbox.KeyEsc) || s.menu.IsActive() {
return s.menu.Update(msg)
}
switch msg.Type {
case termbox.EventKey:
if msg.Key == termbox.KeyEnter {
// TODO: Edit the entry
/*
if s.cursor >= 0 && s.cursor < s.timerFilterList.Size()+s.doneFilterList.Size() {
} else {
}
*/
} else if msg.Key == termbox.KeySpace {
// (un)Select the entry
if v := s.selected[s.cursor]; v {
delete(s.selected, s.cursor)
} else {
s.selected[s.cursor] = true
}
if s.cursor < s.fullFilterList.Size()-1 {
s.cursor++
}
} else if msg.Ch == 'T' {
s.activeToggle = (s.activeToggle + 1) % activeToggleErr
s.updateFullFilterList()
} else if msg.Ch == 'A' {
return s.showArchiveSelected()
} else if msg.Key == termbox.KeyArrowUp || msg.Ch == 'k' {
if s.cursor > 0 {
s.cursor--
} else {
s.cursor = 0
}
return nil
} else if msg.Key == termbox.KeyArrowDown || msg.Ch == 'j' {
if s.cursor < s.fullFilterList.Size()-1 {
s.cursor++
} else {
s.cursor = s.fullFilterList.Size() - 1
}
return nil
} else if msg.Ch == 'G' {
s.cursor = s.fullFilterList.Size() - 1
} else if msg.Ch == 'g' {
s.cursor = 0
} else if msg.Ch == 't' {
// Edit tag(s)
return s.showEditTagsChoice()
} else if msg.Ch == 'p' {
// Edit project(s)
// TODO: Prompt for Choice: Add/Edit/Remove
projs := s.fullList.GetProjects()
_ = projs
} else if msg.Ch == 'c' {
// Edit context(s)
// TODO: Prompt for choice: Add/Edit/Remove
ctxts := s.fullList.GetContexts()
_ = ctxts
} else if msg.Key == termbox.KeyCtrlA {
if len(s.selected) != s.fullFilterList.Size() {
// Select None
for k := range s.selected {
delete(s.selected, k)
}
} else {
// Select All
for i := 0; i < s.fullFilterList.Size(); i++ {
s.selected[i] = true
}
}
for i := 0; i < s.fullFilterList.Size(); i++ {
if v := s.selected[i]; v {
delete(s.selected, i)
} else {
s.selected[i] = true
}
}
} else if msg.Key == termbox.KeyCtrlI {
for i := 0; i < s.fullFilterList.Size(); i++ {
if v := s.selected[i]; v {
delete(s.selected, i)
} else {
s.selected[i] = true
}
}
}
}
return nil
}
func (s *listTimersScreen) showArchiveSelected() wandle.Cmd {
return func() wandle.Msg {
if len(s.selected) > 0 {
s.confirm.SetTitle(fmt.Sprintf("Archive %d Timers?", len(s.selected)))
s.confirm.SetMessage("Are you sure you want to archive these timers? (y/n)")
} else {
s.confirm.SetTitle("Archive Timer?")
s.confirm.SetMessage("Are you sure you want to archive this timer? (y/n)")
}
s.confirm.SetOkCommand(func() wandle.Msg {
s.confirm.SetVisible(false)
return s.doArchiveSelected()
})
s.confirm.SetCancelCommand(func() wandle.Msg {
s.confirm.SetVisible(false)
return wandle.EmptyCmd
})
s.confirm.SetVisible(true)
return nil
}
}
func (s *listTimersScreen) doArchiveSelected() wandle.Cmd {
archiveTimer := func(t *timertxt.Timer) error {
if remErr := s.timerList.RemoveTimer(*t); remErr != nil {
return remErr
}
s.doneList.AddTimer(t)
return nil
}
selected := len(s.selected)
if selected == 0 {
if s.cursor < s.fullFilterList.Size() {
var selTimer *timertxt.Timer
if selTimer, s.err = s.fullFilterList.GetTimer(s.cursor); s.err == nil {
if archErr := archiveTimer(selTimer); archErr != nil {
s.err = archErr
return wandle.EmptyCmd
}
}
}
} else {
for i := range s.selected {
if tmr, err := s.fullFilterList.GetTimer(i); err == nil {
if err := archiveTimer(tmr); err != nil {
s.err = err
return wandle.EmptyCmd
}
}
}
}
return wandle.EmptyCmd
}
func (s *listTimersScreen) showEditTagsChoice() wandle.Cmd {
tags := s.getSelectedTimerTags()
var showTagEditor = func(key, val string, multival bool) wandle.Cmd {
return func() wandle.Msg {
s.tagEditor.SetTag(key, val)
s.tagEditor.SetDoneCommand(func() wandle.Msg {
s.updateTagOnSelectedTimers(s.tagEditor.GetTag())
s.tagEditor.Done()
return nil
})
s.tagEditor.SetCancelCommand(func() wandle.Msg {
s.tagEditor.SetActive(false)
return nil
})
s.choiceMenu.SetActive(false)
s.tagEditor.SetActive(true)
s.tagEditor.SetMultiVal(multival)
return wandle.EmptyCmd
}
}
s.choiceMenu.SetTitle("")
s.choiceMenu.ClearOptions()
addTag := widdles.NewMenuItem("[A]dd New Tag")
addTag.SetHotkey(widdles.NewHotkeyCh('a'))
addTag.SetCommand(showTagEditor("", "", false))
s.choiceMenu.AddOption(addTag)
editTag := widdles.NewMenuItem("[E]dit Tag")
editTag.SetHotkey(widdles.NewHotkeyCh('e'))
editTag.SetEnabled(len(tags) > 0)
editTag.SetCommand(func() wandle.Msg {
s.choiceMenu.ClearOptions()
s.choiceMenu.SetTitle("Choose Tag to Edit")
for k, v := range tags {
var vals string
var multival bool
if len(v) == 1 {
vals = v[0]
} else {
vals = ""
multival = true
}
opt := widdles.NewMenuItem(fmt.Sprintf("%s (%s)", k, vals))
opt.SetCommand(showTagEditor(k, vals, multival))
s.choiceMenu.AddOption(opt)
}
return wandle.EmptyCmd
})
s.choiceMenu.AddOption(editTag)
removeTag := widdles.NewMenuItem("[R]emove Tag")
removeTag.SetHotkey(widdles.NewHotkeyCh('r'))
removeTag.SetCommand(func() wandle.Msg {
s.choiceMenu.ClearOptions()
s.choiceMenu.SetTitle("Choose Tag to Remove")
for k, v := range tags {
opt := widdles.NewMenuItem(fmt.Sprintf("%s: %s", k, v))
opt.SetCommand(func() wandle.Msg {
s.removeTagOnSelectedTimers(k)
return wandle.EmptyCmd
})
s.choiceMenu.AddOption(opt)
}
s.choiceMenu.SetActive(true)
return wandle.EmptyCmd
})
s.choiceMenu.AddOption(removeTag)
s.choiceMenu.SetActive(true)
return wandle.EmptyCmd
}
func (s *listTimersScreen) updateFullFilterList() {
s.fullFilterList = timertxt.NewTimerList()
switch s.activeToggle {
case activeToggleAll:
s.fullFilterList.Combine(s.timerFilterList)
s.fullFilterList.Combine(s.doneFilterList)
case activeToggleActive:
s.fullFilterList.Combine(s.timerFilterList)
case activeToggleInactive:
s.fullFilterList.Combine(s.doneFilterList)
}
if s.cursor >= s.fullFilterList.Size() {
s.cursor = s.fullFilterList.Size() - 1
}
}
func (s *listTimersScreen) gotoSettingsScreen() wandle.Msg {
return ScreenMsg{
source: ListTimersId,
command: CmdGotoSettings,
}
}
// Writes the lists through the program, putting errors in s.err
func (s *listTimersScreen) writeLists() {
var errText string
if err := s.ui.program.WriteLists(); err != nil {
errText = fmt.Sprintf("Errors Writing Lists (%v)", err)
}
if len(errText) > 0 {
s.err = errors.New(errText)
}
}
func (s *listTimersScreen) getSelectedTimers() []*timertxt.Timer {
var ret []*timertxt.Timer
selected := len(s.selected)
if selected == 0 {
if s.cursor < s.fullFilterList.Size() {
var selTimer *timertxt.Timer
if selTimer, s.err = s.fullFilterList.GetTimer(s.cursor); s.err == nil {
ret = append(ret, selTimer)
}
}
} else {
for i := range s.selected {
if tmr, err := s.fullFilterList.GetTimer(i); err == nil {
ret = append(ret, tmr)
}
}
}
return ret
}
func (s *listTimersScreen) getSelectedTimerTags() map[string][]string {
ret := make(map[string][]string)
sel := s.getSelectedTimers()
for _, tmr := range sel {
for k, v := range tmr.AdditionalTags {
ret[k] = util.AppendStringIfDistinct(ret[k], v)
}
}
return ret
}
func (s *listTimersScreen) updateTagOnSelectedTimers(key, val string) {
sel := s.getSelectedTimers()
for _, tmr := range sel {
tmr.AdditionalTags[key] = val
}
s.writeLists()
}
func (s *listTimersScreen) removeTagOnSelectedTimers(key string) {
sel := s.getSelectedTimers()
for _, tmr := range sel {
if _, ok := tmr.AdditionalTags[key]; ok {
delete(tmr.AdditionalTags, key)
}
}
s.writeLists()
}
func (s *listTimersScreen) getSelectedTimerProjects() []string {
var ret []string
sel := s.getSelectedTimers()
for _, tmr := range sel {
for _, v := range tmr.Contexts {
ret = util.AppendStringIfDistinct(ret, v)
}
}
return ret
}
func (s *listTimersScreen) getSelectedTimerContexts() []string {
var ret []string
sel := s.getSelectedTimers()
for _, tmr := range sel {
for _, v := range tmr.Contexts {
ret = util.AppendStringIfDistinct(ret, v)
}
}
return ret
}
func (s *listTimersScreen) getSelectedTimerDuration() time.Duration {
sel := s.getSelectedTimers()
var ret time.Duration
for _, tmr := range sel {
ret = util.AddDurations(ret, util.Round(tmr.Duration()))
}
return ret
}
// Returns true if all selected timers are done
func (s *listTimersScreen) areSelectedInDoneList() bool {
sel := s.getSelectedTimers()
for i := range sel {
if s.timerList.Contains(sel[i]) {
return false
}
}
return true
}
// Return true if all selected timers are from the same list (file)
func (s *listTimersScreen) areSelectedInSameList() bool {
sel := s.getSelectedTimers()
var inActive, inDone int
for i := range sel {
if s.timerList.Contains(sel[i]) {
inActive++
}
if s.doneList.Contains(sel[i]) {
inDone++
}
}
return inActive == 0 || inDone == 0
}