|Node $node * @param class-string $rectorClass */ public function decorate($node, Node $originalNode, string $rectorClass) : void { if ($node instanceof Node && $node === $originalNode) { $this->createByRule($node, $rectorClass); return; } if ($node instanceof Node) { $node = [$node]; } foreach ($node as $singleNode) { if (\get_class($singleNode) === \get_class($originalNode)) { $this->createByRule($singleNode, $rectorClass); } } $this->createByRule($originalNode, $rectorClass); } /** * @param class-string $rectorClass */ private function createByRule(Node $node, string $rectorClass) : void { /** @var class-string[] $createdByRule */ $createdByRule = $node->getAttribute(AttributeKey::CREATED_BY_RULE) ?? []; // empty array, insert if ($createdByRule === []) { $node->setAttribute(AttributeKey::CREATED_BY_RULE, [$rectorClass]); return; } // consecutive, no need to refill if (\end($createdByRule) === $rectorClass) { return; } // filter out when exists, then append $createdByRule = \array_filter($createdByRule, static function (string $rectorRule) use($rectorClass) : bool { return $rectorRule !== $rectorClass; }); $node->setAttribute(AttributeKey::CREATED_BY_RULE, \array_merge($createdByRule, [$rectorClass])); } }