custom/plugins/WbfkExtensions/src/Subscriber/SaveTradeLicenseForCustomer.php line 31

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\Event\CustomerRegisterEvent;
  9. use Shopware\Core\Checkout\Customer\Event\GuestCustomerRegisterEvent;
  10. use Shopware\Core\Framework\Context;
  11. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepository;
  12. use Shopware\Core\Framework\Event\DataMappingEvent;
  13. use Shopware\Core\Framework\Routing\Event\SalesChannelContextResolvedEvent;
  14. use Shopware\Storefront\Framework\Media\StorefrontMediaUploader;
  15. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  16. use Symfony\Component\HttpFoundation\File\UploadedFile;
  17. use Symfony\Component\HttpFoundation\RequestStack;
  19. class SaveTradeLicenseForCustomer implements EventSubscriberInterface
  20. {
  21.     private ?CustomerEntity $currentCustomer;
  23.     public function __construct(
  24.         private readonly RequestStack $requestStack,
  25.         private readonly StorefrontMediaUploader $storefrontMediaUploader,
  26.         private readonly EntityRepository $customerRepository,
  27.         private readonly EntityRepository $mediaRepository,
  28.     ) {
  29.     }
  31.     public function getCustomerInfo(SalesChannelContextResolvedEvent $event)
  32.     {
  33.         $this->currentCustomer = $event->getSalesChannelContext()->getCustomer();
  34.     }
  36.     public static function getSubscribedEvents(): array
  37.     {
  38.         return [
  39.             CustomerRegisterEvent::class => ['saveTradeLicenseIfUploaded', 4000],
  40.             GuestCustomerRegisterEvent::class => ['saveTradeLicenseIfUploaded', 4000],
  41. //            'frontend.account.profile.save.request'=> 'getCustomerInfo',
  42.             'checkout.customer.sales_channel.profile.update' => 'saveTradeLicenseIfUploadedDuringProfileUpdate',
  43.             SalesChannelContextResolvedEvent::class => 'getCustomerInfo',
  44.         ];
  45.     }
  47.     public function saveTradeLicenseIfUploaded(CustomerRegisterEvent|GuestCustomerRegisterEvent $event): void
  48.     {
  49.         /** @var ?UploadedFile $tradeLicenseFile */
  50.         $tradeLicenseFile = $this->requestStack->getCurrentRequest()->files->get('tradeLicense');
  51.         if (!$tradeLicenseFile) {
  52.             return;
  53.         }
  54.         $mediaId = $this->storefrontMediaUploader->upload($tradeLicenseFile, 'user', 'trade_license', $event->getContext(), true);
  55.         $customer = $event->getCustomer();
  57.         $this->customerRepository->update([
  58.             [
  59.                 'id' => $customer->getId(),
  60.                 'customFields' => [
  61.                     'wbfk_customer_trade_license' => $mediaId,
  62.                 ],
  63.             ],
  64.         ], $event->getContext());
  66.         // Add custom field to customer entity so that it can be utilized to send mail to admin
  67.         $customer->setCustomFields([
  68.             'wbfk_customer_trade_license' => $mediaId,
  69.             ...$customer->getCustomFields() ?? [],
  70.         ]);
  71.     }
  74.     public function saveTradeLicenseIfUploadedDuringProfileUpdate(DataMappingEvent $event): void
  75.     {
  76.         $tradeLicenseFile = $this->requestStack->getCurrentRequest()->files->get('tradeLicense');
  77.         if (!$tradeLicenseFile) {
  78.             return;
  79.         }
  81.         $this->deletePreviousTradeLicense($event->getContext());
  83.         $mediaId = $this->storefrontMediaUploader->upload($tradeLicenseFile, 'user', 'trade_license', $event->getContext(), true);
  84.         $customer = $event->getOutput();
  85.         if (!isset($customer['customFields'])) {
  86.             $customer['customFields'] = [];
  87.         }
  88.         $customer['customFields']['wbfk_customer_trade_license'] = $mediaId;
  89.         $event->setOutput($customer);
  90.     }
  92.     private function deletePreviousTradeLicense(Context $context): void
  93.     {
  94.         if (!$this->currentCustomer) {
  95.             return;
  96.         }
  97.         $previousTradeLicenseMediaId = $this->currentCustomer->getCustomFields()['wbfk_customer_trade_license'] ?? null;
  98.         if (!$previousTradeLicenseMediaId) {
  99.             return;
  100.         }
  101.         $context->scope(Context::SYSTEM_SCOPE, function (Context $context) use ($previousTradeLicenseMediaId) {
  102.             $this->mediaRepository->delete([['id' => $previousTradeLicenseMediaId]], $context);
  103.         });
  104.     }
  105. }