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.
106 lines
2.9 KiB
Go
106 lines
2.9 KiB
Go
package model
|
|
|
|
import (
|
|
"fmt"
|
|
"math/rand/v2"
|
|
"time"
|
|
|
|
"github.com/synctv-org/synctv/utils"
|
|
"github.com/zijiren233/stream"
|
|
"golang.org/x/crypto/bcrypt"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type Role uint8
|
|
|
|
const (
|
|
RoleBanned Role = 1
|
|
RolePending Role = 2
|
|
RoleUser Role = 3
|
|
RoleAdmin Role = 4
|
|
RoleRoot Role = 5
|
|
)
|
|
|
|
func (r Role) String() string {
|
|
switch r {
|
|
case RoleBanned:
|
|
return "banned"
|
|
case RolePending:
|
|
return "pending"
|
|
case RoleUser:
|
|
return "user"
|
|
case RoleAdmin:
|
|
return "admin"
|
|
case RoleRoot:
|
|
return "root"
|
|
default:
|
|
return "unknown"
|
|
}
|
|
}
|
|
|
|
type User struct {
|
|
ID string `gorm:"primaryKey;type:char(32)" json:"id"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
Username string `gorm:"not null;uniqueIndex;type:varchar(32)"`
|
|
Email EmptyNullString `gorm:"type:varchar(64);uniqueIndex"`
|
|
HashedPassword []byte `gorm:"not null"`
|
|
BilibiliVendor *BilibiliVendor `gorm:"foreignKey:UserID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE"`
|
|
Movies []*Movie `gorm:"foreignKey:CreatorID;constraint:OnUpdate:CASCADE,OnDelete:SET NULL"`
|
|
UserProviders []*UserProvider `gorm:"foreignKey:UserID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE"`
|
|
RoomMembers []*RoomMember `gorm:"foreignKey:UserID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE"`
|
|
Rooms []*Room `gorm:"foreignKey:CreatorID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE"`
|
|
AlistVendor []*AlistVendor `gorm:"foreignKey:UserID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE"`
|
|
EmbyVendor []*EmbyVendor `gorm:"foreignKey:UserID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE"`
|
|
Role Role `gorm:"not null;default:2"`
|
|
RegisteredByProvider bool `gorm:"not null;default:false"`
|
|
RegisteredByEmail bool `gorm:"not null;default:false"`
|
|
autoAddUsernameSuffix bool
|
|
}
|
|
|
|
func (u *User) EnableAutoAddUsernameSuffix() {
|
|
u.autoAddUsernameSuffix = true
|
|
}
|
|
|
|
func (u *User) DisableAutoAddUsernameSuffix() {
|
|
u.autoAddUsernameSuffix = false
|
|
}
|
|
|
|
func (u *User) CheckPassword(password string) bool {
|
|
return bcrypt.CompareHashAndPassword(u.HashedPassword, stream.StringToBytes(password)) == nil
|
|
}
|
|
|
|
func (u *User) BeforeCreate(tx *gorm.DB) error {
|
|
if u.autoAddUsernameSuffix {
|
|
var existingUser User
|
|
err := tx.Select("username").Where("username = ?", u.Username).First(&existingUser).Error
|
|
if err == nil {
|
|
u.Username = fmt.Sprintf("%s#%d", u.Username, rand.IntN(9999))
|
|
}
|
|
}
|
|
if u.ID == "" {
|
|
u.ID = utils.SortUUID()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (u *User) IsRoot() bool {
|
|
return u.Role == RoleRoot
|
|
}
|
|
|
|
func (u *User) IsAdmin() bool {
|
|
return u.Role == RoleAdmin || u.IsRoot()
|
|
}
|
|
|
|
func (u *User) IsUser() bool {
|
|
return u.Role == RoleUser || u.IsAdmin()
|
|
}
|
|
|
|
func (u *User) IsPending() bool {
|
|
return u.Role == RolePending
|
|
}
|
|
|
|
func (u *User) IsBanned() bool {
|
|
return u.Role == RoleBanned
|
|
}
|