custom/plugins/VioB2BLogin/src/Core/Services/PrivilegeService.php line 104

Open in your IDE?
  1. <?php
  2. declare(strict_types=1);
  5. namespace VioB2BLogin\Core\Services;
  8. use Shopware\Core\Checkout\Customer\CustomerEntity;
  9. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Search\Filter\EqualsFilter;
  12. use Shopware\Core\System\SalesChannel\SalesChannelContext;
  13. use VioB2BLogin\Core\System\SalesChannel\Context\SalesChannelContextFactory;
  14. use VioB2BLogin\Entity\Employee\EmployeeEntity;
  15. use VioB2BLogin\Entity\Privilege\PrivilegeCollection;
  16. use VioB2BLogin\Entity\Privilege\PrivilegeEntity;
  17. use VioB2BLogin\Entity\Privilege\PrivilegeNamespaces;
  19. class PrivilegeService
  20. {
  22.     private EntityRepositoryInterface $privilegeRepository;
  24.     /**
  25.      * PrivilegeService constructor.
  26.      * @param EntityRepositoryInterface $privilegeRepository
  27.      */
  28.     public function __construct(
  29.         EntityRepositoryInterface $privilegeRepository
  30.     )
  31.     {
  32.         $this->privilegeRepository = $privilegeRepository;
  33.     }
  35.     public function getAllPrivileges(SalesChannelContext $context): PrivilegeCollection
  36.     {
  37.         $criteria = new Criteria();
  38.         $criteria->addFilter(new EqualsFilter('active', true));
  39.         $result = $this->privilegeRepository->search($criteria, $context->getContext());
  40.         $result->sort(function (PrivilegeEntity $a, PrivilegeEntity $b) {
  41.             return $a->getNamespace() . $a->getName() <=> $b->getNamespace() . $b->getName();
  42.         });
  43.         /** @noinspection PhpIncompatibleReturnTypeInspection */
  44.         return $result->getEntities();
  45.     }
  47.     /**
  48.      * @param SalesChannelContext $context
  49.      * @return array<string,string>
  50.      */
  51.     public function getAllGroupedPrivileges(SalesChannelContext $context): array
  52.     {
  54.         $privileges = $this->getAllPrivileges($context);
  55.         $allNamespaces = [];
  56.         $allPermissions = [];
  58.         // project to array
  59.         $projection = [];
  60.         foreach ($privileges as $privilege) {
  62.             if (!in_array($privilege->getNamespace(), $allNamespaces, true)) {
  63.                 $allNamespaces[] = $privilege->getNamespace();
  64.             }
  65.             if (!in_array($privilege->getName(), $allPermissions, true)) {
  66.                 $allPermissions[] = $privilege->getName();
  67.             }
  69.             if (!array_key_exists($privilege->getNamespace(), $projection)) {
  70.                 $projection[$privilege->getNamespace()] = [];
  71.             }
  73.             $projection[$privilege->getNamespace()][$privilege->getName()] = $privilege;
  74.         }
  76.         foreach ( $allNamespaces as $namespace )
  77.         {
  78.             if( !array_key_exists($namespace, $projection ) ) {
  79.                 $projection[$namespace] = [];
  80.             }
  81.             foreach ($allPermissions as $permission)
  82.             {
  83.                 if( !array_key_exists($permission, $projection[$namespace] ) ) {
  84.                     $projection[$namespace][$permission] = null;
  85.                 }
  86.             }
  87.             ksort($projection[$namespace]);
  88.         }
  90.         return $projection;
  91.     }
  93.     /**
  94.      * @param SalesChannelContext $context
  95.      * @return PrivilegeCollection
  96.      */
  97.     public function getPrivilegesOfUser(SalesChannelContext $context): PrivilegeCollection
  98.     {
  99.         $user = $context->getCustomer();
  100.         /** @var EmployeeEntity $employee */
  101.         $employee = $context->getExtensionOfType(SalesChannelContextFactory::EMPLOYEE_KEY, EmployeeEntity::class);
  103.         if(!$user instanceof CustomerEntity) {
  104.             $collection = $this->getAllActivePriveleges($context);
  105.             $collection->filterAndReduceByProperty('namespace', PrivilegeNamespaces::EMPLOYEE);
  106.             return $collection;
  107.         }
  109.         if(!$employee instanceof EmployeeEntity || $employee->getPrivileges()->count() === 0) {
  110.             // if customer is directly logged in all privileges are granted
  111.             return $this->getAllActivePriveleges($context);
  112.         }
  114.         return $employee->getPrivileges();
  115.     }
  117.     public function checkPrivilege(SalesChannelContext $context, string $namespace, string $name): bool
  118.     {
  119.         return $this->checkPrivilegeByCollection($this->getPrivilegesOfUser($context), $namespace, $name);
  120.     }
  122.     public function checkPrivilegeByCollection(PrivilegeCollection $privileges, string $namespace, string $name): bool
  123.     {
  124.         $privilege = $privileges->filter(
  125.         /**
  126.          * @param $p PrivilegeEntity
  127.          * @return bool
  128.          */
  129.             function (PrivilegeEntity $p) use ($namespace, $name) {
  130.                 return ($p->getNamespace() === $namespace && $p->getName() === $name);
  131.             })->first();
  133.         return ($privilege instanceof PrivilegeEntity);
  134.     }
  136.     /**
  137.      * @param SalesChannelContext $context
  138.      * @return PrivilegeCollection
  139.      */
  140.     private function getAllActivePriveleges(SalesChannelContext $context): PrivilegeCollection
  141.     {
  142.         $criteria = new Criteria();
  143.         $criteria->addFilter(new EqualsFilter('active', true));
  144.         $result = $this->privilegeRepository->search($criteria, $context->getContext());
  145.         /** @noinspection PhpIncompatibleReturnTypeInspection */
  146.         return $result->getEntities();
  147.     }
  149. }