mirror of https://github.com/synctv-org/synctv
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package model
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type Role uint8
|
|
|
|
const (
|
|
RoleBanned Role = iota
|
|
RoleUser
|
|
RoleAdmin
|
|
)
|
|
|
|
func (r Role) String() string {
|
|
switch r {
|
|
case RoleBanned:
|
|
return "banned"
|
|
case RoleUser:
|
|
return "user"
|
|
case RoleAdmin:
|
|
return "admin"
|
|
default:
|
|
return "unknown"
|
|
}
|
|
}
|
|
|
|
type User struct {
|
|
ID uint `gorm:"primarykey"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
Providers []UserProvider `gorm:"foreignKey:UserID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE"`
|
|
Username string `gorm:"not null;uniqueIndex"`
|
|
Role Role `gorm:"not null"`
|
|
GroupUserRelations []RoomUserRelation `gorm:"foreignKey:UserID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE"`
|
|
Rooms []Room `gorm:"foreignKey:CreatorID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE"`
|
|
Movies []Movie `gorm:"foreignKey:CreatorID;constraint:OnUpdate:CASCADE,OnDelete:SET NULL"`
|
|
}
|
|
|
|
func (u *User) BeforeCreate(tx *gorm.DB) error {
|
|
var existingUser User
|
|
err := tx.Where("username = ?", u.Username).First(&existingUser).Error
|
|
if err == nil {
|
|
u.Username = fmt.Sprintf("%s#%d", u.Username, rand.Intn(9999))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (u *User) IsAdmin() bool {
|
|
return u.Role == RoleAdmin
|
|
}
|
|
|
|
func (u *User) IsBanned() bool {
|
|
return u.Role == RoleBanned
|
|
}
|