126 lines
3.6 KiB
Go
126 lines
3.6 KiB
Go
package z
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/jinzhu/now"
|
|
"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 = now.Parse(since)
|
|
if err != nil {
|
|
fmt.Printf("%s %+v\n", CharError, err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
if until != "" {
|
|
untilTime, err = now.Parse(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)
|
|
}
|
|
}
|