| Current Path : /home/rtorresani/www/vendor/rector/rector/src/NodeManipulator/ |
| Current File : //home/rtorresani/www/vendor/rector/rector/src/NodeManipulator/StmtsManipulator.php |
<?php
declare (strict_types=1);
namespace Rector\Core\NodeManipulator;
use PhpParser\Node;
use PhpParser\Node\Stmt;
use PhpParser\Node\Stmt\ClassMethod;
use PhpParser\Node\Stmt\Expression;
use Rector\Core\PhpParser\Comparing\NodeComparator;
use Rector\PhpDocParser\NodeTraverser\SimpleCallableNodeTraverser;
final class StmtsManipulator
{
/**
* @readonly
* @var \Rector\PhpDocParser\NodeTraverser\SimpleCallableNodeTraverser
*/
private $simpleCallableNodeTraverser;
/**
* @readonly
* @var \Rector\Core\PhpParser\Comparing\NodeComparator
*/
private $nodeComparator;
public function __construct(SimpleCallableNodeTraverser $simpleCallableNodeTraverser, NodeComparator $nodeComparator)
{
$this->simpleCallableNodeTraverser = $simpleCallableNodeTraverser;
$this->nodeComparator = $nodeComparator;
}
/**
* @param Stmt[] $stmts
*/
public function getUnwrappedLastStmt(array $stmts) : ?Node
{
\end($stmts);
$lastStmtKey = \key($stmts);
$lastStmt = $stmts[$lastStmtKey];
if ($lastStmt instanceof Expression) {
return $lastStmt->expr;
}
return $lastStmt;
}
/**
* @param Stmt[] $stmts
* @return Stmt[]
*/
public function filterOutExistingStmts(ClassMethod $classMethod, array $stmts) : array
{
$this->simpleCallableNodeTraverser->traverseNodesWithCallable((array) $classMethod->stmts, function (Node $node) use(&$stmts) {
foreach ($stmts as $key => $assign) {
if (!$this->nodeComparator->areNodesEqual($node, $assign)) {
continue;
}
unset($stmts[$key]);
}
return null;
});
return $stmts;
}
}