72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/jawher/mow.cli"
|
||
|
|
"os"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func maybe(err error) {
|
||
|
|
if err != nil {
|
||
|
|
fmt.Printf("Error: %s", err)
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func startTask(task Task, prompter Prompter, db *Store) {
|
||
|
|
taskID, err := db.AddTask(task)
|
||
|
|
maybe(err)
|
||
|
|
for i := 0; i < task.count; i++ {
|
||
|
|
// Create a new "Pomo" to represent
|
||
|
|
// this particular stent of work
|
||
|
|
pomo := &Pomo{TaskID: taskID}
|
||
|
|
// Prompt the client
|
||
|
|
maybe(prompter.Prompt("Begin Working!"))
|
||
|
|
pomo.Start = time.Now()
|
||
|
|
// Wait the specified interval
|
||
|
|
time.Sleep(task.duration)
|
||
|
|
maybe(prompter.Prompt("Take a Break!"))
|
||
|
|
// Record how long the user waited
|
||
|
|
// until closing the notification
|
||
|
|
pomo.End = time.Now()
|
||
|
|
maybe(db.AddPomo(*pomo))
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func start(cmd *cli.Cmd) {
|
||
|
|
cmd.Spec = "[OPTIONS] NAME"
|
||
|
|
var (
|
||
|
|
duration = cmd.StringOpt("d duration", "25m", "duration of each stent")
|
||
|
|
count = cmd.IntOpt("c count", 4, "number of working stents")
|
||
|
|
name = cmd.StringArg("NAME", "", "descriptive name of the given task")
|
||
|
|
path = cmd.StringOpt("d db", "~/.pomo/state.db", "path to the pomo sqlite database")
|
||
|
|
)
|
||
|
|
cmd.Action = func() {
|
||
|
|
parsed, err := time.ParseDuration(*duration)
|
||
|
|
maybe(err)
|
||
|
|
db, err := NewStore(*path)
|
||
|
|
maybe(err)
|
||
|
|
task := Task{
|
||
|
|
Name: *name,
|
||
|
|
count: *count,
|
||
|
|
duration: parsed,
|
||
|
|
}
|
||
|
|
startTask(task, &I3{}, db)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func initialize(cmd *cli.Cmd) {}
|
||
|
|
|
||
|
|
func list(cmd *cli.Cmd) {}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
app := cli.App("pomo", "Pomodoro CLI")
|
||
|
|
app.Spec = "[OPTIONS]"
|
||
|
|
app.Command("start", "start a new task", start)
|
||
|
|
app.Command("init", "initialize the sqlite database", initialize)
|
||
|
|
app.Command("ls", "list historical tasks", list)
|
||
|
|
app.Run(os.Args)
|
||
|
|
}
|