src/Security/DeviceVoter.php line 18

Open in your IDE?
  1. <?php
  2. /**
  3.  * @license SILK SOFTWARE HOUSE SP Z O O
  4.  */
  6. namespace App\Security;
  8. use App\Entity\Customer;
  9. use App\Entity\Device;
  10. use App\Entity\User;
  11. use App\Repository\LicenseRepository;
  12. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  13. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  15. /**
  16.  * Class DeviceVoter.
  17.  */
  18. class DeviceVoter extends Voter
  19. {
  20.     const SHOW = 'show';
  21.     const NEW = 'new';
  22.     const EDIT = 'edit';
  23.     const DELETE = 'delete';
  25.     /**
  26.      * License repository.
  27.      *
  28.      * @var LicenseRepository
  29.      */
  30.     private $licenseRepository;
  32.     /**
  33.      * DeviceVoter constructor.
  34.      *
  35.      * @param LicenseRepository $licenseRepository
  36.      */
  37.     public function __construct(LicenseRepository $licenseRepository)
  38.     {
  39.         $this->licenseRepository = $licenseRepository;
  40.     }
  42.     /**
  43.      * Supports.
  44.      *
  45.      * @param string $attribute
  46.      * @param object $subject
  47.      *
  48.      * @return bool
  49.      */
  50.     protected function supports($attribute, $subject)
  51.     {
  52.         // if the attribute isn't one we support, return false
  53.         if (!in_array($attribute, [self::SHOW, self::NEW, self::EDIT, self::DELETE])) {
  54.             return false;
  55.         }
  56.         if ($subject && !$subject instanceof Device) {
  57.             return false;
  58.         }
  60.         return true;
  61.     }
  63.     /**
  64.      * Vote on attribute.
  65.      *
  66.      * @param string         $attribute
  67.      * @param object         $subject
  68.      * @param TokenInterface $token
  69.      *
  70.      * @return bool
  71.      */
  72.     protected function voteOnAttribute($attribute, $subject, TokenInterface $token)
  73.     {
  74.         $user = $token->getUser();
  75.         if (!$user instanceof User) {
  76.             // the user must be logged in; if not, deny access
  77.             return false;
  78.         }
  79.         if (!$subject instanceof Device && self::NEW !== $attribute) {
  80.             return false;
  81.         }
  83.         // you know $subject is a Device object, thanks to `supports()`
  84.         /** @var Device $device */
  85.         $device = $subject;
  86.         if (in_array($attribute, [self::NEW, self::DELETE])) {
  87.             return $user->hasRole('ROLE_ADMIN');
  88.         }
  89.         if (self::EDIT === $attribute) {
  90.             return $this->canEdit($device, $user);
  91.         }
  93.         return $this->canView($device, $user);
  94.     }
  96.     /**
  97.      * Can view.
  98.      *
  99.      * @param Device $device
  100.      * @param User   $user
  101.      *
  102.      * @return bool
  103.      */
  104.     private function canView(Device $device, User $user)
  105.     {
  106.         if ($user->hasRole('ROLE_ADMIN') || $this->canEdit($device, $user)) {
  107.             return true;
  108.         }
  110.         $deviceHasCustomersMeasurement = $this->hasMeasurement($device, $user->getCustomer());
  112.         return null !== $this->licenseRepository->findOneBy([
  113.                 'device' => $device,
  114.                 'customer' => $user->getActiveCustomer(),
  115.             ]) || $deviceHasCustomersMeasurement;
  116.     }
  118.     /**
  119.      * Can edit.
  120.      *
  121.      * @param Device $device
  122.      * @param User   $user
  123.      *
  124.      * @return bool
  125.      */
  126.     private function canEdit(Device $device, User $user)
  127.     {
  128.         if ($user->hasRole('ROLE_ADMIN')) {
  129.             return true;
  130.         }
  132.         $deviceCustomer = $device->getCustomer();
  133.         $userCustomer = $user->getActiveCustomer();
  135.         return ($deviceCustomer && $userCustomer && $deviceCustomer->getId() === $userCustomer->getId());
  136.     }
  138.     /**
  139.      * @param Device        $device
  140.      * @param Customer|null $customer
  141.      *
  142.      * @return bool
  143.      */
  144.     private function hasMeasurement(Device $device, ?Customer $customer): bool
  145.     {
  146.         foreach ($device->getMeasurements() ?? [] as $measurement) {
  147.             return $measurement->getCustomer() === $customer;
  148.         }
  150.         return false;
  151.     }
  152. }