mirror of https://github.com/statping/statping
56 lines
1.0 KiB
Go
56 lines
1.0 KiB
Go
package users
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/hunterlong/statping/database"
|
|
"github.com/hunterlong/statping/utils"
|
|
"time"
|
|
)
|
|
|
|
func DB() database.Database {
|
|
return database.DB().Model(&User{})
|
|
}
|
|
|
|
func Find(id int64) (*User, error) {
|
|
var user User
|
|
db := DB().Where("id = ?", id).Find(user)
|
|
return &user, db.Error()
|
|
}
|
|
|
|
func FindByUsername(username string) (*User, error) {
|
|
var user *User
|
|
db := DB().Where("username = ?", username).Find(user)
|
|
return user, db.Error()
|
|
}
|
|
|
|
func All() []*User {
|
|
var users []*User
|
|
DB().Find(&users)
|
|
return users
|
|
}
|
|
|
|
func (u *User) Create() error {
|
|
u.CreatedAt = time.Now().UTC()
|
|
if u.Password == "" {
|
|
return errors.New("did not supply user password")
|
|
}
|
|
u.Password = utils.HashPassword(u.Password)
|
|
u.ApiKey = utils.NewSHA1Hash(5)
|
|
u.ApiSecret = utils.NewSHA1Hash(10)
|
|
|
|
db := DB().Create(u)
|
|
return db.Error()
|
|
}
|
|
|
|
func (u *User) Update() error {
|
|
u.ApiKey = utils.NewSHA1Hash(5)
|
|
u.ApiSecret = utils.NewSHA1Hash(10)
|
|
db := DB().Update(&u)
|
|
return db.Error()
|
|
}
|
|
|
|
func (u *User) Delete() error {
|
|
db := DB().Delete(u)
|
|
return db.Error()
|
|
}
|