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

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\FlowEvent;
  9. use Shopware\Core\Framework\Uuid\Uuid;
  11. class AddCustomerAffiliateAndCampaignCodeAction extends FlowAction
  12. {
  13.     private Connection $connection;
  15.     private EntityRepositoryInterface $customerRepository;
  17.     public function __construct(
  18.         Connection $connection,
  19.         EntityRepositoryInterface $customerRepository
  20.     ) {
  21.         $this->connection = $connection;
  22.         $this->customerRepository = $customerRepository;
  23.     }
  25.     public static function getName(): string
  26.     {
  27.         return 'action.add.customer.affiliate.and.campaign.code';
  28.     }
  30.     public static function getSubscribedEvents(): array
  31.     {
  32.         return [
  33.             self::getName() => 'handle',
  34.         ];
  35.     }
  37.     public function requirements(): array
  38.     {
  39.         return [CustomerAware::class];
  40.     }
  42.     public function handle(FlowEvent $event): void
  43.     {
  44.         $config = $event->getConfig();
  46.         if (!\array_key_exists('affiliateCode', $config) || !\array_key_exists('campaignCode', $config)) {
  47.             return;
  48.         }
  50.         $baseEvent = $event->getEvent();
  52.         if (!$baseEvent instanceof CustomerAware) {
  53.             return;
  54.         }
  56.         $customerId = $baseEvent->getCustomerId();
  57.         $customerData = $this->getAffiliateAndCampaignCodeFromCustomerId($customerId);
  59.         if (empty($customerData)) {
  60.             return;
  61.         }
  63.         $affiliateCode = $customerData['affiliate_code'];
  64.         if ($affiliateCode === null || $config['affiliateCode']['upsert']) {
  65.             $affiliateCode = $config['affiliateCode']['value'];
  66.         }
  68.         $campaignCode = $customerData['campaign_code'];
  69.         if ($campaignCode === null || $config['campaignCode']['upsert']) {
  70.             $campaignCode = $config['campaignCode']['value'];
  71.         }
  73.         $data = [];
  74.         if ($affiliateCode !== $customerData['affiliate_code']) {
  75.             $data['affiliateCode'] = $affiliateCode;
  76.         }
  78.         if ($campaignCode !== $customerData['campaign_code']) {
  79.             $data['campaignCode'] = $campaignCode;
  80.         }
  82.         if (empty($data)) {
  83.             return;
  84.         }
  86.         $data['id'] = $customerId;
  88.         $this->customerRepository->update([$data], $baseEvent->getContext());
  89.     }
  91.     private function getAffiliateAndCampaignCodeFromCustomerId(string $customerId): array
  92.     {
  93.         $query = $this->connection->createQueryBuilder();
  94.         $query->select(['affiliate_code', 'campaign_code']);
  95.         $query->from('customer');
  96.         $query->where('id = :id');
  97.         $query->setParameter('id', Uuid::fromHexToBytes($customerId));
  99.         if (!$data = $query->execute()->fetchAssociative()) {
  100.             return [];
  101.         }
  103.         return $data;
  104.     }
  105. }