Add strict scalar type support to AddReturnDocblockForScalarArrayFromAssignsRector

Author: TomasVotruba

rules-tests/TypeDeclaration/Rector/ClassMethod/AddReturnDocblockForScalarArrayFromAssignsRector/Fixture/nested_assign.php.inc

@@ -4,6 +4,24 @@ namespace Rector\Tests\TypeDeclaration\Rector\ClassMethod\AddReturnDocblockForSc
 
 class NestedAssign
 {
+    private function someMethod(string $path): array
+    {
+        $stringLength = strlen($path);
+
+        $items = [];
+        $currentdItem = '';
+
+        for ($i = 0; $i < $stringLength; $i++) {
+            $currentdItem = (string) $path[$i];
+        }
+
+        if ($currentdItem !== '') {
+            $items[] = $currentdItem;
+        }
+
+        return $items;
+    }
+
 }
 
 ?>
@@ -14,6 +32,27 @@ namespace Rector\Tests\TypeDeclaration\Rector\ClassMethod\AddReturnDocblockForSc
 
 class NestedAssign
 {
+    /**
+     * @return string[]
+     */
+    private function someMethod(string $path): array
+    {
+        $stringLength = strlen($path);
+
+        $items = [];
+        $currentdItem = '';
+
+        for ($i = 0; $i < $stringLength; $i++) {
+            $currentdItem = (string) $path[$i];
+        }
+
+        if ($currentdItem !== '') {
+            $items[] = $currentdItem;
+        }
+
+        return $items;
+    }
+
 }
 
 ?>

rules/TypeDeclaration/Rector/ClassMethod/AddReturnDocblockForScalarArrayFromAssignsRector.php

@@ -135,6 +135,7 @@ public function refactor(Node $node): ?Node
         $scalarArrayTypes = [];
         foreach ($returnedVariableNames as $returnedVariableName) {
             $scalarType = $this->resolveScalarArrayTypeForVariable($node, $returnedVariableName);
+
             if ($scalarType instanceof Type) {
                 $scalarArrayTypes[] = $scalarType;
             } else {
@@ -219,6 +220,7 @@ private function resolveScalarArrayTypeForVariable(ClassMethod|Function_ $node,
             $arrayHasDimAssigns = true;
 
             $scalarType = $this->resolveScalarType($assign->expr);
+
             if ($scalarType instanceof Type) {
                 $scalarTypes[] = $scalarType;
             } else {
@@ -258,6 +260,11 @@ private function resolveScalarType(Expr $expr): ?Type
             return new FloatType();
         }
 
+        $exprType = $this->nodeTypeResolver->getNativeType($expr);
+        if ($exprType->isScalar()->yes()) {
+            return $exprType;
+        }
+
         return null;
     }
 }