diff --git a/postgres/ynab-export.go b/postgres/ynab-export.go index 95e5290..636b455 100644 --- a/postgres/ynab-export.go +++ b/postgres/ynab-export.go @@ -67,31 +67,33 @@ func (ynab *YNABExport) ExportTransactions(context context.Context, w io.Writer) csv := csv.NewWriter(w) csv.Comma = '\t' - transactions, err := ynab.queries.GetTransactionsForBudget(context, ynab.budgetID) + transactions, err := ynab.queries.GetAllTransactionsForBudget(context, ynab.budgetID) if err != nil { return fmt.Errorf("load transactions: %w", err) } + header := []string{ + "Account", + "Flag", + "Date", + "Payee", + "Category Group/Category", + "Category Group", + "Category", + "Memo", + "Outflow", + "Inflow", + "Cleared", + } + + err = csv.Write(header) + if err != nil { + return fmt.Errorf("write transaction: %w", err) + } + count := 0 for _, transaction := range transactions { - row := []string{ - transaction.Account, - "", // Flag - transaction.Date.Format("02.01.2006"), - transaction.Payee, - transaction.CategoryGroup + " : " + transaction.Category, - transaction.CategoryGroup, - transaction.Category, - transaction.Memo, - } - - if transaction.Amount.IsPositive() { - row = append(row, transaction.Amount.String()+"€", "0,00€") - } else { - row = append(row, "0,00€", transaction.Amount.String()[1:]+"€") - } - - row = append(row, string(transaction.Status)) + row := GetTransactionRow(transaction) err := csv.Write(row) if err != nil { @@ -106,3 +108,31 @@ func (ynab *YNABExport) ExportTransactions(context context.Context, w io.Writer) return nil } + +func GetTransactionRow(transaction GetAllTransactionsForBudgetRow) []string { + row := []string{ + transaction.Account, + "", // Flag + transaction.Date.Format("02.01.2006"), + transaction.Payee, + } + + if transaction.CategoryGroup != "" && transaction.Category != "" { + row = append(row, + transaction.CategoryGroup+" : "+transaction.Category, + transaction.CategoryGroup, + transaction.Category) + } else { + row = append(row, "", "", "") + } + + row = append(row, transaction.Memo) + + if transaction.Amount.IsPositive() { + row = append(row, transaction.Amount.String()+"€", "0,00€") + } else { + row = append(row, "0,00€", transaction.Amount.String()[1:]+"€") + } + + return append(row, string(transaction.Status)) +} diff --git a/server/http.go b/server/http.go index dbc623f..a80f55c 100644 --- a/server/http.go +++ b/server/http.go @@ -67,7 +67,8 @@ func (h *Handler) LoadRoutes(router *gin.Engine) { authenticated.GET("/budget/:budgetid/autocomplete/categories", h.autocompleteCategories) authenticated.DELETE("/budget/:budgetid", h.deleteBudget) authenticated.POST("/budget/:budgetid/import/ynab", h.importYNAB) - authenticated.POST("/budget/:budgetid/export/ynab", h.exportYNAB) + authenticated.POST("/budget/:budgetid/export/ynab/transactions", h.exportYNABTransactions) + authenticated.POST("/budget/:budgetid/export/ynab/assignments", h.exportYNABAssignments) authenticated.POST("/budget/:budgetid/settings/clear", h.clearBudget) budget := authenticated.Group("/budget") diff --git a/server/ynab-import.go b/server/ynab-import.go index 11d0a4c..ac0da74 100644 --- a/server/ynab-import.go +++ b/server/ynab-import.go @@ -64,7 +64,7 @@ func (h *Handler) importYNAB(c *gin.Context) { } } -func (h *Handler) exportYNAB(c *gin.Context) { +func (h *Handler) exportYNABTransactions(c *gin.Context) { budgetID, succ := c.Params.Get("budgetid") if !succ { c.AbortWithStatusJSON(http.StatusBadRequest, ErrorResponse{"no budget_id specified"}) @@ -88,6 +88,26 @@ func (h *Handler) exportYNAB(c *gin.Context) { c.AbortWithError(http.StatusInternalServerError, err) return } +} + +func (h *Handler) exportYNABAssignments(c *gin.Context) { + budgetID, succ := c.Params.Get("budgetid") + if !succ { + c.AbortWithStatusJSON(http.StatusBadRequest, ErrorResponse{"no budget_id specified"}) + return + } + + budgetUUID, err := uuid.Parse(budgetID) + if !succ { + c.AbortWithError(http.StatusBadRequest, err) + return + } + + ynab, err := postgres.NewYNABExport(c.Request.Context(), h.Service.Queries, budgetUUID) + if err != nil { + c.AbortWithError(http.StatusInternalServerError, err) + return + } err = ynab.ExportAssignments(c.Request.Context(), c.Writer) if err != nil {