]>
Commit | Line | Data |
---|---|---|
558d9aab JB |
1 | <?php |
2 | ||
3 | namespace Wallabag\CoreBundle\Helper; | |
4 | ||
5 | use Graby\Graby; | |
45fd7e09 | 6 | use Psr\Log\LoggerInterface; |
558d9aab | 7 | use Wallabag\CoreBundle\Entity\Entry; |
da3d4998 | 8 | use Wallabag\CoreBundle\Tools\Utils; |
8d7b4f0e | 9 | use Symfony\Component\HttpFoundation\File\MimeType\MimeTypeExtensionGuesser; |
558d9aab JB |
10 | |
11 | /** | |
12 | * This kind of proxy class take care of getting the content from an url | |
f1e29e69 | 13 | * and update the entry with what it found. |
558d9aab JB |
14 | */ |
15 | class ContentProxy | |
16 | { | |
17 | protected $graby; | |
c3510620 | 18 | protected $tagger; |
1c9cd2a7 | 19 | protected $logger; |
8d7b4f0e | 20 | protected $mimeGuesser; |
29dca432 | 21 | protected $fetchingErrorMessage; |
6bc6fb1f | 22 | protected $eventDispatcher; |
558d9aab | 23 | |
6bc6fb1f | 24 | public function __construct(Graby $graby, RuleBasedTagger $tagger, LoggerInterface $logger, $fetchingErrorMessage) |
558d9aab | 25 | { |
347fa6be | 26 | $this->graby = $graby; |
c3510620 | 27 | $this->tagger = $tagger; |
1c9cd2a7 | 28 | $this->logger = $logger; |
8d7b4f0e | 29 | $this->mimeGuesser = new MimeTypeExtensionGuesser(); |
29dca432 | 30 | $this->fetchingErrorMessage = $fetchingErrorMessage; |
558d9aab JB |
31 | } |
32 | ||
33 | /** | |
34 | * Fetch content using graby and hydrate given entry with results information. | |
f1e29e69 | 35 | * In case we couldn't find content, we'll try to use Open Graph data. |
558d9aab | 36 | * |
4d0ec0e7 JB |
37 | * We can also force the content, in case of an import from the v1 for example, so the function won't |
38 | * fetch the content from the website but rather use information given with the $content parameter. | |
39 | * | |
40 | * @param Entry $entry Entry to update | |
41 | * @param string $url Url to grab content for | |
42 | * @param array $content An array with AT LEAST keys title, html, url, language & content_type to skip the fetchContent from the url | |
558d9aab JB |
43 | * |
44 | * @return Entry | |
45 | */ | |
4d0ec0e7 | 46 | public function updateEntry(Entry $entry, $url, array $content = []) |
558d9aab | 47 | { |
e668a812 JB |
48 | // ensure content is a bit cleaned up |
49 | if (!empty($content['html'])) { | |
74a75f7d JB |
50 | $extractor = $this->graby->getExtractor(); |
51 | $contentExtracted = $extractor->process($content['html'], $url); | |
52 | ||
53 | if ($contentExtracted) { | |
54 | $contentBlock = $extractor->getContent(); | |
55 | $contentBlock->normalize(); | |
56 | ||
57 | $content['html'] = trim($contentBlock->innerHTML); | |
58 | } | |
59 | ||
e668a812 JB |
60 | $content['html'] = htmLawed($content['html'], [ |
61 | 'safe' => 1, | |
62 | // which means: do not remove iframe elements | |
63 | 'elements' => '*+iframe', | |
64 | 'deny_attribute' => 'style', | |
65 | 'comment' => 1, | |
66 | 'cdata' => 1, | |
67 | ]); | |
68 | } | |
69 | ||
4d0ec0e7 JB |
70 | // do we have to fetch the content or the provided one is ok? |
71 | if (empty($content) || false === $this->validateContent($content)) { | |
29dca432 | 72 | $fetchedContent = $this->graby->fetchContent($url); |
106bdbcd JB |
73 | |
74 | // when content is imported, we have information in $content | |
75 | // in case fetching content goes bad, we'll keep the imported information instead of overriding them | |
29dca432 JC |
76 | if (empty($content) || $fetchedContent['html'] !== $this->fetchingErrorMessage) { |
77 | $content = $fetchedContent; | |
78 | } | |
4d0ec0e7 | 79 | } |
558d9aab JB |
80 | |
81 | $title = $content['title']; | |
e668a812 | 82 | if (!$title && !empty($content['open_graph']['og_title'])) { |
558d9aab JB |
83 | $title = $content['open_graph']['og_title']; |
84 | } | |
85 | ||
86 | $html = $content['html']; | |
87 | if (false === $html) { | |
36e6ef52 | 88 | $html = $this->fetchingErrorMessage; |
558d9aab | 89 | |
e668a812 | 90 | if (!empty($content['open_graph']['og_description'])) { |
558d9aab JB |
91 | $html .= '<p><i>But we found a short description: </i></p>'; |
92 | $html .= $content['open_graph']['og_description']; | |
93 | } | |
94 | } | |
95 | ||
96 | $entry->setUrl($content['url'] ?: $url); | |
97 | $entry->setTitle($title); | |
48656e0e | 98 | $entry->setContent($html); |
10b35097 | 99 | $entry->setHttpStatus(isset($content['status']) ? $content['status'] : ''); |
419214d7 | 100 | |
e668a812 JB |
101 | if (!empty($content['date'])) { |
102 | try { | |
103 | $entry->setPublishedAt(new \DateTime($content['date'])); | |
104 | } catch (\Exception $e) { | |
105 | $this->logger->warn('Error while defining date', ['e' => $e, 'url' => $url, 'date' => $content['date']]); | |
106 | } | |
5e9009ce NL |
107 | } |
108 | ||
7b0b3622 NL |
109 | if (!empty($content['authors'])) { |
110 | $entry->setPublishedBy($content['authors']); | |
111 | } | |
112 | ||
dda6a6ad NL |
113 | if (!empty($content['all_headers'])) { |
114 | $entry->setHeaders($content['all_headers']); | |
115 | } | |
116 | ||
e858018f JC |
117 | $entry->setLanguage(isset($content['language']) ? $content['language'] : ''); |
118 | $entry->setMimetype(isset($content['content_type']) ? $content['content_type'] : ''); | |
da3d4998 | 119 | $entry->setReadingTime(Utils::getReadingTime($html)); |
4d0ec0e7 JB |
120 | |
121 | $domainName = parse_url($entry->getUrl(), PHP_URL_HOST); | |
122 | if (false !== $domainName) { | |
123 | $entry->setDomainName($domainName); | |
124 | } | |
558d9aab | 125 | |
e668a812 | 126 | if (!empty($content['open_graph']['og_image'])) { |
558d9aab JB |
127 | $entry->setPreviewPicture($content['open_graph']['og_image']); |
128 | } | |
129 | ||
8d7b4f0e | 130 | // if content is an image define as a preview too |
e668a812 | 131 | if (!empty($content['content_type']) && in_array($this->mimeGuesser->guess($content['content_type']), ['jpeg', 'jpg', 'gif', 'png'], true)) { |
8d7b4f0e JB |
132 | $entry->setPreviewPicture($content['url']); |
133 | } | |
134 | ||
1c9cd2a7 KG |
135 | try { |
136 | $this->tagger->tag($entry); | |
137 | } catch (\Exception $e) { | |
4094ea47 | 138 | $this->logger->error('Error while trying to automatically tag an entry.', [ |
1c9cd2a7 KG |
139 | 'entry_url' => $url, |
140 | 'error_msg' => $e->getMessage(), | |
4094ea47 | 141 | ]); |
1c9cd2a7 | 142 | } |
c3510620 | 143 | |
558d9aab JB |
144 | return $entry; |
145 | } | |
c2656f96 | 146 | |
4d0ec0e7 JB |
147 | /** |
148 | * Validate that the given content as enough value to be used | |
149 | * instead of fetch the content from the url. | |
150 | * | |
151 | * @param array $content | |
152 | * | |
153 | * @return bool true if valid otherwise false | |
154 | */ | |
155 | private function validateContent(array $content) | |
156 | { | |
e668a812 | 157 | return !empty($content['title']) && !empty($content['html']) && !empty($content['url']); |
4d0ec0e7 | 158 | } |
558d9aab | 159 | } |