Added the resolution of class names of PHPDoc for SchemaClassDocRefle…

…ctor (#605) * added the resolution of names in PHPDoc for SchemaClassDocReflector * Fix styling --------- Co-authored-by: romalytvynenko <[email protected]>
dedoc · Oct 27, 2024 · d32b280 · d32b280
1 parent 1a7547b
commit d32b280
Showing 1 changed file with 44 additions and 5 deletions.
diff --git a/src/Support/SchemaClassDocReflector.php b/src/Support/SchemaClassDocReflector.php
@@ -2,10 +2,14 @@
 
 namespace Dedoc\Scramble\Support;
 
+use Dedoc\Scramble\Infer\Reflector\ClassReflector;
+use Dedoc\Scramble\Infer\Services\FileNameResolver;
+use Dedoc\Scramble\PhpDoc\PhpDocTypeWalker;
+use Dedoc\Scramble\PhpDoc\ResolveFqnPhpDocTypeVisitor;
+use PhpParser\NameContext;
 use PHPStan\PhpDocParser\Ast\PhpDoc\PhpDocChildNode;
 use PHPStan\PhpDocParser\Ast\PhpDoc\PhpDocNode;
 use PHPStan\PhpDocParser\Ast\PhpDoc\PhpDocTextNode;
-use ReflectionClass;
 
 class SchemaClassDocReflector
 {
@@ -33,15 +37,50 @@ static function (PhpDocChildNode $child): string {
         )));
     }
 
-    public static function createFromDocString(string $phpDocString)
+    public static function createFromDocString(string $phpDocString, ?string $definingClassName = null)
     {
-        return new self(PhpDoc::parse($phpDocString ?: '/** */'));
+        $parsedDoc = PhpDoc::parse($phpDocString ?: '/** */');
+
+        if ($definingClassName) {
+            $classReflector = ClassReflector::make($definingClassName);
+
+            $nameContext = $classReflector->getNameContext();
+
+            $parsedDoc = static::replaceClassNamesInDoc($parsedDoc, $nameContext);
+        }
+
+        return new self($parsedDoc);
     }
 
     public static function createFromClassName(string $className)
     {
-        $reflection = new ReflectionClass($className);
+        $classReflector = ClassReflector::make($className);
+
+        $nameContext = $classReflector->getNameContext();
+        $reflection = $classReflector->getReflection();
+
+        $parsedDoc = static::replaceClassNamesInDoc(PhpDoc::parse($reflection->getDocComment() ?: '/** */'), $nameContext);
+
+        return new self($parsedDoc);
+    }
+
+    private static function replaceClassNamesInDoc(PhpDocNode $docNode, NameContext $nameContext): PhpDocNode
+    {
+        $tagValues = [
+            ...$docNode->getReturnTagValues(),
+            ...$docNode->getVarTagValues(),
+            ...$docNode->getThrowsTagValues(),
+        ];
+
+        foreach ($tagValues as $tagValue) {
+            if (! $tagValue->type) {
+                continue;
+            }
+            PhpDocTypeWalker::traverse($tagValue->type, [
+                new ResolveFqnPhpDocTypeVisitor(new FileNameResolver($nameContext)),
+            ]);
+        }
 
-        return new self(PhpDoc::parse($reflection->getDocComment() ?: '/** */'));
+        return $docNode;
     }
 }