src/Infrastructure/CommonBundle/Form/DataTransformer/ObjectToIdTransformer.php line 37

Open in your IDE?
  1. <?php
  2. namespace Whater\Infrastructure\CommonBundle\Form\DataTransformer;
  4. use Doctrine\Persistence\ObjectManager;
  5. use Doctrine\Persistence\ManagerRegistry;
  6. use Doctrine\Common\Persistence\ObjectRepository;
  7. use Doctrine\ORM\EntityManager;
  8. use Doctrine\ORM\EntityRepository;
  9. use Symfony\Component\Form\DataTransformerInterface;
  10. use Symfony\Component\Form\Exception\InvalidConfigurationException;
  11. use Symfony\Component\Form\Exception\TransformationFailedException;
  13. /**
  14. * Class ObjectToIdTransformer
  15. *
  16. * @package Whater\Infrastructure\CommonBundle\Form\DataTransformer
  17. */
  18. class ObjectToIdTransformer implements DataTransformerInterface
  19. {
  20. /** @var string */
  21. protected $class;
  23. /** @var string */
  24. protected $property;
  26. /** @var EntityManager */
  27. protected $em;
  29. /** @var EntityRepository */
  30. protected $repository;
  32. /**
  33. * @param ManagerRegistry $registry
  34. * @param string $class
  35. * @param string $property
  36. */
  37. public function __construct(ManagerRegistry $registry, $em, $class, $property)
  38. {
  39. $this->class = $class;
  40. $this->property = $property;
  41. $this->em = $this->getObjectManager($registry, $em);
  42. $this->repository = $this->getObjectRepository($this->em, $this->class);
  43. }
  45. /**
  46. * @param mixed $entity
  47. *
  48. * @return mixed|null
  49. */
  50. public function transform($entity)
  51. {
  52. if (null === $entity) {
  53. return null;
  54. }
  56. $className = $this->repository->getClassName();
  57. if (!$entity instanceof $className) {
  58. throw new TransformationFailedException(sprintf('Object must be instance of %s, instance of %s has given.', $className, get_class($entity)));
  59. }
  61. $methodName = $this->property;
  62. if (!method_exists($entity, $methodName)) {
  63. $methodName = 'get' . ucfirst($this->property);
  64. if (!method_exists($entity, $methodName)) {
  65. throw new InvalidConfigurationException(sprintf('There is no getter for property "%s" in class "%s".', $this->property, $this->class));
  66. }
  67. }
  69. return $entity->{$methodName}();
  70. }
  72. /**
  73. * @param mixed $id
  74. *
  75. * @return mixed|null|object
  76. */
  77. public function reverseTransform($id)
  78. {
  79. if (!$id) {
  80. return null;
  81. }
  83. $property = $this->property;
  84. if(strcmp($property, "id") == 0){
  85. $property = "uuid";
  86. }
  87. $entity = $this->repository->findOneBy([$property => $id]);
  88. if (null === $entity) {
  89. throw new TransformationFailedException(sprintf('Can\'t find entity of class "%s" with property "%s" = "%s".', $this->class, $this->property, $id));
  90. }
  92. return $entity;
  93. }
  95. /**
  96. * @param ManagerRegistry $registry
  97. * @param ObjectManager|string $omName
  98. *
  99. * @return ObjectManager
  100. */
  101. private function getObjectManager(ManagerRegistry $registry, $omName)
  102. {
  103. if ($omName instanceof ObjectManager) {
  104. return $omName;
  105. }
  107. $omName = (string)$omName;
  108. if ($om = $registry->getManager($omName)) {
  109. return $om;
  110. }
  112. throw new InvalidConfigurationException(sprintf('Doctrine Manager named "%s" does not exist.', $omName));
  113. }
  115. /**
  116. * @param ObjectManager $om
  117. * @param string $class
  118. *
  119. * @return ObjectRepository
  120. */
  121. private function getObjectRepository(ObjectManager $om, $class)
  122. {
  123. if ($repo = $om->getRepository($class)) {
  124. return $repo;
  125. }
  126. throw new InvalidConfigurationException(sprintf('Repository for class "%s" does not exist.', $class));
  127. }
  128. }