rector/rules/CodingStyle/Rector/Encapsed/WrapEncapsedVariableInCurlyBracesRector.php
2021-07-02 10:36:52 +00:00

64 lines
2.2 KiB
PHP

<?php
declare (strict_types=1);
namespace Rector\CodingStyle\Rector\Encapsed;
use PhpParser\Node;
use PhpParser\Node\Expr\Variable;
use PhpParser\Node\Scalar\Encapsed;
use Rector\Core\Rector\AbstractRector;
use Rector\NodeTypeResolver\Node\AttributeKey;
use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
/**
* @see \Rector\Tests\CodingStyle\Rector\Encapsed\WrapEncapsedVariableInCurlyBracesRector\WrapEncapsedVariableInCurlyBracesRectorTest
*/
final class WrapEncapsedVariableInCurlyBracesRector extends \Rector\Core\Rector\AbstractRector
{
public function getRuleDefinition() : \Symplify\RuleDocGenerator\ValueObject\RuleDefinition
{
return new \Symplify\RuleDocGenerator\ValueObject\RuleDefinition('Wrap encapsed variables in curly braces', [new \Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample(<<<'CODE_SAMPLE'
function run($world)
{
echo "Hello $world!";
}
CODE_SAMPLE
, <<<'CODE_SAMPLE'
function run($world)
{
echo "Hello {$world}!";
}
CODE_SAMPLE
)]);
}
/**
* @return array<class-string<Node>>
*/
public function getNodeTypes() : array
{
return [\PhpParser\Node\Scalar\Encapsed::class];
}
/**
* @param Encapsed $node
*/
public function refactor(\PhpParser\Node $node) : ?\PhpParser\Node
{
$startTokenPos = $node->getStartTokenPos();
$hasVariableBeenWrapped = \false;
foreach ($node->parts as $index => $nodePart) {
if ($nodePart instanceof \PhpParser\Node\Expr\Variable) {
$previousNode = $nodePart->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::PREVIOUS_NODE);
$previousNodeEndTokenPosition = $previousNode instanceof \PhpParser\Node ? $previousNode->getEndTokenPos() : $startTokenPos;
if ($previousNodeEndTokenPosition + 1 === $nodePart->getStartTokenPos()) {
$hasVariableBeenWrapped = \true;
$node->parts[$index] = new \PhpParser\Node\Expr\Variable($nodePart->name);
}
}
}
if (!$hasVariableBeenWrapped) {
return null;
}
return $node;
}
}