-
-
Notifications
You must be signed in to change notification settings - Fork 680
/
RemoveTraitUseRector.php
81 lines (79 loc) · 2.15 KB
/
RemoveTraitUseRector.php
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
<?php
declare (strict_types=1);
namespace Rector\Removing\Rector\Class_;
use PhpParser\Node;
use PhpParser\Node\Stmt\Class_;
use PhpParser\Node\Stmt\Trait_;
use PhpParser\Node\Stmt\TraitUse;
use Rector\Contract\Rector\ConfigurableRectorInterface;
use Rector\Rector\AbstractRector;
use Symplify\RuleDocGenerator\ValueObject\CodeSample\ConfiguredCodeSample;
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition;
use RectorPrefix202409\Webmozart\Assert\Assert;
/**
* @see \Rector\Tests\Removing\Rector\Class_\RemoveTraitUseRector\RemoveTraitUseRectorTest
*/
final class RemoveTraitUseRector extends AbstractRector implements ConfigurableRectorInterface
{
/**
* @var string[]
*/
private $traitsToRemove = [];
public function getRuleDefinition() : RuleDefinition
{
return new RuleDefinition('Remove specific traits from code', [new ConfiguredCodeSample(<<<'CODE_SAMPLE'
class SomeClass
{
use SomeTrait;
}
CODE_SAMPLE
, <<<'CODE_SAMPLE'
class SomeClass
{
}
CODE_SAMPLE
, ['TraitNameToRemove'])]);
}
/**
* @return array<class-string<Node>>
*/
public function getNodeTypes() : array
{
return [Class_::class, Trait_::class];
}
/**
* @param Class_|Trait_ $node
*/
public function refactor(Node $node) : ?Node
{
$hasChanged = \false;
foreach ($node->stmts as $key => $stmt) {
if (!$stmt instanceof TraitUse) {
continue;
}
foreach ($stmt->traits as $traitKey => $trait) {
if (!$this->isNames($trait, $this->traitsToRemove)) {
continue;
}
unset($stmt->traits[$traitKey]);
$hasChanged = \true;
}
// remove empty trait uses
if ($stmt->traits === []) {
unset($node->stmts[$key]);
}
}
if ($hasChanged) {
return $node;
}
return null;
}
/**
* @param mixed[] $configuration
*/
public function configure(array $configuration) : void
{
Assert::allString($configuration);
$this->traitsToRemove = $configuration;
}
}