src/Wallabag/CoreBundle/Subscriber/SQLiteCascadeDeleteSubscriber.php

   1 <?php
   2
   3 namespace Wallabag\CoreBundle\Subscriber;
   4
   5 use Doctrine\Common\EventSubscriber;
   6 use Doctrine\ORM\Event\LifecycleEventArgs;
   7 use Wallabag\CoreBundle\Entity\Entry;
   8 use Doctrine\Bundle\DoctrineBundle\Registry;
   9
  10 /**
  11  * SQLite doesn't care about cascading remove, so we need to manually remove associated stuf for an Entry.
  12  * Foreign Key Support can be enabled by running `PRAGMA foreign_keys = ON;` at runtime (AT RUNTIME !).
  13  * But it needs a compilation flag that not all SQLite instance has ...
  14  *
  15  * @see https://www.sqlite.org/foreignkeys.html#fk_enable
  16  */
  17 class SQLiteCascadeDeleteSubscriber implements EventSubscriber
  18 {
  19     private $doctrine;
  20
  21     /**
  22      * @param \Doctrine\Bundle\DoctrineBundle\Registry $doctrine
  23      */
  24     public function __construct(Registry $doctrine)
  25     {
  26         $this->doctrine = $doctrine;
  27     }
  28
  29     /**
  30      * @return array
  31      */
  32     public function getSubscribedEvents()
  33     {
  34         return [
  35             'preRemove',
  36         ];
  37     }
  38
  39     /**
  40      * We removed everything related to the upcoming removed entry because SQLite can't handle it on it own.
  41      * We do it in the preRemove, because we can't retrieve tags in the postRemove (because the entry id is gone).
  42      *
  43      * @param LifecycleEventArgs $args
  44      */
  45     public function preRemove(LifecycleEventArgs $args)
  46     {
  47         $entity = $args->getEntity();
  48
  49         if (!$this->doctrine->getConnection()->getDriver() instanceof \Doctrine\DBAL\Driver\PDOSqlite\Driver ||
  50             !$entity instanceof Entry) {
  51             return;
  52         }
  53
  54         $em = $this->doctrine->getManager();
  55
  56         if (null !== $entity->getTags()) {
  57             foreach ($entity->getTags() as $tag) {
  58                 $entity->removeTag($tag);
  59             }
  60         }
  61
  62         if (null !== $entity->getAnnotations()) {
  63             foreach ($entity->getAnnotations() as $annotation) {
  64                 $em->remove($annotation);
  65             }
  66         }
  67
  68         $em->flush();
  69     }
  70 }