<?php
namespace App\Security\Voter;
use App\Entity\File;
use App\Entity\User;
use App\Repository\FileRepository;
use App\Utils\Commons;
use LogicException;
use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
use Symfony\Component\Security\Core\Authorization\Voter\Voter;
use Symfony\Component\Security\Core\Security;
use Symfony\Component\Security\Core\User\UserInterface;
class FileVoter extends Voter
{
const FILE_VIEW = 'FILE_VIEW';
const FILE_SHOW = 'FILE_SHOW';
const FILE_CREATE = 'FILE_CREATE';
const FILE_EDIT = 'FILE_EDIT';
const FILE_DELETE = 'FILE_DELETE';
const FILE_MULTI_DELETE = 'FILE_MULTI_DELETE';
/**
* @var Security
*/
private $security;
/**
* @var Commons
*/
private $commons;
/**
* @var array
*/
private $permissions;
/**
* @var FileRepository
*/
private $fileRepository;
/**
* FileGroupVoter constructor.
* @param Security $security
* @param Commons $commons
* @param FileRepository $fileRepository
*/
public function __construct(Security $security, Commons $commons, FileRepository $fileRepository)
{
$this->security = $security;
$this->commons = $commons;
$this->fileRepository = $fileRepository;
$this->permissions = [];
}
/**
* @param string $attribute
* @param mixed $subject
* @return bool
*/
protected function supports($attribute, $subject): bool
{
// if the attribute isn't one we support, return false
if (!in_array($attribute,
[self::FILE_VIEW, self::FILE_EDIT, self::FILE_CREATE, self::FILE_DELETE, self::FILE_MULTI_DELETE, self::FILE_SHOW]
)) {
return false;
}
return true;
}
/**
* @param string $attribute
* @param mixed $subject
* @param TokenInterface $token
* @return bool
*/
protected function voteOnAttribute($attribute, $subject, TokenInterface $token): bool
{
/** @var User $user */
$user = $token->getUser();
// if the user is anonymous, do not grant access
if (!$user instanceof UserInterface) {
return false;
}
$this->permissions = $this->commons->getUserPermissions($user);
// ... (check conditions and return true to grant permission) ...
switch ($attribute) {
case self::FILE_VIEW:
return $this->canView();
case self::FILE_CREATE:
return $this->canCreate();
case self::FILE_EDIT:
return ($this->canEdit() && $this->isOwner($user, $subject));
case self::FILE_SHOW:
return ($this->canShow() && $this->isOwner($user, $subject));
case self::FILE_DELETE:
return ($this->canDelete() && $this->isOwner($user, $subject));
case self::FILE_MULTI_DELETE:
return $this->canMultiDelete();
}
throw new LogicException('Invalid attribute: ' . $attribute);
}
/**
* @param User|null $user
* @param $subject
* @return bool
*/
private function isOwner(User $user, $subject): bool
{
if (!is_object($subject)) {
$subject = $this->fileRepository->findOneBy(["id" => intval($subject)]);
}
if (!$subject instanceof File) {
return false;
}
return $user->getWorkspace()->getId() === $subject->getWorkspace()->getId();
}
/**
* Return True if have View Permission else return false
*/
private function canView(): bool
{
if ($this->security->isGranted('ROLE_ADMIN')) {
return true;
}
if (array_key_exists('file', $this->permissions)) {
if (in_array('View', $this->permissions['file'])) {
return true;
}
}
return false;
}
/**
* Return True if have Show Permission else return false
* @return bool
*/
private function canShow(): bool
{
if ($this->security->isGranted('ROLE_ADMIN')) {
return true;
}
if (array_key_exists('file', $this->permissions)) {
if (in_array('Show', $this->permissions['file'])) {
return true;
}
}
return false;
}
/**
* Return True if have Edit Permission else return false
* @return bool
*/
private function canEdit(): bool
{
if ($this->security->isGranted('ROLE_ADMIN')) {
return true;
}
if (array_key_exists('file', $this->permissions)) {
if (in_array('Edit', $this->permissions['file'])) {
return true;
}
}
return false;
}
/**
* Return True if have Create Permission else return false
* @return bool
*/
private function canCreate(): bool
{
if ($this->security->isGranted('ROLE_ADMIN')) {
return true;
}
if (array_key_exists('file', $this->permissions)) {
if (in_array('Create', $this->permissions['file'])) {
return true;
}
}
return false;
}
/** Return True if have Delete Permission else return false
* @return bool
*/
private function canDelete(): bool
{
if ($this->security->isGranted('ROLE_ADMIN')) {
return true;
}
if (array_key_exists('file', $this->permissions)) {
if (in_array('Delete', $this->permissions['file'])) {
return true;
}
}
return false;
}
/**
* @return bool
*/
private function canMultiDelete(): bool
{
if ($this->security->isGranted('ROLE_ADMIN')) {
return true;
}
if (array_key_exists('file', $this->permissions)) {
if (in_array('Delete', $this->permissions['file'])) {
return true;
}
}
return false;
}
}