custom/plugins/WbfkExtensions/src/Subscriber/AutoApproveOrSetFallbackB2BCustomerGroup.php line 32

Open in your IDE?
  1. <?php
  3. declare(strict_types=1);
  5. namespace WbfkExtensions\Subscriber;
  7. use Shopware\Core\Checkout\Customer\CustomerEntity;
  8. use Shopware\Core\Checkout\Customer\CustomerEvents;
  9. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  11. use Shopware\Core\Framework\Event\DataMappingEvent;
  12. use Shopware\Core\System\SalesChannel\SalesChannelEntity;
  13. use Shopware\Core\System\SystemConfig\SystemConfigService;
  14. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  16. class AutoApproveOrSetFallbackB2BCustomerGroup implements EventSubscriberInterface
  17. {
  18.     public function __construct(
  19.         private readonly SystemConfigService $systemConfigService,
  20.         private readonly EntityRepositoryInterface $salesChannelRepository,
  21.     ) {
  22.     }
  24.     public static function getSubscribedEvents(): array
  25.     {
  26.         return [
  27.             CustomerEvents::MAPPING_REGISTER_CUSTOMER => 'customerGroupAssignmentWorkflow',
  28.             CustomerEvents::MAPPING_CUSTOMER_PROFILE_SAVE => 'customerGroupAssignmentWorkflow',
  29.         ];
  30.     }
  32.     public function customerGroupAssignmentWorkflow(DataMappingEvent $event): void
  33.     {
  34.         if ($event->getInput()->get('accountType') === CustomerEntity::ACCOUNT_TYPE_PRIVATE) {
  35.             $this->setDefaultCustomerGroup($event);
  36.             return;
  37.         }
  38.         $requestedCustomerGroupId = $event->getInput()->get('requestedGroupId');
  39.         if (!$requestedCustomerGroupId) {
  40.             return;
  41.         }
  43.         if ($this->isAutoApprovedB2BCustomerGroup($requestedCustomerGroupId)) {
  44.             $this->autoApproveB2BCustomerGroup($event, $requestedCustomerGroupId);
  46.             return;
  47.         }
  48.         $this->setFallBackB2BCustomerGroup($event);
  49.     }
  51.     private
  52.     function autoApproveB2BCustomerGroup(
  53.         DataMappingEvent $event,
  54.         string $requestedCustomerGroupId
  55.     ): void {
  56.         $output = $event->getOutput();
  57.         $output['groupId'] = $requestedCustomerGroupId;
  58.         $output['requestedGroupId'] = null;
  60.         $event->setOutput($output);
  61.     }
  64.     public
  65.     function isAutoApprovedB2BCustomerGroup(
  66.         string $requestedCustomerGroupId
  67.     ): bool {
  68.         $autoApprovedB2BCustomerGroupIds = $this->systemConfigService->get(
  69.             'WbfkExtensions.config.autoApprovedB2BCustomerGroupIds'
  70.         );
  72.         return in_array($requestedCustomerGroupId, $autoApprovedB2BCustomerGroupIds);
  73.     }
  75.     private
  76.     function setFallBackB2BCustomerGroup(
  77.         DataMappingEvent $event
  78.     ): void {
  79.         $input = $event->getInput();
  80.         $output = $event->getOutput();
  81.         $output['groupId'] = $this->systemConfigService->get('WbfkExtensions.config.fallbackB2BCustomerGroupId');
  82.         $output['requestedGroupId'] = $input->get('requestedGroupId');
  83.         $event->setOutput($output);
  84.     }
  86.     private function setDefaultCustomerGroup(DataMappingEvent $event): void
  87.     {
  88.         $criteria = new Criteria([$event->getContext()->getSource()->getSalesChannelId()]);
  89.         /** @var SalesChannelEntity $salesChannel */
  90.         $salesChannel = $this->salesChannelRepository->search($criteria, $event->getContext())->first();
  92.         $output = $event->getOutput();
  93.         $output['groupId'] = $salesChannel->getCustomerGroupId();
  94.         $output['requestedGroupId'] = null;
  95.         $event->setOutput($output);
  96.     }
  97. }