95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/gosuri/uilive"
|
|
"time"
|
|
)
|
|
|
|
type TaskRunner struct {
|
|
count int
|
|
task *Task
|
|
store *Store
|
|
writer *uilive.Writer
|
|
timer *time.Timer
|
|
ticker *time.Ticker
|
|
notifier Notifier
|
|
}
|
|
|
|
func NewTaskRunner(task *Task, store *Store) (*TaskRunner, error) {
|
|
taskID, err := store.CreateTask(*task)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
task.ID = taskID
|
|
tr := &TaskRunner{
|
|
task: task,
|
|
store: store,
|
|
notifier: NewLibNotifier(),
|
|
writer: uilive.New(),
|
|
timer: time.NewTimer(task.Duration),
|
|
ticker: time.NewTicker(RefreshInterval),
|
|
}
|
|
tr.writer.Start()
|
|
return tr, nil
|
|
}
|
|
|
|
func (t *TaskRunner) Run() error {
|
|
for t.count < t.task.NPomodoros {
|
|
// ASCII spinner
|
|
wheel := &Wheel{}
|
|
// This pomodoro
|
|
pomodoro := &Pomodoro{}
|
|
prompt("press enter to begin")
|
|
// Emit a desktop notification
|
|
// that the task is beginning.
|
|
t.notifier.Begin(t.count, *t.task)
|
|
// Record task as started
|
|
pomodoro.Start = time.Now()
|
|
// Reset the timer
|
|
t.timer.Reset(t.task.Duration)
|
|
// Wait for either a tick to update
|
|
// the UI for the timer to complete
|
|
loop:
|
|
select {
|
|
case <-t.ticker.C:
|
|
t.updateUI(Message{
|
|
Start: pomodoro.Start,
|
|
Duration: t.task.Duration,
|
|
Pomodoros: t.task.NPomodoros,
|
|
Wheel: wheel,
|
|
CurrentPomodoro: t.count,
|
|
})
|
|
goto loop
|
|
case <-t.timer.C:
|
|
// Send a notification for the
|
|
// user to take a break. We record
|
|
// how long it actually takes for
|
|
// them to initiate the break.
|
|
t.notifier.Break(*t.task)
|
|
prompt("press enter to take a break")
|
|
// Record the task as complete
|
|
pomodoro.End = time.Now()
|
|
// Record the session in the db
|
|
err := t.store.CreatePomodoro(t.task.ID, *pomodoro)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// Increment the count of completed pomodoros
|
|
t.count++
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (t *TaskRunner) updateUI(msg Message) {
|
|
fmt.Fprintf(
|
|
t.writer,
|
|
"%s %s remaining [ pomodoro %d/%d ]\n",
|
|
msg.Wheel,
|
|
(msg.Duration - time.Since(msg.Start)).Truncate(time.Second),
|
|
msg.CurrentPomodoro,
|
|
msg.Pomodoros,
|
|
)
|
|
}
|