vendor/lexik/form-filter-bundle/Event/Listener/DoctrineApplyFilterListener.php line 40

Open in your IDE?
  1. <?php
  3. namespace Lexik\Bundle\FormFilterBundle\Event\Listener;
  5. use Doctrine\ORM\Query\Expr\Composite;
  6. use Doctrine\DBAL\Query\Expression\CompositeExpression;
  7. use Lexik\Bundle\FormFilterBundle\Event\ApplyFilterConditionEvent;
  8. use Lexik\Bundle\FormFilterBundle\Filter\Condition\ConditionInterface;
  9. use Lexik\Bundle\FormFilterBundle\Filter\Condition\ConditionNodeInterface;
  10. use Lexik\Bundle\FormFilterBundle\Filter\Doctrine\DoctrineQueryBuilderAdapter;
  12. /**
  13.  * Add filter conditions on a Doctrine ORM or DBAL query builder.
  14.  *
  15.  * @author Cédric Girard <[email protected]>
  16.  */
  17. class DoctrineApplyFilterListener
  18. {
  19.     /**
  20.      * @var array
  21.      */
  22.     private $parameters;
  24.     /**
  25.      * @var string
  26.      */
  27.     private $whereMethod;
  29.     /**
  30.      * @param string $whereMethod
  31.      */
  32.     public function __construct($whereMethod)
  33.     {
  34.         $this->whereMethod = empty($whereMethod) ? 'where' : sprintf('%sWhere', strtolower($whereMethod));
  35.     }
  37.     /**
  38.      * @param ApplyFilterConditionEvent $event
  39.      */
  40.     public function onApplyFilterCondition(ApplyFilterConditionEvent $event)
  41.     {
  42.         $qbAdapter = new DoctrineQueryBuilderAdapter($event->getQueryBuilder());
  43.         $conditionBuilder = $event->getConditionBuilder();
  45.         $this->parameters = array();
  46.         $expression = $this->computeExpression($qbAdapter, $conditionBuilder->getRoot());
  48.         if (null !== $expression && $expression->count()) {
  49.             $qbAdapter->{$this->whereMethod}($expression);
  51.             foreach ($this->parameters as $name => $value) {
  52.                 if (is_array($value)) {
  53.                     list($value, $type) = $value;
  54.                     $qbAdapter->setParameter($name, $value, $type);
  55.                 } else {
  56.                     $qbAdapter->setParameter($name, $value);
  57.                 }
  58.             }
  59.         }
  60.     }
  62.     /**
  63.      * @param DoctrineQueryBuilderAdapter $queryBuilder
  64.      * @param ConditionNodeInterface      $node
  65.      * @return Composite|CompositeExpression|null
  66.      */
  67.     protected function computeExpression(DoctrineQueryBuilderAdapter $queryBuilder, ConditionNodeInterface $node)
  68.     {
  69.         if (count($node->getFields()) == 0 && count($node->getChildren()) == 0) {
  70.             return null;
  71.         }
  73.         $method = ($node->getOperator() == ConditionNodeInterface::EXPR_AND) ? 'andX' : 'orX';
  75.         $expression = $queryBuilder->{$method}();
  77.         foreach ($node->getFields() as $condition) {
  78.             if (null !== $condition) {
  79.                 /** @var ConditionInterface $condition */
  80.                 $expression->add($condition->getExpression());
  82.                 $this->parameters = array_merge($this->parameters, $condition->getParameters());
  83.             }
  84.         }
  86.         foreach ($node->getChildren() as $child) {
  87.             $subExpr = $this->computeExpression($queryBuilder, $child);
  89.             if (null !== $subExpr && $subExpr->count()) {
  90.                 $expression->add($subExpr);
  91.             }
  92.         }
  94.         return $expression->count() ? $expression : null;
  95.     }
  96. }