kevinschoon-pomo/task.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,
	)
}

/*

// Task Starting..
//
// 20min remaning [pomodoro 1/4]
// ..
// 15min remaining [pomodoro 2/4]
// ..
// Task Completed!
func display(writer io.Writer, msg Message) {
	fmt.Fprintf(
		writer,
		"%s %s remaining [ pomodoro %d/%d ]\n",
		msg.Wheel,
		(msg.Duration - time.Since(msg.Start)).Truncate(time.Second),
		msg.CurrentPomodoro,
		msg.Pomodoros,
	)
}

func run(task Task, notifier Notifier, db *Store) {
	taskID, err := db.CreateTask(task)
	maybe(err)
	writer := uilive.New()
	writer.Start()
	ticker := time.NewTicker(RefreshInterval)
	timer := time.NewTimer(task.Duration)
	wheel := &Wheel{}
	var p int
	for p < task.NPomodoros {
		pomodoro := &Pomodoro{}
		maybe(notifier.Begin(task))
		pomodoro.Start = time.Now()
		timer.Reset(task.Duration)
	loop:
		select {
		case <-ticker.C:
			display(writer, Message{
				Start:           pomodoro.Start,
				Duration:        task.Duration,
				Pomodoros:       task.NPomodoros,
				Wheel:           wheel,
				CurrentPomodoro: p,
			})
			goto loop
		case <-timer.C:
			maybe(notifier.Break(task))
			fmt.Println("press enter to take a break")
			pomodoro.End = time.Now()
			maybe(db.CreatePomodoro(taskID, *pomodoro))
			p++
		}
	}
	maybe(notifier.Finish(task))
	writer.Stop()
}
*/