Passed
Push — master ( e2a719...9eb666 )
by Alan
03:22
created

PaginationExtension.php$0 ➔ addCountToContext()   A

Complexity

Conditions 4

Size

Total Lines 11

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
cc 4
dl 0
loc 11
rs 9.9
c 0
b 0
f 0
1
<?php
2
3
/*
4
 * This file is part of the API Platform project.
5
 *
6
 * (c) Kévin Dunglas <[email protected]>
7
 *
8
 * For the full copyright and license information, please view the LICENSE
9
 * file that was distributed with this source code.
10
 */
11
12
declare(strict_types=1);
13
14
namespace ApiPlatform\Core\Bridge\Doctrine\Orm\Extension;
15
16
use ApiPlatform\Core\Bridge\Doctrine\Orm\AbstractPaginator;
17
use ApiPlatform\Core\Bridge\Doctrine\Orm\Paginator;
18
use ApiPlatform\Core\Bridge\Doctrine\Orm\Util\QueryChecker;
19
use ApiPlatform\Core\Bridge\Doctrine\Orm\Util\QueryNameGeneratorInterface;
20
use ApiPlatform\Core\DataProvider\Pagination;
21
use ApiPlatform\Core\Exception\InvalidArgumentException;
22
use ApiPlatform\Core\Metadata\Resource\Factory\ResourceMetadataFactoryInterface;
23
use ApiPlatform\Core\Metadata\Resource\ResourceMetadata;
24
use Doctrine\Common\Persistence\ManagerRegistry;
25
use Doctrine\ORM\QueryBuilder;
26
use Doctrine\ORM\Tools\Pagination\Paginator as DoctrineOrmPaginator;
27
use Symfony\Component\HttpFoundation\Request;
28
use Symfony\Component\HttpFoundation\RequestStack;
29
30
/**
31
 * Applies pagination on the Doctrine query for resource collection when enabled.
32
 *
33
 * @author Kévin Dunglas <[email protected]>
34
 * @author Samuel ROZE <[email protected]>
35
 */
36
final class PaginationExtension implements ContextAwareQueryResultCollectionExtensionInterface
37
{
38
    private $managerRegistry;
39
    private $requestStack;
40
    private $resourceMetadataFactory;
41
    private $enabled;
42
    private $clientEnabled;
43
    private $clientItemsPerPage;
44
    private $itemsPerPage;
45
    private $pageParameterName;
46
    private $enabledParameterName;
47
    private $itemsPerPageParameterName;
48
    private $maximumItemPerPage;
49
    private $partial;
50
    private $clientPartial;
51
    private $partialParameterName;
52
    private $pagination;
53
54
    /**
55
     * @param ResourceMetadataFactoryInterface|RequestStack $resourceMetadataFactory
56
     * @param Pagination|ResourceMetadataFactoryInterface   $pagination
57
     */
58
    public function __construct(ManagerRegistry $managerRegistry, /* ResourceMetadataFactoryInterface */ $resourceMetadataFactory, /* Pagination */ $pagination)
59
    {
60
        if ($resourceMetadataFactory instanceof RequestStack && $pagination instanceof ResourceMetadataFactoryInterface) {
61
            @trigger_error(sprintf('Passing an instance of "%s" as second argument of "%s" is deprecated since API Platform 2.4 and will not be possible anymore in API Platform 3. Pass an instance of "%s" instead.', RequestStack::class, self::class, ResourceMetadataFactoryInterface::class), E_USER_DEPRECATED);
62
            @trigger_error(sprintf('Passing an instance of "%s" as third argument of "%s" is deprecated since API Platform 2.4 and will not be possible anymore in API Platform 3. Pass an instance of "%s" instead.', ResourceMetadataFactoryInterface::class, self::class, Pagination::class), E_USER_DEPRECATED);
63
64
            $this->requestStack = $resourceMetadataFactory;
65
            $resourceMetadataFactory = $pagination;
66
            $pagination = null;
67
68
            $args = \array_slice(\func_get_args(), 3);
69
            $legacyPaginationArgs = [
70
                ['arg_name' => 'enabled', 'type' => 'bool', 'default' => true],
71
                ['arg_name' => 'clientEnabled', 'type' => 'bool', 'default' => false],
72
                ['arg_name' => 'clientItemsPerPage', 'type' => 'bool', 'default' => false],
73
                ['arg_name' => 'itemsPerPage', 'type' => 'int', 'default' => 30],
74
                ['arg_name' => 'pageParameterName', 'type' => 'string', 'default' => 'page'],
75
                ['arg_name' => 'enabledParameterName', 'type' => 'string', 'default' => 'pagination'],
76
                ['arg_name' => 'itemsPerPageParameterName', 'type' => 'string', 'default' => 'itemsPerPage'],
77
                ['arg_name' => 'maximumItemPerPage', 'type' => 'int', 'default' => null],
78
                ['arg_name' => 'partial', 'type' => 'bool', 'default' => false],
79
                ['arg_name' => 'clientPartial', 'type' => 'bool', 'default' => false],
80
                ['arg_name' => 'partialParameterName', 'type' => 'string', 'default' => 'partial'],
81
            ];
82
83
            foreach ($legacyPaginationArgs as $pos => $arg) {
84
                if (\array_key_exists($pos, $args)) {
85
                    @trigger_error(sprintf('Passing "$%s" arguments is deprecated since API Platform 2.4 and will not be possible anymore in API Platform 3. Pass an instance of "%s" as third argument instead.', implode('", "$', array_column($legacyPaginationArgs, 'arg_name')), Paginator::class), E_USER_DEPRECATED);
86
87
                    if (!((null === $arg['default'] && null === $args[$pos]) || \call_user_func("is_{$arg['type']}", $args[$pos]))) {
88
                        throw new InvalidArgumentException(sprintf('The "$%s" argument is expected to be a %s%s.', $arg['arg_name'], $arg['type'], null === $arg['default'] ? ' or null' : ''));
89
                    }
90
91
                    $value = $args[$pos];
92
                } else {
93
                    $value = $arg['default'];
94
                }
95
96
                $this->{$arg['arg_name']} = $value;
97
            }
98
        } elseif (!$resourceMetadataFactory instanceof ResourceMetadataFactoryInterface) {
99
            throw new InvalidArgumentException(sprintf('The "$resourceMetadataFactory" argument is expected to be an implementation of the "%s" interface.', ResourceMetadataFactoryInterface::class));
100
        } elseif (!$pagination instanceof Pagination) {
101
            throw new InvalidArgumentException(sprintf('The "$pagination" argument is expected to be an instance of the "%s" class.', Pagination::class));
102
        }
103
104
        $this->managerRegistry = $managerRegistry;
105
        $this->resourceMetadataFactory = $resourceMetadataFactory;
106
        $this->pagination = $pagination;
107
    }
108
109
    /**
110
     * {@inheritdoc}
111
     */
112
    public function applyToCollection(QueryBuilder $queryBuilder, QueryNameGeneratorInterface $queryNameGenerator, string $resourceClass, string $operationName = null, array $context = [])
113
    {
114
        if (null === $pagination = $this->getPagination($queryBuilder, $resourceClass, $operationName, $context)) {
115
            return;
116
        }
117
118
        [$offset, $limit] = $pagination;
119
120
        $queryBuilder
121
            ->setFirstResult($offset)
122
            ->setMaxResults($limit);
123
    }
124
125
    /**
126
     * {@inheritdoc}
127
     */
128
    public function supportsResult(string $resourceClass, string $operationName = null, array $context = []): bool
129
    {
130
        if (null === $this->requestStack) {
131
            return $this->pagination->isEnabled($resourceClass, $operationName, $context);
132
        }
133
134
        if (null === $request = $this->requestStack->getCurrentRequest()) {
135
            return false;
136
        }
137
138
        return $this->isPaginationEnabled($request, $this->resourceMetadataFactory->create($resourceClass), $operationName);
139
    }
140
141
    /**
142
     * {@inheritdoc}
143
     */
144
    public function getResult(QueryBuilder $queryBuilder, string $resourceClass = null, string $operationName = null, array $context = [])
145
    {
146
        $doctrineOrmPaginator = new DoctrineOrmPaginator($queryBuilder, $this->useFetchJoinCollection($queryBuilder, $resourceClass, $operationName));
147
        $doctrineOrmPaginator->setUseOutputWalkers($this->useOutputWalkers($queryBuilder));
148
149
        if (null === $this->requestStack) {
150
            $isPartialEnabled = $this->pagination->isPartialEnabled($resourceClass, $operationName, $context);
151
        } else {
152
            $isPartialEnabled = $this->isPartialPaginationEnabled(
153
                $this->requestStack->getCurrentRequest(),
154
                null === $resourceClass ? null : $this->resourceMetadataFactory->create($resourceClass),
155
                $operationName
156
            );
157
        }
158
159
        if ($isPartialEnabled) {
160
            return new class($doctrineOrmPaginator) extends AbstractPaginator {
161
            };
162
        }
163
164
        return new Paginator($doctrineOrmPaginator);
165
    }
166
167
    /**
168
     * @throws InvalidArgumentException
169
     */
170
    private function getPagination(QueryBuilder $queryBuilder, string $resourceClass, ?string $operationName, array $context): ?array
171
    {
172
        $request = null;
173
        if (null !== $this->requestStack && null === $request = $this->requestStack->getCurrentRequest()) {
174
            return null;
175
        }
176
177
        if (null === $request) {
178
            if (!$this->pagination->isEnabled($resourceClass, $operationName, $context)) {
179
                return null;
180
            }
181
182
            $context = $this->addCountToContext($queryBuilder, $context);
183
184
            return \array_slice($this->pagination->getPagination($resourceClass, $operationName, $context), 1);
185
        }
186
187
        $resourceMetadata = $this->resourceMetadataFactory->create($resourceClass);
188
        if (!$this->isPaginationEnabled($request, $resourceMetadata, $operationName)) {
189
            return null;
190
        }
191
192
        $itemsPerPage = $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_items_per_page', $this->itemsPerPage, true);
193
        if ($request->attributes->get('_graphql')) {
194
            $collectionArgs = $request->attributes->get('_graphql_collections_args', []);
195
            $itemsPerPage = $collectionArgs[$resourceClass]['first'] ?? $itemsPerPage;
196
        }
197
198
        if ($resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_client_items_per_page', $this->clientItemsPerPage, true)) {
199
            $maxItemsPerPage = $resourceMetadata->getCollectionOperationAttribute($operationName, 'maximum_items_per_page', $this->maximumItemPerPage, true);
200
            $itemsPerPage = (int) $this->getPaginationParameter($request, $this->itemsPerPageParameterName, $itemsPerPage);
201
            $itemsPerPage = (null !== $maxItemsPerPage && $itemsPerPage >= $maxItemsPerPage ? $maxItemsPerPage : $itemsPerPage);
202
        }
203
204
        if (0 > $itemsPerPage) {
205
            throw new InvalidArgumentException('Item per page parameter should not be less than 0');
206
        }
207
208
        $page = (int) $this->getPaginationParameter($request, $this->pageParameterName, 1);
209
210
        if (1 > $page) {
211
            throw new InvalidArgumentException('Page should not be less than 1');
212
        }
213
214
        if (0 === $itemsPerPage && 1 < $page) {
215
            throw new InvalidArgumentException('Page should not be greater than 1 if itemsPerPage is equal to 0');
216
        }
217
218
        $firstResult = ($page - 1) * $itemsPerPage;
219
        if ($request->attributes->get('_graphql')) {
220
            $collectionArgs = $request->attributes->get('_graphql_collections_args', []);
221
            if (isset($collectionArgs[$resourceClass]['after'])) {
222
                $after = base64_decode($collectionArgs[$resourceClass]['after'], true);
223
                $firstResult = (int) $after;
224
                $firstResult = false === $after ? $firstResult : ++$firstResult;
225
            }
226
        }
227
228
        return [$firstResult, $itemsPerPage];
229
    }
230
231
    private function isPartialPaginationEnabled(Request $request = null, ResourceMetadata $resourceMetadata = null, string $operationName = null): bool
232
    {
233
        $enabled = $this->partial;
234
        $clientEnabled = $this->clientPartial;
235
236
        if ($resourceMetadata) {
237
            $enabled = $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_partial', $enabled, true);
238
239
            if ($request) {
240
                $clientEnabled = $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_client_partial', $clientEnabled, true);
241
            }
242
        }
243
244
        if ($clientEnabled && $request) {
245
            $enabled = filter_var($this->getPaginationParameter($request, $this->partialParameterName, $enabled), FILTER_VALIDATE_BOOLEAN);
246
        }
247
248
        return $enabled;
249
    }
250
251
    private function isPaginationEnabled(Request $request, ResourceMetadata $resourceMetadata, string $operationName = null): bool
252
    {
253
        $enabled = $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_enabled', $this->enabled, true);
254
        $clientEnabled = $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_client_enabled', $this->clientEnabled, true);
255
256
        if ($clientEnabled) {
257
            $enabled = filter_var($this->getPaginationParameter($request, $this->enabledParameterName, $enabled), FILTER_VALIDATE_BOOLEAN);
258
        }
259
260
        return $enabled;
261
    }
262
263
    private function getPaginationParameter(Request $request, string $parameterName, $default = null)
264
    {
265
        if (null !== $paginationAttribute = $request->attributes->get('_api_pagination')) {
266
            return \array_key_exists($parameterName, $paginationAttribute) ? $paginationAttribute[$parameterName] : $default;
267
        }
268
269
        return $request->query->get($parameterName, $default);
270
    }
271
272
    private function addCountToContext(QueryBuilder $queryBuilder, array $context): array
273
    {
274
        if (!($context['graphql'] ?? false)) {
275
            return $context;
276
        }
277
278
        if (isset($context['filters']['last']) && !isset($context['filters']['before'])) {
279
            $context['count'] = (new DoctrineOrmPaginator($queryBuilder))->count();
280
        }
281
282
        return $context;
283
    }
284
285
    /**
286
     * Determines whether the Paginator should fetch join collections, if the root entity uses composite identifiers it should not.
287
     *
288
     * @see https://github.com/doctrine/doctrine2/issues/2910
289
     */
290
    private function useFetchJoinCollection(QueryBuilder $queryBuilder, string $resourceClass = null, string $operationName = null): bool
291
    {
292
        if (QueryChecker::hasRootEntityWithCompositeIdentifier($queryBuilder, $this->managerRegistry)) {
293
            return false;
294
        }
295
296
        if (null === $resourceClass) {
297
            return true;
298
        }
299
300
        $resourceMetadata = $this->resourceMetadataFactory->create($resourceClass);
301
302
        return $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_fetch_join_collection', true, true);
303
    }
304
305
    /**
306
     * Determines whether output walkers should be used.
307
     */
308
    private function useOutputWalkers(QueryBuilder $queryBuilder): bool
309
    {
310
        /*
311
         * "Cannot count query that uses a HAVING clause. Use the output walkers for pagination"
312
         *
313
         * @see https://github.com/doctrine/doctrine2/blob/900b55d16afdcdeb5100d435a7166d3a425b9873/lib/Doctrine/ORM/Tools/Pagination/CountWalker.php#L50
314
         */
315
        if (QueryChecker::hasHavingClause($queryBuilder)) {
316
            return true;
317
        }
318
319
        /*
320
         * "Paginating an entity with foreign key as identifier only works when using the Output Walkers. Call Paginator#setUseOutputWalkers(true) before iterating the paginator."
321
         *
322
         * @see https://github.com/doctrine/doctrine2/blob/900b55d16afdcdeb5100d435a7166d3a425b9873/lib/Doctrine/ORM/Tools/Pagination/LimitSubqueryWalker.php#L87
323
         */
324
        if (QueryChecker::hasRootEntityWithForeignKeyIdentifier($queryBuilder, $this->managerRegistry)) {
325
            return true;
326
        }
327
328
        /*
329
         * "Cannot select distinct identifiers from query with LIMIT and ORDER BY on a column from a fetch joined to-many association. Use output walkers."
330
         *
331
         * @see https://github.com/doctrine/doctrine2/blob/900b55d16afdcdeb5100d435a7166d3a425b9873/lib/Doctrine/ORM/Tools/Pagination/LimitSubqueryWalker.php#L149
332
         */
333
        if (
334
            QueryChecker::hasMaxResults($queryBuilder) &&
335
            QueryChecker::hasOrderByOnToManyJoin($queryBuilder, $this->managerRegistry)
336
        ) {
337
            return true;
338
        }
339
340
        /*
341
         * When using composite identifiers pagination will need Output walkers
342
         */
343
        if (QueryChecker::hasRootEntityWithCompositeIdentifier($queryBuilder, $this->managerRegistry)) {
344
            return true;
345
        }
346
347
        // Disable output walkers by default (performance)
348
        return false;
349
    }
350
}
351