%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 = [TraitUse::class, Property::class, ClassMethod::class];
/**
* @readonly
* @var \Rector\Core\PhpParser\Node\NodeFactory
*/
private $nodeFactory;
/**
* @readonly
* @var \Rector\NodeNameResolver\NodeNameResolver
*/
private $nodeNameResolver;
public function __construct(NodeFactory $nodeFactory, NodeNameResolver $nodeNameResolver)
{
$this->nodeFactory = $nodeFactory;
$this->nodeNameResolver = $nodeNameResolver;
}
/**
* @api
* @param \PhpParser\Node\Stmt\Property|\PhpParser\Node\Stmt\ClassConst|\PhpParser\Node\Stmt\ClassMethod $stmt
*/
public function addAsFirstMethod(Class_ $class, $stmt) : void
{
if ($this->isSuccessToInsertBeforeFirstMethod($class, $stmt)) {
return;
}
if ($this->isSuccessToInsertAfterLastProperty($class, $stmt)) {
return;
}
$class->stmts[] = $stmt;
}
public function addConstantToClass(Class_ $class, string $constantName, ClassConst $classConst) : void
{
if ($this->hasClassConstant($class, $constantName)) {
return;
}
$this->addAsFirstMethod($class, $classConst);
}
/**
* @param Property[] $properties
*/
public function addPropertiesToClass(Class_ $class, array $properties) : void
{
foreach ($properties as $property) {
$this->addAsFirstMethod($class, $property);
}
}
/**
* @internal Use PropertyAdder service instead
*/
public function addPropertyToClass(Class_ $class, string $name, ?Type $type) : void
{
$existingProperty = $class->getProperty($name);
if ($existingProperty instanceof Property) {
return;
}
$property = $this->nodeFactory->createPrivatePropertyFromNameAndType($name, $type);
$this->addAsFirstMethod($class, $property);
}
public function addInjectPropertyToClass(Class_ $class, PropertyMetadata $propertyMetadata) : void
{
$existingProperty = $class->getProperty($propertyMetadata->getName());
if ($existingProperty instanceof Property) {
return;
}
$property = $this->nodeFactory->createPublicInjectPropertyFromNameAndType($propertyMetadata->getName(), $propertyMetadata->getType());
$this->addAsFirstMethod($class, $property);
}
public function addAsFirstTrait(Class_ $class, string $traitName) : void
{
$traitUse = new TraitUse([new FullyQualified($traitName)]);
$this->addTraitUse($class, $traitUse);
}
/**
* @param Stmt[] $stmts
* @return Stmt[]
*/
private function insertBefore(array $stmts, 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(Class_ $class, $stmt) : bool
{
foreach ($class->stmts as $key => $classStmt) {
if (!$classStmt instanceof 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(Class_ $class, $stmt) : bool
{
$previousElement = null;
foreach ($class->stmts as $key => $classStmt) {
if ($previousElement instanceof Property && !$classStmt instanceof Property) {
$class->stmts = $this->insertBefore($class->stmts, $stmt, $key);
return \true;
}
$previousElement = $classStmt;
}
return \false;
}
private function hasClassConstant(Class_ $class, string $constantName) : bool
{
foreach ($class->getConstants() as $classConst) {
if ($this->nodeNameResolver->isName($classConst, $constantName)) {
return \true;
}
}
return \false;
}
private function addTraitUse(Class_ $class, 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