rector/rules/Php73/Rector/String_/SensitiveHereNowDocRector.php
Tomas Votruba 8ef1dfb7af Updated Rector to commit ed8191fb64
ed8191fb64 [Docs] Fixes rector_rules_overview.md docs link in each rules (#92)
2021-05-23 09:33:26 +00:00

79 lines
2.7 KiB
PHP

<?php
declare (strict_types=1);
namespace Rector\Php73\Rector\String_;
use RectorPrefix20210523\Nette\Utils\Strings;
use PhpParser\Node;
use PhpParser\Node\Scalar\String_;
use Rector\Core\Rector\AbstractRector;
use Rector\NodeTypeResolver\Node\AttributeKey;
use Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample;
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
/**
* @changelog https://wiki.php.net/rfc/flexible_heredoc_nowdoc_syntaxes
* @see \Rector\Tests\Php73\Rector\String_\SensitiveHereNowDocRector\SensitiveHereNowDocRectorTest
*/
final class SensitiveHereNowDocRector extends \Rector\Core\Rector\AbstractRector
{
/**
* @var string
*/
private const WRAP_SUFFIX = '_WRAP';
/**
* @var string
*/
private const ATTRIBUTE_DOC_LABEL = 'docLabel';
public function getRuleDefinition() : \Symplify\RuleDocGenerator\ValueObject\RuleDefinition
{
return new \Symplify\RuleDocGenerator\ValueObject\RuleDefinition('Changes heredoc/nowdoc that contains closing word to safe wrapper name', [new \Symplify\RuleDocGenerator\ValueObject\CodeSample\CodeSample(<<<'CODE_SAMPLE'
$value = <<<A
A
A
CODE_SAMPLE
, <<<'CODE_SAMPLE'
$value = <<<A_WRAP
A
A_WRAP
CODE_SAMPLE
)]);
}
/**
* @return array<class-string<Node>>
*/
public function getNodeTypes() : array
{
return [\PhpParser\Node\Scalar\String_::class];
}
/**
* @param String_ $node
*/
public function refactor(\PhpParser\Node $node) : ?\PhpParser\Node
{
$kind = $node->getAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::KIND);
if (!\in_array($kind, [\PhpParser\Node\Scalar\String_::KIND_HEREDOC, \PhpParser\Node\Scalar\String_::KIND_NOWDOC], \true)) {
return null;
}
// the doc label is not in the string → ok
/** @var string $docLabel */
$docLabel = $node->getAttribute(self::ATTRIBUTE_DOC_LABEL);
if (!\RectorPrefix20210523\Nette\Utils\Strings::contains($node->value, $docLabel)) {
return null;
}
$node->setAttribute(self::ATTRIBUTE_DOC_LABEL, $this->uniquateDocLabel($node->value, $docLabel));
// invoke redraw
$node->setAttribute(\Rector\NodeTypeResolver\Node\AttributeKey::ORIGINAL_NODE, null);
return $node;
}
private function uniquateDocLabel(string $value, string $docLabel) : string
{
$docLabel .= self::WRAP_SUFFIX;
$docLabelCounterTemplate = $docLabel . '_%d';
$i = 0;
while (\RectorPrefix20210523\Nette\Utils\Strings::contains($value, $docLabel)) {
$docLabel = \sprintf($docLabelCounterTemplate, ++$i);
}
return $docLabel;
}
}