src/Security/CategoryVoter.php line 12

Open in your IDE?
  1. <?php
  3. namespace App\Security;
  5. use App\Entity\Category;
  6. use App\Entity\Security\AppUser;
  7. use App\Repository\CategoryRepository;
  8. use App\Repository\MalletteFileRepository;
  9. use Symfony\Component\Security\Core\Authentication\Token\TokenInterface;
  10. use Symfony\Component\Security\Core\Authorization\Voter\Voter;
  12. class CategoryVoter extends Voter
  13. {
  14.     public const VIEW = 'view';
  15.     public const VIEW_MENU = 'view_menu';
  17.     /** @var null|array<Category> */
  18.     private ?array $cachedAccessibleCategories = null;
  20.     private ?int $cachedUserId = null;
  21.     private ?string $cachedAttribute = null;
  23.     public function __construct(
  24.         private MalletteFileRepository $malletteFileRepository,
  25.         private CategoryRepository $categoryRepository,
  26.     ) {
  27.     }
  29.     protected function supports(string $attribute, $subject): bool
  30.     {
  31.         if (!in_array($attribute, [self::VIEW, self::VIEW_MENU])) {
  32.             return false;
  33.         }
  35.         return $subject instanceof Category;
  36.     }
  38.     /**
  39.      * @param Category $subject
  40.      */
  41.     protected function voteOnAttribute(string $attribute, $subject, TokenInterface $token): bool
  42.     {
  43.         if (!in_array($attribute, [self::VIEW, self::VIEW_MENU])) {
  44.             return false;
  45.         }
  47.         $user = $token->getUser();
  49.         // the user must be logged in; if not, deny access
  50.         if (!$user instanceof AppUser) {
  51.             return false;
  52.         }
  53.         $this->loadAccessibleCategoriesIfNeeded($user, $attribute);
  55.         if (in_array($subject, $this->cachedAccessibleCategories, true)) {
  56.             return true;
  57.         }
  59.         return $attribute === self::VIEW_MENU && $this->hasAccessibleChild($subject);
  60.     }
  62.     private function loadAccessibleCategoriesIfNeeded(AppUser $user, string $attribute): void
  63.     {
  64.         if ($this->cachedUserId === $user->getId()
  65.             && $this->cachedAttribute === $attribute
  66.             && $this->cachedAccessibleCategories !== null) {
  67.             return;
  68.         }
  70.         $groups = $user->getClient()->getGroups();
  71.         $malletteFiles = $this->malletteFileRepository->findByCategoryAccess($groups, $user);
  72.         $this->cachedAccessibleCategories = $this->categoryRepository->findByGroupsAndMalletteFiles($groups, $malletteFiles);
  73.         $this->cachedUserId = $user->getId();
  74.         $this->cachedAttribute = $attribute;
  75.     }
  77.     private function hasAccessibleChild(Category $category): bool
  78.     {
  79.         foreach ($category->getSubCategories() as $subCategory) {
  80.             if (in_array($subCategory, $this->cachedAccessibleCategories, true)) {
  81.                 return true;
  82.             }
  83.             if ($this->hasAccessibleDescendant($subCategory)) {
  84.                 return true;
  85.             }
  86.         }
  88.         return false;
  89.     }
  91.     private function hasAccessibleDescendant(Category $category): bool
  92.     {
  93.         foreach ($category->getSubCategories() as $subCategory) {
  94.             if (in_array($subCategory, $this->cachedAccessibleCategories, true)) {
  95.                 return true;
  96.             }
  97.             if ($this->hasAccessibleDescendant($subCategory)) {
  98.                 return true;
  99.             }
  100.         }
  102.         return false;
  103.     }
  104. }