PiouPiou\RibsAdminBundle\EventListener\CreateUpdateAwareListener

Complexity

Total Complexity

5

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	1

3 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	3	1
A	preUpdate()	0	4	1
A	prePersist()	0	9	3

<?php

namespace PiouPiou\RibsAdminBundle\EventListener;

use PiouPiou\RibsAdminBundle\Entity\User;
use Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorage;

class CreateUpdateAwareListener
{
	/**
	 * @var User
	 */
	private $user;

    /**
     * CreateUpdateAwareListener constructor.
     * @param TokenStorage $tokenStorage
     */
    public function __construct(TokenStorage $tokenStorage)
	{
        $this->user = $tokenStorage->getToken()->getUser()->getUser();
	}
	
	public function prePersist($entity)
	{
		if ($entity->getCreatedBy() === null) {
            $entity->setCreatedAt(new \DateTime());
            $entity->setCreatedBy($this->user);
		}
        if ($entity->getUpdatedBy() === null) {
            $entity->setUpdatedAt(new \DateTime());
            $entity->setUpdatedBy($this->user);
        }
	}

    public function preUpdate($entity)
    {
        $entity->setUpdatedAt(new \DateTime());
        $entity->setUpdatedBy($this->user);
    }
}