custom/plugins/CioBudget/src/Subscriber/FilterCustomerEntityBudgetsSubscriber.php line 37

Open in your IDE?
  1. <?php
  3. namespace CioBudget\Subscriber;
  5. use CioBudget\Definition\Budget\BudgetEntity;
  6. use CioBudget\Service\BudgetLoaderService;
  7. use Shopware\Core\Checkout\Customer\CustomerEntity;
  8. use Shopware\Core\Checkout\Customer\CustomerEvents;
  9. use Shopware\Core\Framework\Api\Context\SalesChannelApiSource;
  10. use Shopware\Core\Framework\Context;
  11. use Shopware\Core\Framework\DataAbstractionLayer\Event\EntityLoadedEvent;
  12. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  13. use Shopware\Core\Framework\DataAbstractionLayer\Search\Filter\RangeFilter;
  14. use Symfony\Component\DependencyInjection\Container;
  15. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  17. class FilterCustomerEntityBudgetsSubscriber implements EventSubscriberInterface
  18. {
  19.     private BudgetLoaderService $budgetLoaderService;
  20.     private Container $container;
  22.     public function __construct(
  23.         Container $container,
  24.         BudgetLoaderService $budgetLoaderService
  25.     ) {
  26.         $this->budgetLoaderService = $budgetLoaderService;
  27.         $this->container = $container;
  28.     }
  30.     public static function getSubscribedEvents(): array
  31.     {
  32.         return [
  33.             CustomerEvents::CUSTOMER_LOADED_EVENT => 'onLoaded'
  34.         ];
  35.     }
  37.     public function onLoaded(EntityLoadedEvent $event)
  38.     {
  39.         /** @var BudgetLoaderService $budgetLoaderService */
  41.         if ($event->getContext()->getScope() !== 'user' || !($event->getContext()->getSource() instanceof SalesChannelApiSource)) {
  42.             return;
  43.         }
  45.         /** @var CustomerEntity $customer */
  46.         $customers = $event->getEntities();
  48.         foreach ($customers as $customer) {
  49.             $customFields = $customer->getCustomFields();
  50.             $customerBudgets = $this->budgetLoaderService->getActiveBudgetsByCustomer($customer, $event->getContext());
  52.             if ($customerBudgets) {
  53.                 $customFields['cio_active_budgets'] = array_map(function ($customerBudget) {
  54.                     return $customerBudget->getId();
  55.                 }, $customerBudgets->getElements());
  56.             }
  57.             $customer->setCustomFields($customFields);
  58.         }
  59.     }
  60. }