Go\Instrument\Transformer\NodeFinderVisitor

Complexity

Total Complexity

6

Size/Duplication

Total Lines	82
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Test Coverage

Coverage

100%

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	getFoundNodes()	0	4	1
A	beforeTraverse()	0	6	1
A	enterNode()	0	10	3

<?php
/*
 * Go! AOP framework
 *
 * @copyright Copyright 2017, Lisachenko Alexander <lisachenko.it@gmail.com>
 *
 * This source file is subject to the license that is bundled
 * with this source code in the file LICENSE.
 */

namespace Go\Instrument\Transformer;

use PhpParser\Node;
use PhpParser\NodeVisitorAbstract;

/**
 * Node finder visitor for compatibility with PHP-Parser < 4.0
 */
class NodeFinderVisitor extends NodeVisitorAbstract
{
    /**
     * List of node class names to search
     *
     * @var string[]
     */
    protected $searchNodes;

    /**
     * List of found nodes
     *
     * @var Node[] Found nodes
     */
    protected $foundNodes;

    /**
     * @param array $searchNodes List of node names to search in AST
     */
    public function __construct(array $searchNodes)
    {
        $this->searchNodes = $searchNodes;
    }

    /**
     * Get found nodes.
     *
     * Nodes are returned in pre-order.
     *
     * @return Node[] Found nodes
     */
    public function getFoundNodes()
    {
        return $this->foundNodes;
    }

    /**
     * Called once before traversal.
     *
     * Return value semantics:
     *  * null:      $nodes stays as-is
     *  * otherwise: $nodes is set to the return value
     *
     * @param Node[] $nodes Array of nodes
     *
     * @return null|Node[] Array of nodes
     */
    public function beforeTraverse(array $nodes)
    {
        $this->foundNodes = [];

        return null;
    }

    /**
     * Called when entering a node.
     *
     * Return value semantics:
     *  * null
     *        => $node stays as-is
     *  * NodeTraverser::DONT_TRAVERSE_CHILDREN
     *        => Children of $node are not traversed. $node stays as-is
     *  * NodeTraverser::STOP_TRAVERSAL
     *        => Traversal is aborted. $node stays as-is
     *  * otherwise
     *        => $node is set to the return value
     *
     * @param Node $node Node
     *
     * @return null|int|Node Node
     */
    public function enterNode(Node $node)
    {
        foreach ($this->searchNodes as $nodeClassName) {
            if ($node instanceof $nodeClassName) {
                $this->foundNodes[] = $node;
            }
        }

        return null;
    }
}