statping/types/users/database.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()
}