Selector.php

<?php

declare(strict_types=1);

namespace PHPHtmlParser\Selector;

use PHPHtmlParser\Contracts\Selector\ParserInterface;
use PHPHtmlParser\Contracts\Selector\SeekerInterface;
use PHPHtmlParser\Contracts\Selector\SelectorInterface;
use PHPHtmlParser\Discovery\SeekerDiscovery;
use PHPHtmlParser\Discovery\SelectorParserDiscovery;
use PHPHtmlParser\Dom\Node\AbstractNode;
use PHPHtmlParser\Dom\Node\Collection;
use PHPHtmlParser\DTO\Selector\ParsedSelectorCollectionDTO;
use PHPHtmlParser\DTO\Selector\RuleDTO;
use PHPHtmlParser\Exceptions\ChildNotFoundException;

/**
 * Class Selector.
 */
class Selector implements SelectorInterface
{
    /**
     * @var ParsedSelectorCollectionDTO
     */
    private $ParsedSelectorCollectionDTO;

    /**
     * @var SeekerInterface
     */
    private $seeker;

    /**
     * Constructs with the selector string.
     */
    public function __construct(string $selector, ?ParserInterface $parser = null, ?SeekerInterface $seeker = null)
    {
        if ($parser == null) {
            $parser = SelectorParserDiscovery::find();
        }
        if ($seeker == null) {
            $seeker = SeekerDiscovery::find();
        }

        $this->ParsedSelectorCollectionDTO = $parser->parseSelectorString($selector);
        $this->seeker = $seeker;
    }

    /**
     * Returns the selectors that where found in __construct.
     */
    public function getParsedSelectorCollectionDTO(): ParsedSelectorCollectionDTO
    {
        return $this->ParsedSelectorCollectionDTO;
    }

    /**
     * Attempts to find the selectors starting from the given
     * node object.
     *
     * @throws ChildNotFoundException
     */
    public function find(AbstractNode $node): Collection
    {
        $results = new Collection();
        foreach ($this->ParsedSelectorCollectionDTO->getParsedSelectorDTO() as $selector) {
            $nodes = [$node];
            if (\count($selector->getRules()) == 0) {
                continue;
            }

            $options = [];
            $lastRule = null;
            foreach ($selector->getRules() as $rule) {
                if ($rule->getTag() == '*' && $lastRule && ($lastRule->getTag() == '+' || $lastRule->getTag() == '~')) {
                    continue;
                }
                if ($rule->isAlterNext() && $rule->getTag() == '>') {
                    $options[] = $this->alterNext($rule);
                    continue;
                }
                $nodes = $this->seeker->seek($nodes, $rule, $options);
                // clear the options
                $options = [];
                $lastRule = $rule;
            }

            // this is the final set of nodes
            foreach ($nodes as $result) {
                $results[] = $result;
            }
        }

        return $results;
    }

    /**
     * Attempts to figure out what the alteration will be for
     * the next element.
     */
    private function alterNext(RuleDTO $rule): array
    {
        $options = [];
        if ($rule->getTag() == '>') {
            $options['checkGrandChildren'] = false;
        }

        return $options;
    }
}