2018-02-09 00:14:45 +00:00
|
|
|
<?php declare(strict_types=1);
|
|
|
|
|
|
|
|
namespace Rector\Rector\Dynamic;
|
|
|
|
|
|
|
|
use PhpParser\Node;
|
|
|
|
use PhpParser\Node\Stmt\ClassMethod;
|
|
|
|
use PhpParser\Node\Stmt\Property;
|
|
|
|
use Rector\Node\Attribute;
|
|
|
|
use Rector\Rector\AbstractPHPUnitRector;
|
|
|
|
use Rector\ReflectionDocBlock\NodeAnalyzer\DocBlockAnalyzer;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Before:
|
|
|
|
* - @scenario
|
|
|
|
*
|
|
|
|
* After:
|
|
|
|
* - @test
|
|
|
|
*/
|
|
|
|
final class AnnotationReplacerRector extends AbstractPHPUnitRector
|
|
|
|
{
|
|
|
|
/**
|
|
|
|
* @var string[][]
|
|
|
|
*/
|
2018-02-09 00:16:54 +00:00
|
|
|
private $classToAnnotationMap = [];
|
2018-02-09 00:14:45 +00:00
|
|
|
|
|
|
|
/**
|
|
|
|
* @var DocBlockAnalyzer
|
|
|
|
*/
|
|
|
|
private $docBlockAnalyzer;
|
|
|
|
|
2018-02-09 09:27:48 +00:00
|
|
|
/**
|
|
|
|
* @var string[]
|
|
|
|
*/
|
|
|
|
private $activeAnnotationMap = [];
|
|
|
|
|
2018-02-09 00:14:45 +00:00
|
|
|
/**
|
|
|
|
* @param string[][] $classToAnnotationMap
|
|
|
|
*/
|
|
|
|
public function __construct(array $classToAnnotationMap, DocBlockAnalyzer $docBlockAnalyzer)
|
|
|
|
{
|
|
|
|
$this->docBlockAnalyzer = $docBlockAnalyzer;
|
|
|
|
$this->classToAnnotationMap = $classToAnnotationMap;
|
|
|
|
}
|
|
|
|
|
|
|
|
public function isCandidate(Node $node): bool
|
|
|
|
{
|
2018-02-09 09:27:48 +00:00
|
|
|
if ($this->shouldSkip($node)) {
|
2018-02-09 00:14:45 +00:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
$parentNode = $node->getAttribute(Attribute::PARENT_NODE);
|
|
|
|
foreach ($this->classToAnnotationMap as $type => $annotationMap) {
|
|
|
|
if (! in_array($type, $parentNode->getAttribute(Attribute::TYPES), true)) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2018-02-09 09:27:48 +00:00
|
|
|
$this->activeAnnotationMap = $annotationMap;
|
|
|
|
|
|
|
|
if ($this->hasAnyAnnotation($node)) {
|
|
|
|
return true;
|
|
|
|
}
|
2018-02-09 00:14:45 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
2018-02-09 09:27:48 +00:00
|
|
|
* @param ClassMethod|Property $node
|
2018-02-09 00:14:45 +00:00
|
|
|
*/
|
2018-02-09 09:27:48 +00:00
|
|
|
public function refactor(Node $node): ?Node
|
2018-02-09 00:14:45 +00:00
|
|
|
{
|
2018-02-09 09:27:48 +00:00
|
|
|
foreach ($this->activeAnnotationMap as $oldAnnotation => $newAnnotation) {
|
|
|
|
$this->docBlockAnalyzer->replaceAnnotationInNode($node, $oldAnnotation, $newAnnotation);
|
|
|
|
}
|
2018-02-09 00:14:45 +00:00
|
|
|
|
2018-02-09 09:27:48 +00:00
|
|
|
return $node;
|
|
|
|
}
|
|
|
|
|
|
|
|
private function shouldSkip(Node $node): bool
|
|
|
|
{
|
|
|
|
if (! $node instanceof ClassMethod && ! $node instanceof Property) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
/** @var Node|null $parentNode */
|
|
|
|
$parentNode = $node->getAttribute(Attribute::PARENT_NODE);
|
|
|
|
if (! $parentNode) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
private function hasAnyAnnotation(Node $node): bool
|
|
|
|
{
|
|
|
|
foreach ($this->activeAnnotationMap as $oldAnnotation => $newAnnotation) {
|
|
|
|
if ($this->docBlockAnalyzer->hasAnnotation($node, $oldAnnotation)) {
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
2018-02-09 00:14:45 +00:00
|
|
|
}
|
|
|
|
}
|