BooleanFilter.php 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. <?php
  2. /*
  3. * This file is part of the Sonata package.
  4. *
  5. * (c) Thomas Rabaix <thomas.rabaix@sonata-project.org>
  6. *
  7. * For the full copyright and license information, please view the LICENSE
  8. * file that was distributed with this source code.
  9. */
  10. namespace Sonata\AdminBundle\Filter\ORM;
  11. use Sonata\AdminBundle\Form\Type\Filter\BooleanType;
  12. class BooleanFilter extends Filter
  13. {
  14. /**
  15. * @param QueryBuilder $queryBuilder
  16. * @param string $alias
  17. * @param string $field
  18. * @param mixed $value
  19. * @return
  20. */
  21. public function filter($queryBuilder, $alias, $field, $value)
  22. {
  23. if (is_array($value)) {
  24. $values = array();
  25. foreach ($value as $v) {
  26. if (!in_array($v, array(BooleanType::TYPE_NO, BooleanType::TYPE_YES))) {
  27. continue;
  28. }
  29. $values[] = $v == ($value == BooleanType::TYPE_YES) ? 1 : 0;
  30. }
  31. if (count($values) == 0) {
  32. return;
  33. }
  34. $queryBuilder->andWhere($queryBuilder->expr()->in(sprintf('%s.%s', $alias, $field), $values));
  35. } else {
  36. if (!in_array($value, array(BooleanType::TYPE_NO, BooleanType::TYPE_YES))) {
  37. return;
  38. }
  39. $queryBuilder->andWhere(sprintf('%s.%s = :%s', $alias, $field, $this->getName()));
  40. $queryBuilder->setParameter($this->getName(), ($value == BooleanType::TYPE_YES) ? 1 : 0);
  41. }
  42. }
  43. }