rector/vendor/rector/rector-downgrade-php/rules/DowngradePhp70/Rector/New_/DowngradeAnonymousClassRector.php
2023-05-13 16:44:37 +00:00

132 lines
3.5 KiB
PHP

<?php
declare (strict_types=1);
namespace Rector\DowngradePhp70\Rector\New_;
use PhpParser\Node;
use PhpParser\Node\Expr\New_;
use PhpParser\Node\Name;
use PhpParser\Node\Stmt\Class_;
use Rector\Core\NodeAnalyzer\ClassAnalyzer;
use Rector\Core\Rector\AbstractRector;
use Rector\DowngradePhp70\NodeFactory\ClassFromAnonymousFactory;
use Rector\NodeManipulator\NamespacedNameDecorator;
use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
/**
* @see \Rector\Tests\DowngradePhp70\Rector\New_\DowngradeAnonymousClassRector\DowngradeAnonymousClassRectorTest
*/
final class DowngradeAnonymousClassRector extends AbstractRector
{
/**
* @var string
*/
private const ANONYMOUS_CLASS_PREFIX = 'Anonymous__';
/**
* @var Class_[]
*/
private $classes = [];
/**
* @readonly
* @var \Rector\Core\NodeAnalyzer\ClassAnalyzer
*/
private $classAnalyzer;
/**
* @readonly
* @var \Rector\DowngradePhp70\NodeFactory\ClassFromAnonymousFactory
*/
private $classFromAnonymousFactory;
/**
* @readonly
* @var \Rector\NodeManipulator\NamespacedNameDecorator
*/
private $namespacedNameDecorator;
public function __construct(ClassAnalyzer $classAnalyzer, ClassFromAnonymousFactory $classFromAnonymousFactory, NamespacedNameDecorator $namespacedNameDecorator)
{
$this->classAnalyzer = $classAnalyzer;
$this->classFromAnonymousFactory = $classFromAnonymousFactory;
$this->namespacedNameDecorator = $namespacedNameDecorator;
}
/**
* @return array<class-string<Node>>
*/
public function getNodeTypes() : array
{
return [New_::class];
}
public function getRuleDefinition() : RuleDefinition
{
return new RuleDefinition('Remove anonymous class', [new CodeSample(<<<'CODE_SAMPLE'
class SomeClass
{
public function run()
{
return new class {
public function execute()
{
}
};
}
}
CODE_SAMPLE
, <<<'CODE_SAMPLE'
class Anonymous
{
public function execute()
{
}
}
class SomeClass
{
public function run()
{
return new Anonymous();
}
}
CODE_SAMPLE
)]);
}
/**
* @param Node[] $nodes
* @return Node[]|null
*/
public function beforeTraverse(array $nodes) : ?array
{
$this->classes = [];
return parent::beforeTraverse($nodes);
}
/**
* @param Node[] $nodes
* @return Node[]
*/
public function afterTraverse(array $nodes)
{
if ($this->classes === []) {
return $nodes;
}
return \array_merge($nodes, $this->classes);
}
/**
* @param New_ $node
*/
public function refactor(Node $node) : ?Node
{
if (!$this->classAnalyzer->isAnonymousClass($node->class)) {
return null;
}
if (!$node->class instanceof Class_) {
return null;
}
$className = $this->createAnonymousClassName();
$class = $this->classFromAnonymousFactory->create($className, $node->class);
$this->classes[] = $class;
$this->namespacedNameDecorator->decorate($class);
return new New_(new Name($className), $node->getArgs());
}
private function createAnonymousClassName() : string
{
$filePathHash = \md5($this->file->getFilePath());
return self::ANONYMOUS_CLASS_PREFIX . $filePathHash . '__' . \count($this->classes);
}
}