%PDF-
%PDF-
Mini Shell
Mini Shell
<?php
declare (strict_types=1);
namespace Rector\Core\NodeManipulator;
use PhpParser\Node\Name\FullyQualified;
use PhpParser\Node\Stmt;
use PhpParser\Node\Stmt\Class_;
use PhpParser\Node\Stmt\ClassConst;
use PhpParser\Node\Stmt\ClassMethod;
use PhpParser\Node\Stmt\Property;
use PhpParser\Node\Stmt\TraitUse;
use PHPStan\Type\Type;
use Rector\Core\PhpParser\Node\NodeFactory;
use Rector\NodeNameResolver\NodeNameResolver;
use Rector\PostRector\ValueObject\PropertyMetadata;
final class ClassInsertManipulator
{
/**
* @var array<class-string<Stmt>>
*/
private const BEFORE_TRAIT_TYPES = [\PhpParser\Node\Stmt\TraitUse::class, \PhpParser\Node\Stmt\Property::class, \PhpParser\Node\Stmt\ClassMethod::class];
/**
* @readonly
* @var \Rector\Core\PhpParser\Node\NodeFactory
*/
private $nodeFactory;
/**
* @readonly
* @var \Rector\NodeNameResolver\NodeNameResolver
*/
private $nodeNameResolver;
public function __construct(\Rector\Core\PhpParser\Node\NodeFactory $nodeFactory, \Rector\NodeNameResolver\NodeNameResolver $nodeNameResolver)
{
$this->nodeFactory = $nodeFactory;
$this->nodeNameResolver = $nodeNameResolver;
}
/**
* @param \PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $stmt
*/
public function addAsFirstMethod(\PhpParser\Node\Stmt\Class_ $class, $stmt) : void
{
if ($this->isSuccessToInsertBeforeFirstMethod($class, $stmt)) {
return;
}
if ($this->isSuccessToInsertAfterLastProperty($class, $stmt)) {
return;
}
$class->stmts[] = $stmt;
}
public function addConstantToClass(\PhpParser\Node\Stmt\Class_ $class, string $constantName, \PhpParser\Node\Stmt\ClassConst $classConst) : void
{
if ($this->hasClassConstant($class, $constantName)) {
return;
}
$this->addAsFirstMethod($class, $classConst);
}
/**
* @param Property[] $properties
*/
public function addPropertiesToClass(\PhpParser\Node\Stmt\Class_ $class, array $properties) : void
{
foreach ($properties as $property) {
$this->addAsFirstMethod($class, $property);
}
}
/**
* @internal Use PropertyAdder service instead
*/
public function addPropertyToClass(\PhpParser\Node\Stmt\Class_ $class, string $name, ?\PHPStan\Type\Type $type) : void
{
$existingProperty = $class->getProperty($name);
if ($existingProperty instanceof \PhpParser\Node\Stmt\Property) {
return;
}
$property = $this->nodeFactory->createPrivatePropertyFromNameAndType($name, $type);
$this->addAsFirstMethod($class, $property);
}
public function addInjectPropertyToClass(\PhpParser\Node\Stmt\Class_ $class, \Rector\PostRector\ValueObject\PropertyMetadata $propertyMetadata) : void
{
$existingProperty = $class->getProperty($propertyMetadata->getName());
if ($existingProperty instanceof \PhpParser\Node\Stmt\Property) {
return;
}
$property = $this->nodeFactory->createPublicInjectPropertyFromNameAndType($propertyMetadata->getName(), $propertyMetadata->getType());
$this->addAsFirstMethod($class, $property);
}
public function addAsFirstTrait(\PhpParser\Node\Stmt\Class_ $class, string $traitName) : void
{
$traitUse = new \PhpParser\Node\Stmt\TraitUse([new \PhpParser\Node\Name\FullyQualified($traitName)]);
$this->addTraitUse($class, $traitUse);
}
/**
* @param Stmt[] $stmts
* @return Stmt[]
*/
public function insertBefore(array $stmts, \PhpParser\Node\Stmt $stmt, int $key) : array
{
\array_splice($stmts, $key, 0, [$stmt]);
return $stmts;
}
/**
* @param \PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $stmt
*/
private function isSuccessToInsertBeforeFirstMethod(\PhpParser\Node\Stmt\Class_ $class, $stmt) : bool
{
foreach ($class->stmts as $key => $classStmt) {
if (!$classStmt instanceof \PhpParser\Node\Stmt\ClassMethod) {
continue;
}
$class->stmts = $this->insertBefore($class->stmts, $stmt, $key);
return \true;
}
return \false;
}
/**
* @param \PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod|\PhpParser\Node\Stmt\Property $stmt
*/
private function isSuccessToInsertAfterLastProperty(\PhpParser\Node\Stmt\Class_ $class, $stmt) : bool
{
$previousElement = null;
foreach ($class->stmts as $key => $classStmt) {
if ($previousElement instanceof \PhpParser\Node\Stmt\Property && !$classStmt instanceof \PhpParser\Node\Stmt\Property) {
$class->stmts = $this->insertBefore($class->stmts, $stmt, $key);
return \true;
}
$previousElement = $classStmt;
}
return \false;
}
private function hasClassConstant(\PhpParser\Node\Stmt\Class_ $class, string $constantName) : bool
{
foreach ($class->getConstants() as $classConst) {
if ($this->nodeNameResolver->isName($classConst, $constantName)) {
return \true;
}
}
return \false;
}
private function addTraitUse(\PhpParser\Node\Stmt\Class_ $class, \PhpParser\Node\Stmt\TraitUse $traitUse) : void
{
foreach (self::BEFORE_TRAIT_TYPES as $type) {
foreach ($class->stmts as $key => $classStmt) {
if (!$classStmt instanceof $type) {
continue;
}
$class->stmts = $this->insertBefore($class->stmts, $traitUse, $key);
return;
}
}
$class->stmts[] = $traitUse;
}
}
Zerion Mini Shell 1.0