fix @method union return type annotation

packages/attribute-aware-php-doc/src/AttributeAwareNodeFactory/PhpDoc/AttributeAwareMethodTagValueNodeFactory.php

@@ -6,12 +6,20 @@
 
 use PHPStan\PhpDocParser\Ast\Node;
 use PHPStan\PhpDocParser\Ast\PhpDoc\MethodTagValueNode;
+use PHPStan\PhpDocParser\Ast\Type\TypeNode;
 use Rector\AttributeAwarePhpDoc\Ast\PhpDoc\AttributeAwareMethodTagValueNode;
+use Rector\AttributeAwarePhpDoc\Contract\AttributeNodeAwareFactory\AttributeAwareNodeFactoryAwareInterface;
 use Rector\AttributeAwarePhpDoc\Contract\AttributeNodeAwareFactory\AttributeNodeAwareFactoryInterface;
+use Rector\BetterPhpDocParser\Attributes\Ast\AttributeAwareNodeFactory;
 use Rector\BetterPhpDocParser\Contract\PhpDocNode\AttributeAwareNodeInterface;
 
-final class AttributeAwareMethodTagValueNodeFactory implements AttributeNodeAwareFactoryInterface
+final class AttributeAwareMethodTagValueNodeFactory implements AttributeNodeAwareFactoryInterface, AttributeAwareNodeFactoryAwareInterface
 {
+    /**
+     * @var AttributeAwareNodeFactory
+     */
+    private $attributeAwareNodeFactory;
+
     public function getOriginalNodeClass(): string
     {
         return MethodTagValueNode::class;
@@ -27,12 +35,31 @@ public function isMatch(Node $node): bool
      */
     public function create(Node $node): AttributeAwareNodeInterface
     {
+        $returnType = $this->createAttributeAwareReturnType($node);
+
         return new AttributeAwareMethodTagValueNode(
             $node->isStatic,
-            $node->returnType,
+            $returnType,
             $node->methodName,
             $node->parameters,
             $node->description
         );
     }
+
+    public function setAttributeAwareNodeFactory(AttributeAwareNodeFactory $attributeAwareNodeFactory): void
+    {
+        $this->attributeAwareNodeFactory = $attributeAwareNodeFactory;
+    }
+
+    /**
+     * @return TypeNode&AttributeAwareNodeInterface
+     */
+    private function createAttributeAwareReturnType(MethodTagValueNode $methodTagValueNode)
+    {
+        if ($methodTagValueNode->returnType !== null) {
+            return $this->attributeAwareNodeFactory->createFromNode($methodTagValueNode->returnType);
+        }
+
+        return $methodTagValueNode->returnType;
+    }
 }

packages/better-php-doc-parser/tests/PhpDocInfo/PhpDocInfoPrinter/FixtureBasic/method_union.txt

@@ -0,0 +1,3 @@
+/**
+ * @method int|bool|null|void main(...$args)
+ */