2019-10-13 05:59:52 +00:00
< ? php
2021-05-09 20:15:43 +00:00
declare ( strict_types = 1 );
2021-01-29 08:08:55 +00:00
namespace Rector\Removing\Rector\Class_ ;
2019-03-09 17:36:27 +00:00
use PhpParser\Node ;
use PhpParser\Node\Stmt\Class_ ;
use PhpParser\Node\Stmt\Trait_ ;
2020-07-29 23:39:41 +00:00
use Rector\Core\Contract\Rector\ConfigurableRectorInterface ;
2020-02-06 21:48:18 +00:00
use Rector\Core\Rector\AbstractRector ;
2019-04-13 09:20:27 +00:00
use Rector\NodeTypeResolver\Node\AttributeKey ;
2020-11-16 17:50:38 +00:00
use Symplify\RuleDocGenerator\ValueObject\CodeSample\ConfiguredCodeSample ;
use Symplify\RuleDocGenerator\ValueObject\RuleDefinition ;
2022-02-18 00:42:23 +00:00
use RectorPrefix20220218\Webmozart\Assert\Assert ;
2019-09-03 09:11:45 +00:00
/**
2021-07-05 09:49:56 +00:00
* @ see \Rector\Tests\Removing\Rector\Class_\RemoveTraitUseRector\RemoveTraitUseRectorTest
2019-09-03 09:11:45 +00:00
*/
2021-07-05 09:49:56 +00:00
final class RemoveTraitUseRector extends \Rector\Core\Rector\AbstractRector implements \Rector\Core\Contract\Rector\ConfigurableRectorInterface
2019-03-09 17:36:27 +00:00
{
2020-07-29 23:39:41 +00:00
/**
2021-12-10 00:25:09 +00:00
* @ deprecated
2020-07-29 23:39:41 +00:00
* @ var string
*/
2021-01-29 08:08:55 +00:00
public const TRAITS_TO_REMOVE = 'traits_to_remove' ;
2019-03-09 17:36:27 +00:00
/**
* @ var bool
*/
2021-05-09 20:15:43 +00:00
private $classHasChanged = \false ;
2019-03-09 17:36:27 +00:00
/**
* @ var string []
*/
private $traitsToRemove = [];
2021-05-10 22:23:08 +00:00
public function getRuleDefinition () : \Symplify\RuleDocGenerator\ValueObject\RuleDefinition
2019-03-09 17:36:27 +00:00
{
2021-05-10 22:23:08 +00:00
return new \Symplify\RuleDocGenerator\ValueObject\RuleDefinition ( 'Remove specific traits from code' , [ new \Symplify\RuleDocGenerator\ValueObject\CodeSample\ConfiguredCodeSample ( <<< 'CODE_SAMPLE'
2019-03-09 17:36:27 +00:00
class SomeClass
{
use SomeTrait ;
}
2020-09-15 08:23:13 +00:00
CODE_SAMPLE
2021-05-09 20:15:43 +00:00
, <<< 'CODE_SAMPLE'
2019-03-09 17:36:27 +00:00
class SomeClass
{
}
2020-09-15 08:23:13 +00:00
CODE_SAMPLE
2021-12-06 21:12:03 +00:00
, [ 'TraitNameToRemove' ])]);
2019-03-09 17:36:27 +00:00
}
/**
2021-02-27 00:06:15 +00:00
* @ return array < class - string < Node >>
2019-03-09 17:36:27 +00:00
*/
2021-05-09 20:15:43 +00:00
public function getNodeTypes () : array
2019-03-09 17:36:27 +00:00
{
2021-05-10 22:23:08 +00:00
return [ \PhpParser\Node\Stmt\Class_ :: class , \PhpParser\Node\Stmt\Trait_ :: class ];
2019-03-09 17:36:27 +00:00
}
/**
2021-12-10 10:22:23 +00:00
* @ param Class_ | Trait_ $node
2019-03-09 17:36:27 +00:00
*/
2021-12-10 10:22:23 +00:00
public function refactor ( \PhpParser\Node $node ) : ? \PhpParser\Node
2019-03-09 17:36:27 +00:00
{
2021-05-09 20:15:43 +00:00
$this -> classHasChanged = \false ;
2021-07-05 09:49:56 +00:00
foreach ( $node -> getTraitUses () as $traitUse ) {
foreach ( $traitUse -> traits as $trait ) {
if ( ! $this -> isNames ( $trait , $this -> traitsToRemove )) {
continue ;
}
$this -> removeNode ( $traitUse );
$this -> classHasChanged = \true ;
}
}
2019-03-09 17:36:27 +00:00
// invoke re-print
if ( $this -> classHasChanged ) {
2021-05-10 22:23:08 +00:00
$node -> setAttribute ( \Rector\NodeTypeResolver\Node\AttributeKey :: ORIGINAL_NODE , null );
2021-01-29 08:08:55 +00:00
return $node ;
2019-03-09 17:36:27 +00:00
}
2021-01-29 08:08:55 +00:00
return null ;
2019-03-09 17:36:27 +00:00
}
2021-05-02 10:46:55 +00:00
/**
2021-11-27 13:43:28 +00:00
* @ param mixed [] $configuration
2021-05-02 10:46:55 +00:00
*/
2021-12-10 10:22:23 +00:00
public function configure ( array $configuration ) : void
2020-07-29 23:39:41 +00:00
{
2021-11-28 17:01:20 +00:00
$traitsToRemove = $configuration [ self :: TRAITS_TO_REMOVE ] ? ? $configuration ;
2022-02-18 00:42:23 +00:00
\RectorPrefix20220218\Webmozart\Assert\Assert :: allString ( $traitsToRemove );
2021-11-27 13:43:28 +00:00
/** @var string[] $traitsToRemove */
$this -> traitsToRemove = $traitsToRemove ;
2020-07-29 23:39:41 +00:00
}
2019-03-09 17:36:27 +00:00
}