108 lines
2.8 KiB
Go
108 lines
2.8 KiB
Go
package controllers
|
|
|
|
import (
|
|
"context"
|
|
"encoding/base64"
|
|
"time"
|
|
|
|
"github.com/SowinskiBraeden/BugBeGone/database"
|
|
"github.com/SowinskiBraeden/BugBeGone/models"
|
|
"github.com/gofiber/fiber/v2"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
)
|
|
|
|
var userCollection *mongo.Collection = database.OpenCollection(database.Client, "users")
|
|
|
|
func toBase64(b []byte) string {
|
|
return base64.StdEncoding.EncodeToString(b)
|
|
}
|
|
|
|
func Register(c *fiber.Ctx) error {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
firstname := c.FormValue("firstname")
|
|
lastname := c.FormValue("lastname")
|
|
username := c.FormValue("username")
|
|
email := c.FormValue("email")
|
|
password := c.FormValue("password")
|
|
|
|
if email == "" {
|
|
cancel()
|
|
return c.Status(fiber.StatusBadRequest).Render("register", fiber.Map{
|
|
"errorMsg": "Missing Email",
|
|
})
|
|
}
|
|
if password == "" {
|
|
cancel()
|
|
return c.Status(fiber.StatusBadRequest).Render("register", fiber.Map{
|
|
"errorMsg": "Please enter your password",
|
|
})
|
|
}
|
|
if username == "" {
|
|
cancel()
|
|
return c.Status(fiber.StatusBadRequest).Render("register", fiber.Map{
|
|
"errorMsg": "Missing Username",
|
|
})
|
|
}
|
|
|
|
// Check if email or username is previously registered
|
|
count, err := userCollection.CountDocuments(ctx, bson.M{
|
|
"$or": []bson.M{
|
|
bson.M{"email": email},
|
|
bson.M{"username": username},
|
|
}},
|
|
)
|
|
if err != nil {
|
|
cancel()
|
|
return c.Status(fiber.StatusInternalServerError).Render("register", fiber.Map{
|
|
"errorMsg": "Failed to search database",
|
|
})
|
|
}
|
|
if count > 0 {
|
|
cancel()
|
|
return c.Status(fiber.StatusInternalServerError).Render("register", fiber.Map{
|
|
"errorMsg": "An account already exists with that email or username",
|
|
})
|
|
}
|
|
|
|
var user models.User
|
|
if user.CheckPasswordStrength(password) {
|
|
cancel()
|
|
return c.Status(fiber.StatusBadRequest).Render("register", fiber.Map{
|
|
"errorMsg": "Your password must contain at least 1 lowercase, 1 uppercase & 1 special character",
|
|
"username": username,
|
|
"email": email,
|
|
})
|
|
}
|
|
|
|
user.Username = username
|
|
user.Firstname = firstname
|
|
user.Lastname = lastname
|
|
user.Email = email
|
|
user.Password = user.HashPassword(password)
|
|
user.TempPassword = false
|
|
user.ID = primitive.NewObjectID()
|
|
user.Created_at, _ = time.Parse(time.RFC3339, time.Now().Format(time.RFC3339))
|
|
user.Updated_at, _ = time.Parse(time.RFC3339, time.Now().Format(time.RFC3339))
|
|
|
|
_, insertErr := userCollection.InsertOne(ctx, user)
|
|
if insertErr != nil {
|
|
cancel()
|
|
return c.Status(fiber.StatusInternalServerError).Render("register", fiber.Map{
|
|
"errorMsg": "the user could not be inserted",
|
|
})
|
|
}
|
|
|
|
defer cancel()
|
|
|
|
return c.Render("login", fiber.Map{
|
|
"msg": "Successfully registered an account",
|
|
"errorMsg": "",
|
|
})
|
|
}
|
|
|
|
func Login(c *fiber.Ctx) error {
|
|
return c.Status(fiber.StatusNotImplemented).JSON(fiber.Map{})
|
|
}
|