Odiseo\SyliusReferralsPlugin\Generator\AffiliateReferralGenerator

Complexity

Total Complexity

3

Size/Duplication

Total Lines	38
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	generate()	0	29	2

<?php

declare(strict_types=1);

namespace Odiseo\SyliusReferralsPlugin\Generator;

use Odiseo\SyliusReferralsPlugin\Entity\AffiliateInterface;
use Odiseo\SyliusReferralsPlugin\Entity\AffiliateReferralInterface;
use Odiseo\SyliusReferralsPlugin\Event\AffiliateReferralEvent;
use Odiseo\SyliusReferralsPlugin\Factory\AffiliateReferralFactoryInterface;
use Odiseo\SyliusReferralsPlugin\Repository\AffiliateReferralRepositoryInterface;
use Sylius\Component\Core\Model\ProductInterface;
use Symfony\Contracts\EventDispatcher\EventDispatcherInterface;

final class AffiliateReferralGenerator implements AffiliateReferralGeneratorInterface
{
    public function __construct(
        private AffiliateReferralFactoryInterface $affiliateReferralFactory,
        private AffiliateReferralRepositoryInterface $affiliateReferralRepository,
        private EventDispatcherInterface $eventDispatcher,
    ) {
    }

    public function generate(
        AffiliateInterface $affiliate,
        ?ProductInterface $product = null,
    ): AffiliateReferralInterface {
        $this->eventDispatcher->dispatch(
            new AffiliateReferralEvent(),
            AffiliateReferralEvent::PRE_GENERATE,
        );

        $affiliateReferral = $this->affiliateReferralFactory->createForAffiliate($affiliate);
        $affiliateReferral->setRewardType(AffiliateReferralInterface::REWARD_TYPE_PROMOTION);
        if ($product instanceof ProductInterface) {
            $affiliateReferral->setProduct($product);
            $affiliateReferral->setExpiresAt(new \DateTime('+15 day'));
        }

        $this->eventDispatcher->dispatch(
            new AffiliateReferralEvent(),
            AffiliateReferralEvent::GENERATE,
        );

        $this->affiliateReferralRepository->add($affiliateReferral);

        $this->eventDispatcher->dispatch(
            new AffiliateReferralEvent(),
            AffiliateReferralEvent::POST_GENERATE,
        );

        return $affiliateReferral;
    }
}