DoS\UserBundle\EventListener\UserAwareResourceListener

Complexity

Total Complexity

10

Size/Duplication

Total Lines	52
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	6

Importance

Changes	1
Bugs	0	Features	0

3 Methods

Rating	Name	Duplication	Size	Complexity
C	onFlush()	0	28	8
A	getUser()	0	4	1
A	__construct()	0	4	1

<?php

namespace DoS\UserBundle\EventListener;

use Doctrine\ORM\Event\OnFlushEventArgs;
use DoS\UserBundle\Security\Security;
use DoS\UserBundle\Model\UserUpdaterAwareInterface;
use DoS\UserBundle\Model\UserInterface;
use Sylius\Component\User\Model\UserAwareInterface;

class UserAwareResourceListener
{
    /**
     * @var Security
     */
    protected $security;

    public function __construct(Security $security)
    {
        $this->security = $security;
    }

    /**
     * @param OnFlushEventArgs $onFlushEventArgs
     */
    public function onFlush(OnFlushEventArgs $onFlushEventArgs)
    {
        $entityManager = $onFlushEventArgs->getEntityManager();
        $unitOfWork = $entityManager->getUnitOfWork();

        $entities = array_merge(
            $unitOfWork->getScheduledEntityInsertions(),
            $unitOfWork->getScheduledEntityUpdates()
        );

        foreach ($entities as $entity) {
            if (!$entity instanceof UserAwareInterface && !$entity instanceof UserUpdaterAwareInterface) {
                continue;
            }

            if ($entity instanceof UserAwareInterface && !$entity->getUser()) {
                $entity->setUser($this->getUser());
            }

            if ($entity instanceof UserUpdaterAwareInterface && !$entity->getUpdater()) {
                $entity->setUpdater($this->getUser());
            }

            $entityManager->persist($entity);
            $metadata = $entityManager->getClassMetadata(get_class($entity));
            $unitOfWork->recomputeSingleEntityChangeSet($metadata, $entity);
        }
    }

    /**
     * @return UserInterface|null
     */
    private function getUser()
    {
        return $this->security->getUser();
    }
}