mirror of https://github.com/gophish/gophish
42 lines
1.2 KiB
Go
42 lines
1.2 KiB
Go
package models
|
|
|
|
// User represents the user model for gophish.
|
|
type User struct {
|
|
Id int64 `json:"id"`
|
|
Username string `json:"username" sql:"not null;unique"`
|
|
Hash string `json:"-"`
|
|
ApiKey string `json:"api_key" sql:"not null;unique"`
|
|
Role Role `json:"role" gorm:"association_autoupdate:false;association_autocreate:false"`
|
|
RoleID int64 `json:"-"`
|
|
}
|
|
|
|
// GetUser returns the user that the given id corresponds to. If no user is found, an
|
|
// error is thrown.
|
|
func GetUser(id int64) (User, error) {
|
|
u := User{}
|
|
err := db.Preload("Role").Where("id=?", id).First(&u).Error
|
|
return u, err
|
|
}
|
|
|
|
// GetUserByAPIKey returns the user that the given API Key corresponds to. If no user is found, an
|
|
// error is thrown.
|
|
func GetUserByAPIKey(key string) (User, error) {
|
|
u := User{}
|
|
err := db.Preload("Role").Where("api_key = ?", key).First(&u).Error
|
|
return u, err
|
|
}
|
|
|
|
// GetUserByUsername returns the user that the given username corresponds to. If no user is found, an
|
|
// error is thrown.
|
|
func GetUserByUsername(username string) (User, error) {
|
|
u := User{}
|
|
err := db.Preload("Role").Where("username = ?", username).First(&u).Error
|
|
return u, err
|
|
}
|
|
|
|
// PutUser updates the given user
|
|
func PutUser(u *User) error {
|
|
err := db.Save(u).Error
|
|
return err
|
|
}
|