feat: add loan handler

api/handler/loan.go

@@ -16,7 +16,7 @@ import (
 	"github.com/gorilla/mux"
 )
 
-func borrow(bService book.UseCase, uService user.UseCase, loanService loan.UseCase) http.Handler {
+func borrowBook(bService book.UseCase, uService user.UseCase, loanService loan.UseCase) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		errorMessage := "Error borrowing book"
 		vars := mux.Vars(r)
@@ -32,21 +32,60 @@ func borrow(bService book.UseCase, uService user.UseCase, loanService loan.UseCa
 			w.Write([]byte(errorMessage))
 			return
 		}
+		if b == nil {
+			w.WriteHeader(http.StatusNotFound)
+			w.Write([]byte(errorMessage))
+			return
+		}
 		uID, err := entity.StringToID(vars["user_id"])
 		if err != nil {
 			w.WriteHeader(http.StatusInternalServerError)
 			w.Write([]byte(errorMessage))
 			return
 		}
-
 		u, err := uService.Get(uID)
 		if err != nil && err != domain.ErrNotFound {
 			w.WriteHeader(http.StatusInternalServerError)
 			w.Write([]byte(errorMessage))
 			return
 		}
-
+		if u == nil {
+			w.WriteHeader(http.StatusNotFound)
+			w.Write([]byte(errorMessage))
+			return
+		}
 		err = loanService.Borrow(u, b)
+		if err != nil {
+			w.WriteHeader(http.StatusInternalServerError)
+			w.Write([]byte(errorMessage))
+			return
+		}
+		w.WriteHeader(http.StatusCreated)
+	})
+}
+
+func returnBook(bService book.UseCase, loanService loan.UseCase) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		errorMessage := "Error returning book"
+		vars := mux.Vars(r)
+		bID, err := entity.StringToID(vars["book_id"])
+		if err != nil {
+			w.WriteHeader(http.StatusInternalServerError)
+			w.Write([]byte(errorMessage))
+			return
+		}
+		b, err := bService.Get(bID)
+		if err != nil && err != domain.ErrNotFound {
+			w.WriteHeader(http.StatusInternalServerError)
+			w.Write([]byte(errorMessage))
+			return
+		}
+		if b == nil {
+			w.WriteHeader(http.StatusNotFound)
+			w.Write([]byte(errorMessage))
+			return
+		}
+		err = loanService.Return(b)
 		if err != nil && err != domain.ErrNotFound {
 			w.WriteHeader(http.StatusInternalServerError)
 			w.Write([]byte(errorMessage))
@@ -59,10 +98,10 @@ func borrow(bService book.UseCase, uService user.UseCase, loanService loan.UseCa
 //MakeLoanHandlers make url handlers
 func MakeLoanHandlers(r *mux.Router, n negroni.Negroni, bService book.UseCase, uService user.UseCase, loanService loan.UseCase) {
 	r.Handle("/v1/loan/borrow/{book_id}/{user_id}", n.With(
-		negroni.Wrap(borrow(bService, uService, loanService)),
-	)).Methods("GET", "OPTIONS").Name("borrow")
+		negroni.Wrap(borrowBook(bService, uService, loanService)),
+	)).Methods("GET", "OPTIONS").Name("borrowBook")
 
-	//r.Handle("/v1/book/{book_id}/return", n.With(
-	//	negroni.Wrap(returnBook(service)),
-	//)).Methods("GET", "OPTIONS").Name("returnBook")
+	r.Handle("/v1/loan/return/{book_id}", n.With(
+		negroni.Wrap(returnBook(bService, loanService)),
+	)).Methods("GET", "OPTIONS").Name("returnBook")
 }