consent_controller.go

package controllers

import (
	"net/http"
	"time"

	"github.com/amundlrohne/dcs-medication-sharing/services/consent/configs"
	"github.com/amundlrohne/dcs-medication-sharing/services/consent/models"
	"github.com/amundlrohne/dcs-medication-sharing/services/consent/responses"

	"github.com/go-playground/validator/v10"
	"github.com/labstack/echo/v4"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/primitive"
	"go.mongodb.org/mongo-driver/mongo"
	"golang.org/x/net/context"
)

var consentCollection *mongo.Collection = configs.GetCollection(configs.DB, "consents")
var validate = validator.New()

func CreateConsent(c echo.Context) error {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	var consent models.Consent
	defer cancel()

	//validate the request body
	if err := c.Bind(&consent); err != nil {
		return c.JSON(http.StatusBadRequest, responses.ConsentResponse{Status: http.StatusBadRequest, Message: "error", Data: &echo.Map{"data": err.Error()}})
	}

	//use the validator library to validate required fields
	if validationErr := validate.Struct(&consent); validationErr != nil {
		return c.JSON(http.StatusBadRequest, responses.ConsentResponse{Status: http.StatusBadRequest, Message: "error", Data: &echo.Map{"data": validationErr.Error()}})
	}

	newConsent := models.Consent{
		ID:            primitive.NewObjectID(),
		ToPublicKey:   consent.ToPublicKey,
		FromPublicKey: consent.FromPublicKey,
		ExpDate:       consent.ExpDate,
		DateCreated:   consent.DateCreated,
	}

	result, err := consentCollection.InsertOne(ctx, newConsent)
	if err != nil {
		return c.JSON(http.StatusInternalServerError, responses.ConsentResponse{Status: http.StatusInternalServerError, Message: "error", Data: &echo.Map{"data": err.Error()}})
	}

	return c.JSON(http.StatusCreated, responses.ConsentResponse{Status: http.StatusCreated, Message: "success", Data: &echo.Map{"data": result}})
}

func GetConsentRequests(c echo.Context) error {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	fromPublicKey := c.Param("from_public_key")
	var consents []models.Consent
	defer cancel()

	// objId, _ := primitive.ObjectIDFromHex(fromPublicKey)

	results, err := consentCollection.Find(ctx, bson.M{"frompublickey": fromPublicKey})

	if err != nil {
		return c.JSON(http.StatusInternalServerError, responses.ConsentResponse{Status: http.StatusInternalServerError, Message: "error", Data: &echo.Map{"data": err.Error()}})
	}

	defer results.Close(ctx)
	for results.Next(ctx) {
		var singleConsent models.Consent
		if err = results.Decode(&singleConsent); err != nil {
			return c.JSON(http.StatusInternalServerError, responses.ConsentResponse{Status: http.StatusInternalServerError, Message: "error", Data: &echo.Map{"data": err.Error()}})
		}

		consents = append(consents, singleConsent)
	}

	return c.JSON(http.StatusOK, responses.ConsentResponse{Status: http.StatusOK, Message: "success", Data: &echo.Map{"data": consents}})
}

func GetConsentIncoming(c echo.Context) error {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	toPublicKey := c.Param("to_public_key")
	var consents []models.Consent
	defer cancel()

	// objId, _ := primitive.ObjectIDFromHex(fromPublicKey)

	results, err := consentCollection.Find(ctx, bson.M{"topublickey": toPublicKey})

	if err != nil {
		return c.JSON(http.StatusInternalServerError, responses.ConsentResponse{Status: http.StatusInternalServerError, Message: "error", Data: &echo.Map{"data": err.Error()}})
	}

	defer results.Close(ctx)
	for results.Next(ctx) {
		var singleConsent models.Consent
		if err = results.Decode(&singleConsent); err != nil {
			return c.JSON(http.StatusInternalServerError, responses.ConsentResponse{Status: http.StatusInternalServerError, Message: "error", Data: &echo.Map{"data": err.Error()}})
		}

		consents = append(consents, singleConsent)
	}

	return c.JSON(http.StatusOK, responses.ConsentResponse{Status: http.StatusOK, Message: "success", Data: &echo.Map{"data": consents}})
}

func GetAllConsents(c echo.Context) error {
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	var consents []models.Consent
	defer cancel()

	results, err := consentCollection.Find(ctx, bson.M{})

	if err != nil {
		return c.JSON(http.StatusInternalServerError, responses.ConsentResponse{Status: http.StatusInternalServerError, Message: "error", Data: &echo.Map{"data": err.Error()}})
	}

	//reading from the db in an optimal way
	defer results.Close(ctx)
	for results.Next(ctx) {
		var singleConsent models.Consent
		if err = results.Decode(&singleConsent); err != nil {
			return c.JSON(http.StatusInternalServerError, responses.ConsentResponse{Status: http.StatusInternalServerError, Message: "error", Data: &echo.Map{"data": err.Error()}})
		}

		consents = append(consents, singleConsent)
	}

	return c.JSON(http.StatusOK, responses.ConsentResponse{Status: http.StatusOK, Message: "success", Data: &echo.Map{"data": consents}})
}