mirror of
https://git.rwth-aachen.de/acs/public/villas/web-backend-go/
synced 2025-03-30 00:00:12 +01:00
f3a7ed0e61
- rename common package to database - move all code not related to database to new helper package - add more test in database package to improve code coverage - add a new (own) package for AMQP client
78 lines
1.8 KiB
Go
78 lines
1.8 KiB
Go
package user
|
|
|
|
import (
|
|
"fmt"
|
|
"git.rwth-aachen.de/acs/public/villas/villasweb-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
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (u *User) ByID(id uint) error {
|
|
db := database.GetDB()
|
|
err := db.Find(u, id).Error
|
|
if err != nil {
|
|
return fmt.Errorf("User with id=%v does not exist", id)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
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 {
|
|
|
|
// TODO: if the field is empty string member shouldn't be updated
|
|
u.Username = updatedUser.Username
|
|
u.Password = updatedUser.Password
|
|
u.Mail = updatedUser.Mail
|
|
u.Role = updatedUser.Role
|
|
|
|
db := database.GetDB()
|
|
err := db.Model(u).Update(updatedUser).Error
|
|
return err
|
|
}
|