package http import ( "fmt" "net/http" "github.com/gin-gonic/gin" "github.com/google/uuid" "github.com/pressly/goose/v3" ) func (h *Handler) clearDatabase(c *gin.Context) { if err := goose.Reset(h.Service.DB, "schema"); err != nil { c.AbortWithError(http.StatusInternalServerError, err) } if err := goose.Up(h.Service.DB, "schema"); err != nil { c.AbortWithError(http.StatusInternalServerError, err) } } func (h *Handler) deleteBudget(c *gin.Context) { budgetID := c.Param("budgetid") budgetUUID, err := uuid.Parse(budgetID) if err != nil { c.AbortWithStatus(http.StatusBadRequest) return } h.clearBudgetData(c, budgetUUID) err = h.Service.DeleteBudget(c.Request.Context(), budgetUUID) if err != nil { c.AbortWithError(http.StatusInternalServerError, err) return } } func (h *Handler) clearBudgetData(c *gin.Context, budgetUUID uuid.UUID) { rows, err := h.Service.DeleteAllAssignments(c.Request.Context(), budgetUUID) if err != nil { c.AbortWithError(http.StatusInternalServerError, err) return } fmt.Printf("Deleted %d assignments\n", rows) rows, err = h.Service.DeleteAllTransactions(c.Request.Context(), budgetUUID) if err != nil { c.AbortWithError(http.StatusInternalServerError, err) return } fmt.Printf("Deleted %d transactions\n", rows) } func (h *Handler) clearBudget(c *gin.Context) { budgetID := c.Param("budgetid") budgetUUID, err := uuid.Parse(budgetID) if err != nil { c.AbortWithStatus(http.StatusBadRequest) return } h.clearBudgetData(c, budgetUUID) } func (h *Handler) cleanNegativeBudget(c *gin.Context) { /*budgetID := c.Param("budgetid") budgetUUID, err := uuid.Parse(budgetID) if err != nil { c.Redirect(http.StatusTemporaryRedirect, "/login") return }*/ /*min_date, err := h.Service.GetFirstActivity(c.Request.Context(), budgetUUID) date := getFirstOfMonthTime(min_date) for { nextDate := date.AddDate(0, 1, 0) params := postgres.GetCategoriesWithBalanceParams{ BudgetID: budgetUUID, ToDate: nextDate, FromDate: date, } categories, err := h.Service.GetCategoriesWithBalance(c.Request.Context(), params) if err != nil { c.AbortWithError(http.StatusInternalServerError, err) return } for _, category := range categories { available := category.Available.GetFloat64() if available >= 0 { continue } var negativeAvailable postgres.Numeric negativeAvailable.Set(-available) createAssignment := postgres.CreateAssignmentParams{ Date: nextDate.AddDate(0, 0, -1), Amount: negativeAvailable, CategoryID: category.ID, } h.Service.CreateAssignment(c.Request.Context(), createAssignment) } if nextDate.Before(time.Now()) { date = nextDate } else { break } }*/ }