feat: support imported array shapes (@phpstan-import-type)

schmittjoh · Jul 19, 2024 · 0190dfc · 0190dfc
1 parent 13292e1
commit 0190dfc
Show file tree

Hide file tree

Showing 4 changed files with 70 additions and 2 deletions.
diff --git a/src/Metadata/Driver/DocBlockDriver/DocBlockTypeResolver.php b/src/Metadata/Driver/DocBlockDriver/DocBlockTypeResolver.php
@@ -7,6 +7,7 @@
 use PHPStan\PhpDocParser\Ast\PhpDoc\ParamTagValueNode;
 use PHPStan\PhpDocParser\Ast\PhpDoc\PhpDocTagNode;
 use PHPStan\PhpDocParser\Ast\PhpDoc\ReturnTagValueNode;
+use PHPStan\PhpDocParser\Ast\PhpDoc\TypeAliasImportTagValueNode;
 use PHPStan\PhpDocParser\Ast\PhpDoc\TypeAliasTagValueNode;
 use PHPStan\PhpDocParser\Ast\PhpDoc\VarTagValueNode;
 use PHPStan\PhpDocParser\Ast\Type\ArrayTypeNode;
@@ -266,7 +267,7 @@ private function expandClassNameUsingUseStatements(string $typeHint, \Reflection
         }
 
         if ($declaringClass->getDocComment()) {
-            $phpstanArrayType = $this->getPhpstanType($declaringClass, $typeHint, $reflector);
+            $phpstanArrayType = $this->getPhpstanArrayType($declaringClass, $typeHint, $reflector);
 
             if ($phpstanArrayType) {
                 return $phpstanArrayType;
@@ -402,7 +403,7 @@ private function resolveTypeFromDocblock($reflector): array
     /**
      * @param \ReflectionMethod|\ReflectionProperty $reflector
      */
-    private function getPhpstanType(\ReflectionClass $declaringClass, string $typeHint, $reflector): ?string
+    private function getPhpstanArrayType(\ReflectionClass $declaringClass, string $typeHint, $reflector): ?string
     {
         $tokens = $this->lexer->tokenize($declaringClass->getDocComment());
         $phpDocNode = $this->phpDocParser->parse(new TokenIterator($tokens));
@@ -425,6 +426,14 @@ private function getPhpstanType(\ReflectionClass $declaringClass, string $typeHi
                         array_map(static fn (string $type) => $self->resolveType(trim($type), $reflector), $types),
                     ));
                 }
+            } elseif ($node instanceof PhpDocTagNode && $node->value instanceof TypeAliasImportTagValueNode) {
+                $importedFromFqn = $this->resolveType($node->value->importedFrom->name, $reflector);
+
+                return $this->getPhpstanArrayType(
+                    new \ReflectionClass($importedFromFqn),
+                    $node->value->importedAlias,
+                    $reflector,
+                );
             }
         }
 

diff --git a/tests/Fixtures/DocBlockType/Phpstan/PhpstanArrayShapeImported.php b/tests/Fixtures/DocBlockType/Phpstan/PhpstanArrayShapeImported.php
@@ -0,0 +1,24 @@
+<?php
+
+declare(strict_types=1);
+
+namespace JMS\Serializer\Tests\Fixtures\DocBlockType\Phpstan;
+
+use JMS\Serializer\Tests\Fixtures\DocBlockType\Phpstan\Sub\PhpstanArrayShapeToImport;
+
+/**
+ * @phpstan-import-type Settings from PhpstanArrayShapeToImport
+ * @phpstan-import-type Settings from PhpstanArrayShapeToImport as AliasedSettings
+ */
+final class PhpstanArrayShapeImported
+{
+    /**
+     * @var Settings
+     */
+    public $data;
+
+    /**
+     * @var AliasedSettings
+     */
+    public $dataAliased;
+}
diff --git a/tests/Fixtures/DocBlockType/Phpstan/Sub/PhpstanArrayShapeToImport.php b/tests/Fixtures/DocBlockType/Phpstan/Sub/PhpstanArrayShapeToImport.php
@@ -0,0 +1,19 @@
+<?php
+
+declare(strict_types=1);
+
+namespace JMS\Serializer\Tests\Fixtures\DocBlockType\Phpstan\Sub;
+
+/**
+ * @phpstan-type Settings array{
+ *  method: string,
+ *  amount: array{type: string, value: string|null}
+ * }
+ */
+final class PhpstanArrayShapeToImport
+{
+    /**
+     * @var Settings
+     */
+    public $data;
+}
diff --git a/tests/Metadata/Driver/DocBlockDriverTest.php b/tests/Metadata/Driver/DocBlockDriverTest.php
@@ -41,6 +41,7 @@
 use JMS\Serializer\Tests\Fixtures\DocBlockType\Collection\Vehicle;
 use JMS\Serializer\Tests\Fixtures\DocBlockType\Phpstan\PhpstanArrayCollectionShape;
 use JMS\Serializer\Tests\Fixtures\DocBlockType\Phpstan\PhpstanArrayShape;
+use JMS\Serializer\Tests\Fixtures\DocBlockType\Phpstan\PhpstanArrayShapeImported;
 use JMS\Serializer\Tests\Fixtures\DocBlockType\Phpstan\PhpstanMultipleArrayShapes;
 use JMS\Serializer\Tests\Fixtures\DocBlockType\Phpstan\PhpstanNestedArrayShape;
 use JMS\Serializer\Tests\Fixtures\DocBlockType\Phpstan\ProductType;
@@ -349,6 +350,21 @@ public function testInferTypeForPhpstanArray()
         );
     }
 
+    public function testInferImportedTypeForPhpstanArray()
+    {
+        $m = $this->resolve(PhpstanArrayShapeImported::class);
+
+        self::assertEquals(
+            ['name' => 'array', 'params' => []],
+            $m->propertyMetadata['data']->type,
+        );
+
+        self::assertEquals(
+            ['name' => 'array', 'params' => []],
+            $m->propertyMetadata['dataAliased']->type,
+        );
+    }
+
     public function testInferTypeForPhpstanNestedArrayShape()
     {
         $m = $this->resolve(PhpstanNestedArrayShape::class);