Capstone/ui/controllers/auth.go

package controllers

import (
	"crypto/sha256"
	"encoding/hex"
	"fmt"
	"regexp"
	"time"

	"git.preston-baxter.com/Preston_PLB/capstone/frontend-service/config"
	"git.preston-baxter.com/Preston_PLB/capstone/frontend-service/db/models"
	"git.preston-baxter.com/Preston_PLB/capstone/frontend-service/templates"
	"github.com/gin-gonic/gin"
	"github.com/golang-jwt/jwt/v5"
	"golang.org/x/crypto/bcrypt"
)

const AUTH_COOKIE_NAME = "authorization"

var VALIDATE_EMAIL_REGEX = regexp.MustCompile(`^[\w-\.]+@([\w-]+\.)+[\w-]{2,4}$`)

type LoginPostBody struct {
	Email    string `json:"email"`
	Password string `json:"password"`
}

func SignUpHandler(c *gin.Context) {
	//get uname and password.
	conf := config.Config()
	reqBody := &LoginPostBody{}
	c.Request.ParseForm()
	reqBody.Email = c.Request.FormValue("email")
	reqBody.Password = c.Request.FormValue("password")

	if reqBody.Email == "" {
		log.Warn("Request contained no email")
		renderTempl(c, templates.SignupPage("Please provide an email"))
		return
	}

	if reqBody.Password == "" {
		log.Warn("Request contained no password")
		renderTempl(c, templates.SignupPage("Please provide a password"))
		return
	}

	//Verify username and password
	if ok := VALIDATE_EMAIL_REGEX.Match([]byte(reqBody.Email)); !ok {
		log.Warnf("User provided email field is not valid: %s", reqBody.Email)
		renderTempl(c, templates.SignupPage("You eamil is invalid. Please try again"))
		return
	}

	user, err := mongo.FindUserByEmail(reqBody.Email)
	if err != nil {
		log.WithError(err).Errorf("Failed to lookup user: %s", reqBody.Email)
		renderTempl(c, templates.SignupPage("Error occured. Please try again later"))
		return
	}

	if user != nil {
		log.Warnf("User: %s, already exists", reqBody.Email)
		renderTempl(c, templates.SignupPage(fmt.Sprintf("user already exists for %s", reqBody.Email)))
		return
	}

	//create new user
	user = &models.User{}

	passHash, err := bcrypt.GenerateFromPassword([]byte(reqBody.Password), 10)
	if err != nil {
		log.WithError(err).Errorf("Passowrd hash failed for user: %s", reqBody.Email)
		renderTempl(c, templates.SignupPage("Signup failed. Please try again later"))
		return
	}

	user.PassowrdHash = string(passHash)
	user.Email = reqBody.Email

	err = mongo.SaveModel(user)
	if err != nil {
		log.WithError(err).Errorf("Failed to write user to DB for user: %s", reqBody.Email)
		renderTempl(c, templates.SignupPage("Signup failed. Please try again later"))
		return
	}

	now := time.Now().Unix()
	exp := time.Now().Add(12 * time.Hour).Unix()
	//build jwt
	token := jwt.NewWithClaims(jwt.SigningMethodHS256,
		AuthClaims{
			Subject:   user.MongoId().Hex(),
			Expires:   exp,
			IssuedAt:  now,
			NotBefore: now,
			Issuer:    "capstone.preston-baxter.com",
			Audience:  "capstone.preston-baxter.com",
		},
	)

	jwtStr, err := token.SignedString([]byte(conf.JwtSecret))
	if err != nil {
		log.WithError(err).Errorf("Failed to encode jwt for user: %s", reqBody.Email)
		renderTempl(c, templates.SignupPage("Signup failed. Please try again later"))
		return
	}

	//store jwt as cookie
	//TODO: Make sure set secure for prd deployment
	c.SetCookie(AUTH_COOKIE_NAME, jwtStr, 3600*24, "", "", false, true)

	c.Redirect(302, "/dashboard")
}

func LoginHandler(c *gin.Context) {
	//get uname and password.
	conf := config.Config()
	reqBody := &LoginPostBody{}
	c.Request.ParseForm()
	reqBody.Email = c.Request.FormValue("email")
	reqBody.Password = c.Request.FormValue("password")

	if reqBody.Email == "" {
		log.Warn("Request contained no email")
		renderTempl(c, templates.LoginPage("Please provide an email"))
		return
	}

	if reqBody.Password == "" {
		log.Warn("Request contained no password")
		renderTempl(c, templates.LoginPage("Please provide a password"))
		return
	}

	//Verify username and password
	if ok := VALIDATE_EMAIL_REGEX.Match([]byte(reqBody.Email)); !ok {
		log.Warnf("User provided email field is not valid: %s", reqBody.Email)
		renderTempl(c, templates.SignupPage("You eamil is invalid. Please try again"))
		return
	}

	user, err := mongo.FindUserByEmail(reqBody.Email)
	if err != nil {
		log.WithError(err).Errorf("Failed to lookup user: %s", reqBody.Email)
		renderTempl(c, templates.LoginPage(err.Error()))
		return
	}


	if user == nil {
		log.Warnf("No user was found for: %s", reqBody.Email)
		renderTempl(c, templates.LoginPage(fmt.Sprintf("No user found for %s", reqBody.Email)))
		return
	}

	if err := bcrypt.CompareHashAndPassword([]byte(user.PassowrdHash), []byte(reqBody.Password)); err != nil {
		log.Warnf("Password does not match for user: %s", reqBody.Email)
		renderTempl(c, templates.LoginPage("Email or password are incorrect"))
		return
	}

	now := time.Now().Unix()
	exp := time.Now().Add(12 * time.Hour).Unix()
	//build jwt
	token := jwt.NewWithClaims(jwt.SigningMethodHS256,
		AuthClaims{
			Subject:   user.MongoId().Hex(),
			Expires:   exp,
			IssuedAt:  now,
			NotBefore: now,
			Issuer:    "capstone.preston-baxter.com",
			Audience:  "capstone.preston-baxter.com",
		},
	)

	jwtStr, err := token.SignedString([]byte(conf.JwtSecret))
	if err != nil {
		renderTempl(c, templates.LoginPage("An error occured. Please try again later"))
	}

	//store jwt as cookie
	var secure bool
	if conf.Env == "dev" {
		secure = false
	} else {
		secure = true
	}
	c.SetCookie(AUTH_COOKIE_NAME, jwtStr, 3600*24, "", "", secure, true)

	c.Redirect(302, "/dashboard")
}

func LogoutHandler(c *gin.Context) {
	conf := config.Config()

	var secure bool
	if conf.Env == "dev" {
		secure = false
	} else {
		secure = true
	}
	c.SetCookie("authorization", "", 3600*24, "", "", secure, true)

	c.Redirect(302, "/login")
}

func getAuthHash(c *gin.Context) string {
	jwtToken, err := c.Cookie(AUTH_COOKIE_NAME)
	if err != nil {
		return ""
	}

	h := sha256.New()
	h.Write([]byte(jwtToken))

	return hex.EncodeToString(h.Sum(nil))
}