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

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.         $activations = [];
  47.         $ids = array_filter(array_map(fn($payload) => ($payload['id'] ?? ''), $payloads));
  48.         if ($ids) {
  49.             foreach ($this->userRepository->search(
  50.                 (new Criteria())->addFilter(new EqualsAnyFilter('id', $ids)),
  51.                 $event->getContext()
  52.             ) as $userDB) {
  53.                 $customFields = (array) $userDB->getCustomFields();
  55.                 if (array_key_exists('iwvTwoFactor', $customFields)) {
  56.                     if (empty($customFields['iwvTwoFactor'])) {
  57.                         unset($customFields['iwvTwoFactor']);
  58.                         unset($customFields['iwvTwoFactorCookie']);
  59.                         unset($customFields['iwvTwoFactorSession']);
  60.                         $updateUsers[] = [
  61.                             'id' => $userDB->getId(),
  62.                             'customFields' => $customFields
  63.                         ];
  64.                     } elseif (!array_key_exists('iwvTwoFactorCookie', $customFields)) {
  65.                         $customFields['iwvTwoFactorCookie'] = bin2hex(random_bytes(25));
  66.                         $updateUsers[] = [
  67.                             'id' => $userDB->getId(),
  68.                             'customFields' => $customFields
  69.                         ];
  70.                     }
  71.                 }
  73.                 if (!array_key_exists('iwvTwoFactor', $customFields) || !$customFields) {
  74.                     $customFields['iwvTwoFactor']['emailAuth']['status'] = 'active';
  75.                    
  76.                     $activations[] = [
  77.                         'id' => $userDB->getId(),
  78.                         'customFields' => $customFields
  79.                     ];
  80.                 }
  81.             }
  83.             if ($activations) {
  84.                 $this->userRepository->update($activations, $event->getContext());
  86.             }
  88.             if ($updateUsers) {
  89.                 $this->userRepository->update($updateUsers, $event->getContext());
  90.             }
  91.         }
  92.     }
  94.     public function onCustomerWrittenEvent(EntityWrittenEvent $event): void
  95.     {
  97.         $payloads = $event->getPayloads();
  98.         /** @var \Shopware\Core\Checkout\Customer\CustomerEntity $userDB */
  99.         $updateCustomers = [];
  101.         $ids = array_map(fn($payload) => $payload['id'] ?? null, $payloads);
  102.         $ids = array_filter($ids, fn($id) => $id !== null);
  104.         if (!empty($ids)) {
  105.             $criteria = new Criteria();
  106.             $criteria->addFilter(new EqualsAnyFilter('id', $ids));
  108.             foreach ($this->customerRepository->search($criteria, $event->getContext()) as $userDB) {
  109.                 $customFields = (array) $userDB->getCustomFields();
  110.                 if (array_key_exists('iwvTwoFactor', $customFields)) {
  111.                     if (empty($customFields['iwvTwoFactor'])) {
  112.                         unset($customFields['iwvTwoFactor']);
  113.                         unset($customFields['iwvTwoFactorCookie']);
  114.                         unset($customFields['iwvTwoFactorSession']);
  115.                         $updateCustomers[] = [
  116.                             'id' => $userDB->getId(),
  117.                             'customFields' => $customFields
  118.                         ];
  119.                     } elseif (!array_key_exists('iwvTwoFactorCookie', $customFields)) {
  120.                         $customFields['iwvTwoFactorCookie'] = bin2hex(random_bytes(25));
  121.                         $updateCustomers[] = [
  122.                             'id' => $userDB->getId(),
  123.                             'customFields' => $customFields
  124.                         ];
  125.                     }
  126.                 }
  127.             }
  128.         }
  130.         if ($updateCustomers) {
  131.             $this->customerRepository->update($updateCustomers, $event->getContext());
  132.         }
  133.     }
  134. }