diff options
Diffstat (limited to 'invoice')
| -rw-r--r-- | invoice/db_actions.go | 12 | ||||
| -rw-r--r-- | invoice/router.go | 88 |
2 files changed, 90 insertions, 10 deletions
diff --git a/invoice/db_actions.go b/invoice/db_actions.go index a4880fa..0cbeec1 100644 --- a/invoice/db_actions.go +++ b/invoice/db_actions.go @@ -117,3 +117,15 @@ func getTransports(filter bson.M) ([]Transport, error) { err = cursor.All(context.TODO(), &transports) return transports, err } + +func getInvoiceByNumber(invoiceNumber int) (Invoice, error) { + var invoice Invoice + err := db.Collection("Invoices").FindOne(context.TODO(), bson.M{"InvoiceNumber": invoiceNumber}).Decode(&invoice) + return invoice, err +} + +func getInvoiceById(invoiceId primitive.ObjectID) (Invoice, error) { + var invoice Invoice + err := db.Collection("Invoices").FindOne(context.TODO(), bson.M{"_id": invoiceId}).Decode(&invoice) + return invoice, err +} diff --git a/invoice/router.go b/invoice/router.go index 4e33e18..ba6665f 100644 --- a/invoice/router.go +++ b/invoice/router.go @@ -20,10 +20,12 @@ package invoice import ( "github.com/gin-gonic/gin" "log" + "errors" "net/http" "strconv" "go.mongodb.org/mongo-driver/bson/primitive" - "go.mongodb.org/mongo-driver/bson" + "go.mongodb.org/mongo-driver/mongo" + //"go.mongodb.org/mongo-driver/bson" ) func Routes(route *gin.Engine) { @@ -41,29 +43,95 @@ func Routes(route *gin.Engine) { ctx.JSON(http.StatusOK, invoices) }) - // preview invoice - i.GET("/preview/:invoiceNumber", func(ctx *gin.Context) { + // TODO: /preview routes should send error codes as HTML + // send invoice as HTML, filtering by InvoiceNumber + i.GET("/preview/by-num/:invoiceNumber", func(ctx *gin.Context) { num := ctx.Param("invoiceNumber") numInt, _ := strconv.Atoi(num) - invoice, err := getInvoices(bson.M{"InvoiceNumber": numInt}) + invoice, err := getInvoiceByNumber(numInt) if err != nil { - ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) - log.Printf("ERROR: Failed to read invoice %v from DB: %v\n", numInt, err.Error()) + if errors.Is(err, mongo.ErrNoDocuments) { + ctx.JSON(http.StatusNotFound, gin.H{"error": err.Error()}) + } else { + ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + } + log.Printf("ERROR: Failed to read invoice #%d from DB: %v\n", numInt, err.Error()) return } - if len(invoice) == 0 { - ctx.JSON(http.StatusNotFound, gin.H{"error": "no invoice with this invoice number"}) - log.Printf("WARN: No invoice with number %v found", numInt) + ctx.HTML(http.StatusOK, "invoice.html", gin.H{ + "Invoice": invoice, + }) + }) + + // send invoice as HTML, filtering by ID + i.GET("/preview/by-id/:invoiceId", func(ctx *gin.Context) { + id, err := primitive.ObjectIDFromHex(ctx.Param("invoiceId")) + if err != nil { + ctx.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) + log.Printf("ERROR: Failed to get invoice with ID, Error parsing ID: %v\n", err.Error()) + return + } + + invoice, err := getInvoiceById(id) + if err != nil { + if errors.Is(err, mongo.ErrNoDocuments) { + ctx.JSON(http.StatusNotFound, gin.H{"error": err.Error()}) + } else { + ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + } + log.Printf("ERROR: Failed to read invoice %v from DB: %v\n", id, err.Error()) return } ctx.HTML(http.StatusOK, "invoice.html", gin.H{ - "Invoice": invoice[0], + "Invoice": invoice, }) }) + // send invoice as JSON, filtering by InvoiceNumber + i.GET("/by-num/:invoiceNumber", func(ctx *gin.Context) { + num := ctx.Param("invoiceNumber") + numInt, _ := strconv.Atoi(num) + + invoice, err := getInvoiceByNumber(numInt) + if err != nil { + if errors.Is(err, mongo.ErrNoDocuments) { + ctx.JSON(http.StatusNotFound, gin.H{"error": err.Error()}) + } else { + ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + } + log.Printf("ERROR: Failed to read invoice #%d from DB: %v\n", numInt, err.Error()) + return + } + + ctx.JSON(http.StatusOK, invoice) + }) + + // send invoice as JSON, filtering by ID + i.GET("/by-id/:invoiceId", func(ctx *gin.Context) { + id, err := primitive.ObjectIDFromHex(ctx.Param("invoiceId")) + if err != nil { + ctx.JSON(http.StatusBadRequest, gin.H{"error": err.Error()}) + log.Printf("ERROR: Failed to get invoice with ID, Error parsing ID: %v\n", err.Error()) + return + } + + invoice, err := getInvoiceById(id) + if err != nil { + if errors.Is(err, mongo.ErrNoDocuments) { + ctx.JSON(http.StatusNotFound, gin.H{"error": err.Error()}) + } else { + ctx.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()}) + } + log.Printf("ERROR: Failed to read invoice %v from DB: %v\n", id, err.Error()) + return + } + + ctx.JSON(http.StatusOK, invoice) + }) + i.POST("/new", func(ctx *gin.Context) { var i Invoice ctx.BindJSON(&i) |