synctv/internal/model/relation.go

package model

import "gorm.io/gorm"

type Role uint32

const (
	RoleBanned Role = iota + 1
	RoleUser
	RoleAdmin
	RoleCreater
)

type Permission uint32

const (
	CanRenameRoom Permission = 1 << iota
	CanSetAdmin
	CanSetRoomPassword
	CanSetRoomSetting
	CanSetUserPermission
	CanSetUserPassword
	CanCreateUserPublishKey
	CanEditUserMovies
	CanDeleteUserMovies
	CanCreateMovie
	CanChangeCurrentMovie
	CanChangeMovieStatus
	CanDeleteRoom
	AllPermissions Permission = 0xffffffff
)

const (
	DefaultPermissions = CanCreateMovie | CanChangeCurrentMovie | CanChangeMovieStatus
)

func (p Permission) Has(permission Permission) bool {
	return p&permission == permission
}

type RoomUserRelation struct {
	gorm.Model
	UserID      uint `gorm:"not null;uniqueIndex:idx_user_room"`
	RoomID      uint `gorm:"not null;uniqueIndex:idx_user_room"`
	Role        Role `gorm:"not null"`
	Permissions Permission
}

func (r *RoomUserRelation) HasPermission(permission Permission) bool {
	switch r.Role {
	case RoleCreater:
		return true
	case RoleAdmin:
		return r.Permissions.Has(permission) // TODO: add more permissions
	case RoleUser:
		return r.Permissions.Has(permission)
	default:
		return false
	}
}