]>
Commit | Line | Data |
---|---|---|
7f559418 JB |
1 | <?php |
2 | ||
3 | namespace Wallabag\CoreBundle\Event\Subscriber; | |
4 | ||
5 | use Doctrine\Common\EventSubscriber; | |
6 | use Doctrine\ORM\Event\LifecycleEventArgs; | |
7 | use Psr\Log\LoggerInterface; | |
8 | use Wallabag\CoreBundle\Helper\DownloadImages; | |
9 | use Wallabag\CoreBundle\Entity\Entry; | |
10 | use Doctrine\ORM\EntityManager; | |
11 | use Craue\ConfigBundle\Util\Config; | |
12 | ||
13 | class DownloadImagesSubscriber implements EventSubscriber | |
14 | { | |
15 | private $configClass; | |
16 | private $downloadImages; | |
17 | private $logger; | |
18 | ||
19 | /** | |
20 | * We inject the class instead of the service otherwise it generates a circular reference with the EntityManager. | |
21 | * So we build the service ourself when we got the EntityManager (in downloadImages). | |
22 | */ | |
23 | public function __construct(DownloadImages $downloadImages, $configClass, LoggerInterface $logger) | |
24 | { | |
25 | $this->downloadImages = $downloadImages; | |
26 | $this->configClass = $configClass; | |
27 | $this->logger = $logger; | |
28 | } | |
29 | ||
30 | public function getSubscribedEvents() | |
31 | { | |
32 | return array( | |
33 | 'prePersist', | |
34 | 'preUpdate', | |
35 | ); | |
36 | } | |
37 | ||
38 | /** | |
39 | * In case of an entry has been updated. | |
40 | * We won't update the content field if it wasn't updated. | |
41 | * | |
42 | * @param LifecycleEventArgs $args | |
43 | */ | |
44 | public function preUpdate(LifecycleEventArgs $args) | |
45 | { | |
46 | $entity = $args->getEntity(); | |
47 | ||
48 | if (!$entity instanceof Entry) { | |
49 | return; | |
50 | } | |
51 | ||
52 | $em = $args->getEntityManager(); | |
53 | ||
54 | // field content has been updated | |
55 | if ($args->hasChangedField('content')) { | |
56 | $html = $this->downloadImages($em, $entity); | |
57 | ||
58 | if (null !== $html) { | |
59 | $args->setNewValue('content', $html); | |
60 | } | |
61 | } | |
62 | ||
63 | // field preview picture has been updated | |
64 | if ($args->hasChangedField('previewPicture')) { | |
65 | $previewPicture = $this->downloadPreviewImage($em, $entity); | |
66 | ||
67 | if (null !== $previewPicture) { | |
68 | $entity->setPreviewPicture($previewPicture); | |
69 | } | |
70 | } | |
71 | } | |
72 | ||
73 | /** | |
74 | * When a new entry is saved. | |
75 | * | |
76 | * @param LifecycleEventArgs $args | |
77 | */ | |
78 | public function prePersist(LifecycleEventArgs $args) | |
79 | { | |
80 | $entity = $args->getEntity(); | |
81 | ||
82 | if (!$entity instanceof Entry) { | |
83 | return; | |
84 | } | |
85 | ||
86 | $config = new $this->configClass(); | |
87 | $config->setEntityManager($args->getEntityManager()); | |
88 | ||
89 | // update all images inside the html | |
90 | $html = $this->downloadImages($config, $entity); | |
91 | if (null !== $html) { | |
92 | $entity->setContent($html); | |
93 | } | |
94 | ||
95 | // update preview picture | |
96 | $previewPicture = $this->downloadPreviewImage($config, $entity); | |
97 | if (null !== $previewPicture) { | |
98 | $entity->setPreviewPicture($previewPicture); | |
99 | } | |
100 | } | |
101 | ||
102 | public function downloadImages(Config $config, Entry $entry) | |
103 | { | |
104 | // if ($config->get('download_images_with_rabbitmq')) { | |
105 | ||
106 | // } else if ($config->get('download_images_with_redis')) { | |
107 | ||
108 | // } | |
109 | ||
110 | return $this->downloadImages->processHtml( | |
111 | $entry->getContent(), | |
112 | $entry->getUrl() | |
113 | ); | |
114 | } | |
115 | ||
116 | public function downloadPreviewImage(Config $config, Entry $entry) | |
117 | { | |
118 | // if ($config->get('download_images_with_rabbitmq')) { | |
119 | ||
120 | // } else if ($config->get('download_images_with_redis')) { | |
121 | ||
122 | // } | |
123 | ||
124 | return $this->downloadImages->processSingleImage( | |
125 | $entry->getPreviewPicture(), | |
126 | $entry->getUrl() | |
127 | ); | |
128 | } | |
129 | } |