You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

222 lines
5.2KB

  1. <?php
  2. namespace Lc\SovBundle\Repository;
  3. use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepository;
  4. use Doctrine\ORM\QueryBuilder;
  5. use Knp\Component\Pager\PaginatorInterface;
  6. use Lc\SovBundle\Doctrine\EntityInterface;
  7. abstract class AbstractRepositoryQuery implements RepositoryQueryInterface
  8. {
  9. protected ServiceEntityRepository $repository;
  10. protected QueryBuilder $query;
  11. protected PaginatorInterface $paginator;
  12. protected string $id;
  13. public function __construct(ServiceEntityRepository $repository, string $id, PaginatorInterface $paginator = null)
  14. {
  15. $this->repository = $repository;
  16. $this->query = $repository->createQueryBuilder($id);
  17. $this->paginator = $paginator;
  18. $this->id = $id;
  19. }
  20. public function __call(string $name, $params): self
  21. {
  22. foreach ($params as $key => $value) {
  23. $this->populateDqlId($params[$key]);
  24. }
  25. call_user_func_array([$this->query, $name], $params);
  26. return $this;
  27. }
  28. public function create()
  29. {
  30. $class = get_called_class();
  31. return new $class($this->repository, $this->paginator);
  32. }
  33. public function call(callable $fn): self
  34. {
  35. $fn($this->query, $this);
  36. return $this;
  37. }
  38. public function count()
  39. {
  40. return $this->query->getQuery()
  41. ->getSingleScalarResult();
  42. }
  43. public function findOne()
  44. {
  45. return $this->query->getQuery()
  46. ->setMaxResults(1)
  47. ->getOneOrNullResult();
  48. }
  49. public function find(): array
  50. {
  51. return $this->query->getQuery()->getResult();
  52. }
  53. public function limit(int $maxResults):self
  54. {
  55. $this->query->setMaxResults($maxResults);
  56. return $this;
  57. }
  58. public function paginate(int $page = 1, int $limit = 20)
  59. {
  60. return $this->paginator->paginate($this->query->getQuery(), $page, $limit);
  61. }
  62. public function getRepository(): ServiceEntityRepository
  63. {
  64. return $this->repository;
  65. }
  66. public function getQueryBuilder(): QueryBuilder
  67. {
  68. return $this->query;
  69. }
  70. protected function populateDqlId(&$data)
  71. {
  72. if (is_string($data)) {
  73. $words = explode(' ', $data);
  74. foreach ($words as $k => $v) {
  75. if (isset($v[0]) && '.' === $v[0]) {
  76. $words[$k] = $this->id . $v;
  77. }
  78. }
  79. $data = implode(' ', $words);
  80. } elseif (is_array($data)) {
  81. foreach ($data as $k => $v) {
  82. $this->populateDqlId($data[$k]);
  83. }
  84. }
  85. return $data;
  86. }
  87. public function groupBy(string $field): self
  88. {
  89. if (strpos($field, '.')!==false) {
  90. $this->addGroupBy($field) ;
  91. } else {
  92. $this->addGroupBy('.'.$field) ;
  93. }
  94. return $this;
  95. }
  96. /*
  97. public function addGroupBy(string $field): self
  98. {
  99. if (strpos($field, '.')!==false) {
  100. $this->query->addGroupBy($field) ;
  101. } else {
  102. $this->query->addGroupBy('.'.$field) ;
  103. }
  104. return $this;
  105. }*/
  106. // @TODO : créer un addOrderBy et un orderBy
  107. public function orderBy(string $field, string $sort = 'ASC'): self
  108. {
  109. if (strpos($field, '.')!==false) {
  110. return $this->addOrderBy($field, $sort);
  111. } else {
  112. return $this->addOrderBy('.' . $field, $sort);
  113. }
  114. }
  115. public function filterById(int $id):self
  116. {
  117. return $this
  118. ->andWhere('.id = :id')
  119. ->setParameter('id', $id);
  120. }
  121. public function andWhereEqual($field, $value)
  122. {
  123. return $this->andWhere('.'.$field.' = :'.$field)->setParameter($field, $value);
  124. }
  125. public function filterByOldUrl(string $oldUrl): self
  126. {
  127. return $this->andWhere('.oldUrls LIKE :oldUrl')->setParameter('oldUrl', '%'.$oldUrl.'%');
  128. }
  129. /*
  130. * DEVALIAS
  131. */
  132. public function filterByDevAlias(string $devAlias): self
  133. {
  134. return $this
  135. ->andWhere('.devAlias = :devAlias')
  136. ->setParameter('devAlias', $devAlias);
  137. }
  138. /*
  139. * SLUG
  140. */
  141. public function filterBySlug(string $slug):self
  142. {
  143. return $this
  144. ->andWhere('.slug = :slug')
  145. ->setParameter('slug', $slug);
  146. }
  147. /*
  148. * TREE
  149. */
  150. public function filterIsParent():self
  151. {
  152. return $this->andWhere('.parent IS NULL');
  153. }
  154. public function filterIsChildren():self
  155. {
  156. return $this->andWhere('.parent IS NOT NULL');
  157. }
  158. public function filterByParent(EntityInterface $parent = null):self
  159. {
  160. return $this->andWhere('.parent = :parent')->setParameter('parent', $parent);
  161. }
  162. /*
  163. * STATUS
  164. */
  165. public function filterIsOffline():self
  166. {
  167. return $this->andWhereStatus($this->id, 0);
  168. }
  169. public function filterIsOnline():self
  170. {
  171. return $this->andWhereStatus($this->id, 1);
  172. }
  173. public function filterIsDeleted():self
  174. {
  175. return $this->andWhereStatus($this->id, -1);
  176. }
  177. public function filterIsOnlineAndOffline():self
  178. {
  179. return $this->andWhere('.status >= 0');
  180. }
  181. }