vendor/shopware/core/Content/Flow/Dispatching/Action/AddCustomerAffiliateAndCampaignCodeAction.php line 46

Open in your IDE?
  1. <?php declare(strict_types=1);
  3. namespace Shopware\Core\Content\Flow\Dispatching\Action;
  5. use Doctrine\DBAL\Connection;
  6. use Shopware\Core\Framework\DataAbstractionLayer\EntityRepositoryInterface;
  7. use Shopware\Core\Framework\Event\CustomerAware;
  8. use Shopware\Core\Framework\Event\DelayAware;
  9. use Shopware\Core\Framework\Event\FlowEvent;
  10. use Shopware\Core\Framework\Uuid\Uuid;
  12. class AddCustomerAffiliateAndCampaignCodeAction extends FlowAction
  13. {
  14.     private Connection $connection;
  16.     private EntityRepositoryInterface $customerRepository;
  18.     /**
  19.      * @internal
  20.      */
  21.     public function __construct(
  22.         Connection $connection,
  23.         EntityRepositoryInterface $customerRepository
  24.     ) {
  25.         $this->connection = $connection;
  26.         $this->customerRepository = $customerRepository;
  27.     }
  29.     public static function getName(): string
  30.     {
  31.         return 'action.add.customer.affiliate.and.campaign.code';
  32.     }
  34.     public static function getSubscribedEvents(): array
  35.     {
  36.         return [
  37.             self::getName() => 'handle',
  38.         ];
  39.     }
  41.     public function requirements(): array
  42.     {
  43.         return [CustomerAware::class, DelayAware::class];
  44.     }
  46.     public function handle(FlowEvent $event): void
  47.     {
  48.         $config = $event->getConfig();
  50.         if (!\array_key_exists('affiliateCode', $config) || !\array_key_exists('campaignCode', $config)) {
  51.             return;
  52.         }
  54.         $baseEvent = $event->getEvent();
  56.         if (!$baseEvent instanceof CustomerAware) {
  57.             return;
  58.         }
  60.         $customerId = $baseEvent->getCustomerId();
  61.         $customerData = $this->getAffiliateAndCampaignCodeFromCustomerId($customerId);
  63.         if (empty($customerData)) {
  64.             return;
  65.         }
  67.         $affiliateCode = $customerData['affiliate_code'];
  68.         if ($affiliateCode === null || $config['affiliateCode']['upsert']) {
  69.             $affiliateCode = $config['affiliateCode']['value'];
  70.         }
  72.         $campaignCode = $customerData['campaign_code'];
  73.         if ($campaignCode === null || $config['campaignCode']['upsert']) {
  74.             $campaignCode = $config['campaignCode']['value'];
  75.         }
  77.         $data = [];
  78.         if ($affiliateCode !== $customerData['affiliate_code']) {
  79.             $data['affiliateCode'] = $affiliateCode;
  80.         }
  82.         if ($campaignCode !== $customerData['campaign_code']) {
  83.             $data['campaignCode'] = $campaignCode;
  84.         }
  86.         if (empty($data)) {
  87.             return;
  88.         }
  90.         $data['id'] = $customerId;
  92.         $this->customerRepository->update([$data], $baseEvent->getContext());
  93.     }
  95.     private function getAffiliateAndCampaignCodeFromCustomerId(string $customerId): array
  96.     {
  97.         $query = $this->connection->createQueryBuilder();
  98.         $query->select(['affiliate_code', 'campaign_code']);
  99.         $query->from('customer');
  100.         $query->where('id = :id');
  101.         $query->setParameter('id', Uuid::fromHexToBytes($customerId));
  103.         if (!$data = $query->execute()->fetchAssociative()) {
  104.             return [];
  105.         }
  107.         return $data;
  108.     }
  109. }