|
1
|
|
|
<?php |
|
2
|
|
|
|
|
3
|
|
|
declare(strict_types=1); |
|
4
|
|
|
|
|
5
|
|
|
namespace Atlance\HttpDoctrineOrmFilter; |
|
6
|
|
|
|
|
7
|
|
|
use Doctrine\ORM\Mapping\ClassMetadata; |
|
8
|
|
|
use Doctrine\ORM\Query\Expr; |
|
9
|
|
|
use Doctrine\ORM\Query\Expr\From; |
|
10
|
|
|
use Doctrine\ORM\Query\Expr\Join; |
|
11
|
|
|
use Doctrine\ORM\QueryBuilder; |
|
12
|
|
|
use InvalidArgumentException; |
|
13
|
|
|
use Psr\SimpleCache\CacheInterface; |
|
14
|
|
|
use Psr\SimpleCache\InvalidArgumentException as PsrException; |
|
15
|
|
|
use Symfony\Component\Validator\Exception\ValidatorException; |
|
16
|
|
|
|
|
17
|
|
|
final class Filter |
|
18
|
|
|
{ |
|
19
|
|
|
/** @var string */ |
|
20
|
|
|
public const CACHE_KEY_METADATA = 'metadata'; |
|
21
|
|
|
|
|
22
|
|
|
/** @var string */ |
|
23
|
|
|
public const CACHE_KEY_FIELD = 'field'; |
|
24
|
|
|
|
|
25
|
78 |
|
public function __construct(private readonly Query\Validator $validator, private readonly ?CacheInterface $cache = null) |
|
26
|
|
|
{ |
|
27
|
78 |
|
} |
|
28
|
|
|
|
|
29
|
|
|
/** |
|
30
|
|
|
* @throws InvalidArgumentException if the query arguments logic exception |
|
31
|
|
|
* @throws ValidatorException if not valid the query arguments value |
|
32
|
|
|
* @throws PsrException if cache problem |
|
33
|
|
|
*/ |
|
34
|
76 |
|
public function apply(QueryBuilder $qb, Query\Configuration $configuration): void |
|
35
|
|
|
{ |
|
36
|
76 |
|
$this |
|
37
|
76 |
|
->select($qb, $configuration->filter) |
|
38
|
76 |
|
->order($qb, $configuration->order); |
|
39
|
|
|
} |
|
40
|
|
|
|
|
41
|
|
|
/** |
|
42
|
|
|
* @throws InvalidArgumentException if the query arguments logic exception |
|
43
|
|
|
* @throws ValidatorException if not valid the query arguments value |
|
44
|
|
|
* @throws PsrException if cache problem |
|
45
|
|
|
*/ |
|
46
|
76 |
|
private function select(QueryBuilder $qb, array $conditions): self |
|
47
|
|
|
{ |
|
48
|
|
|
/** |
|
49
|
|
|
* @var string $expr |
|
50
|
|
|
* @var array $aliases |
|
51
|
|
|
*/ |
|
52
|
76 |
|
foreach ($conditions as $expr => $aliases) { |
|
53
|
|
|
/** |
|
54
|
|
|
* @var string $alias |
|
55
|
|
|
* @var array $values |
|
56
|
|
|
*/ |
|
57
|
75 |
|
foreach ($aliases as $alias => $values) { |
|
58
|
75 |
|
[$cacheKey,] = Cache\Keys\Generator::generate( |
|
59
|
75 |
|
self::CACHE_KEY_FIELD, |
|
60
|
75 |
|
$qb->getDQL(), |
|
61
|
75 |
|
['query' => sprintf('[%s][%s]', $expr, $alias)] |
|
62
|
75 |
|
); |
|
63
|
|
|
/** @var Query\Field|null $field */ |
|
64
|
75 |
|
$field = $this->cache?->get($cacheKey); |
|
65
|
75 |
|
if (!$field instanceof Query\Field) { |
|
66
|
75 |
|
$field = $this->createField($qb, $alias, $expr); |
|
67
|
|
|
} |
|
68
|
|
|
|
|
69
|
73 |
|
$this->andWhere($qb, $field, $values, $cacheKey); |
|
70
|
|
|
} |
|
71
|
|
|
} |
|
72
|
|
|
|
|
73
|
67 |
|
if (!$this->validator->isValid()) { |
|
74
|
3 |
|
throw new ValidatorException((string) json_encode($this->validator->getAllViolations(), \JSON_UNESCAPED_UNICODE)); |
|
75
|
|
|
} |
|
76
|
|
|
|
|
77
|
64 |
|
return $this; |
|
78
|
|
|
} |
|
79
|
|
|
|
|
80
|
64 |
|
private function order(QueryBuilder $qb, array $conditions): void |
|
81
|
|
|
{ |
|
82
|
|
|
/** |
|
83
|
|
|
* @var string $alias |
|
84
|
|
|
* @var string $value |
|
85
|
|
|
*/ |
|
86
|
64 |
|
foreach ($conditions as $alias => $value) { |
|
87
|
1 |
|
$snakeCaseExprMethod = 'order_by'; |
|
88
|
1 |
|
[$cacheKey,] = Cache\Keys\Generator::generate( |
|
89
|
1 |
|
self::CACHE_KEY_FIELD, |
|
90
|
1 |
|
$qb->getDQL(), |
|
91
|
1 |
|
['query' => sprintf('[%s][%s]', $snakeCaseExprMethod, $alias)] |
|
92
|
1 |
|
); |
|
93
|
|
|
/** @var Query\Field|null $field */ |
|
94
|
1 |
|
$field = $this->cache?->get($cacheKey); |
|
95
|
1 |
|
if (!$field instanceof Query\Field) { |
|
96
|
1 |
|
$field = $this->createField($qb, $alias, $snakeCaseExprMethod); |
|
97
|
|
|
} |
|
98
|
|
|
|
|
99
|
1 |
|
$this->andWhere($qb, $field, [$value], $cacheKey); |
|
100
|
|
|
} |
|
101
|
|
|
} |
|
102
|
|
|
|
|
103
|
74 |
|
private function andWhere(QueryBuilder $qb, Query\Field $field, array $values, string $cacheKey): void |
|
104
|
|
|
{ |
|
105
|
74 |
|
if ($this->isValid($field, $values)) { |
|
106
|
71 |
|
(new Query\Builder($qb))->andWhere($field->setValues($values)); |
|
107
|
64 |
|
$this->cache?->set($cacheKey, $field); |
|
108
|
|
|
} |
|
109
|
|
|
} |
|
110
|
|
|
|
|
111
|
76 |
|
private function createField(QueryBuilder $qb, string $tableAliasAndColumnName, string $expr): Query\Field |
|
112
|
|
|
{ |
|
113
|
76 |
|
foreach ($this->getAliasesAndMetadata($qb) as $alias => $metadata) { |
|
114
|
76 |
|
if (0 === strncasecmp($tableAliasAndColumnName, $alias . '_', mb_strlen($alias . '_'))) { |
|
115
|
75 |
|
$columnName = mb_substr($tableAliasAndColumnName, mb_strlen($alias . '_')); |
|
116
|
|
|
|
|
117
|
75 |
|
if (\array_key_exists($columnName, $metadata->fieldNames)) { |
|
118
|
74 |
|
return (new Query\Field($expr, $metadata->getName(), $alias)) |
|
119
|
74 |
|
->initProperties($metadata->getFieldMapping($metadata->getFieldForColumn($columnName))); |
|
120
|
|
|
} |
|
121
|
|
|
} |
|
122
|
|
|
} |
|
123
|
|
|
|
|
124
|
2 |
|
throw new \InvalidArgumentException($tableAliasAndColumnName . ' not allowed'); |
|
125
|
|
|
} |
|
126
|
|
|
|
|
127
|
|
|
/** |
|
128
|
|
|
* @psalm-suppress MixedReturnTypeCoercion |
|
129
|
|
|
* @psalm-suppress MixedAssignment |
|
130
|
|
|
* |
|
131
|
|
|
* @return array<string, ClassMetadata> |
|
132
|
|
|
* @throws PsrException |
|
133
|
|
|
*/ |
|
134
|
76 |
|
private function getAliasesAndMetadata(QueryBuilder $qb): array |
|
135
|
|
|
{ |
|
136
|
76 |
|
[$cacheKey,] = Cache\Keys\Generator::generate(self::CACHE_KEY_METADATA, $qb->getDQL()); |
|
137
|
76 |
|
if (!\is_array($aliasesAndMetadata = $this->cache?->get($cacheKey))) { |
|
138
|
76 |
|
$aliasesAndMetadata = []; |
|
139
|
|
|
|
|
140
|
76 |
|
foreach ($qb->getAllAliases() as $alias) { |
|
141
|
76 |
|
$metadata = $this->getMetadataByAlias($qb, $alias); |
|
142
|
76 |
|
if ($metadata instanceof ClassMetadata) { |
|
143
|
76 |
|
$aliasesAndMetadata[$alias] = $metadata; |
|
144
|
|
|
} |
|
145
|
|
|
} |
|
146
|
|
|
|
|
147
|
76 |
|
$this->cache?->set($cacheKey, $aliasesAndMetadata); |
|
148
|
|
|
} |
|
149
|
|
|
|
|
150
|
76 |
|
return $aliasesAndMetadata; |
|
151
|
|
|
} |
|
152
|
|
|
|
|
153
|
76 |
|
private function getMetadataByAlias(QueryBuilder $qb, string $alias): ?ClassMetadata |
|
154
|
|
|
{ |
|
155
|
76 |
|
$metadata = null; |
|
156
|
76 |
|
foreach ($this->getParts($qb) as $part) { |
|
157
|
76 |
|
if ($part->getAlias() !== $alias) { |
|
158
|
76 |
|
continue; |
|
159
|
|
|
} |
|
160
|
|
|
|
|
161
|
76 |
|
$metadata = $this->getMetadataByDQLPart($qb, $part); |
|
162
|
|
|
} |
|
163
|
|
|
|
|
164
|
76 |
|
return $metadata; |
|
165
|
|
|
} |
|
166
|
|
|
|
|
167
|
|
|
/** |
|
168
|
|
|
* @psalm-suppress MixedAssignment |
|
169
|
|
|
* @psalm-suppress MixedOperand |
|
170
|
|
|
* @psalm-suppress MixedInferredReturnType |
|
171
|
|
|
* @psalm-suppress MixedArrayAssignment |
|
172
|
|
|
* @psalm-suppress MixedReturnStatement |
|
173
|
|
|
* |
|
174
|
|
|
* @return Expr\From[]|Expr\Join[] |
|
175
|
|
|
*/ |
|
176
|
76 |
|
private function getParts(QueryBuilder $qb): array |
|
177
|
|
|
{ |
|
178
|
76 |
|
$parts = []; |
|
179
|
|
|
/** @var Expr\From[]|Expr\Join[] $tmp */ |
|
180
|
76 |
|
$tmp = $qb->getDQLPart('join') + $qb->getDQLPart('from'); |
|
181
|
76 |
|
array_walk_recursive($tmp, static function ($part) use (&$parts): void {$parts[] = $part; }); |
|
182
|
76 |
|
unset($tmp); |
|
183
|
|
|
|
|
184
|
76 |
|
return $parts; |
|
185
|
|
|
} |
|
186
|
|
|
|
|
187
|
|
|
/** |
|
188
|
|
|
* For join without class name. |
|
189
|
|
|
* Example: ->leftJoin('users.cards', 'cards', Join::WITH). |
|
190
|
|
|
*/ |
|
191
|
76 |
|
private function getMetadataByDQLPart(QueryBuilder $qb, Join | From $partData): ClassMetadata |
|
192
|
|
|
{ |
|
193
|
76 |
|
if (!class_exists($class = $partData instanceof From ? $partData->getFrom() : $partData->getJoin())) { |
|
194
|
76 |
|
$joinAlias = explode('.', $class)[1]; |
|
195
|
|
|
|
|
196
|
76 |
|
foreach ($qb->getRootEntities() as $rootEntity) { |
|
197
|
76 |
|
$class = $qb->getEntityManager()->getClassMetadata($rootEntity)->getAssociationTargetClass($joinAlias); |
|
198
|
|
|
} |
|
199
|
|
|
} |
|
200
|
|
|
|
|
201
|
76 |
|
return $qb->getEntityManager()->getClassMetadata($class); |
|
202
|
|
|
} |
|
203
|
|
|
|
|
204
|
74 |
|
private function isValid(Query\Field $field, array $values): bool |
|
205
|
|
|
{ |
|
206
|
74 |
|
$exp = $field->getSnakeCaseExprMethod(); |
|
207
|
|
|
|
|
208
|
74 |
|
if ($this->skipValidate($exp)) { |
|
209
|
32 |
|
return true; |
|
210
|
|
|
} |
|
211
|
|
|
|
|
212
|
42 |
|
$this->validator->validatePropertyValue( |
|
213
|
42 |
|
sprintf('[%s][%s]', $field->getExprMethod(), $field->generateParameter()), |
|
214
|
42 |
|
$field->getClass(), |
|
215
|
42 |
|
$field->getFieldName(), |
|
216
|
42 |
|
$values |
|
217
|
42 |
|
); |
|
218
|
|
|
|
|
219
|
42 |
|
return $this->validator->isValid(); |
|
220
|
|
|
} |
|
221
|
|
|
|
|
222
|
74 |
|
private function skipValidate(string $exp): bool |
|
223
|
|
|
{ |
|
224
|
74 |
|
return \in_array($exp, ['is_null', 'is_not_null', 'like', 'ilike', 'not_like', 'between', 'order_by'], true); |
|
225
|
|
|
} |
|
226
|
|
|
} |
|
227
|
|
|
|