Scraper.php 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282
  1. <?php
  2. namespace PicoFeed\Scraper;
  3. use PicoFeed\Base;
  4. use PicoFeed\Client\Client;
  5. use PicoFeed\Client\ClientException;
  6. use PicoFeed\Client\Url;
  7. use PicoFeed\Encoding\Encoding;
  8. use PicoFeed\Filter\Filter;
  9. use PicoFeed\Logging\Logger;
  10. use PicoFeed\Parser\XmlParser;
  11. /**
  12. * Scraper class.
  13. *
  14. * @author Frederic Guillot
  15. */
  16. class Scraper extends Base
  17. {
  18. /**
  19. * URL.
  20. *
  21. * @var string
  22. */
  23. private $url = '';
  24. /**
  25. * Relevant content.
  26. *
  27. * @var string
  28. */
  29. private $content = '';
  30. /**
  31. * HTML content.
  32. *
  33. * @var string
  34. */
  35. private $html = '';
  36. /**
  37. * HTML content encoding.
  38. *
  39. * @var string
  40. */
  41. private $encoding = '';
  42. /**
  43. * Flag to enable candidates parsing.
  44. *
  45. * @var bool
  46. */
  47. private $enableCandidateParser = true;
  48. /**
  49. * Disable candidates parsing.
  50. *
  51. * @return Scraper
  52. */
  53. public function disableCandidateParser()
  54. {
  55. $this->enableCandidateParser = false;
  56. return $this;
  57. }
  58. /**
  59. * Get encoding.
  60. *
  61. * @return string
  62. */
  63. public function getEncoding()
  64. {
  65. return $this->encoding;
  66. }
  67. /**
  68. * Set encoding.
  69. *
  70. * @param string $encoding
  71. *
  72. * @return Scraper
  73. */
  74. public function setEncoding($encoding)
  75. {
  76. $this->encoding = $encoding;
  77. return $this;
  78. }
  79. /**
  80. * Get URL to download.
  81. *
  82. * @return string
  83. */
  84. public function getUrl()
  85. {
  86. return $this->url;
  87. }
  88. /**
  89. * Set URL to download.
  90. *
  91. * @param string $url URL
  92. *
  93. * @return Scraper
  94. */
  95. public function setUrl($url)
  96. {
  97. $this->url = $url;
  98. return $this;
  99. }
  100. /**
  101. * Return true if the scraper found relevant content.
  102. *
  103. * @return bool
  104. */
  105. public function hasRelevantContent()
  106. {
  107. return !empty($this->content);
  108. }
  109. /**
  110. * Get relevant content.
  111. *
  112. * @return string
  113. */
  114. public function getRelevantContent()
  115. {
  116. return $this->content;
  117. }
  118. /**
  119. * Get raw content (unfiltered).
  120. *
  121. * @return string
  122. */
  123. public function getRawContent()
  124. {
  125. return $this->html;
  126. }
  127. /**
  128. * Set raw content (unfiltered).
  129. *
  130. * @param string $html
  131. *
  132. * @return Scraper
  133. */
  134. public function setRawContent($html)
  135. {
  136. $this->html = $html;
  137. return $this;
  138. }
  139. /**
  140. * Get filtered relevant content.
  141. *
  142. * @return string
  143. */
  144. public function getFilteredContent()
  145. {
  146. $filter = Filter::html($this->content, $this->url);
  147. $filter->setConfig($this->config);
  148. return $filter->execute();
  149. }
  150. /**
  151. * Download the HTML content.
  152. *
  153. * @return bool
  154. */
  155. public function download()
  156. {
  157. if (!empty($this->url)) {
  158. // Clear everything
  159. $this->html = '';
  160. $this->content = '';
  161. $this->encoding = '';
  162. try {
  163. $client = Client::getInstance();
  164. $client->setConfig($this->config);
  165. $client->setTimeout($this->config->getGrabberTimeout());
  166. $client->setUserAgent($this->config->getGrabberUserAgent());
  167. $client->execute($this->url);
  168. $this->url = $client->getUrl();
  169. $this->html = $client->getContent();
  170. $this->encoding = $client->getEncoding();
  171. return true;
  172. } catch (ClientException $e) {
  173. Logger::setMessage(get_called_class().': '.$e->getMessage());
  174. }
  175. }
  176. return false;
  177. }
  178. /**
  179. * Execute the scraper.
  180. *
  181. * @param string $pageContent
  182. * @param int $recursionDepth
  183. */
  184. public function execute($pageContent = '', $recursionDepth = 0)
  185. {
  186. $this->html = '';
  187. $this->encoding = '';
  188. $this->content = '';
  189. $this->download();
  190. $this->prepareHtml();
  191. $parser = $this->getParser();
  192. if ($parser !== null) {
  193. $maxRecursions = $this->config->getMaxRecursions();
  194. if(!isset($maxRecursions)){
  195. $maxRecursions = 25;
  196. }
  197. $pageContent .= $parser->execute();
  198. // check if there is a link to next page and recursively get content (max 25 pages)
  199. if((($nextLink = $parser->findNextLink()) !== null) && $recursionDepth < $maxRecursions){
  200. $nextLink = Url::resolve($nextLink,$this->url);
  201. $this->setUrl($nextLink);
  202. $this->execute($pageContent,$recursionDepth+1);
  203. }
  204. else{
  205. $this->content = $pageContent;
  206. }
  207. Logger::setMessage(get_called_class().': Content length: '.strlen($this->content).' bytes');
  208. }
  209. }
  210. /**
  211. * Get the parser.
  212. *
  213. * @return ParserInterface
  214. */
  215. public function getParser()
  216. {
  217. $ruleLoader = new RuleLoader($this->config);
  218. $rules = $ruleLoader->getRules($this->url);
  219. if (!empty($rules['grabber'])) {
  220. Logger::setMessage(get_called_class().': Parse content with rules');
  221. foreach ($rules['grabber'] as $pattern => $rule) {
  222. $url = new Url($this->url);
  223. $sub_url = $url->getFullPath();
  224. if (preg_match($pattern, $sub_url)) {
  225. Logger::setMessage(get_called_class().': Matched url '.$sub_url);
  226. return new RuleParser($this->html, $rule);
  227. }
  228. }
  229. } elseif ($this->enableCandidateParser) {
  230. Logger::setMessage(get_called_class().': Parse content with candidates');
  231. }
  232. return new CandidateParser($this->html);
  233. }
  234. /**
  235. * Normalize encoding and strip head tag.
  236. */
  237. public function prepareHtml()
  238. {
  239. $html_encoding = XmlParser::getEncodingFromMetaTag($this->html);
  240. $this->html = Encoding::convert($this->html, $html_encoding ?: $this->encoding);
  241. $this->html = Filter::stripHeadTags($this->html);
  242. Logger::setMessage(get_called_class().': HTTP Encoding "'.$this->encoding.'" ; HTML Encoding "'.$html_encoding.'"');
  243. }
  244. }