Nie możesz wybrać więcej, niż 25 tematów Tematy muszą się zaczynać od litery lub cyfry, mogą zawierać myślniki ('-') i mogą mieć do 35 znaków.

221 lines
5.1KB

  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. public function orderBy(string $field, string $sort = 'ASC'): self
  107. {
  108. if (strpos($field, '.')!==false) {
  109. return $this->addOrderBy($field, $sort);
  110. } else {
  111. return $this->addOrderBy('.' . $field, $sort);
  112. }
  113. }
  114. public function filterById(int $id):self
  115. {
  116. return $this
  117. ->andWhere('.id = :id')
  118. ->setParameter('id', $id);
  119. }
  120. public function andWhereEqual($field, $value)
  121. {
  122. return $this->andWhere('.'.$field.' = :'.$field)->setParameter($field, $value);
  123. }
  124. public function filterByOldUrl(string $oldUrl): self
  125. {
  126. return $this->andWhere('.oldUrls LIKE :oldUrl')->setParameter('oldUrl', '%'.$oldUrl.'%');
  127. }
  128. /*
  129. * DEVALIAS
  130. */
  131. public function filterByDevAlias(string $devAlias): self
  132. {
  133. return $this
  134. ->andWhere('.devAlias = :devAlias')
  135. ->setParameter('devAlias', $devAlias);
  136. }
  137. /*
  138. * SLUG
  139. */
  140. public function filterBySlug(string $slug):self
  141. {
  142. return $this
  143. ->andWhere('.slug = :slug')
  144. ->setParameter('slug', $slug);
  145. }
  146. /*
  147. * TREE
  148. */
  149. public function filterIsParent():self
  150. {
  151. return $this->andWhere('.parent IS NULL');
  152. }
  153. public function filterIsChildren():self
  154. {
  155. return $this->andWhere('.parent IS NOT NULL');
  156. }
  157. public function filterByParent(EntityInterface $parent = null):self
  158. {
  159. return $this->andWhere('.parent = :parent')->setParameter('parent', $parent);
  160. }
  161. /*
  162. * STATUS
  163. */
  164. public function filterIsOffline():self
  165. {
  166. return $this->andWhereStatus($this->id, 0);
  167. }
  168. public function filterIsOnline():self
  169. {
  170. return $this->andWhereStatus($this->id, 1);
  171. }
  172. public function filterIsDeleted():self
  173. {
  174. return $this->andWhereStatus($this->id, -1);
  175. }
  176. public function filterIsOnlineAndOffline():self
  177. {
  178. return $this->andWhere('.status >= 0');
  179. }
  180. }