VDB\Spider\Filter\Prefetch\AllowedHostsFilter

Complexity

Total Complexity

5

Size/Duplication

Total Lines	38
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	match()	0	10	2
A	__construct()	0	13	3

<?php
namespace VDB\Spider\Filter\Prefetch;

use VDB\Spider\Filter\PreFetchFilterInterface;
use VDB\Uri\UriInterface;

/**
 * @author matthijs
 */
class AllowedHostsFilter implements PreFetchFilterInterface
{
    /** @var array The hostnames to filter links with */
    private $allowedHosts;

    private $allowSubDomains;

    /**
     * @param string[] $seeds
     * @param bool $allowSubDomains
     */
    public function __construct(array $seeds, $allowSubDomains = false)
    {
        $this->allowSubDomains = $allowSubDomains;

        foreach ($seeds as $seed) {
            $hostname = parse_url($seed, PHP_URL_HOST);

            if ($this->allowSubDomains) {
                // only use hostname.tld for comparison
                $this->allowedHosts[] = join('.', array_slice(explode('.', $hostname), -2));
            } else {
                // user entire *.hostname.tld for comparison
                $this->allowedHosts[] = $hostname;
            }
        }
    }

    public function match(UriInterface $uri)
    {
        $currentHostname = $uri->getHost();

        if ($this->allowSubDomains) {
            // only use hostname.tld for comparison
            $currentHostname = join('.', array_slice(explode('.', $currentHostname), -2));
        }

        return !in_array($currentHostname, $this->allowedHosts);
    }
}