/lib/PHPParser/NodeTraverser.php
PHP | 118 lines | 80 code | 20 blank | 18 comment | 10 complexity | 93c6d5874870bd5465c06b3ce0d52ef5 MD5 | raw file
Possible License(s): BSD-3-Clause
- <?php
- class PHPParser_NodeTraverser
- {
- /**
- * @var PHPParser_NodeVisitor[] Visitors
- */
- protected $visitors;
- /**
- * Constructs a node traverser.
- */
- public function __construct() {
- $this->visitors = array();
- }
- /**
- * Adds a visitor.
- *
- * @param PHPParser_NodeVisitor $visitor Visitor to add
- */
- public function addVisitor(PHPParser_NodeVisitor $visitor) {
- $this->visitors[] = $visitor;
- }
- /**
- * Traverses an array of nodes using the registered visitors.
- *
- * @param PHPParser_Node[] $nodes Array of nodes
- *
- * @return PHPParser_Node[] Traversed array of nodes
- */
- public function traverse(array $nodes) {
- foreach ($this->visitors as $visitor) {
- if (null !== $return = $visitor->beforeTraverse($nodes)) {
- $nodes = $return;
- }
- }
- $nodes = $this->traverseArray($nodes);
- foreach ($this->visitors as $visitor) {
- if (null !== $return = $visitor->afterTraverse($nodes)) {
- $nodes = $return;
- }
- }
- return $nodes;
- }
- protected function traverseNode(PHPParser_Node $node) {
- $node = clone $node;
- foreach ($node->getSubNodeNames() as $name) {
- $subNode =& $node->$name;
- if (is_array($subNode)) {
- $subNode = $this->traverseArray($subNode);
- } elseif ($subNode instanceof PHPParser_Node) {
- foreach ($this->visitors as $visitor) {
- if (null !== $return = $visitor->enterNode($subNode)) {
- $subNode = $return;
- }
- }
- $subNode = $this->traverseNode($subNode);
- foreach ($this->visitors as $visitor) {
- if (null !== $return = $visitor->leaveNode($subNode)) {
- $subNode = $return;
- }
- }
- }
- }
- return $node;
- }
- protected function traverseArray(array $nodes) {
- $doNodes = array();
- foreach ($nodes as $i => &$node) {
- if (is_array($node)) {
- $node = $this->traverseArray($node);
- } elseif ($node instanceof PHPParser_Node) {
- foreach ($this->visitors as $visitor) {
- if (null !== $return = $visitor->enterNode($node)) {
- $node = $return;
- }
- }
- $node = $this->traverseNode($node);
- foreach ($this->visitors as $visitor) {
- $return = $visitor->leaveNode($node);
- if (false === $return) {
- $doNodes[] = array($i, array());
- break;
- } elseif (is_array($return)) {
- $doNodes[] = array($i, $return);
- break;
- } elseif (null !== $return) {
- $node = $return;
- }
- }
- }
- }
- if (!empty($doNodes)) {
- while (list($i, $replace) = array_pop($doNodes)) {
- array_splice($nodes, $i, 1, $replace);
- }
- }
- return $nodes;
- }
- }