User roles.

[perso/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Front.git] / db / user.go

package db

import (
	"golang.org/x/crypto/bcrypt"
)

type UserStatus uint8

const (
	Confirmed UserStatus = iota + 1
	AwaitingConfirmation
)

type UserRole string

const RoleUser UserRole = "user"
const RoleAdmin UserRole = "admin"

type User struct {
	Id           int64
	Role         UserRole
	Email        string
	PasswordHash string
	OtpSecret    string
	IsOtpSetup   bool
	Status       UserStatus
}

func HashPassword(password string) (string, error) {
	b, err := bcrypt.GenerateFromPassword([]byte(password), 10)

	return string(b), err
}

func ValidatePassword(password string, hash string) error {
	return bcrypt.CompareHashAndPassword([]byte(hash), []byte(password))
}

func InsertUser(user *User) error {
	return DB.Insert(user)
}

func ConfirmUserByEmail(email string) error {
	_, err := DB.Model(&User{}).Set("status=?", Confirmed).Where("email=?", email).Returning("*").Update()

	return err
}

func GetUserById(id int64) (*User, error) {
	user := User{Id: id}

	err := DB.Select(&user)

	return &user, err
}

func GetUserByEmail(email string) (*User, error) {
	var users []User

	err := DB.Model(&users).Where("email = ?", email).Select()

	if err != nil {
		return nil, err
	}

	if len(users) == 0 {
		return nil, nil
	}

	return &users[0], nil
}

func SetOtpSecret(user *User, secret string, temporary bool) error {
	user.OtpSecret = secret
	user.IsOtpSetup = !temporary

	return DB.Update(user)
}

func SetPassword(user *User, password string) error {
	var err error
	user.PasswordHash, err = HashPassword(password)
	if err != nil {
		return err
	}

	return DB.Update(user)
}

func SetUserStatus(user *User, status UserStatus) error {
	user.Status = status

	return DB.Update(user)
}