NikicPhpParser: support custom NodeVisitors

qossmic · Mar 19, 2024 · 8e7f14f · 8e7f14f
1 parent 104f17a
commit 8e7f14f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/config/services.php b/config/services.php
@@ -161,6 +161,7 @@
         ->set(NikicPhpParser::class)
         ->args([
             '$extractors' => tagged_iterator('reference_extractors'),
+            '$visitors' => tagged_iterator('node_visitors'),
         ]);
     $services->alias(ParserInterface::class, NikicPhpParser::class);
     $services->set(TypeResolver::class);

diff --git a/src/Core/Ast/Parser/NikicPhpParser/NikicPhpParser.php b/src/Core/Ast/Parser/NikicPhpParser/NikicPhpParser.php
@@ -10,6 +10,7 @@
 use PhpParser\Node\Identifier;
 use PhpParser\Node\Stmt\ClassLike;
 use PhpParser\NodeTraverser;
+use PhpParser\NodeVisitor;
 use PhpParser\NodeVisitor\FindingVisitor;
 use PhpParser\NodeVisitor\NameResolver;
 use PhpParser\Parser;
@@ -35,15 +36,20 @@ class NikicPhpParser implements ParserInterface
 
     /**
      * @param ReferenceExtractorInterface[] $extractors
+     * @param NodeVisitor[] $visitors
      */
     public function __construct(
         private readonly Parser $parser,
         private readonly AstFileReferenceCacheInterface $cache,
         private readonly TypeResolver $typeResolver,
-        private readonly iterable $extractors
+        private readonly iterable $extractors,
+        private readonly iterable $visitors
     ) {
         $this->traverser = new NodeTraverser();
         $this->traverser->addVisitor(new NameResolver());
+        foreach ($this->visitors as $visitor) {
+            $this->traverser->addVisitor($visitor);
+        }
     }
 
     public function parseFile(string $file): FileReference