<?php
declare (strict_types=1);
namespace MailPoetVendor\Doctrine\ORM\Tools\Pagination;
if (!defined('ABSPATH')) exit;
use MailPoetVendor\Doctrine\ORM\Query\AST\ArithmeticExpression;
use MailPoetVendor\Doctrine\ORM\Query\AST\ConditionalExpression;
use MailPoetVendor\Doctrine\ORM\Query\AST\ConditionalPrimary;
use MailPoetVendor\Doctrine\ORM\Query\AST\ConditionalTerm;
use MailPoetVendor\Doctrine\ORM\Query\AST\InListExpression;
use MailPoetVendor\Doctrine\ORM\Query\AST\InputParameter;
use MailPoetVendor\Doctrine\ORM\Query\AST\NullComparisonExpression;
use MailPoetVendor\Doctrine\ORM\Query\AST\PathExpression;
use MailPoetVendor\Doctrine\ORM\Query\AST\SelectStatement;
use MailPoetVendor\Doctrine\ORM\Query\AST\SimpleArithmeticExpression;
use MailPoetVendor\Doctrine\ORM\Query\AST\WhereClause;
use MailPoetVendor\Doctrine\ORM\Query\TreeWalkerAdapter;
use RuntimeException;
use function count;
use function reset;
class WhereInWalker extends TreeWalkerAdapter
{
 public const HINT_PAGINATOR_HAS_IDS = 'doctrine.paginator_has_ids';
 public const PAGINATOR_ID_ALIAS = 'dpid';
 public function walkSelectStatement(SelectStatement $AST)
 {
 // Get the root entity and alias from the AST fromClause
 $from = $AST->fromClause->identificationVariableDeclarations;
 if (count($from) > 1) {
 throw new RuntimeException('Cannot count query which selects two FROM components, cannot make distinction');
 }
 $fromRoot = reset($from);
 $rootAlias = $fromRoot->rangeVariableDeclaration->aliasIdentificationVariable;
 $rootClass = $this->getMetadataForDqlAlias($rootAlias);
 $identifierFieldName = $rootClass->getSingleIdentifierFieldName();
 $pathType = PathExpression::TYPE_STATE_FIELD;
 if (isset($rootClass->associationMappings[$identifierFieldName])) {
 $pathType = PathExpression::TYPE_SINGLE_VALUED_ASSOCIATION;
 }
 $pathExpression = new PathExpression(PathExpression::TYPE_STATE_FIELD | PathExpression::TYPE_SINGLE_VALUED_ASSOCIATION, $rootAlias, $identifierFieldName);
 $pathExpression->type = $pathType;
 $hasIds = $this->_getQuery()->getHint(self::HINT_PAGINATOR_HAS_IDS);
 if ($hasIds) {
 $arithmeticExpression = new ArithmeticExpression();
 $arithmeticExpression->simpleArithmeticExpression = new SimpleArithmeticExpression([$pathExpression]);
 $expression = new InListExpression($arithmeticExpression, [new InputParameter(':' . self::PAGINATOR_ID_ALIAS)]);
 } else {
 $expression = new NullComparisonExpression($pathExpression);
 }
 $conditionalPrimary = new ConditionalPrimary();
 $conditionalPrimary->simpleConditionalExpression = $expression;
 if ($AST->whereClause) {
 if ($AST->whereClause->conditionalExpression instanceof ConditionalTerm) {
 $AST->whereClause->conditionalExpression->conditionalFactors[] = $conditionalPrimary;
 } elseif ($AST->whereClause->conditionalExpression instanceof ConditionalPrimary) {
 $AST->whereClause->conditionalExpression = new ConditionalExpression([new ConditionalTerm([$AST->whereClause->conditionalExpression, $conditionalPrimary])]);
 } else {
 $tmpPrimary = new ConditionalPrimary();
 $tmpPrimary->conditionalExpression = $AST->whereClause->conditionalExpression;
 $AST->whereClause->conditionalExpression = new ConditionalTerm([$tmpPrimary, $conditionalPrimary]);
 }
 } else {
 $AST->whereClause = new WhereClause(new ConditionalExpression([new ConditionalTerm([$conditionalPrimary])]));
 }
 }
}
