first push

Author: TudorEsan

.DS_Store

@@ -0,0 +1,17 @@
+server/.env
+
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/

.gitignore

@@ -0,0 +1,14 @@
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/

server/.gitignore

@@ -0,0 +1,63 @@
+package config
+
+import (
+	"os"
+
+	"github.com/hashicorp/go-hclog"
+	"github.com/joho/godotenv"
+)
+
+var l = hclog.Default().Named("Config")
+
+func verifyAllEnvVars() {
+	envVars := []string{"MONGO_URL", "JWT_SECRET", "SENDGRID_API_KEY", "DOMAIN_NAME"}
+	for _, envVar := range envVars {
+		if os.Getenv(envVar) == "" {
+			l.Error(("Missing env var: " + envVar))
+		}
+	}
+}
+
+func init() {
+	godotenv.Load(".env")
+	verifyAllEnvVars()
+}
+
+type Config struct {
+	MongoUrl     string
+	JwtSecret    []byte
+	SmtpUsername string
+	SmtpPassword string
+	DomainName   string
+}
+
+func getConfig() *Config {
+	jwtSecret := []byte("secret")
+	mongoUrl := "mongodb://localhost:27017"
+
+	if os.Getenv("TESTING") != "" {
+		goto ReturnConfig
+	}
+
+	if os.Getenv("JWT_SECRET") != "" {
+		jwtSecret = []byte(os.Getenv("JWT_SECRET"))
+	}
+	
+	if os.Getenv("MONGO_URL") != "" {
+		mongoUrl = os.Getenv("MONGO_URL")
+	}
+
+ReturnConfig:
+
+	return &Config{
+		MongoUrl:     mongoUrl,
+		JwtSecret:    jwtSecret,
+		SmtpUsername: os.Getenv("SMTP_USERNAME"),
+		SmtpPassword: os.Getenv("SMTP_PASSWORD"),
+		DomainName:   os.Getenv("DOMAIN_NAME"),
+	}
+}
+
+func New() *Config {
+	return getConfig()
+}

server/config/config.go

@@ -0,0 +1,154 @@
+package controllers
+
+import (
+	"context"
+	"strconv"
+	"time"
+
+	"github.com/TudorEsan/QPerior-Hackhaton/database"
+	"github.com/TudorEsan/QPerior-Hackhaton/helpers"
+	"github.com/TudorEsan/QPerior-Hackhaton/models"
+	"github.com/gin-gonic/gin"
+	"github.com/hashicorp/go-hclog"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+)
+
+type ReservationController struct {
+	l                     hclog.Logger
+	userCollection        *mongo.Collection
+	reservationCollection *mongo.Collection
+	roomsCollection       *mongo.Collection
+}
+
+func NewReservationController(l hclog.Logger, mongoClient *mongo.Client) *ReservationController {
+	userCollection := database.OpenCollection(mongoClient, "users")
+	reservationCollection := database.OpenCollection(mongoClient, "reservations")
+	roomCollection := database.OpenCollection(mongoClient, "rooms")
+	return &ReservationController{l, userCollection, reservationCollection, roomCollection}
+}
+
+func (controller *ReservationController) AddReservation() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		var reservationForm models.ReservationForm
+		if err := c.Bind(&reservationForm); err != nil {
+			controller.l.Error("Could not bind reservation", err)
+			c.JSON(400, gin.H{"message": err.Error()})
+			return
+		}
+
+		reservation := models.NewReservation(reservationForm)
+		userId, _ := primitive.ObjectIDFromHex(c.MustGet("claims").(*models.SignedDetails).Id)
+		reservation.UserId = userId
+
+		// validate room
+		ctx, cancel := context.WithTimeout(context.Background(), time.Second*15)
+		defer cancel()
+
+		var room models.Room
+		err := controller.roomsCollection.FindOne(ctx, bson.M{"_id": reservation.RoomId}).Decode(&room)
+		if err != nil {
+			controller.l.Error("Could not get room", err)
+			c.JSON(400, gin.H{"message": err.Error()})
+			return
+		}
+
+		reservation.Id = primitive.NewObjectID()
+
+		reservationFromThatDay, err := helpers.GetReservationsFromDate(controller.reservationCollection, reservation.From.Year(), int(reservation.From.Month()), reservation.From.Day())
+		if err != nil {
+			controller.l.Error("Could not get reservations from that day", err)
+			c.JSON(400, gin.H{"message": err.Error()})
+			return
+		}
+
+		reservationAvalabile := helpers.IsReservationValid(reservation, reservationFromThatDay)
+		if !reservationAvalabile {
+			controller.l.Error("Reservation is not valid", err)
+			c.JSON(400, gin.H{"message": "Reservation is not valid"})
+			return
+		}
+
+		_, err = controller.reservationCollection.InsertOne(ctx, reservation)
+		if err != nil {
+			controller.l.Error("Could not add reservation", err)
+			c.JSON(400, gin.H{"message": err.Error()})
+			return
+		}
+
+		c.JSON(200, gin.H{"message": "Reservation added", "reservation": reservation})
+		// get all reservations
+		// return reservations
+	}
+}
+
+func (controller *ReservationController) GetReservations() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		ctx, cancel := context.WithTimeout(context.Background(), time.Second*15)
+		defer cancel()
+
+		var reservations []models.Reservation
+		cursor, err := controller.reservationCollection.Find(ctx, bson.M{})
+		if err != nil {
+			controller.l.Error("Could not get reservations", err)
+			c.JSON(400, gin.H{"message": err.Error()})
+			return
+		}
+
+		defer cursor.Close(ctx)
+		for cursor.Next(ctx) {
+			var reservation models.Reservation
+			cursor.Decode(&reservation)
+			reservations = append(reservations, reservation)
+		}
+
+		c.JSON(200, reservations)
+	}
+}
+
+func (controller *ReservationController) FilterReservations() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		var day, month, year int
+		// get date from query
+		day, _ = strconv.Atoi(c.Query("day"))
+		month, _ = strconv.Atoi(c.Query("month"))
+		year, _ = strconv.Atoi(c.Query("year"))
+
+		reservations, err := helpers.GetReservationsFromDate(controller.reservationCollection, year, month, day)
+		if err != nil {
+			controller.l.Error("Could not get reservations", err)
+			c.JSON(400, gin.H{"message": err.Error()})
+			return
+		}
+
+		c.JSON(200, reservations)
+	}
+}
+
+func (controller *ReservationController) GetUserReservations() gin.HandlerFunc {
+	return func(c *gin.Context) {
+		ctx, cancel := context.WithTimeout(context.Background(), time.Second*15)
+		defer cancel()
+
+		claims := c.MustGet("claims").(models.SignedDetails)
+		idS := claims.Id
+
+		userId, err := primitive.ObjectIDFromHex(idS)
+		if err != nil {
+			controller.l.Error("Could not get user id", err)
+			c.JSON(400, gin.H{"message": err.Error()})
+			return
+		}
+
+		var reservation models.Reservation
+		err = controller.reservationCollection.FindOne(ctx, bson.M{"userId": userId}).Decode(&reservation)
+		if err != nil {
+			controller.l.Error("Could not get reservation", err)
+			c.JSON(400, gin.H{"message": err.Error()})
+			return
+		}
+
+		c.JSON(200, reservation)
+	}
+}