60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/gosuri/uilive"
|
||
|
|
"io"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Task Starting..
|
||
|
|
//
|
||
|
|
// 20min remaning [stent 1/4]
|
||
|
|
// ..
|
||
|
|
// 15min remaining [stent 2/4]
|
||
|
|
// ..
|
||
|
|
// Task Completed!
|
||
|
|
func display(writer io.Writer, msg Message) {
|
||
|
|
fmt.Fprintf(
|
||
|
|
writer,
|
||
|
|
"%s remaining [ stent %d/%d ]\n",
|
||
|
|
(msg.Duration - time.Since(msg.Start)).Truncate(time.Second),
|
||
|
|
msg.CurrentStent,
|
||
|
|
msg.Stents,
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
func run(task Task, prompter Prompter, db *Store) {
|
||
|
|
taskID, err := db.CreateTask(task)
|
||
|
|
maybe(err)
|
||
|
|
writer := uilive.New()
|
||
|
|
writer.Start()
|
||
|
|
ticker := time.NewTicker(RefreshInterval)
|
||
|
|
timer := time.NewTimer(task.duration)
|
||
|
|
var currentStent int
|
||
|
|
for currentStent < task.stents {
|
||
|
|
record := &Record{}
|
||
|
|
maybe(prompter.Prompt("Begin working!"))
|
||
|
|
record.Start = time.Now()
|
||
|
|
timer.Reset(task.duration)
|
||
|
|
loop:
|
||
|
|
select {
|
||
|
|
case <-ticker.C:
|
||
|
|
display(writer, Message{
|
||
|
|
Start: record.Start,
|
||
|
|
Duration: task.duration,
|
||
|
|
Stents: task.stents,
|
||
|
|
CurrentStent: currentStent,
|
||
|
|
})
|
||
|
|
goto loop
|
||
|
|
case <-timer.C:
|
||
|
|
maybe(prompter.Prompt("Take a break!"))
|
||
|
|
record.End = time.Now()
|
||
|
|
maybe(db.CreateRecord(taskID, *record))
|
||
|
|
currentStent++
|
||
|
|
}
|
||
|
|
maybe(db.CreateRecord(taskID, *record))
|
||
|
|
}
|
||
|
|
writer.Stop()
|
||
|
|
}
|