vendor/dachcom-digital/members/src/MembersBundle/EventListener/TreeListener.php line 96

Open in your IDE?
  1. <?php
  3. namespace MembersBundle\EventListener;
  5. use MembersBundle\Configuration\Configuration;
  6. use MembersBundle\Manager\RestrictionManager;
  7. use MembersBundle\Restriction\Restriction;
  8. use MembersBundle\Security\RestrictionUri;
  9. use Pimcore\Event\AdminEvents;
  10. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  11. use Symfony\Component\EventDispatcher\GenericEvent;
  13. class TreeListener implements EventSubscriberInterface
  14. {
  15.     public function __construct(
  16.         protected Configuration $configuration,
  17.         protected RestrictionManager $restrictionManager
  18.     )
  19.     {
  20.     }
  22.     public static function getSubscribedEvents(): array
  23.     {
  24.         $defaultEvents = [
  25.             AdminEvents::OBJECT_TREE_GET_CHILDREN_BY_ID_PRE_SEND_DATA => ['handleObjectTree', 0]
  26.         ];
  28.         if (defined('\Pimcore\Event\AdminEvents::ASSET_TREE_GET_CHILDREN_BY_ID_PRE_SEND_DATA')) {
  29.             $defaultEvents[AdminEvents::ASSET_TREE_GET_CHILDREN_BY_ID_PRE_SEND_DATA] = ['handleAssetTree', 0];
  30.         }
  32.         if (defined('\Pimcore\Event\AdminEvents::DOCUMENT_TREE_GET_CHILDREN_BY_ID_PRE_SEND_DATA')) {
  33.             $defaultEvents[AdminEvents::DOCUMENT_TREE_GET_CHILDREN_BY_ID_PRE_SEND_DATA] = ['handleDocumentTree', 0];
  34.         }
  36.         return $defaultEvents;
  37.     }
  39.     public function handleObjectTree(GenericEvent $event): void
  40.     {
  41.         $objects = $event->getArgument('objects');
  43.         $restrictionConfig = $this->configuration->getConfig('restriction');
  44.         $allowedTypes = $restrictionConfig['allowed_objects'];
  46.         foreach ($objects as &$object) {
  47.             if (!isset($object['className'])) {
  48.                 continue;
  49.             }
  51.             if (!in_array($object['className'], $allowedTypes, true)) {
  52.                 continue;
  53.             }
  55.             $restriction = $this->getRestriction($object['id'], 'object');
  56.             if (!$restriction instanceof Restriction) {
  57.                 continue;
  58.             }
  60.             $lockClass = $restriction->isInherited() ? 'members-locked-inherit' : 'members-locked-main';
  62.             $currentClass = $object['cls'] ?? '';
  63.             $object['cls'] = implode(' ', [$currentClass, 'members-locked', $lockClass]);
  64.         }
  66.         $event->setArgument('objects', $objects);
  67.     }
  69.     public function handleAssetTree(GenericEvent $event): void
  70.     {
  71.         $assets = $event->getArgument('assets');
  73.         foreach ($assets as &$asset) {
  75.             if (!isset($asset['basePath'])) {
  76.                 continue;
  77.             }
  79.             if ($this->restrictionManager->pathIsInProtectedStorageFolder($asset['basePath']) === false) {
  80.                 continue;
  81.             }
  83.             $restriction = $this->getRestriction($asset['id'], 'asset');
  84.             if (!$restriction instanceof Restriction) {
  85.                 continue;
  86.             }
  88.             $lockClass = $restriction->isInherited() ? 'members-locked-inherit' : 'members-locked-main';
  89.             $currentClass = $asset['cls'] ?? '';
  90.             $asset['cls'] = implode(' ', [$currentClass, 'members-locked', $lockClass]);
  91.         }
  93.         $event->setArgument('assets', $assets);
  94.     }
  96.     public function handleDocumentTree(GenericEvent $event): void
  97.     {
  98.         $documents = $event->getArgument('documents');
  100.         foreach ($documents as &$document) {
  101.             $restriction = $this->getRestriction($document['id'], 'page');
  102.             if (!$restriction instanceof Restriction) {
  103.                 continue;
  104.             }
  106.             $lockClass = $restriction->isInherited() ? 'members-locked-inherit' : 'members-locked-main';
  107.             $currentClass = $document['cls'] ?? '';
  108.             $document['cls'] = implode(' ', [$currentClass, 'members-locked', $lockClass]);
  109.         }
  111.         $event->setArgument('documents', $documents);
  112.     }
  114.     private function getRestriction(int $id, string $type): ?Restriction
  115.     {
  116.         try {
  117.             $restriction = Restriction::getByTargetId($id, $type);
  118.         } catch (\Exception $e) {
  119.             return null;
  120.         }
  122.         return $restriction;
  123.     }
  124. }