Viewing file:  LinkAdder.php (3.23 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
<?php
namespace Spatie\Crawler;
use GuzzleHttp\Psr7\Uri;
use Illuminate\Support\Collection;
use InvalidArgumentException;
use Psr\Http\Message\UriInterface;
use Symfony\Component\DomCrawler\Crawler as DomCrawler;
use Symfony\Component\DomCrawler\Link;
use Tree\Node\Node;
class LinkAdder
{
protected Crawler $crawler;
public function __construct(Crawler $crawler)
{
$this->crawler = $crawler;
}
public function addFromHtml(string $html, UriInterface $foundOnUrl): void
{
$allLinks = $this->extractLinksFromHtml($html, $foundOnUrl);
collect($allLinks)
->filter(fn (UriInterface $url) => $this->hasCrawlableScheme($url))
->map(fn (UriInterface $url) => $this->normalizeUrl($url))
->filter(function (UriInterface $url) use ($foundOnUrl) {
if (! $node = $this->crawler->addToDepthTree($url, $foundOnUrl)) {
return false;
}
return $this->shouldCrawl($node);
})
->filter(fn (UriInterface $url) => ! str_contains($url->getPath(), '/tel:'))
->each(function (UriInterface $url) use ($foundOnUrl) {
$crawlUrl = CrawlUrl::create($url, $foundOnUrl);
$this->crawler->addToCrawlQueue($crawlUrl);
});
}
protected function extractLinksFromHtml(string $html, UriInterface $foundOnUrl): ?Collection
{
$domCrawler = new DomCrawler($html, $foundOnUrl);
return collect($domCrawler->filterXpath('//a | //link[@rel="next" or @rel="prev"]')->links())
->reject(function (Link $link) {
if ($this->isInvalidHrefNode($link)) {
return true;
}
if ($this->crawler->mustRejectNofollowLinks() && $link->getNode()->getAttribute('rel') === 'nofollow') {
return true;
}
return false;
})
->map(function (Link $link) {
try {
return new Uri($link->getUri());
} catch (InvalidArgumentException $exception) {
return;
}
})
->filter();
}
protected function hasCrawlableScheme(UriInterface $uri): bool
{
return in_array($uri->getScheme(), ['http', 'https']);
}
protected function normalizeUrl(UriInterface $url): UriInterface
{
return $url->withFragment('');
}
protected function shouldCrawl(Node $node): bool
{
if ($this->crawler->mustRespectRobots() && ! $this->crawler->getRobotsTxt()->allows($node->getValue(), $this->crawler->getUserAgent())) {
return false;
}
$maximumDepth = $this->crawler->getMaximumDepth();
if (is_null($maximumDepth)) {
return true;
}
return $node->getDepth() <= $maximumDepth;
}
protected function isInvalidHrefNode(Link $link): bool
{
if ($link->getNode()->nodeName !== 'a') {
return false;
}
if ($link->getNode()->nextSibling !== null) {
return false;
}
if ($link->getNode()->childNodes->length !== 0) {
return false;
}
return true;
}
}
|