feat(cmd/report_month): add optional flag for month filtering

cmd/report_month.go

@@ -1,6 +1,7 @@
 package cmd
 
 import (
+	"errors"
 	"fmt"
 	"time"
 
@@ -35,15 +36,26 @@ func reportMonth(cmd *cobra.Command, args []string) error {
 	if err != nil {
 		return err
 	}
-	now := time.Now()
-	currentMonth, err := journal.FetchEntriesByMonthDate(entries, now)
+
+	// Check if the month flag has been set
+	monthFlag, _ := cmd.Flags().GetString("month")
+	var monthFilter time.Time
+	if monthFlag != "" {
+		monthFilter, err = time.Parse("2006-01", monthFlag)
+		if err != nil {
+			return errors.New("invalid month format, expected YYYY-MM")
+		}
+	} else {
+		monthFilter = time.Now()
+	}
+	currentMonth, err := journal.FetchEntriesByMonthDate(entries, monthFilter)
 	if err != nil {
 		return err
 	}
 	for _, entry := range currentMonth {
 		fmt.Printf("%s\n---\n", entry.String())
 	}
-	month := now.Format("January 2006")
+	month := monthFilter.Format("January 2006")
 	lunchTime, err := time.ParseDuration(viper.GetString("lunchTime"))
 	if err != nil {
 		return err
@@ -59,5 +71,6 @@ func reportMonth(cmd *cobra.Command, args []string) error {
 }
 
 func init() {
+	reportMonthCmd.Flags().StringP("month", "m", "", "Specify the month and year in the format YYYY-MM")
 	reportCmd.AddCommand(reportMonthCmd)
 }

internal/journal/utils.go

@@ -81,14 +81,14 @@ func FetchEntriesByWeekDate(journalEntries []JournalEntry, currentDate time.Time
 //	    log.Fatal(err)
 //	}
 //	fmt.Println(currentMonthEntries) // Prints the entries for the current week
-func FetchEntriesByMonthDate(journalEntries []JournalEntry, currentDate time.Time) ([]JournalEntry, error) {
+func FetchEntriesByMonthDate(journalEntries []JournalEntry, filterDate time.Time) ([]JournalEntry, error) {
 	if len(journalEntries) == 0 {
 		return nil, fmt.Errorf("No entries were passed")
 	}
 	var currentMonthEntries []JournalEntry
 
 	// Get the current year and month.
-	currentYear, currentMonth := currentDate.Year(), currentDate.Month()
+	currentYear, currentMonth := filterDate.Year(), filterDate.Month()
 
 	for _, entry := range journalEntries {
 		// get the year and month of the entry.