vendor/gedmo/doctrine-extensions/lib/Gedmo/SoftDeleteable/Filter/SoftDeleteableFilter.php line 51

Open in your IDE?
  1. <?php
  3. namespace Gedmo\SoftDeleteable\Filter;
  5. use Doctrine\ORM\EntityManagerInterface;
  6. use Doctrine\ORM\Mapping\ClassMetadata;
  7. use Doctrine\ORM\Query\Filter\SQLFilter;
  8. use Gedmo\SoftDeleteable\SoftDeleteableListener;
  10. /**
  11.  * The SoftDeleteableFilter adds the condition necessary to
  12.  * filter entities which were deleted "softly"
  13.  *
  14.  * @author Gustavo Falco <comfortablynumb84@gmail.com>
  15.  * @author Gediminas Morkevicius <gediminas.morkevicius@gmail.com>
  16.  * @author Patrik Votoček <patrik@votocek.cz>
  17.  * @license MIT License (http://www.opensource.org/licenses/mit-license.php)
  18.  */
  20. class SoftDeleteableFilter extends SQLFilter
  21. {
  22.     /**
  23.      * @var SoftDeleteableListener
  24.      */
  25.     protected $listener;
  27.     /**
  28.      * @var EntityManagerInterface
  29.      */
  30.     protected $entityManager;
  32.     /**
  33.      * @var string[bool]
  34.      */
  35.     protected $disabled = array();
  37.     /**
  38.      * @param ClassMetadata $targetEntity
  39.      * @param string        $targetTableAlias
  40.      * @return string
  41.      */
  42.     public function addFilterConstraint(ClassMetadata $targetEntity, $targetTableAlias)
  43.     {
  44.         $class = $targetEntity->getName();
  45.         if (array_key_exists($class, $this->disabled) && $this->disabled[$class] === true) {
  46.             return '';
  47.         } elseif (array_key_exists($targetEntity->rootEntityName, $this->disabled) && $this->disabled[$targetEntity->rootEntityName] === true) {
  48.             return '';
  49.         }
  51.         $config = $this->getListener()->getConfiguration($this->getEntityManager(), $targetEntity->name);
  53.         if (!isset($config['softDeleteable']) || !$config['softDeleteable']) {
  54.             return '';
  55.         }
  57.         $conn = $this->getEntityManager()->getConnection();
  58.         $platform = $conn->getDatabasePlatform();
  59.         $column = $targetEntity->getQuotedColumnName($config['fieldName'], $platform);
  61.         $addCondSql = $platform->getIsNullExpression($targetTableAlias.'.'.$column);
  62.         if (isset($config['timeAware']) && $config['timeAware']) {
  63.             $addCondSql = "({$addCondSql} OR {$targetTableAlias}.{$column} > {$platform->getCurrentTimestampSQL()})";
  64.         }
  66.         return $addCondSql;
  67.     }
  69.     /**
  70.      * @param string $class
  71.      */
  72.     public function disableForEntity($class)
  73.     {
  74.         $this->disabled[$class] = true;
  75.         // Make sure the hash (@see SQLFilter::__toString()) for this filter will be changed to invalidate the query cache.
  76.         $this->setParameter(sprintf('disabled_%s', $class), true);
  77.     }
  79.     /**
  80.      * @param string $class
  81.      */
  82.     public function enableForEntity($class)
  83.     {
  84.         $this->disabled[$class] = false;
  85.         // Make sure the hash (@see SQLFilter::__toString()) for this filter will be changed to invalidate the query cache.
  86.         $this->setParameter(sprintf('disabled_%s', $class), false);
  87.     }
  89.     /**
  90.      * @return SoftDeleteableListener
  91.      * @throws \RuntimeException
  92.      */
  93.     protected function getListener()
  94.     {
  95.         if ($this->listener === null) {
  96.             $em = $this->getEntityManager();
  97.             $evm = $em->getEventManager();
  99.             foreach ($evm->getListeners() as $listeners) {
  100.                 foreach ($listeners as $listener) {
  101.                     if ($listener instanceof SoftDeleteableListener) {
  102.                         $this->listener = $listener;
  104.                         break 2;
  105.                     }
  106.                 }
  107.             }
  109.             if ($this->listener === null) {
  110.                 throw new \RuntimeException('Listener "SoftDeleteableListener" was not added to the EventManager!');
  111.             }
  112.         }
  114.         return $this->listener;
  115.     }
  117.     /**
  118.      * @return EntityManagerInterface
  119.      */
  120.     protected function getEntityManager()
  121.     {
  122.         if ($this->entityManager === null) {
  123.             $refl = new \ReflectionProperty('Doctrine\ORM\Query\Filter\SQLFilter', 'em');
  124.             $refl->setAccessible(true);
  125.             $this->entityManager = $refl->getValue($this);
  126.         }
  128.         return $this->entityManager;
  129.     }
  130. }