diff options
Diffstat (limited to 'invoice/controller.go')
| -rw-r--r-- | invoice/controller.go | 27 |
1 files changed, 26 insertions, 1 deletions
diff --git a/invoice/controller.go b/invoice/controller.go index 7260834..6bd5ad5 100644 --- a/invoice/controller.go +++ b/invoice/controller.go @@ -74,7 +74,32 @@ func handleGetInvoices (ctx *gin.Context) { userId := uId.(uint) - err := getInvoices(&invoices, userId) + err := getInvoices(&invoices, userId, false) + if err != nil { + ctx.Error(err) + ctx.Abort() + return + } + + ctx.JSON(http.StatusOK, gin.H{ + "message": "success", + "data": invoices, + }) +} + +func handleGetDrafts (ctx *gin.Context) { + var invoices []Invoice + + uId, ok := ctx.Get("UserID") + if !ok { + ctx.Error(e.ErrUnauthorized) + ctx.Abort() + return + } + + userId := uId.(uint) + + err := getInvoices(&invoices, userId, true) if err != nil { ctx.Error(err) ctx.Abort() |