src/Wallabag/CoreBundle/Event/Subscriber/DownloadImagesSubscriber.php

   1 <?php
   2
   3 namespace Wallabag\CoreBundle\Event\Subscriber;
   4
   5 use Doctrine\ORM\EntityManager;
   6 use Psr\Log\LoggerInterface;
   7 use Symfony\Component\EventDispatcher\EventSubscriberInterface;
   8 use Wallabag\CoreBundle\Entity\Entry;
   9 use Wallabag\CoreBundle\Event\EntryDeletedEvent;
  10 use Wallabag\CoreBundle\Event\EntrySavedEvent;
  11 use Wallabag\CoreBundle\Helper\DownloadImages;
  12
  13 class DownloadImagesSubscriber implements EventSubscriberInterface
  14 {
  15     private $em;
  16     private $downloadImages;
  17     private $enabled;
  18     private $logger;
  19
  20     public function __construct(EntityManager $em, DownloadImages $downloadImages, $enabled, LoggerInterface $logger)
  21     {
  22         $this->em = $em;
  23         $this->downloadImages = $downloadImages;
  24         $this->enabled = $enabled;
  25         $this->logger = $logger;
  26     }
  27
  28     public static function getSubscribedEvents()
  29     {
  30         return [
  31             EntrySavedEvent::NAME => 'onEntrySaved',
  32             EntryDeletedEvent::NAME => 'onEntryDeleted',
  33         ];
  34     }
  35
  36     /**
  37      * Download images and updated the data into the entry.
  38      *
  39      * @param EntrySavedEvent $event
  40      */
  41     public function onEntrySaved(EntrySavedEvent $event)
  42     {
  43         if (!$this->enabled) {
  44             $this->logger->debug('DownloadImagesSubscriber: disabled.');
  45
  46             return;
  47         }
  48
  49         $entry = $event->getEntry();
  50
  51         $html = $this->downloadImages($entry);
  52         if (false !== $html) {
  53             $this->logger->debug('DownloadImagesSubscriber: updated html.');
  54
  55             $entry->setContent($html);
  56         }
  57
  58         // update preview picture
  59         $previewPicture = $this->downloadPreviewImage($entry);
  60         if (false !== $previewPicture) {
  61             $this->logger->debug('DownloadImagesSubscriber: update preview picture.');
  62
  63             $entry->setPreviewPicture($previewPicture);
  64         }
  65
  66         $this->em->persist($entry);
  67         $this->em->flush();
  68     }
  69
  70     /**
  71      * Remove images related to the entry.
  72      *
  73      * @param EntryDeletedEvent $event
  74      */
  75     public function onEntryDeleted(EntryDeletedEvent $event)
  76     {
  77         if (!$this->enabled) {
  78             $this->logger->debug('DownloadImagesSubscriber: disabled.');
  79
  80             return;
  81         }
  82
  83         $this->downloadImages->removeImages($event->getEntry()->getId());
  84     }
  85
  86     /**
  87      * Download all images from the html.
  88      *
  89      * @todo If we want to add async download, it should be done in that method
  90      *
  91      * @param Entry $entry
  92      *
  93      * @return string|false False in case of async
  94      */
  95     private function downloadImages(Entry $entry)
  96     {
  97         return $this->downloadImages->processHtml(
  98             $entry->getId(),
  99             $entry->getContent(),
 100             $entry->getUrl()
 101         );
 102     }
 103
 104     /**
 105      * Download the preview picture.
 106      *
 107      * @todo If we want to add async download, it should be done in that method
 108      *
 109      * @param Entry $entry
 110      *
 111      * @return string|false False in case of async
 112      */
 113     private function downloadPreviewImage(Entry $entry)
 114     {
 115         return $this->downloadImages->processSingleImage(
 116             $entry->getId(),
 117             $entry->getPreviewPicture(),
 118             $entry->getUrl()
 119         );
 120     }
 121 }