Allow to extend the phpNodeVisitor #5

Author: Oscar Otero

src/PhpFunctionsScanner.php

@@ -4,6 +4,7 @@
 namespace Gettext\Scanner;
 
 use PhpParser\NodeTraverser;
+use PhpParser\NodeVisitor;
 use PhpParser\Parser;
 use PhpParser\ParserFactory;
 
@@ -27,10 +28,15 @@ public function scan(string $code, string $filename): array
         }
 
         $traverser = new NodeTraverser();
-        $visitor = new PhpNodeVisitor($filename, $this->validFunctions);
+        $visitor = $this->createNodeVisitor($filename);
         $traverser->addVisitor($visitor);
         $traverser->traverse($ast);
 
         return $visitor->getFunctions();
     }
+
+    protected function createNodeVisitor(string $filename): NodeVisitor
+    {
+        return new PhpNodeVisitor($filename, $this->validFunctions);
+    }
 }

src/PhpNodeVisitor.php

@@ -14,6 +14,7 @@ class PhpNodeVisitor implements NodeVisitor
     protected $validFunctions;
     protected $filename;
     protected $functions = [];
+    protected $argumentsHandlers = [];
 
     public function __construct(string $filename, array $validFunctions = null)
     {
@@ -76,7 +77,14 @@ protected function createFunction(FuncCall $node): ParsedFunction
                 $function->addComment(static::getComment($comment));
             }
 
-            switch ($value->getType()) {
+            $type = $value->getType();
+
+            if (isset($this->argumentsHandlers[$type])) {
+                call_user_func($this->argumentsHandlers[$type], $function, $value);
+                continue;
+            }
+
+            switch ($type) {
                 case 'Scalar_String':
                 case 'Scalar_LNumber':
                 case 'Scalar_DNumber':
@@ -90,6 +98,11 @@ protected function createFunction(FuncCall $node): ParsedFunction
         return $function;
     }
 
+    public function setArgumentsHandler(string $type, callable $handler)
+    {
+        $this->argumentsHandlers[$type] = $handler;
+    }
+
     protected static function getComment(Comment $comment): string
     {
         $text = $comment->getReformattedText();

tests/assets/code.php

@@ -125,3 +125,5 @@ public function test()
         $test();
     }
 }
+
+t("Translatable string","",["context"=>"Context string"]);