Chris@13: <?php declare(strict_types=1);
Chris@13: 
Chris@13: namespace PhpParser\NodeVisitor;
Chris@13: 
Chris@13: use PhpParser\Node;
Chris@13: use PhpParser\NodeTraverser;
Chris@13: use PhpParser\NodeVisitorAbstract;
Chris@13: 
Chris@13: /**
Chris@13:  * This visitor can be used to find the first node satisfying some criterion determined by
Chris@13:  * a filter callback.
Chris@13:  */
Chris@13: class FirstFindingVisitor extends NodeVisitorAbstract
Chris@13: {
Chris@13:     /** @var callable Filter callback */
Chris@13:     protected $filterCallback;
Chris@13:     /** @var null|Node Found node */
Chris@13:     protected $foundNode;
Chris@13: 
Chris@13:     public function __construct(callable $filterCallback) {
Chris@13:         $this->filterCallback = $filterCallback;
Chris@13:     }
Chris@13: 
Chris@13:     /**
Chris@13:      * Get found node satisfying the filter callback.
Chris@13:      *
Chris@13:      * Returns null if no node satisfies the filter callback.
Chris@13:      *
Chris@13:      * @return null|Node Found node (or null if not found)
Chris@13:      */
Chris@13:     public function getFoundNode() {
Chris@13:         return $this->foundNode;
Chris@13:     }
Chris@13: 
Chris@13:     public function beforeTraverse(array $nodes) {
Chris@13:         $this->foundNode = null;
Chris@13: 
Chris@13:         return null;
Chris@13:     }
Chris@13: 
Chris@13:     public function enterNode(Node $node) {
Chris@13:         $filterCallback = $this->filterCallback;
Chris@13:         if ($filterCallback($node)) {
Chris@13:             $this->foundNode = $node;
Chris@13:             return NodeTraverser::STOP_TRAVERSAL;
Chris@13:         }
Chris@13: 
Chris@13:         return null;
Chris@13:     }
Chris@13: }