%PDF-
%PDF-
Mini Shell
Mini Shell
<?php
declare (strict_types=1);
namespace Rector\PostRector\Collector;
use PhpParser\Node;
use PhpParser\Node\Arg;
use PhpParser\Node\Expr\MethodCall;
use PhpParser\Node\Expr\StaticCall;
use PhpParser\Node\Expr\Variable;
use PhpParser\Node\Param;
use PhpParser\Node\Stmt;
use PhpParser\Node\Stmt\ClassMethod;
use PhpParser\Node\Stmt\Expression;
use Rector\ChangesReporting\Collector\AffectedFilesCollector;
use Rector\Core\Exception\ShouldNotHappenException;
use Rector\Core\PhpParser\Comparing\NodeComparator;
use Rector\Core\PhpParser\Node\BetterNodeFinder;
use Rector\Core\Provider\CurrentFileProvider;
use Rector\Core\ValueObject\Application\File;
use Rector\NodeRemoval\BreakingRemovalGuard;
use Rector\NodeTypeResolver\Node\AttributeKey;
use Rector\PostRector\Contract\Collector\NodeCollectorInterface;
final class NodesToRemoveCollector implements NodeCollectorInterface
{
/**
* @var Stmt[]|Node[]
*/
private $nodesToRemove = [];
/**
* @readonly
* @var \Rector\ChangesReporting\Collector\AffectedFilesCollector
*/
private $affectedFilesCollector;
/**
* @readonly
* @var \Rector\NodeRemoval\BreakingRemovalGuard
*/
private $breakingRemovalGuard;
/**
* @readonly
* @var \Rector\Core\PhpParser\Node\BetterNodeFinder
*/
private $betterNodeFinder;
/**
* @readonly
* @var \Rector\Core\PhpParser\Comparing\NodeComparator
*/
private $nodeComparator;
/**
* @readonly
* @var \Rector\Core\Provider\CurrentFileProvider
*/
private $currentFileProvider;
public function __construct(AffectedFilesCollector $affectedFilesCollector, BreakingRemovalGuard $breakingRemovalGuard, BetterNodeFinder $betterNodeFinder, NodeComparator $nodeComparator, CurrentFileProvider $currentFileProvider)
{
$this->affectedFilesCollector = $affectedFilesCollector;
$this->breakingRemovalGuard = $breakingRemovalGuard;
$this->betterNodeFinder = $betterNodeFinder;
$this->nodeComparator = $nodeComparator;
$this->currentFileProvider = $currentFileProvider;
}
public function addNodeToRemove(Node $node) : void
{
/** Node|null $parentNode */
$parentNode = $node->getAttribute(AttributeKey::PARENT_NODE);
if ($parentNode !== null && $this->isUsedInArg($node, $parentNode)) {
return;
}
// chain call: "->method()->another()"
$this->ensureIsNotPartOfChainMethodCall($node);
if (!$node instanceof Expression && $parentNode instanceof Expression) {
// only expressions can be removed
$node = $parentNode;
} else {
$this->breakingRemovalGuard->ensureNodeCanBeRemove($node);
}
$file = $this->currentFileProvider->getFile();
if ($file instanceof File) {
$this->affectedFilesCollector->addFile($file);
}
/** @var Stmt $node */
$this->nodesToRemove[] = $node;
}
public function isNodeRemoved(Node $node) : bool
{
return \in_array($node, $this->nodesToRemove, \true);
}
public function isActive() : bool
{
return $this->getCount() > 0;
}
public function getCount() : int
{
return \count($this->nodesToRemove);
}
/**
* @return array<int, Node>
*/
public function getNodesToRemove() : array
{
return $this->nodesToRemove;
}
public function unset(int $key) : void
{
unset($this->nodesToRemove[$key]);
}
private function isUsedInArg(Node $node, Node $parentNode) : bool
{
if (!$node instanceof Param) {
return \false;
}
if (!$parentNode instanceof ClassMethod) {
return \false;
}
$paramVariable = $node->var;
if ($paramVariable instanceof Variable) {
return (bool) $this->betterNodeFinder->findFirst((array) $parentNode->stmts, function (Node $node) use($paramVariable) : bool {
if (!$this->nodeComparator->areNodesEqual($node, $paramVariable)) {
return \false;
}
$hasArgParent = (bool) $this->betterNodeFinder->findParentType($node, Arg::class);
if (!$hasArgParent) {
return \false;
}
return !(bool) $this->betterNodeFinder->findParentType($node, StaticCall::class);
});
}
return \false;
}
private function ensureIsNotPartOfChainMethodCall(Node $node) : void
{
if (!$node instanceof MethodCall) {
return;
}
if (!$node->var instanceof MethodCall) {
return;
}
throw new ShouldNotHappenException('Chain method calls cannot be removed this way. It would remove the whole tree of calls. Remove them manually by creating new parent node with no following method.');
}
}
Zerion Mini Shell 1.0