1
|
|
|
<?php |
2
|
|
|
|
3
|
|
|
namespace Spatie\Crawler\Handlers; |
4
|
|
|
|
5
|
|
|
use Spatie\Crawler\Crawler; |
6
|
|
|
use Spatie\Crawler\CrawlUrl; |
7
|
|
|
use Spatie\Crawler\LinkAdder; |
8
|
|
|
use Spatie\Crawler\CrawlerRobots; |
9
|
|
|
use Psr\Http\Message\UriInterface; |
10
|
|
|
use Spatie\Crawler\CrawlSubdomains; |
11
|
|
|
use Psr\Http\Message\StreamInterface; |
12
|
|
|
use Psr\Http\Message\ResponseInterface; |
13
|
|
|
use function GuzzleHttp\Psr7\stream_for; |
14
|
|
|
|
15
|
|
|
class CrawlRequestFulfilled |
16
|
|
|
{ |
17
|
|
|
/** @var \Spatie\Crawler\Crawler */ |
18
|
|
|
protected $crawler; |
19
|
|
|
|
20
|
|
|
/** @var \Spatie\Crawler\LinkAdder */ |
21
|
|
|
protected $linkAdder; |
22
|
|
|
|
23
|
|
|
public function __construct(Crawler $crawler) |
24
|
|
|
{ |
25
|
|
|
$this->crawler = $crawler; |
26
|
|
|
|
27
|
|
|
$this->linkAdder = new LinkAdder($this->crawler); |
28
|
|
|
} |
29
|
|
|
|
30
|
|
|
public function __invoke(ResponseInterface $response, $index) |
31
|
|
|
{ |
32
|
|
|
$robots = new CrawlerRobots($response, $this->crawler->mustRespectRobots()); |
33
|
|
|
|
34
|
|
|
if (! $robots->mayIndex()) { |
35
|
|
|
return; |
36
|
|
|
} |
37
|
|
|
|
38
|
|
|
$crawlUrl = $this->crawler->getCrawlQueue()->getUrlById($index); |
39
|
|
|
|
40
|
|
|
if ($this->crawler->mayExecuteJavaScript()) { |
41
|
|
|
$html = $this->getBodyAfterExecutingJavaScript($crawlUrl->url); |
42
|
|
|
|
43
|
|
|
$response = $response->withBody(stream_for($html)); |
44
|
|
|
} |
45
|
|
|
|
46
|
|
|
$this->handleCrawled($response, $crawlUrl); |
47
|
|
|
|
48
|
|
|
if (! $this->crawler->getCrawlProfile() instanceof CrawlSubdomains) { |
49
|
|
|
if ($crawlUrl->url->getHost() !== $this->crawler->getBaseUrl()->getHost()) { |
50
|
|
|
return; |
51
|
|
|
} |
52
|
|
|
} |
53
|
|
|
|
54
|
|
|
if (! $robots->mayFollow()) { |
55
|
|
|
return; |
56
|
|
|
} |
57
|
|
|
|
58
|
|
|
$body = $this->convertBodyToString($response->getBody(), $this->crawler->getMaximumResponseSize()); |
59
|
|
|
|
60
|
|
|
$this->linkAdder->addFromHtml($body, $crawlUrl->url); |
61
|
|
|
} |
62
|
|
|
|
63
|
|
|
protected function handleCrawled(ResponseInterface $response, CrawlUrl $crawlUrl) |
64
|
|
|
{ |
65
|
|
|
$this->crawler->getCrawlObservers()->crawled($crawlUrl, $response); |
66
|
|
|
} |
67
|
|
|
|
68
|
|
|
protected function convertBodyToString(StreamInterface $bodyStream, $readMaximumBytes = 1024 * 1024 * 2): string |
69
|
|
|
{ |
70
|
|
|
$bodyStream->rewind(); |
71
|
|
|
|
72
|
|
|
$body = $bodyStream->read($readMaximumBytes); |
73
|
|
|
|
74
|
|
|
return $body; |
75
|
|
|
} |
76
|
|
|
|
77
|
|
|
protected function getBodyAfterExecutingJavaScript(UriInterface $url): string |
78
|
|
|
{ |
79
|
|
|
$browsershot = $this->crawler->getBrowsershot(); |
80
|
|
|
|
81
|
|
|
$html = $browsershot->setUrl((string) $url)->bodyHtml(); |
82
|
|
|
|
83
|
|
|
return html_entity_decode($html); |
84
|
|
|
} |
85
|
|
|
} |
86
|
|
|
|