From a6087d0a4f177973b9d27e544f85736d90039089 Mon Sep 17 00:00:00 2001
From: Vidhu Kant Sharma <vidhukant@vidhukant.com>
Date: Mon, 4 Dec 2023 01:46:17 +0530
Subject: added API route to get drafts only

---
 invoice/controller.go | 27 ++++++++++++++++++++++++++-
 1 file changed, 26 insertions(+), 1 deletion(-)

(limited to 'invoice/controller.go')

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()
-- 
cgit v1.2.3