69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package model
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/labstack/echo-contrib/session"
|
|
"github.com/labstack/echo/v4"
|
|
"gitnet.fr/deblan/budget/database/manager"
|
|
"golang.org/x/crypto/bcrypt"
|
|
)
|
|
|
|
type User struct {
|
|
ID uint `gorm:"primaryKey" json:"id"`
|
|
Username string `gorm:"unique;not null" json:"username"`
|
|
Password string `gorm:"not null" json:"-"`
|
|
DisplayName string `gorm:"not null" json:"display_name"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
}
|
|
|
|
func NewUser(username, password, displayName string) *User {
|
|
user := User{
|
|
Username: username,
|
|
DisplayName: displayName,
|
|
}
|
|
|
|
user.UpdatePassword(password)
|
|
|
|
return &user
|
|
}
|
|
|
|
func (u *User) UpdatePassword(password string) {
|
|
hashedPassword, _ := bcrypt.GenerateFromPassword([]byte(password), bcrypt.DefaultCost)
|
|
|
|
u.Password = string(hashedPassword)
|
|
}
|
|
|
|
func (u *User) HasPassword(password string) bool {
|
|
return bcrypt.CompareHashAndPassword([]byte(u.Password), []byte(password)) == nil
|
|
}
|
|
|
|
func LoadSessionUser(c echo.Context) *User {
|
|
sess, err := session.Get("session", c)
|
|
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
|
|
value := sess.Values["user"]
|
|
|
|
if value == nil {
|
|
return nil
|
|
}
|
|
|
|
id := value.(uint)
|
|
|
|
var count int64
|
|
db := manager.Get()
|
|
db.Db.Model(User{}).Where("id = ?", id).Count(&count)
|
|
|
|
if count == 0 {
|
|
return nil
|
|
}
|
|
|
|
var user User
|
|
manager.Get().Db.Model(User{}).Where("id = ?", id).Find(&user)
|
|
|
|
return &user
|
|
}
|