113 lines
2.7 KiB
Go
113 lines
2.7 KiB
Go
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
|
|
}
|
|
}*/
|
|
}
|