loophp\phptree\Modifier\Filter

Complexity

Total Complexity

5

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	2
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	modify()	0	15	4
A	__construct()	0	4	1

<?php

/**
 * For the full copyright and license information, please view
 * the LICENSE file that was distributed with this source code.
 */

declare(strict_types=1);

namespace loophp\phptree\Modifier;

use loophp\phptree\Node\NodeInterface;
use loophp\phptree\Traverser\PostOrder;
use loophp\phptree\Traverser\PreOrder;
use loophp\phptree\Traverser\TraverserInterface;

/**
 * Class Filter.
 */
class Filter implements ModifierInterface
{
    /**
     * @var callable
     */
    private $filter;

    /**
     * @var PreOrder|TraverserInterface
     */
    private $traverser;

    /**
     * Filter constructor.
     */
    public function __construct(callable $filter, ?TraverserInterface $traverser = null)
    {
        $this->filter = $filter;
        $this->traverser = $traverser ?? new PostOrder();
    }

    public function modify(NodeInterface $tree): NodeInterface
    {
        foreach ($this->traverser->traverse($tree) as $item) {
            if (null === $parent = $item->getParent()) {
                continue;
            }

            if (!(bool) ($this->filter)($item)) {
                continue;
            }

            $parent->remove($item);
        }

        return $tree;
    }
}