src/EventSubscriber/ECommerce/ClientSubscriber.php line 57

Open in your IDE?
  1. <?php
  3. namespace App\EventSubscriber\ECommerce;
  5. use Doctrine\ORM\Event\LifecycleEventArgs;
  6. use Doctrine\ORM\EntityManagerInterface;
  7. use App\Entity\ECommerce\Currency;
  8. use Symfony\Component\EventDispatcher\EventSubscriberInterface;
  9. use App\Entity\ECommerce\Client;
  10. use App\Entity\ECommerce\Firm;
  11. use Doctrine\ORM\Events;
  13. /**
  14. * Class ClientSubscriber
  15. *
  16. * @package App\EventSubscriber\ECommerce
  17. */
  18. class ClientSubscriber implements EventSubscriberInterface
  19. {
  20. /**
  21. * @var EntityManagerInterface
  22. */
  23. protected $entityManager;
  25. /**
  26. * ClientSubscriber constructor.
  27. *
  28. * @param EntityManagerInterface $entityManager
  29. */
  30. public function __construct(EntityManagerInterface $entityManager)
  31. {
  32. $this->entityManager = $entityManager;
  33. }
  35. /**
  36. * @return array
  37. */
  38. public static function getSubscribedEvents(): array
  39. {
  40. return array(
  41. Events::postLoad => 'postLoad',
  42. Events::prePersist => 'prePersist',
  43. );
  44. }
  46. /**
  47. * @param LifecycleEventArgs $args
  48. */
  49. public function postLoad(LifecycleEventArgs $args)
  50. {
  51. $this->checkFirm($args);
  52. }
  54. /**
  55. * @param LifecycleEventArgs $args
  56. */
  57. public function prePersist(LifecycleEventArgs $args)
  58. {
  59. $this->checkFirm($args);
  60. $this->checkCurrency($args);
  61. }
  63. /**
  64. * @param LifecycleEventArgs $args
  65. */
  66. protected function checkCurrency(LifecycleEventArgs $args)
  67. {
  68. $client = $args->getObject();
  69. if (!$client instanceof Client) {
  70. return;
  71. }
  73. $client->setCurrency($this->entityManager->getRepository(Currency::class)->findFirst());
  74. }
  76. /**
  77. * @param LifecycleEventArgs $args
  78. */
  79. protected function checkFirm(LifecycleEventArgs $args)
  80. {
  81. $client = $args->getObject();
  82. if (!$client instanceof Client) {
  83. return;
  84. }
  86. $this->setFirm($client);
  87. }
  89. /**
  90. * @param Client $client
  91. */
  92. public function setFirm(Client $client)
  93. {
  94. if ($client->getFirm() instanceof Firm) {
  95. return $client;
  96. }
  98. $firm = $this->entityManager->getRepository(Firm::class)->findOneByCurrency($client->getCurrency());
  99. return $client->setFirm($firm);
  100. }
  101. }