From 545953e6fd2dd55e21d52615eeaf24150d41c459 Mon Sep 17 00:00:00 2001
From: Abdul Malik Ikhsan <samsonasik@gmail.com>
Date: Thu, 21 May 2026 01:15:07 +0700
Subject: [PATCH] Fix compatibility with latest phpstan patch 2.1.x-dev

---
 .../ArrowFunction/AddArrowFunctionReturnTypeRector.php      | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/rules/TypeDeclaration/Rector/ArrowFunction/AddArrowFunctionReturnTypeRector.php b/rules/TypeDeclaration/Rector/ArrowFunction/AddArrowFunctionReturnTypeRector.php
index 8ef89b529e6..c1aa2aacb4a 100644
--- a/rules/TypeDeclaration/Rector/ArrowFunction/AddArrowFunctionReturnTypeRector.php
+++ b/rules/TypeDeclaration/Rector/ArrowFunction/AddArrowFunctionReturnTypeRector.php
@@ -5,6 +5,7 @@
 namespace Rector\TypeDeclaration\Rector\ArrowFunction;
 
 use PhpParser\Node;
+use PhpParser\Node\Expr\ArrayDimFetch;
 use PhpParser\Node\Expr\ArrowFunction;
 use PHPStan\Type\MixedType;
 use PHPStan\Type\NullType;
@@ -58,7 +59,10 @@ public function refactor(Node $node): ?Node
             return null;
         }
 
-        $type = $this->nodeTypeResolver->getNativeType($node->expr);
+        // to allow array shape
+        $type = $node->expr instanceof ArrayDimFetch
+            ? $this->getType($node->expr)
+            : $this->nodeTypeResolver->getNativeType($node->expr);
 
         // not valid to add explicit type in PHP
         if ($type->isVoid()->yes()) {