classAnalyzer = $classAnalyzer;
}
public function getRuleDefinition() : RuleDefinition
{
return new RuleDefinition('Replaces parent class to specific traits', [new ConfiguredCodeSample(<<<'CODE_SAMPLE'
class SomeClass extends Nette\Object
{
}
CODE_SAMPLE
, <<<'CODE_SAMPLE'
class SomeClass
{
use Nette\SmartObject;
}
CODE_SAMPLE
, [new ParentClassToTraits('Nette\\Object', ['Nette\\SmartObject'])])]);
}
/**
* @return array>
*/
public function getNodeTypes() : array
{
return [Class_::class];
}
/**
* @param Class_ $node
*/
public function refactor(Node $node) : ?Node
{
$parentExtends = $node->extends;
if (!$parentExtends instanceof Name) {
return null;
}
if ($this->classAnalyzer->isAnonymousClass($node)) {
return null;
}
$traitUses = [];
foreach ($this->parentClassToTraits as $parentClassToTrait) {
if (!$this->isName($parentExtends, $parentClassToTrait->getParentType())) {
continue;
}
foreach ($parentClassToTrait->getTraitNames() as $traitName) {
$traitUses[] = new TraitUse([new FullyQualified($traitName)]);
}
$this->removeParentClass($node);
}
if ($traitUses === []) {
return null;
}
$node->stmts = \array_merge($traitUses, $node->stmts);
return $node;
}
/**
* @param mixed[] $configuration
*/
public function configure(array $configuration) : void
{
Assert::allIsAOf($configuration, ParentClassToTraits::class);
$this->parentClassToTraits = $configuration;
}
private function removeParentClass(Class_ $class) : void
{
$class->extends = null;
}
}