zeit/z/listCmd.go

package z

import (
  "fmt"
  "os"
  "time"

  "github.com/shopspring/decimal"
  "github.com/spf13/cobra"
)

var listTotalTime bool
var listOnlyProjectsAndTasks bool
var listOnlyTasks bool
var appendProjectIDToTask bool


var listCmd = &cobra.Command{
  Use:   "list",
  Short: "List activities",
  Long:  "List all tracked activities.",
  Run: func(cmd *cobra.Command, args []string) {
    user := GetCurrentUser()

    entries, err := database.ListEntries(user)
    if err != nil {
      fmt.Printf("%s %+v\n", CharError, err)
      os.Exit(1)
    }

    var sinceTime time.Time
    var untilTime time.Time

    if since != "" {
      sinceTime, err = time.Parse(time.RFC3339, since)
      if err != nil {
        fmt.Printf("%s %+v\n", CharError, err)
        os.Exit(1)
      }
    }

    if until != "" {
      untilTime, err = time.Parse(time.RFC3339, until)
      if err != nil {
        fmt.Printf("%s %+v\n", CharError, err)
        os.Exit(1)
      }
    }

    var filteredEntries []Entry
    filteredEntries, err = GetFilteredEntries(entries, project, task, sinceTime, untilTime)
    if err != nil {
      fmt.Printf("%s %+v\n", CharError, err)
      os.Exit(1)
    }

    if listOnlyProjectsAndTasks == true || listOnlyTasks == true {
      var projectsAndTasks = make(map[string]map[string]bool)

      for _, filteredEntry := range filteredEntries {
        taskMap, ok := projectsAndTasks[filteredEntry.Project]

        if !ok {
          taskMap = make(map[string]bool)
          projectsAndTasks[filteredEntry.Project] = taskMap
        }

        taskMap[filteredEntry.Task] = true
        projectsAndTasks[filteredEntry.Project] = taskMap
      }

      for project, _ := range projectsAndTasks {
        if listOnlyProjectsAndTasks == true && listOnlyTasks == false {
          fmt.Printf("%s %s\n", CharMore, project)
        }

        for task, _ := range projectsAndTasks[project] {
          if listOnlyProjectsAndTasks == true && listOnlyTasks == false {
            fmt.Printf("%*s└── ", 1, " ")
          }

          if appendProjectIDToTask == true {
            fmt.Printf("%s [%s]\n", task, project)
          } else {
            fmt.Printf("%s\n", task)
          }
        }
      }

      return
    }

    totalHours := decimal.NewFromInt(0)
    for _, entry := range filteredEntries {
      totalHours = totalHours.Add(entry.GetDuration())
      fmt.Printf("%s\n", entry.GetOutput(false))
    }

    if listTotalTime == true {
      fmt.Printf("\nTOTAL: %s H\n\n", fmtHours(totalHours));
    }
    return
  },
}

func init() {
  rootCmd.AddCommand(listCmd)
  listCmd.Flags().StringVar(&since, "since", "", "Date/time to start the list from")
  listCmd.Flags().StringVar(&until, "until", "", "Date/time to list until")
  listCmd.Flags().StringVarP(&project, "project", "p", "", "Project to be listed")
  listCmd.Flags().StringVarP(&task, "task", "t", "", "Task to be listed")
  listCmd.Flags().BoolVar(&fractional, "decimal", false, "Show fractional hours in decimal format instead of minutes")
  listCmd.Flags().BoolVar(&listTotalTime, "total", false, "Show total time of hours for listed activities")
  listCmd.Flags().BoolVar(&listOnlyProjectsAndTasks, "only-projects-and-tasks", false, "Only list projects and their tasks, no entries")
  listCmd.Flags().BoolVar(&listOnlyTasks, "only-tasks", false, "Only list tasks, no projects nor entries")
  listCmd.Flags().BoolVar(&appendProjectIDToTask, "append-project-id-to-task", false, "Append project ID to tasks in the list")

  var err error
  database, err = InitDatabase()
  if err != nil {
    fmt.Printf("%s %+v\n", CharError, err)
    os.Exit(1)
  }
}