First working version. Can add Projects and Tasks

This commit is contained in:
Anis Ahmad
2020-05-25 23:18:48 +06:00
parent 267995124b
commit 6f9580d98a
12 changed files with 610 additions and 0 deletions

243
app/tui.go Normal file
View File

@@ -0,0 +1,243 @@
package main
import (
"fmt"
"reflect"
"time"
"github.com/asdine/storm/v3"
"github.com/gdamore/tcell"
"github.com/rivo/tview"
"github.com/ajaxray/geek-life/model"
"github.com/ajaxray/geek-life/repository"
repo "github.com/ajaxray/geek-life/repository/storm"
"github.com/ajaxray/geek-life/util"
)
var (
app *tview.Application
newProject, newTask *tview.InputField
projectList, taskList *tview.List
projectPane, taskPane *tview.Flex
statusBar *tview.Pages
message *tview.TextView
shortcutsPage, messagePage string = "shortcuts", "message"
db *storm.DB
projectRepo repository.ProjectRepository
taskRepo repository.TaskRepository
projects []model.Project
currentProject model.Project
tasks []model.Task
currentTask model.Task
)
func main() {
app = tview.NewApplication()
db = util.ConnectStorm()
defer db.Close()
projectRepo = repo.NewProjectRepository(db)
taskRepo = repo.NewTaskRepository(db)
titleText := tview.NewTextView().SetText("[lime::b]Geek-life [::-]- life management for geeks!").SetDynamicColors(true)
cloudStatus := tview.NewTextView().SetText("[::d]Cloud Sync: off").SetTextAlign(tview.AlignRight).SetDynamicColors(true)
prepareStatusBar()
titleBar := tview.NewFlex().
AddItem(titleText, 0, 2, false).
AddItem(cloudStatus, 0, 1, false)
projectPane = makeProjectPane()
taskPane = makeTaskPane()
layout := tview.NewFlex().SetDirection(tview.FlexRow).
AddItem(titleBar, 2, 1, false).
AddItem(tview.NewFlex().
AddItem(projectPane, 25, 1, true).
AddItem(taskPane, 0, 2, false).
AddItem(tview.NewBox().SetBorder(true).SetTitle("Details"), 0, 3, false),
0, 2, true).
AddItem(statusBar, 1, 1, false)
setKeyboardShortcuts(projectPane, taskPane)
if err := app.SetRoot(layout, true).EnableMouse(true).Run(); err != nil {
panic(err)
}
}
func makeProjectPane() *tview.Flex {
var err error
projects, err = projectRepo.GetAll()
util.FatalIfError(err, "Could not load Projects")
projectList = tview.NewList()
for i := range projects {
addProjectToList(i, false)
}
newProject = makeLightTextInput("+[New Project]").
SetDoneFunc(func(key tcell.Key) {
switch key {
case tcell.KeyEnter:
project, err := projectRepo.Create(newProject.GetText(), "")
if err != nil {
showMessage("[red::]Failed to create Project:" + err.Error())
} else {
showMessage(fmt.Sprintf("[green::]Project %s created. Press n to start adding new tasks.", newProject.GetText()))
projects = append(projects, project)
addProjectToList(len(projects)-1, true)
newProject.SetText("")
}
case tcell.KeyEsc:
app.SetFocus(projectPane)
}
})
projectBar := tview.NewFlex().SetDirection(tview.FlexRow).
AddItem(projectList, 0, 1, true).
AddItem(newProject, 1, 0, false)
projectBar.SetBorder(true).SetTitle("Projects (p)")
return projectBar
}
func addProjectToList(i int, selectItem bool) {
// To avoid overriding of loop variables - https://www.calhoun.io/gotchas-and-common-mistakes-with-closures-in-go/
projectList.AddItem(projects[i].Title, "", 0, func(idx int) func() {
return func() { loadProject(idx) }
}(i))
if selectItem {
projectList.SetCurrentItem(i)
loadProject(i)
}
}
func makeTaskPane() *tview.Flex {
taskList = tview.NewList().ShowSecondaryText(false)
taskList.SetDoneFunc(func() {
app.SetFocus(projectPane)
})
newTask = makeLightTextInput("+[New Task]").
SetDoneFunc(func(key tcell.Key) {
switch key {
case tcell.KeyEnter:
task, err := taskRepo.Create(currentProject, newTask.GetText(), "", "", time.Now().Unix())
if err != nil {
showMessage("[red::]Could not create Task:" + err.Error())
}
taskList.AddItem(task.Title, "", 0, nil)
newTask.SetText("")
case tcell.KeyEsc:
app.SetFocus(taskPane)
}
})
taskPane := tview.NewFlex().SetDirection(tview.FlexRow).
AddItem(taskList, 0, 1, true).
AddItem(newTask, 1, 0, false)
taskPane.SetBorder(true).SetTitle("Tasks (t)")
return taskPane
}
func loadProject(idx int) {
currentProject = projects[idx]
taskList.Clear()
app.SetFocus(taskPane)
var err error
if tasks, err = taskRepo.GetAllByProject(currentProject); err != nil && err != storm.ErrNotFound {
showMessage("[red::]Error: " + err.Error())
}
for i, task := range tasks {
taskList.AddItem(task.Title, "", 0, func(taskidx int) func() {
return func() { loadTask(taskidx) }
}(i))
}
}
func loadTask(idx int) {
currentTask = tasks[idx]
// taskList.Clear()
// app.SetFocus(taskPane)
}
func makeLightTextInput(placeholder string) *tview.InputField {
return tview.NewInputField().
SetPlaceholder(placeholder).
SetPlaceholderTextColor(tcell.ColorLightSlateGray).
SetFieldTextColor(tcell.ColorBlack).
SetFieldBackgroundColor(tcell.ColorGray)
}
func ignoreKeyEvt() bool {
return reflect.TypeOf(app.GetFocus()).String() == "*tview.InputField"
}
func setKeyboardShortcuts(projectPane *tview.Flex, taskPane *tview.Flex) *tview.Application {
return app.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
if ignoreKeyEvt() {
return event
}
switch event.Rune() {
case 'p':
app.SetFocus(projectPane)
case 't':
app.SetFocus(taskPane)
case 'n':
if projectPane.HasFocus() {
app.SetFocus(newProject)
} else if taskPane.HasFocus() {
app.SetFocus(newTask)
}
}
return event
})
}
func showMessage(text string) {
message.SetText(text)
statusBar.SwitchToPage(messagePage)
go func() {
app.QueueUpdateDraw(func() {
time.Sleep(time.Second * 5)
statusBar.SwitchToPage(shortcutsPage)
})
}()
}
func prepareStatusBar() {
statusBar = tview.NewPages()
message = tview.NewTextView().SetDynamicColors(true).SetText("Loading...")
statusBar.AddPage(messagePage, message, true, true)
statusBar.AddPage(shortcutsPage,
tview.NewGrid().
SetColumns(0, 0, 0, 0).
SetRows(0).
AddItem(tview.NewTextView().SetText("Shortcuts: Alt+.(dot)"), 0, 0, 1, 1, 0, 0, false).
AddItem(tview.NewTextView().SetText("New Project: n").SetTextAlign(tview.AlignCenter), 0, 1, 1, 1, 0, 0, false).
AddItem(tview.NewTextView().SetText("New Task: t").SetTextAlign(tview.AlignCenter), 0, 2, 1, 1, 0, 0, false).
AddItem(tview.NewTextView().SetText("Quit: Ctrl+C").SetTextAlign(tview.AlignRight), 0, 3, 1, 1, 0, 0, false),
true,
true,
)
}