custom/plugins/IwvTwoFactorAuthentication/src/Subscriber/UserEntitySubscriber.php line 77

Open in your IDE?
  1. <?php
  3. declare(strict_types=1);
  5. namespace Iwv\IwvTwoFactorAuthentication\Subscriber;
  7. use Shopware\Core\Checkout\Customer\CustomerEvents;
  8. use Shopware\Core\System\User\UserEvents;
  9. use Shopware\Core\Framework\DataAbstractionLayer\Search\Criteria;
  10. use Shopware\Core\Framework\DataAbstractionLayer\Search\Filter\EqualsAnyFilter;
  11. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  12. use Shopware\Core\Framework\DataAbstractionLayer\Event\EntityWrittenEvent;
  13. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  15. class UserEntitySubscriber implements EventSubscriberInterface
  16. {
  17.     /** @var EntityRepositoryInterface */
  18.     private $userRepository;
  20.     /** @var EntityRepositoryInterface */
  21.     private $customerRepository;
  23.     public function __construct(
  24.         EntityRepositoryInterface $userRepository,
  25.         EntityRepositoryInterface $customerRepository
  26.     ) {
  27.         $this->userRepository = $userRepository;
  28.         $this->customerRepository = $customerRepository;
  29.     }
  31.     public static function getSubscribedEvents()
  32.     {
  33.         return [
  34.             UserEvents::USER_WRITTEN_EVENT => 'onUserWrittenEvent',
  35.             CustomerEvents::CUSTOMER_WRITTEN_EVENT => 'onCustomerWrittenEvent',
  36.         ];
  37.     }
  39.     public function onUserWrittenEvent(EntityWrittenEvent $event): void
  40.     {
  42.         $payloads = $event->getPayloads();
  43.         /** @var \Shopware\Core\System\User\UserEntity $userDB */
  44.         $updateUsers = [];
  45.         $ids = array_filter(array_map(fn($payload) => ($payload['id'] ?? ''), $payloads));
  46.         if ($ids) {
  47.             foreach ($this->userRepository->search(
  48.                 (new Criteria())->addFilter(new EqualsAnyFilter('id', $ids)),
  49.                 $event->getContext()
  50.             ) as $userDB) {
  51.                 $customFields = (array) $userDB->getCustomFields();
  52.                 if (array_key_exists('iwvTwoFactor', $customFields)) {
  53.                     if (empty($customFields['iwvTwoFactor'])) {
  54.                         unset($customFields['iwvTwoFactor']);
  55.                         unset($customFields['iwvTwoFactorCookie']);
  56.                         unset($customFields['iwvTwoFactorSession']);
  57.                         $updateUsers[] = [
  58.                             'id' => $userDB->getId(),
  59.                             'customFields' => $customFields
  60.                         ];
  61.                     } elseif (!array_key_exists('iwvTwoFactorCookie', $customFields)) {
  62.                         $customFields['iwvTwoFactorCookie'] = bin2hex(random_bytes(25));
  63.                         $updateUsers[] = [
  64.                             'id' => $userDB->getId(),
  65.                             'customFields' => $customFields
  66.                         ];
  67.                     }
  68.                 }
  69.             }
  71.             if ($updateUsers) {
  72.                 $this->userRepository->update($updateUsers, $event->getContext());
  73.             }
  74.         }
  75.     }
  77.     public function onCustomerWrittenEvent(EntityWrittenEvent $event): void
  78.     {
  80.         $payloads = $event->getPayloads();
  81.         /** @var \Shopware\Core\Checkout\Customer\CustomerEntity $userDB */
  82.         $updateCustomers = [];
  84.         $ids = array_map(fn($payload) => $payload['id'] ?? null, $payloads);
  85.         $ids = array_filter($ids, fn($id) => $id !== null);
  87.         if (!empty($ids)) {
  88.             $criteria = new Criteria();
  89.             $criteria->addFilter(new EqualsAnyFilter('id', $ids));
  91.             foreach ($this->customerRepository->search($criteria, $event->getContext()) as $userDB) {
  92.                 $customFields = (array) $userDB->getCustomFields();
  93.                 if (array_key_exists('iwvTwoFactor', $customFields)) {
  94.                     if (empty($customFields['iwvTwoFactor'])) {
  95.                         unset($customFields['iwvTwoFactor']);
  96.                         unset($customFields['iwvTwoFactorCookie']);
  97.                         unset($customFields['iwvTwoFactorSession']);
  98.                         $updateCustomers[] = [
  99.                             'id' => $userDB->getId(),
  100.                             'customFields' => $customFields
  101.                         ];
  102.                     } elseif (!array_key_exists('iwvTwoFactorCookie', $customFields)) {
  103.                         $customFields['iwvTwoFactorCookie'] = bin2hex(random_bytes(25));
  104.                         $updateCustomers[] = [
  105.                             'id' => $userDB->getId(),
  106.                             'customFields' => $customFields
  107.                         ];
  108.                     }
  109.                 }
  110.             }
  111.         }
  113.         if ($updateCustomers) {
  114.             $this->customerRepository->update($updateCustomers, $event->getContext());
  115.         }
  116.     }
  117. }