Use doctrine event to download images

1 files changed, 129 insertions, 0 deletions

diff --git a/src/Wallabag/CoreBundle/Event/Subscriber/DownloadImagesSubscriber.php b/src/Wallabag/CoreBundle/Event/Subscriber/DownloadImagesSubscriber.php
new file mode 100644
index 00000000..654edf31
--- /dev/null
+++ b/src/Wallabag/CoreBundle/Event/Subscriber/DownloadImagesSubscriber.php
@@ -0,0 +1,129 @@
+<?php
+
+namespace Wallabag\CoreBundle\Event\Subscriber;
+
+use Doctrine\Common\EventSubscriber;
+use Doctrine\ORM\Event\LifecycleEventArgs;
+use Psr\Log\LoggerInterface;
+use Wallabag\CoreBundle\Helper\DownloadImages;
+use Wallabag\CoreBundle\Entity\Entry;
+use Doctrine\ORM\EntityManager;
+use Craue\ConfigBundle\Util\Config;
+
+class DownloadImagesSubscriber implements EventSubscriber
+{
+    private $configClass;
+    private $downloadImages;
+    private $logger;
+
+    /**
+     * We inject the class instead of the service otherwise it generates a circular reference with the EntityManager.
+     * So we build the service ourself when we got the EntityManager (in downloadImages).
+     */
+    public function __construct(DownloadImages $downloadImages, $configClass, LoggerInterface $logger)
+    {
+        $this->downloadImages = $downloadImages;
+        $this->configClass = $configClass;
+        $this->logger = $logger;
+    }
+
+    public function getSubscribedEvents()
+    {
+        return array(
+            'prePersist',
+            'preUpdate',
+        );
+    }
+
+    /**
+     * In case of an entry has been updated.
+     * We won't update the content field if it wasn't updated.
+     *
+     * @param LifecycleEventArgs $args
+     */
+    public function preUpdate(LifecycleEventArgs $args)
+    {
+        $entity = $args->getEntity();
+
+        if (!$entity instanceof Entry) {
+            return;
+        }
+
+        $em = $args->getEntityManager();
+
+        // field content has been updated
+        if ($args->hasChangedField('content')) {
+            $html = $this->downloadImages($em, $entity);
+
+            if (null !== $html) {
+                $args->setNewValue('content', $html);
+            }
+        }
+
+        // field preview picture has been updated
+        if ($args->hasChangedField('previewPicture')) {
+            $previewPicture = $this->downloadPreviewImage($em, $entity);
+
+            if (null !== $previewPicture) {
+                $entity->setPreviewPicture($previewPicture);
+            }
+        }
+    }
+
+    /**
+     * When a new entry is saved.
+     *
+     * @param LifecycleEventArgs $args
+     */
+    public function prePersist(LifecycleEventArgs $args)
+    {
+        $entity = $args->getEntity();
+
+        if (!$entity instanceof Entry) {
+            return;
+        }
+
+        $config = new $this->configClass();
+        $config->setEntityManager($args->getEntityManager());
+
+        // update all images inside the html
+        $html = $this->downloadImages($config, $entity);
+        if (null !== $html) {
+            $entity->setContent($html);
+        }
+
+        // update preview picture
+        $previewPicture = $this->downloadPreviewImage($config, $entity);
+        if (null !== $previewPicture) {
+            $entity->setPreviewPicture($previewPicture);
+        }
+    }
+
+    public function downloadImages(Config $config, Entry $entry)
+    {
+        // if ($config->get('download_images_with_rabbitmq')) {
+
+        // } else if ($config->get('download_images_with_redis')) {
+
+        // }
+
+        return $this->downloadImages->processHtml(
+            $entry->getContent(),
+            $entry->getUrl()
+        );
+    }
+
+    public function downloadPreviewImage(Config $config, Entry $entry)
+    {
+        // if ($config->get('download_images_with_rabbitmq')) {
+
+        // } else if ($config->get('download_images_with_redis')) {
+
+        // }
+
+        return $this->downloadImages->processSingleImage(
+            $entry->getPreviewPicture(),
+            $entry->getUrl()
+        );
+    }
+}