synctv/internal/model/relation.go

package model

import "time"

type RoomRole uint32

const (
	RoomRoleBanned RoomRole = iota + 1
	RoomRoleUser
	RoomRoleCreator
)

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 {
	CreatedAt   time.Time
	UpdatedAt   time.Time
	UserID      string   `gorm:"not null;primarykey"`
	RoomID      string   `gorm:"not null;primarykey"`
	Role        RoomRole `gorm:"not null"`
	Permissions Permission
}

func (r *RoomUserRelation) HasPermission(permission Permission) bool {
	switch r.Role {
	case RoomRoleCreator:
		return true
	case RoomRoleUser:
		return r.Permissions.Has(permission)
	default:
		return false
	}
}