added API route to get drafts only

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()