forked from nixcn/nixcn-cms
123 lines
3.1 KiB
Go
123 lines
3.1 KiB
Go
package data
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/datatypes"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
// Permission Level
|
|
// Banned User: 0
|
|
// Normal User: 10
|
|
// Admin User: 20
|
|
// Super User: 30
|
|
|
|
type User struct {
|
|
Id uint `json:"id" gorm:"primarykey;autoincrement"`
|
|
UUID uuid.UUID `json:"uuid" gorm:"type:uuid;uniqueindex;not null"`
|
|
UserId uuid.UUID `json:"user_id" gorm:"type:uuid;uniqueindex;not null"`
|
|
Email string `json:"email" gorm:"type:varchar(255);uniqueindex;not null"`
|
|
Type string `json:"type" gorm:"type:varchar(32);index;not null"`
|
|
Nickname string `json:"nickname"`
|
|
Subtitle string `json:"subtitle"`
|
|
Avatar string `json:"avatar"`
|
|
Checkin datatypes.JSONMap `json:"checkin"`
|
|
JoinedEvent datatypes.JSONSlice[uuid.UUID] `json:"joined_event"`
|
|
PermissionLevel uint `json:"permission_level" gorm:"default:10;not null"`
|
|
}
|
|
|
|
func (self *User) GetByEmail(email string) error {
|
|
if err := Database.Where("email = ?", email).First(&self).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (self *User) GetByUserId(userId uuid.UUID) error {
|
|
if err := Database.Where("user_id = ?", userId).First(&self).Error; err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (self *User) UpdateCheckin(userId uuid.UUID, eventId uuid.UUID, time time.Time) error {
|
|
return Database.Transaction(func(tx *gorm.DB) error {
|
|
if err := tx.
|
|
Clauses(clause.Locking{Strength: "UPDATE"}).
|
|
Where("user_id = ?", userId).
|
|
First(self).Error; err != nil {
|
|
return err // if error then rollback
|
|
}
|
|
|
|
self.Checkin = datatypes.JSONMap{eventId.String(): time}
|
|
|
|
if err := tx.Save(self).Error; err != nil {
|
|
return err // rollback
|
|
}
|
|
|
|
return nil // commit
|
|
})
|
|
}
|
|
|
|
func (self *User) Create() error {
|
|
return Database.Transaction(func(tx *gorm.DB) error {
|
|
if self.UUID == uuid.Nil {
|
|
self.UUID = uuid.New()
|
|
}
|
|
if self.UserId == uuid.Nil {
|
|
self.UserId = uuid.New()
|
|
}
|
|
|
|
if err := tx.Create(self).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
}
|
|
|
|
type UserUpdateInput struct {
|
|
Email *string `json:"email"`
|
|
Nickname *string `json:"nickname"`
|
|
Subtitle *string `json:"subtitle"`
|
|
Avatar *string `json:"avatar"`
|
|
Type *string `json:"type"`
|
|
}
|
|
|
|
func (self *User) UpdateByUserID(userID uuid.UUID, in *UserUpdateInput) error {
|
|
return Database.Transaction(func(tx *gorm.DB) error {
|
|
if err := tx.
|
|
Where("user_id = ?", userID).
|
|
First(self).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
updates := map[string]any{}
|
|
|
|
if in.Email != nil {
|
|
updates["email"] = *in.Email
|
|
}
|
|
if in.Nickname != nil {
|
|
updates["nickname"] = *in.Nickname
|
|
}
|
|
if in.Subtitle != nil {
|
|
updates["subtitle"] = *in.Subtitle
|
|
}
|
|
if in.Avatar != nil {
|
|
updates["avatar"] = *in.Avatar
|
|
}
|
|
if in.Type != nil {
|
|
updates["type"] = *in.Type
|
|
}
|
|
|
|
if len(updates) == 0 {
|
|
return nil
|
|
}
|
|
|
|
return tx.Model(self).Updates(updates).Error
|
|
})
|
|
}
|