pelican/internal/ui/ui.go

161 lines
2.9 KiB
Go

package ui
import (
"fmt"
"codeflow.dananglin.me.uk/apollo/canal/internal/board"
"github.com/gdamore/tcell/v2"
"github.com/rivo/tview"
)
const (
shiftLeft int = iota
shiftRight
)
const (
mainPageName string = "main"
quitPageName string = "quit"
addPageName string = "add"
)
type UI struct {
*tview.Application
columns []column
flex *tview.Flex
pages *tview.Pages
focusedColumn int
board board.Board
quit *tview.Modal
add *modalInput
}
// NewUI returns a new UI value.
func NewUI() UI {
u := UI{
Application: tview.NewApplication(),
pages: tview.NewPages(),
flex: tview.NewFlex(),
quit: tview.NewModal(),
add: NewModalInput(),
focusedColumn: 0,
}
u.init()
return u
}
// closeBoard closes the board.
func (u *UI) closeBoard() {
_ = u.board.Close()
}
// init initialises the UI.
func (u *UI) init() {
u.pages.AddPage(mainPageName, u.flex, true, true)
u.initQuitModal()
u.pages.AddPage(quitPageName, u.quit, false, false)
u.initAddInputModal()
u.pages.AddPage(addPageName, u.add, false, false)
u.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
switch event.Rune() {
case 'o':
if u.flex.HasFocus() && len(u.columns) == 0 {
u.openBoard("")
}
}
return event
})
u.SetRoot(u.pages, true)
}
// initAddInputModal initialises the add input modal.
func (u *UI) initAddInputModal() {
doneFunc := func(text string, success bool) {
if success {
_ = u.newCard(text, "")
}
u.pages.HidePage(addPageName)
u.setColumnFocus()
}
u.add.SetDoneFunc(doneFunc)
}
// initQuitModal initialises the quit modal.
func (u *UI) initQuitModal() {
quitDoneFunc := func(_ int, buttonLabel string) {
switch buttonLabel {
case "Quit":
u.shutdown()
default:
u.pages.SwitchToPage(mainPageName)
u.setColumnFocus()
}
}
u.quit.SetText("Do you want to quit the application?").
AddButtons([]string{"Quit", "Cancel"}).
SetDoneFunc(quitDoneFunc)
}
// newCard creates and saves a new card to the database.
func (u *UI) newCard(title, content string) error {
args := board.CardArgs{
NewTitle: title,
NewContent: content,
}
if err := u.board.CreateCard(args); err != nil {
return fmt.Errorf("unable to create card, %w", err)
}
u.refresh()
return nil
}
// openBoard opens the kanban board.
func (u *UI) openBoard(path string) error {
b, err := board.Open(path)
if err != nil {
return fmt.Errorf("unable to load board, %w", err)
}
u.board = b
if err = u.refresh(); err != nil {
return err
}
return nil
}
// refresh refreshes the UI.
func (u *UI) refresh() error {
statusList, err := u.board.StatusList()
if err != nil {
return fmt.Errorf("unable to get the status list, %w", err)
}
u.updateColumns(statusList)
u.setColumnFocus()
// TODO: update move status page here
return nil
}
// shutdown shuts down the application.
func (u *UI) shutdown() {
u.closeBoard()
u.Stop()
}