Soshot/vendor/embed/embed/src/Embed.php

90 lines
2.3 KiB
PHP

<?php
declare(strict_types = 1);
namespace Embed;
use Embed\Http\Crawler;
use Psr\Http\Message\RequestInterface;
use Psr\Http\Message\ResponseInterface;
class Embed
{
private Crawler $crawler;
private ExtractorFactory $extractorFactory;
public function __construct(Crawler $crawler = null, ExtractorFactory $extractorFactory = null)
{
$this->crawler = $crawler ?: new Crawler();
$this->extractorFactory = $extractorFactory ?: new ExtractorFactory();
}
public function get(string $url): Extractor
{
$request = $this->crawler->createRequest('GET', $url);
$response = $this->crawler->sendRequest($request);
return $this->extract($request, $response);
}
/**
* @return Extractor[]
*/
public function getMulti(string ...$urls): array
{
$requests = array_map(
fn ($url) => $this->crawler->createRequest('GET', $url),
$urls
);
$responses = $this->crawler->sendRequests(...$requests);
$return = [];
foreach ($responses as $k => $response) {
$return[] = $this->extract($requests[$k], $responses[$k]);
}
return $return;
}
public function getCrawler(): Crawler
{
return $this->crawler;
}
public function getExtractorFactory(): ExtractorFactory
{
return $this->extractorFactory;
}
public function setSettings(array $settings): void
{
$this->extractorFactory->setSettings($settings);
}
private function extract(RequestInterface $request, ResponseInterface $response, bool $redirect = true): Extractor
{
$uri = $this->crawler->getResponseUri($response) ?: $request->getUri();
$extractor = $this->extractorFactory->createExtractor($uri, $request, $response, $this->crawler);
if (!$redirect || !$this->mustRedirect($extractor)) {
return $extractor;
}
$request = $this->crawler->createRequest('GET', $extractor->redirect);
$response = $this->crawler->sendRequest($request);
return $this->extract($request, $response, false);
}
private function mustRedirect(Extractor $extractor): bool
{
if (!empty($extractor->getOembed()->all())) {
return false;
}
return $extractor->redirect !== null;
}
}