rector/rules/CodingStyle/Rector/ClassMethod/OrderAttributesRector.php
Tomas Votruba e9e85a1300 Updated Rector to commit a8b01f217b
a8b01f217b [automated] Re-Generate Nodes/Rectors Documentation (#619)
2021-08-08 00:26:40 +00:00

95 lines
3.6 KiB
PHP

<?php
declare (strict_types=1);
namespace Rector\CodingStyle\Rector\ClassMethod;
use PhpParser\Node;
use PhpParser\Node\AttributeGroup;
use PhpParser\Node\Expr\ArrowFunction;
use PhpParser\Node\Expr\Closure;
use PhpParser\Node\Param;
use PhpParser\Node\Stmt\Class_;
use PhpParser\Node\Stmt\ClassMethod;
use PhpParser\Node\Stmt\Function_;
use PhpParser\Node\Stmt\Property;
use Rector\Core\Contract\Rector\ConfigurableRectorInterface;
use Rector\Core\Rector\AbstractRector;
use Symplify\RuleDocGenerator\ValueObject\CodeSample\ConfiguredCodeSample;
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
use RectorPrefix20210808\Webmozart\Assert\Assert;
/**
* @see \Rector\Tests\CodingStyle\Rector\ClassMethod\OrderAttributesRector\OrderAttributesRectorTest
*/
final class OrderAttributesRector extends \Rector\Core\Rector\AbstractRector implements \Rector\Core\Contract\Rector\ConfigurableRectorInterface
{
/**
* @var string
*/
public const ATTRIBUTES_ORDER = 'attributes_order';
/**
* @var array<string, int>
*/
private $attributesOrderByName = [];
public function getRuleDefinition() : \Symplify\RuleDocGenerator\ValueObject\RuleDefinition
{
return new \Symplify\RuleDocGenerator\ValueObject\RuleDefinition('Order attributes by desired names', [new \Symplify\RuleDocGenerator\ValueObject\CodeSample\ConfiguredCodeSample(<<<'CODE_SAMPLE'
#[Second]
#[First]
class Someclass
{
}
CODE_SAMPLE
, <<<'CODE_SAMPLE'
#[First]
#[Second]
class Someclass
{
}
CODE_SAMPLE
, [self::ATTRIBUTES_ORDER => ['First', 'Second']])]);
}
/**
* @return array<class-string<Node>>
*/
public function getNodeTypes() : array
{
return [\PhpParser\Node\Stmt\Class_::class, \PhpParser\Node\Stmt\Property::class, \PhpParser\Node\Param::class, \PhpParser\Node\Stmt\ClassMethod::class, \PhpParser\Node\Stmt\Function_::class, \PhpParser\Node\Expr\Closure::class, \PhpParser\Node\Expr\ArrowFunction::class];
}
/**
* @param ClassMethod|Property|Function_|Closure|Param|Class_|ArrowFunction $node
*/
public function refactor(\PhpParser\Node $node) : ?\PhpParser\Node
{
if ($node->attrGroups === []) {
return null;
}
$originalAttrGroups = $node->attrGroups;
$currentAttrGroups = $originalAttrGroups;
\usort($currentAttrGroups, function (\PhpParser\Node\AttributeGroup $firstAttributeGroup, \PhpParser\Node\AttributeGroup $secondAttributeGroup) : int {
$firstAttributePosition = $this->resolveAttributeGroupPosition($firstAttributeGroup);
$secondAttributePosition = $this->resolveAttributeGroupPosition($secondAttributeGroup);
return $firstAttributePosition <=> $secondAttributePosition;
});
if ($currentAttrGroups === $originalAttrGroups) {
return null;
}
$node->attrGroups = $currentAttrGroups;
return $node;
}
/**
* @param array<string, mixed> $configuration
*/
public function configure(array $configuration) : void
{
$attributesOrder = $configuration[self::ATTRIBUTES_ORDER] ?? [];
\RectorPrefix20210808\Webmozart\Assert\Assert::allString($attributesOrder);
$this->attributesOrderByName = \array_flip($attributesOrder);
}
private function resolveAttributeGroupPosition(\PhpParser\Node\AttributeGroup $attributeGroup) : int
{
$attrName = $this->getName($attributeGroup->attrs[0]->name);
// 1000 makes the attribute last, as positioned attributes have a higher priority
return $this->attributesOrderByName[$attrName] ?? 1000;
}
}