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

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.                         $updateUsers[] = [
  57.                             'id' => $userDB->getId(),
  58.                             'customFields' => $customFields
  59.                         ];
  60.                     } elseif (!array_key_exists('iwvTwoFactorCookie', $customFields)) {
  61.                         $customFields['iwvTwoFactorCookie'] = bin2hex(random_bytes(25));
  62.                         $updateUsers[] = [
  63.                             'id' => $userDB->getId(),
  64.                             'customFields' => $customFields
  65.                         ];
  66.                     }
  67.                 }
  68.             }
  70.             if ($updateUsers) {
  71.                 $this->userRepository->update($updateUsers, $event->getContext());
  72.             }
  73.         }
  74.     }
  76.     public function onCustomerWrittenEvent(EntityWrittenEvent $event): void
  77.     {
  79.         $payloads = $event->getPayloads();
  80.         /** @var \Shopware\Core\Checkout\Customer\CustomerEntity $userDB */
  81.         $updateCustomers = [];
  83.         $ids = array_map(fn($payload) => $payload['id'] ?? null, $payloads);
  84.         $ids = array_filter($ids, fn($id) => $id !== null);
  86.         if (!empty($ids)) {
  87.             $criteria = new Criteria();
  88.             $criteria->addFilter(new EqualsAnyFilter('id', $ids));
  90.             foreach ($this->customerRepository->search($criteria, $event->getContext()) as $userDB) {
  91.                 $customFields = (array) $userDB->getCustomFields();
  92.                 if (array_key_exists('iwvTwoFactor', $customFields)) {
  93.                     if (empty($customFields['iwvTwoFactor'])) {
  94.                         unset($customFields['iwvTwoFactor']);
  95.                         unset($customFields['iwvTwoFactorCookie']);
  96.                         $updateCustomers[] = [
  97.                             'id' => $userDB->getId(),
  98.                             'customFields' => $customFields
  99.                         ];
  100.                     } elseif (!array_key_exists('iwvTwoFactorCookie', $customFields)) {
  101.                         $customFields['iwvTwoFactorCookie'] = bin2hex(random_bytes(25));
  102.                         $updateCustomers[] = [
  103.                             'id' => $userDB->getId(),
  104.                             'customFields' => $customFields
  105.                         ];
  106.                     }
  107.                 }
  108.             }
  109.         }
  111.         if ($updateCustomers) {
  112.             $this->customerRepository->update($updateCustomers, $event->getContext());
  113.         }
  114.     }
  115. }