1 files changed, 39 insertions, 0 deletions

diff --git a/src/Wallabag/ImportBundle/Component/AMPQ/EntryConsumer.php b/src/Wallabag/ImportBundle/Component/AMPQ/EntryConsumer.php
new file mode 100644
index 00000000..7775f01c
--- /dev/null
+++ b/src/Wallabag/ImportBundle/Component/AMPQ/EntryConsumer.php
@@ -0,0 +1,39 @@
+<?php
+
+namespace Wallabag\ImportBundle\Component\AMPQ;
+
+use Doctrine\ORM\EntityManager;
+use OldSound\RabbitMqBundle\RabbitMq\ConsumerInterface;
+use PhpAmqpLib\Message\AMQPMessage;
+use Wallabag\CoreBundle\Helper\ContentProxy;
+use Wallabag\CoreBundle\Repository\EntryRepository;
+
+class EntryConsumer implements ConsumerInterface
+{
+    private $em;
+    private $contentProxy;
+    private $entryRepository;
+
+    public function __construct(EntityManager $em, EntryRepository $entryRepository, ContentProxy $contentProxy)
+    {
+        $this->em = $em;
+        $this->entryRepository = $entryRepository;
+        $this->contentProxy = $contentProxy;
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function execute(AMQPMessage $msg)
+    {
+        $storedEntry = unserialize($msg->body);
+        $entry = $this->entryRepository->findByUrlAndUserId($storedEntry['url'], $storedEntry['userId']);
+        if ($entry) {
+            $entry = $this->contentProxy->updateEntry($entry, $entry->getUrl());
+            if ($entry) {
+                $this->em->persist($entry);
+                $this->em->flush();
+            }
+        }
+    }
+}