zeit/z/track.go

package z

import (
  "os"
  "log"
  "fmt"
  "github.com/spf13/cobra"
  "github.com/gookit/color"
)

var trackCmd = &cobra.Command{
  Use:   "track",
  Short: "Tracking time",
  Long: "Add a new tracking entry, which can either be kept running until 'finish' is being called or parameterized to be a finished entry.",
  Run: func(cmd *cobra.Command, args []string) {
    user := GetCurrentUser()

    runningEntryId, err := database.GetRunningEntryId(user)
    if err != nil {
      log.Fatal(err)
    }

    if runningEntryId != "" {
      fmt.Printf("▷ a task is already running\n")
      os.Exit(-1)
    }

    newEntry, err := NewEntry("", begin, finish, project, task, user)
    if err != nil {
      log.Fatal(err)
    }

    _, err = database.AddEntry(user, newEntry, true)
    if err != nil {
      log.Fatal(err)
    }

    if newEntry.Task != "" && newEntry.Project != "" {
      fmt.Printf("▷ began tracking %s on %s\n", color.FgLightWhite.Render(newEntry.Task), color.FgLightWhite.Render(newEntry.Project))
    } else if newEntry.Task != "" && newEntry.Project == "" {
      fmt.Printf("▷ began tracking %s\n", color.FgLightWhite.Render(newEntry.Task))
    } else if newEntry.Task == "" && newEntry.Project != "" {
      fmt.Printf("▷ began tracking task on %s\n", color.FgLightWhite.Render(newEntry.Project))
    } else {
      fmt.Printf("▷ began tracking task\n")
    }
    return
  },
}

func init() {
  rootCmd.AddCommand(trackCmd)
  trackCmd.Flags().StringVarP(&begin, "begin", "b", "", "Time the entry should begin at\n\nEither in the formats 16:00 / 4:00PM \nor relative to the current time, \ne.g. -0:15 (now minus 15 minutes), +1.50 (now plus 1:30h).")
  trackCmd.Flags().StringVarP(&finish, "finish", "s", "", "Time the entry should finish at\n\nEither in the formats 16:00 / 4:00PM \nor relative to the current time, \ne.g. -0:15 (now minus 15 minutes), +1.50 (now plus 1:30h).\nMust be after --begin time.")
  trackCmd.Flags().StringVarP(&project, "project", "p", "", "Project to be assigned")
  trackCmd.Flags().StringVarP(&task, "task", "t", "", "Task to be assigned")
  trackCmd.Flags().BoolVarP(&force, "force", "f", false, "Force begin tracking of a new task \neven though another one is still running \n(ONLY IF YOU KNOW WHAT YOU'RE DOING!)")

  var err error
  database, err = InitDatabase()
  if err != nil {
    log.Fatal(err)
  }
}