1
0
Fork 0
mirror of https://github.com/Luzifer/worktime.git synced 2024-12-23 06:21:16 +00:00
worktime/cmd/track.go

60 lines
1.4 KiB
Go
Raw Permalink Normal View History

2016-09-24 12:59:16 +00:00
package cmd
import (
"crypto/sha1"
"fmt"
"os"
"os/signal"
"syscall"
"time"
"github.com/Luzifer/worktime/schema"
"github.com/spf13/cobra"
)
// trackCmd represents the track command
var trackCmd = &cobra.Command{
Use: "track [tag [tag]]",
Short: "Track a time frame from the command start until interruption",
RunE: func(cmd *cobra.Command, args []string) error {
start := time.Now()
fmt.Println("Press Ctrl+C to stop time tracking...")
c := make(chan os.Signal, 1)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
<-c // Blocks until a signal arrives
end := time.Now()
doc, err := schema.LoadDay(cfg.couchDB, start, true)
if err != nil {
return err
}
doc.Times = append(doc.Times, &schema.Time{
ID: fmt.Sprintf("%x", sha1.Sum([]byte(time.Now().String()))),
Start: start.Format("15:04:05"),
End: end.Format("15:04:05"),
Tags: args,
})
return doc.Save(cfg.couchDB)
},
}
func init() {
timeCmd.AddCommand(trackCmd)
// Here you will define your flags and configuration settings.
// Cobra supports Persistent Flags which will work for this command
// and all subcommands, e.g.:
// trackCmd.PersistentFlags().String("foo", "", "A help for foo")
// Cobra supports local flags which will only run when this command
// is called directly, e.g.:
// trackCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}