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; } }