Samuel Pua
7cf10b07d4
Some checks failed
ktm-booking-bot/ktm-booking-bot/pipeline/head Something is wrong with the build of this commit
72 lines
1.6 KiB
Go
72 lines
1.6 KiB
Go
package user
|
|
|
|
import (
|
|
"errors"
|
|
"log"
|
|
|
|
"github.com/google/uuid"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
const (
|
|
BCRYPTCOST = 12
|
|
)
|
|
|
|
func (env *Env) createUser(username string, password string) (*User, error) {
|
|
var createdUser User
|
|
|
|
passwordByte := []byte(password)
|
|
createdHashBytes, err := bcrypt.GenerateFromPassword(passwordByte, BCRYPTCOST)
|
|
|
|
if err != nil {
|
|
return nil, errors.New("failed to generate bcrypt")
|
|
}
|
|
|
|
createdUser.Username = username
|
|
createdUser.PasswordBcrypt = string(createdHashBytes)
|
|
return &createdUser, nil
|
|
}
|
|
|
|
func (env *Env) registerUser(username string, password string) (*User, error) {
|
|
newUser, err := env.createUser(username, password)
|
|
|
|
if err != nil {
|
|
log.Println(err)
|
|
return nil, errors.New("failed to register user")
|
|
}
|
|
|
|
// Check existing username
|
|
var checkUser User
|
|
env.DB.Where(&User{Username: username}).First(&checkUser)
|
|
if checkUser.ID != uuid.Nil {
|
|
log.Println(err)
|
|
return nil, errors.New("user already exists")
|
|
}
|
|
|
|
err = env.DB.Create(newUser).Error
|
|
if err != nil {
|
|
log.Println(err)
|
|
return nil, errors.New("failed write to database")
|
|
}
|
|
|
|
return newUser, nil
|
|
}
|
|
|
|
func (env *Env) checkLogin(username string, password string) (*User, error) {
|
|
var currUser User
|
|
env.DB.Preload("Profile").Where(&User{Username: username}).First(&currUser)
|
|
|
|
// Prevent username enum by parsing password
|
|
if currUser.ID == uuid.Nil {
|
|
bcrypt.GenerateFromPassword([]byte{}, BCRYPTCOST)
|
|
return nil, errors.New("invalid username or password")
|
|
}
|
|
|
|
err := bcrypt.CompareHashAndPassword([]byte(currUser.PasswordBcrypt), []byte(password))
|
|
if err != nil {
|
|
return nil, errors.New("invalid username or password")
|
|
} else {
|
|
return &currUser, nil
|
|
}
|
|
}
|