Chris@0: <?php declare(strict_types=1);
Chris@0: 
Chris@0: namespace PhpParser\NodeVisitor;
Chris@0: 
Chris@0: use PhpParser\Node;
Chris@0: use PhpParser\NodeVisitorAbstract;
Chris@0: 
Chris@0: /**
Chris@0:  * This visitor can be used to find and collect all nodes satisfying some criterion determined by
Chris@0:  * a filter callback.
Chris@0:  */
Chris@0: class FindingVisitor extends NodeVisitorAbstract
Chris@0: {
Chris@0:     /** @var callable Filter callback */
Chris@0:     protected $filterCallback;
Chris@0:     /** @var Node[] Found nodes */
Chris@0:     protected $foundNodes;
Chris@0: 
Chris@0:     public function __construct(callable $filterCallback) {
Chris@0:         $this->filterCallback = $filterCallback;
Chris@0:     }
Chris@0: 
Chris@0:     /**
Chris@0:      * Get found nodes satisfying the filter callback.
Chris@0:      *
Chris@0:      * Nodes are returned in pre-order.
Chris@0:      *
Chris@0:      * @return Node[] Found nodes
Chris@0:      */
Chris@0:     public function getFoundNodes() : array {
Chris@0:         return $this->foundNodes;
Chris@0:     }
Chris@0: 
Chris@0:     public function beforeTraverse(array $nodes) {
Chris@0:         $this->foundNodes = [];
Chris@0: 
Chris@0:         return null;
Chris@0:     }
Chris@0: 
Chris@0:     public function enterNode(Node $node) {
Chris@0:         $filterCallback = $this->filterCallback;
Chris@0:         if ($filterCallback($node)) {
Chris@0:             $this->foundNodes[] = $node;
Chris@0:         }
Chris@0: 
Chris@0:         return null;
Chris@0:     }
Chris@0: }