opengist/internal/models/user.go

95 lines
2.2 KiB
Go
Raw Normal View History

2023-03-14 15:22:52 +00:00
package models
import "gorm.io/gorm"
2023-03-14 15:22:52 +00:00
type User struct {
ID uint `gorm:"primaryKey"`
Username string `form:"username" gorm:"uniqueIndex" validate:"required,max=24,alphanum,notreserved"`
Password string `form:"password" validate:"required"`
IsAdmin bool
CreatedAt int64
Gists []Gist `gorm:"constraint:OnUpdate:CASCADE,OnDelete:CASCADE;foreignKey:UserID"`
SSHKeys []SSHKey `gorm:"foreignKey:UserID"`
Liked []Gist `gorm:"many2many:likes;constraint:OnUpdate:CASCADE,OnDelete:CASCADE"`
}
func DoesUserExists(userName string, count *int64) error {
return db.Table("users").
Where("username like ?", userName).
Count(count).Error
}
func GetAllUsers(offset int) ([]*User, error) {
var all []*User
err := db.
Limit(11).
Offset(offset * 10).
Order("id asc").
Find(&all).Error
return all, err
}
func GetLoginUser(user *User) error {
return db.
Where("username like ?", user.Username).
First(&user).Error
}
func GetLoginUserById(user *User) error {
return db.
Where("id = ?", user.ID).
First(&user).Error
}
func CreateUser(user *User) error {
return db.Create(&user).Error
}
func DeleteUserByID(userid string) error {
// Decrement likes counter for all gists liked by this user
// The likes will be automatically deleted by the foreign key constraint
err := db.Model(&Gist{}).
Omit("updated_at").
Where("id IN (?)", db.
Select("gist_id").
Table("likes").
Where("user_id = ?", userid),
).
UpdateColumn("nb_likes", gorm.Expr("nb_likes - 1")).
Error
if err != nil {
return err
}
2023-03-14 15:22:52 +00:00
return db.Delete(&User{}, "id = ?", userid).Error
}
func SetAdminUser(user *User) error {
return db.Model(&user).Update("is_admin", true).Error
}
func GetUserBySSHKeyID(sshKeyId uint) (*User, error) {
user := new(User)
err := db.
Preload("SSHKeys").
Joins("join ssh_keys on users.id = ssh_keys.user_id").
Where("ssh_keys.id = ?", sshKeyId).
First(&user).Error
return user, err
}
func UserHasLikedGist(user *User, gist *Gist) (bool, error) {
association := db.Model(&gist).Where("user_id = ?", user.ID).Association("Likes")
if association.Error != nil {
return false, association.Error
}
if association.Count() == 0 {
return false, nil
}
return true, nil
}