feat: print today's total times as well

Signed-off-by: Carlos Alexandro Becker <[email protected]>
caarlos0 · Aug 10, 2021 · 03767d2 · 03767d2
1 parent 8ee89f4
commit 03767d2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/internal/ui/markdown.go b/internal/ui/markdown.go
@@ -25,7 +25,7 @@ func WriteProjectMarkdown(db *badger.DB, project string, w io.Writer) error {
 	_, _ = fmt.Fprintf(
 		w,
 		"> Total time **%s**, timed between **%s** and **%s**\n\n",
-		sumTasksTimes(tasks).Round(time.Second).String(),
+		sumTasksTimes(tasks, time.Time{}).Round(time.Second).String(),
 		tasks[len(tasks)-1].StartAt.Format("2006-01-02"),
 		tasks[0].EndAt.Format("2006-01-02"),
 	)

diff --git a/internal/ui/project_timer.go b/internal/ui/project_timer.go
@@ -25,7 +25,10 @@ func (m projectTimerModel) Update(msg tea.Msg) (projectTimerModel, tea.Cmd) {
 
 func (m projectTimerModel) View() string {
 	return secondaryForeground.Render("total: ") +
-		activeForegroundBold.Render(sumTasksTimes(m.tasks).Round(time.Second).String())
+		activeForegroundBold.Render(sumTasksTimes(m.tasks, time.Time{}).Round(time.Second).String()) +
+		separator +
+		secondaryForeground.Render("today: ") +
+		activeForegroundBold.Render(sumTasksTimes(m.tasks, time.Now().Truncate(time.Hour*24)).Round(time.Second).String())
 }
 
 // msgs and cmds
@@ -40,9 +43,12 @@ func updateProjectTimerCmd(tasks []model.Task) tea.Cmd {
 	}
 }
 
-func sumTasksTimes(tasks []model.Task) time.Duration {
+func sumTasksTimes(tasks []model.Task, since time.Time) time.Duration {
 	d := time.Duration(0)
 	for _, t := range tasks {
+		if t.StartAt.Before(since) {
+			continue
+		}
 		z := t.EndAt
 		if z.IsZero() {
 			z = time.Now()