Bee4\RobotsTxt\Rules

Complexity

Total Complexity

13

Size/Duplication

Total Lines	100
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	2

Test Coverage

Coverage

72.97%

Importance

Changes	8
Bugs	0	Features	3

5 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	add()	0	15	3
A	match()	0	7	2
B	get()	0	23	5
A	handleUa()	0	7	2

<?php

namespace Bee4\RobotsTxt;

use Bee4\RobotsTxt\Exception\DuplicateRuleException;

/**
 * Class Rules
 * Represent a collection of Rules
 *
 * @copyright Bee4 2015
 * @author    Stephane HULARD <s.hulard@chstudio.fr>
 */
class Rules
{
    const DEFAULT_UA = '*';

    /**
     * The collection of rules
     * @var array
     */
    protected $collection = [];

    /**
     * Default rule used if robots.txt is empty
     * @var Rule
     */
    private $defaultRule;

    public function __construct()
    {
        $this->defaultRule = new Rule();
        $this->add(self::DEFAULT_UA, $this->defaultRule);
    }

    /**
     * Add a new rule to the collection
     * @param string $userAgent
     * @param Rule $rule
     * @return Rules
     */
    public function add($userAgent, Rule $rule)
    {
        $userAgent = $this->handleUa($userAgent);
        if (isset($this->collection[$userAgent]) &&
                $this->collection[$userAgent] !== $this->defaultRule ) {
            throw (new DuplicateRuleException(
                'You can\'t add 2 rules for the same UserAgent'
            ))
                ->setRule($rule)
                ->setUserAgent($userAgent);
        }
        $this->collection[$userAgent] = $rule;

        return $this;
    }

    /**
     * Check if the URL match for the given UA or not
     * @param string $userAgent
     * @param string $url
     * @return boolean
     */
    public function match($userAgent, $url)
    {
        if (($rule = $this->get($userAgent)) === null) {
            return false;
        }
        return $rule->match($url);
    }

    /**
     * Retrieve rules for a given UA
     * @param string $userAgent
     * @return null|Rule
     */
    public function get($userAgent)
    {
        $item = null;
        $iterator = new \ArrayIterator($this->collection);
        iterator_apply(
            $iterator,
            function (\ArrayIterator $iterator, $userAgent) use (&$item) {
                if ($iterator->key() != Rules::DEFAULT_UA &&
                        preg_match($iterator->key(), $userAgent) === 1 ) {
                    $item = $iterator->current();
                    return false;
                }
                return true;
            },
            [$iterator, $userAgent]
        );

        return $item!==null?
            $item:
            (isset($this->collection[self::DEFAULT_UA])?
                $this->collection[self::DEFAULT_UA]:
                null);
    }

    /**
     * Update the UA to make a valid regexp
     * @param string $userAgent
     * @return string
     */
    private function handleUa($userAgent)
    {
        if ($userAgent == self::DEFAULT_UA) {
            return $userAgent;
        }
        return '/^'.preg_quote($userAgent).'.*/i';
    }
}