Exctract getDate

http/budgeting.go

@@ -39,6 +39,27 @@ type CategoryWithBalance struct {
 	Assigned           float64
 }
 
+func getDate(c *gin.Context) (time.Time, error) {
+	var year, month int
+	yearString := c.Param("year")
+	monthString := c.Param("month")
+	if yearString == "" && monthString == "" {
+		return getFirstOfMonthTime(time.Now()), nil
+	}
+
+	year, err := strconv.Atoi(yearString)
+	if err != nil {
+		return time.Time{}, fmt.Errorf("parse year: %w", err)
+	}
+
+	month, err = strconv.Atoi(monthString)
+	if err != nil {
+		return time.Time{}, fmt.Errorf("parse month: %w", err)
+	}
+
+	return getFirstOfMonth(year, month, time.Now().Location()), nil
+}
+
 func (h *Handler) budgeting(c *gin.Context) {
 	budgetID := c.Param("budgetid")
 	budgetUUID, err := uuid.Parse(budgetID)
@@ -47,28 +68,11 @@ func (h *Handler) budgeting(c *gin.Context) {
 		return
 	}
 
-	var firstOfMonth time.Time
+	firstOfMonth, err := getDate(c)
-	var year, month int
-	yearString := c.Param("year")
-	monthString := c.Param("month")
-	if yearString != "" && monthString != "" {
-		year, err = strconv.Atoi(yearString)
 	if err != nil {
 		c.Redirect(http.StatusTemporaryRedirect, "/budget/"+budgetUUID.String())
 		return
 	}
-
-		month, err = strconv.Atoi(monthString)
-		if err != nil {
-			c.Redirect(http.StatusTemporaryRedirect, "/budget/"+budgetUUID.String())
-			return
-		}
-
-		firstOfMonth = getFirstOfMonth(year, month, time.Now().Location())
-	} else {
-		firstOfMonth = getFirstOfMonthTime(time.Now())
-	}
-
 	firstOfNextMonth := firstOfMonth.AddDate(0, 1, 0)
 	firstOfPreviousMonth := firstOfMonth.AddDate(0, -1, 0)