Passed
Pull Request — master (#2446)
by
unknown
07:16 queued 03:54
created

php$0 ➔ isPartialPaginationEnabled()   A

Complexity

Conditions 5

Size

Total Lines 18

Duplication

Lines 0
Ratio 0 %

Importance

Changes 0
Metric Value
dl 0
loc 18
rs 9.3554
c 0
b 0
f 0
cc 5
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 = null, string $operationName = null, array $context = [])
113
    {
114
        if (null === $pagination = $this->getPagination($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(?string $resourceClass, ?string $operationName, array $context): ?array
171
    {
172
        if (null === $resourceClass) {
173
            throw new InvalidArgumentException('The "$resourceClass" parameter must not be null');
174
        }
175
176
        $request = null;
177
        if (null !== $this->requestStack && null === $request = $this->requestStack->getCurrentRequest()) {
178
            return null;
179
        }
180
181
        if (null === $request) {
182
            if (!$this->pagination->isEnabled($resourceClass, $operationName, $context)) {
183
                return null;
184
            }
185
186
            return \array_slice($this->pagination->getPagination($resourceClass, $operationName, $context), 1);
187
        }
188
189
        $resourceMetadata = $this->resourceMetadataFactory->create($resourceClass);
190
        if (!$this->isPaginationEnabled($request, $resourceMetadata, $operationName)) {
191
            return null;
192
        }
193
194
        $itemsPerPage = $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_items_per_page', $this->itemsPerPage, true);
195
        if ($request->attributes->get('_graphql')) {
196
            $collectionArgs = $request->attributes->get('_graphql_collections_args', []);
197
            $itemsPerPage = $collectionArgs[$resourceClass]['first'] ?? $itemsPerPage;
198
        }
199
200
        if ($resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_client_items_per_page', $this->clientItemsPerPage, true)) {
201
            $maxItemsPerPage = $resourceMetadata->getCollectionOperationAttribute($operationName, 'maximum_items_per_page', $this->maximumItemPerPage, true);
202
            $itemsPerPage = (int) $this->getPaginationParameter($request, $this->itemsPerPageParameterName, $itemsPerPage);
203
            $itemsPerPage = (null !== $maxItemsPerPage && $itemsPerPage >= $maxItemsPerPage ? $maxItemsPerPage : $itemsPerPage);
204
        }
205
206
        if (0 > $itemsPerPage) {
207
            throw new InvalidArgumentException('Item per page parameter should not be less than 0');
208
        }
209
210
        $page = (int) $this->getPaginationParameter($request, $this->pageParameterName, 1);
211
212
        if (1 > $page) {
213
            throw new InvalidArgumentException('Page should not be less than 1');
214
        }
215
216
        if (0 === $itemsPerPage && 1 < $page) {
217
            throw new InvalidArgumentException('Page should not be greater than 1 if itemsPerPage is equal to 0');
218
        }
219
220
        $firstResult = ($page - 1) * $itemsPerPage;
221
        if ($request->attributes->get('_graphql')) {
222
            $collectionArgs = $request->attributes->get('_graphql_collections_args', []);
223
            if (isset($collectionArgs[$resourceClass]['after'])) {
224
                $after = base64_decode($collectionArgs[$resourceClass]['after'], true);
225
                $firstResult = (int) $after;
226
                $firstResult = false === $after ? $firstResult : ++$firstResult;
227
            }
228
        }
229
230
        return [$firstResult, $itemsPerPage];
231
    }
232
233
    private function isPartialPaginationEnabled(Request $request = null, ResourceMetadata $resourceMetadata = null, string $operationName = null): bool
234
    {
235
        $enabled = $this->partial;
236
        $clientEnabled = $this->clientPartial;
237
238
        if ($resourceMetadata) {
239
            $enabled = $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_partial', $enabled, true);
240
241
            if ($request) {
242
                $clientEnabled = $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_client_partial', $clientEnabled, true);
243
            }
244
        }
245
246
        if ($clientEnabled && $request) {
247
            $enabled = filter_var($this->getPaginationParameter($request, $this->partialParameterName, $enabled), FILTER_VALIDATE_BOOLEAN);
248
        }
249
250
        return $enabled;
251
    }
252
253
    private function isPaginationEnabled(Request $request, ResourceMetadata $resourceMetadata, string $operationName = null): bool
254
    {
255
        $enabled = $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_enabled', $this->enabled, true);
256
        $clientEnabled = $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_client_enabled', $this->clientEnabled, true);
257
258
        if ($clientEnabled) {
259
            $enabled = filter_var($this->getPaginationParameter($request, $this->enabledParameterName, $enabled), FILTER_VALIDATE_BOOLEAN);
260
        }
261
262
        return $enabled;
263
    }
264
265
    private function getPaginationParameter(Request $request, string $parameterName, $default = null)
266
    {
267
        if (null !== $paginationAttribute = $request->attributes->get('_api_pagination')) {
268
            return array_key_exists($parameterName, $paginationAttribute) ? $paginationAttribute[$parameterName] : $default;
269
        }
270
271
        return $request->query->get($parameterName, $default);
272
    }
273
274
    /**
275
     * Determines whether the Paginator should fetch join collections, if the root entity uses composite identifiers it should not.
276
     *
277
     * @see https://github.com/doctrine/doctrine2/issues/2910
278
     */
279
    private function useFetchJoinCollection(QueryBuilder $queryBuilder, string $resourceClass = null, string $operationName = null): bool
280
    {
281
        if (QueryChecker::hasRootEntityWithCompositeIdentifier($queryBuilder, $this->managerRegistry)) {
282
            return false;
283
        }
284
285
        if (null === $resourceClass) {
286
            return true;
287
        }
288
289
        $resourceMetadata = $this->resourceMetadataFactory->create($resourceClass);
290
291
        return $resourceMetadata->getCollectionOperationAttribute($operationName, 'pagination_fetch_join_collection', true, true);
292
    }
293
294
    /**
295
     * Determines whether output walkers should be used.
296
     */
297
    private function useOutputWalkers(QueryBuilder $queryBuilder): bool
298
    {
299
        /*
300
         * "Cannot count query that uses a HAVING clause. Use the output walkers for pagination"
301
         *
302
         * @see https://github.com/doctrine/doctrine2/blob/900b55d16afdcdeb5100d435a7166d3a425b9873/lib/Doctrine/ORM/Tools/Pagination/CountWalker.php#L50
303
         */
304
        if (QueryChecker::hasHavingClause($queryBuilder)) {
305
            return true;
306
        }
307
308
        /*
309
         * "Paginating an entity with foreign key as identifier only works when using the Output Walkers. Call Paginator#setUseOutputWalkers(true) before iterating the paginator."
310
         *
311
         * @see https://github.com/doctrine/doctrine2/blob/900b55d16afdcdeb5100d435a7166d3a425b9873/lib/Doctrine/ORM/Tools/Pagination/LimitSubqueryWalker.php#L87
312
         */
313
        if (QueryChecker::hasRootEntityWithForeignKeyIdentifier($queryBuilder, $this->managerRegistry)) {
314
            return true;
315
        }
316
317
        /*
318
         * "Cannot select distinct identifiers from query with LIMIT and ORDER BY on a column from a fetch joined to-many association. Use output walkers."
319
         *
320
         * @see https://github.com/doctrine/doctrine2/blob/900b55d16afdcdeb5100d435a7166d3a425b9873/lib/Doctrine/ORM/Tools/Pagination/LimitSubqueryWalker.php#L149
321
         */
322
        if (
323
            QueryChecker::hasMaxResults($queryBuilder) &&
324
            QueryChecker::hasOrderByOnToManyJoin($queryBuilder, $this->managerRegistry)
325
        ) {
326
            return true;
327
        }
328
329
        /*
330
         * When using composite identifiers pagination will need Output walkers
331
         */
332
        if (QueryChecker::hasRootEntityWithCompositeIdentifier($queryBuilder, $this->managerRegistry)) {
333
            return true;
334
        }
335
336
        // Disable output walkers by default (performance)
337
        return false;
338
    }
339
}
340