src/EventSubscriber/WorklogEventSubscriber.php line 33

Open in your IDE?
  1. <?php
  2. namespace App\EventSubscriber;
  4. use App\Entity\User;
  5. use Doctrine\ORM\EntityManagerInterface;
  6. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  7. use App\Event\WorklogEvent;
  8. use Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface;
  10. class WorklogEventSubscriber implements EventSubscriberInterface
  11. {
  12.     /** @var EntityManagerInterface */
  13.     protected $entityManager;
  15.     /** @var TokenStorageInterface */
  16.     protected $tokenStorage;
  18.     public function __construct(
  19.         EntityManagerInterface $entityManager,
  20.         TokenStorageInterface $tokenStorage
  21.     ) {
  22.         $this->entityManager = $entityManager;
  23.         $this->tokenStorage = $tokenStorage;
  24.     }
  26.     public static function getSubscribedEvents()
  27.     {
  28.         return array(
  29.             'iris.worklog' => 'handleWorklog',
  30.         );
  31.     }
  33.     public function handleWorklog(WorklogEvent $event)
  34.     {
  35.         $worklog = $event->getWorklog();
  36.         $user = $event->getUser();
  38.         if ($this->tokenStorage->getToken() && $this->tokenStorage->getToken()->getUser()) {
  39.             $worklog->setUserActivity($this->tokenStorage->getToken()->getUser());
  40.         } else if ($user) {
  41.             $worklog->setUserActivity($this->entityManager->getReference(User::class, $user->getId()));
  42.         }
  44.         $this->entityManager->persist($worklog);
  45.         $this->entityManager->flush();
  46.         $this->entityManager->detach($worklog);
  47.     }
  48. }