Add option to return current total time when recording file events

Author: mschenk42

command/commit.go

@@ -55,7 +55,7 @@ func (r GitCommit) Run(args []string) int {
 
 func (r GitCommit) Synopsis() string {
 	return `
-	Usage: gtm commit [--yes]
+	Usage: gtm commit [-yes]
 	Save your logged time with the last commit
 	This is automatically called from the postcommit hook
 	Warning - any time logged will be cleared from your working directory

command/record.go

@@ -1,10 +1,16 @@
 package command
 
 import (
+	"flag"
 	"fmt"
+	"os"
+	"path/filepath"
 
 	"github.com/git-time-metric/gtm/event"
+	"github.com/git-time-metric/gtm/metric"
+	"github.com/git-time-metric/gtm/note"
 	"github.com/git-time-metric/gtm/project"
+	"github.com/git-time-metric/gtm/report"
 
 	"github.com/mitchellh/cli"
 )
@@ -21,25 +27,61 @@ func (r RecordCmd) Help() string {
 }
 
 func (r RecordCmd) Run(args []string) int {
-	if len(args) == 0 {
+	recordFlags := flag.NewFlagSet("record", flag.ExitOnError)
+	status := recordFlags.Bool(
+		"status",
+		false,
+		"After recording, return current total time spent [gtm status -total-only]")
+	if err := recordFlags.Parse(os.Args[2:]); err != nil {
+		fmt.Println(err)
+		return 1
+	}
+
+	if len(recordFlags.Args()) == 0 {
 		fmt.Println("Unable to record, file not provided")
 		return 1
 	}
 
-	//TODO: add an option to turn off silencing ErrFileNotFound errors
-	if err := event.Record(args[0]); err != nil && !(err == project.ErrNotInitialized || err == project.ErrFileNotFound) {
+	if err := event.Record(recordFlags.Args()[0]); err != nil && !(err == project.ErrNotInitialized || err == project.ErrFileNotFound) {
 		if err := project.Log(err); err != nil {
 			fmt.Println(err)
 		}
 		return 1
+	} else if err == nil && *status {
+		var (
+			err        error
+			commitNote note.CommitNote
+			out        string
+			wd         string
+		)
+
+		wd, err = os.Getwd()
+		if err != nil {
+			fmt.Println(err)
+			return 1
+		}
+		defer os.Chdir(wd)
+
+		os.Chdir(filepath.Dir(recordFlags.Args()[0]))
+
+		if commitNote, err = metric.Process(true); err != nil {
+			fmt.Println(err)
+			return 1
+		}
+		out, err = report.Status(commitNote, *status)
+		if err != nil {
+			fmt.Println(err)
+			return 1
+		}
+		fmt.Printf(out)
 	}
 
 	return 0
 }
 
 func (r RecordCmd) Synopsis() string {
 	return `
-	Usage: gtm record <filepath>
+	Usage: gtm record [-status] <path/file>
 	Record a file event
 	`
 }