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