mirror of
https://git.rwth-aachen.de/acs/public/villas/web-backend-go/
synced 2025-03-30 00:00:12 +01:00
76 lines
1.8 KiB
Go
76 lines
1.8 KiB
Go
package user
|
|
|
|
import (
|
|
"fmt"
|
|
"git.rwth-aachen.de/acs/public/villas/web-backend-go/database"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
const bcryptCost = 10
|
|
|
|
// This is ugly but no other way to keep each model on the corresponding
|
|
// package since we have circular dependencies. Methods of a type must
|
|
// live in the same package.
|
|
type User struct {
|
|
database.User // check golang embedding types
|
|
}
|
|
|
|
func (u *User) save() error {
|
|
db := database.GetDB()
|
|
err := db.Create(u).Error
|
|
return err
|
|
}
|
|
|
|
func (u *User) remove() error {
|
|
db := database.GetDB()
|
|
err := db.Delete(u).Error
|
|
return err
|
|
}
|
|
|
|
func (u *User) ByUsername(username string) error {
|
|
db := database.GetDB()
|
|
err := db.Find(u, "Username = ?", username).Error
|
|
return err
|
|
}
|
|
|
|
func (u *User) ByID(id uint) error {
|
|
db := database.GetDB()
|
|
err := db.Find(u, id).Error
|
|
return err
|
|
}
|
|
|
|
func (u *User) setPassword(password string) error {
|
|
if len(password) == 0 {
|
|
return fmt.Errorf("Password cannot be empty")
|
|
}
|
|
newPassword, err :=
|
|
bcrypt.GenerateFromPassword([]byte(password), bcryptCost)
|
|
if err != nil {
|
|
return fmt.Errorf("Failed to generate hash from password")
|
|
}
|
|
u.Password = string(newPassword)
|
|
return nil
|
|
}
|
|
|
|
func (u *User) validatePassword(password string) error {
|
|
loginPassword := []byte(password)
|
|
hashedPassword := []byte(u.Password)
|
|
return bcrypt.CompareHashAndPassword(hashedPassword, loginPassword)
|
|
}
|
|
|
|
func (u *User) update(updatedUser User) error {
|
|
|
|
u.Username = updatedUser.Username
|
|
u.Password = updatedUser.Password
|
|
u.Mail = updatedUser.Mail
|
|
u.Role = updatedUser.Role
|
|
u.Active = updatedUser.Active
|
|
|
|
db := database.GetDB()
|
|
err := db.Model(u).Update(updatedUser).Error
|
|
|
|
// extra update for bool active since it is ignored if false
|
|
err = db.Model(u).Updates(map[string]interface{}{"Active": updatedUser.Active}).Error
|
|
|
|
return err
|
|
}
|