Go-TestAPI/controllers/userController.go

package controllers

import (
	"crypto/rand"
	"encoding/base64"
	"github.com/gin-gonic/gin"
	"github.com/go-playground/validator/v10"
	"golang-test/database"
	"golang-test/libs"
	"golang-test/types"
	"golang-test/validators"
	"golang.org/x/crypto/bcrypt"
	"log"
	"net/http"
)

type TokenStruct struct {
	UserID uint
	token  string
}

func RegisterUser(c *gin.Context, register types.RegisterRequest) {
	db := database.Connector()
	validate := validators.Validate
	var token database.Token

	if err := validate.Struct(register); err != nil {
		c.JSON(http.StatusBadRequest, libs.GetValidationErrors(err.(validator.ValidationErrors)))
	}

	if err := db.Where("email =?", register.Email).Error; err == nil {
		c.JSON(http.StatusBadRequest, types.ErrorResponse{Message: "User already exists"})
		return
	}

	hashedPassword, err := bcrypt.GenerateFromPassword([]byte(register.Password), 12)
	if err != nil {
		log.Println(err.Error())
		c.JSON(http.StatusInternalServerError, types.ErrorResponse{Message: err.Error()})
		return
	}

	user := database.User{
		Name:     register.Name,
		Email:    register.Email,
		Password: string(hashedPassword),
	}

	if err := db.Create(&user).Error; err != nil {
		c.JSON(http.StatusInternalServerError, types.ErrorResponse{Message: err.Error()})
		return
	}

	token.Token = generateToken()
	token.UserID = user.ID
	user.Tokens = append(user.Tokens, token)

	c.JSON(http.StatusCreated, types.TokenResponse{Token: token.Token})
}

func LoginUser(c *gin.Context, login types.LoginRequest) {
	var user database.User
	db := database.Connector()
	validate := validators.Validate
	var token database.Token

	if err := validate.Struct(login); err != nil {
		c.JSON(http.StatusBadRequest, libs.GetValidationErrors(err.(validator.ValidationErrors)))
		return
	}

	if err := db.Where("email = ?", login.Email).First(&user).Error; err != nil {
		c.JSON(http.StatusBadRequest, types.ErrorResponse{Message: "Invalid email or password"})
		return
	}

	if err := bcrypt.CompareHashAndPassword([]byte(user.Password), []byte(login.Password)); err != nil {
		c.JSON(http.StatusBadRequest, types.ErrorResponse{Message: "Invalid email or password"})
		return
	}

	token.Token = generateToken()
	token.UserID = user.ID

	user.Tokens = append(user.Tokens, token)

	db.Save(&user)

	c.JSON(http.StatusOK, types.TokenResponse{Token: token.Token})
}

func generateToken() string {
	b := make([]byte, 25)
	if _, err := rand.Read(b); err != nil {
		return ""
	}
	return base64.StdEncoding.EncodeToString(b)
}

func GetUserByToken(token string) (database.User, error) {
	var tokenObject database.Token
	var err error
	var user database.User

	db := database.Connector()
	if err := db.Where("token = ?", token).First(&tokenObject).Error; err != nil {
		return user, err
	}

	if err := db.First(&user, tokenObject.UserID).Error; err != nil {
		return user, err
	}
	return user, err
}

func GetUser(c *gin.Context) {
	token, err := libs.GetTokenFromHeaders(c)

	if err != nil {
		c.JSON(http.StatusUnauthorized, types.ErrorResponse{Message: "Invalid token. Please login"})
	}

	u, err := GetUserByToken(token)
	if err != nil {
		c.JSON(http.StatusUnauthorized, types.ErrorResponse{Message: "Invalid token. Please login"})
		return
	}

	c.JSON(http.StatusOK, u)
}

func EditUserName(c *gin.Context, userName types.EditUserNameRequest) {
	token, err := libs.GetTokenFromHeaders(c)
	if err != nil {
		c.JSON(http.StatusUnauthorized, types.ErrorResponse{Message: "Invalid token. Please login"})
	}

	db := database.Connector()
	validate := validators.Validate

	if err := validate.Struct(userName); err != nil {
		c.JSON(http.StatusBadRequest, libs.GetValidationErrors(err.(validator.ValidationErrors)))
		return
	}

	u, err := GetUserByToken(token)
	if err != nil {
		c.JSON(http.StatusUnauthorized, types.ErrorResponse{Message: "Invalid token. Please login"})
		return
	}

	u.Name = userName.Name

	if err := db.Save(u).Error; err != nil {
		c.JSON(http.StatusInternalServerError, types.ErrorResponse{Message: err.Error()})
		return
	}

	c.JSON(http.StatusOK, types.MessageResponse{Message: "User name successfully updated"})
}