first commit

api/handlers/api.go

@@ -0,0 +1,77 @@
+package handlers
+
+import (
+	"encoding/json"
+	"io"
+	"net/http"
+
+	"github.com/SowinskiBraeden/SulliCartShare/api"
+	"github.com/SowinskiBraeden/SulliCartShare/models"
+	"github.com/gorilla/mux"
+	"go.uber.org/zap"
+
+	"github.com/SowinskiBraeden/SulliCartShare/config"
+	"github.com/SowinskiBraeden/SulliCartShare/databases"
+)
+
+// App stores the router and db connection so it can be reused
+type App struct {
+	Router   *mux.Router
+	DB       databases.CollectionHelper
+	Config   config.Config
+	dbHelper databases.DatabaseHelper
+}
+
+// New creates a new mux router and all the routes
+func (a *App) New() *mux.Router {
+	r := mux.NewRouter()
+	cow := Cow{DB: databases.NewCowDatabase(a.dbHelper)}
+
+	// healthcheck
+	r.HandleFunc("/health", healthCheckHandler)
+
+	apiCreate := r.PathPrefix("/api/v1").Subrouter()
+
+	apiCreate.Handle("/cow/{cow_id}", api.Middleware(http.HandlerFunc(cow.CowByIDHandler))).Methods("GET") // By Object ID not CowCode
+	apiCreate.Handle("/cows", api.Middleware(http.HandlerFunc(cow.CowHandler))).Methods("GET")
+	apiCreate.Handle("/cows/new", api.Middleware(http.HandlerFunc(cow.NewCowHandler))).Methods("POST")
+	apiCreate.Handle("/cows/update/{cow_id}", api.Middleware(http.HandlerFunc(cow.UpdateCowHandler))).Methods("POST") // By Object ID not CowCode
+
+	return r
+}
+
+func (a *App) Initialize() error {
+	client, err := databases.NewClient(&a.Config)
+	if err != nil {
+		// if we fail to create a new database client, the kill the pod
+		zap.S().With(err).Error("failed to create new client")
+		return err
+	}
+
+	a.dbHelper = databases.NewDatabase(&a.Config, client)
+	err = client.Connect()
+	if err != nil {
+		// if we fail to connect to the database, the kill the pod
+		zap.S().With(err).Error("failed to connect to database")
+		return err
+	}
+	zap.S().Info("SulliCartCheckout has connected to the database")
+
+	// initialize api router
+	a.initializeRoutes()
+	return nil
+
+}
+
+func (a *App) initializeRoutes() {
+	a.Router = a.New()
+}
+
+func healthCheckHandler(w http.ResponseWriter, r *http.Request) {
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(http.StatusOK)
+	b, _ := json.Marshal(models.HealthCheckResponse{
+		Alive: true,
+	})
+	_, _ = io.WriteString(w, string(b))
+}

api/handlers/cow.go

@@ -0,0 +1,142 @@
+package handlers
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/gorilla/mux"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+
+	"github.com/go-playground/validator/v10"
+
+	"github.com/SowinskiBraeden/SulliCartShare/config"
+	"github.com/SowinskiBraeden/SulliCartShare/databases"
+	"github.com/SowinskiBraeden/SulliCartShare/models"
+)
+
+var validate = validator.New()
+
+// Cow exported for testing purposes
+type Cow struct {
+	DB databases.CowDatabase
+}
+
+// CowHandler returns all cows
+func (c Cow) CowHandler(w http.ResponseWriter, r *http.Request) {
+	dbResp, err := c.DB.Find(context.TODO(), bson.M{})
+	if err != nil {
+		config.ErrorStatus("failed to get cows", http.StatusNotFound, w, err)
+		return
+	}
+
+	// If len == 0 then we will just return an empty data object
+	if len(dbResp) == 0 {
+		dbResp = []models.Cow{}
+	}
+	b, err := json.Marshal(dbResp)
+	if err != nil {
+		config.ErrorStatus("failed to marshal response", http.StatusInternalServerError, w, err)
+		return
+	}
+	w.WriteHeader(http.StatusOK)
+	w.Write(b)
+}
+
+// CowByIDHandler returns a cow by ID
+func (c Cow) CowByIDHandler(w http.ResponseWriter, r *http.Request) {
+	cowID := mux.Vars(r)["cow_id"]
+
+	cID, err := primitive.ObjectIDFromHex(cowID)
+	if err != nil {
+		config.ErrorStatus("failed to get objectID from Hex", http.StatusBadRequest, w, err)
+		return
+	}
+
+	dbResp, err := c.DB.FindOne(context.Background(), bson.M{"_id": cID})
+	if err != nil {
+		config.ErrorStatus("failed to get cow by ID", http.StatusNotFound, w, err)
+		return
+	}
+
+	b, err := json.Marshal(dbResp)
+	if err != nil {
+		config.ErrorStatus("failed to marshal responce", http.StatusInternalServerError, w, err)
+		return
+	}
+	w.WriteHeader(http.StatusOK)
+	w.Write(b)
+}
+
+// NewCowHandler inserts a new cow into the collection and returns a result and error
+func (c Cow) NewCowHandler(w http.ResponseWriter, r *http.Request) {
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	var cowDetails models.CowDetails // Json data will represent the cow details model
+	defer cancel()
+
+	//validate the request body
+	if err := json.NewDecoder(r.Body).Decode(&cowDetails); err != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		response := models.UserResponse{Status: http.StatusBadRequest, Message: "error", Data: map[string]interface{}{"error": err.Error()}}
+		json.NewEncoder(w).Encode(response)
+		return
+	}
+
+	//use the validator library to validate required fields
+	if validationErr := validate.Struct(&cowDetails); validationErr != nil {
+		w.WriteHeader(http.StatusBadRequest)
+		response := models.UserResponse{Status: http.StatusBadRequest, Message: "error", Data: map[string]interface{}{"error": validationErr.Error()}}
+		json.NewEncoder(w).Encode(response)
+		return
+	}
+
+	fmt.Print(cowDetails)
+
+	newCow := models.Cow{
+		ID:      primitive.NewObjectID().Hex(),
+		Details: cowDetails,
+	}
+
+	result, err := c.DB.InsertOne(ctx, newCow)
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		response := models.UserResponse{Status: http.StatusInternalServerError, Message: "error", Data: map[string]interface{}{"error": err.Error()}}
+		json.NewEncoder(w).Encode(response)
+		return
+	}
+
+	w.WriteHeader(http.StatusCreated)
+	response := models.UserResponse{Status: http.StatusCreated, Message: "success", Data: map[string]interface{}{"result": result}}
+	json.NewEncoder(w).Encode(response)
+}
+
+// TODO: fix all update handlers because they don't work
+
+// UpdateCowHandler gets updates the data for an existing cow and returns a result and error
+func (c Cow) UpdateCowHandler(w http.ResponseWriter, r *http.Request) {
+	cowID := mux.Vars(r)["cow_id"]
+
+	// TODO: Collect data to update from passed json
+	update := bson.M{
+		"$set": bson.M{
+			"cow.CowCode": "123",
+		},
+	}
+
+	dbResp, err := c.DB.UpdateOne(context.Background(), bson.M{"_id": cowID}, update)
+	if err != nil {
+		config.ErrorStatus("failed to update cow by ID", http.StatusNotFound, w, err)
+		return
+	}
+
+	b, err := json.Marshal(dbResp) // TODO: get rid of this warning cause its bothering me
+	if err != nil {
+		config.ErrorStatus("failed ot marshal response", http.StatusInternalServerError, w, err)
+		return
+	}
+	w.WriteHeader(http.StatusOK)
+	w.Write(b)
+}