Add support for arrow functions in DowngradeUnionTypeDeclarationRector (

#1296)
rectorphp · Nov 24, 2021 · 3d49ef2 · 3d49ef2
1 parent 11ca58a
commit 3d49ef2
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 13 deletions.
diff --git a/...Php80/Rector/FunctionLike/DowngradeStaticTypeDeclarationRector/config/configured_rule.php b/...Php80/Rector/FunctionLike/DowngradeStaticTypeDeclarationRector/config/configured_rule.php
diff --git a/...80/Rector/FunctionLike/DowngradeUnionTypeDeclarationRector/Fixture/array_function.php.inc b/...80/Rector/FunctionLike/DowngradeUnionTypeDeclarationRector/Fixture/array_function.php.inc
@@ -0,0 +1,15 @@
+<?php
+
+namespace Rector\Tests\DowngradePhp80\Rector\FunctionLike\DowngradeUnionTypeDeclarationRector\Fixture;
+
+$fn = fn (stdClass|null $obj): stdClass|null => $obj;
+
+?>
+-----
+<?php
+
+namespace Rector\Tests\DowngradePhp80\Rector\FunctionLike\DowngradeUnionTypeDeclarationRector\Fixture;
+
+$fn = fn ($obj) => $obj;
+
+?>
diff --git a/rules/DowngradePhp80/Rector/FunctionLike/DowngradeUnionTypeDeclarationRector.php b/rules/DowngradePhp80/Rector/FunctionLike/DowngradeUnionTypeDeclarationRector.php
@@ -5,6 +5,7 @@
 namespace Rector\DowngradePhp80\Rector\FunctionLike;
 
 use PhpParser\Node;
+use PhpParser\Node\Expr\ArrowFunction;
 use PhpParser\Node\Expr\Closure;
 use PhpParser\Node\Stmt\ClassMethod;
 use PhpParser\Node\Stmt\Function_;
@@ -31,7 +32,7 @@ public function __construct(
      */
     public function getNodeTypes(): array
     {
-        return [Function_::class, ClassMethod::class, Closure::class];
+        return [Function_::class, ClassMethod::class, Closure::class, ArrowFunction::class];
     }
 
     public function getRuleDefinition(): RuleDefinition
@@ -69,7 +70,7 @@ public function echoInput($input)
     }
 
     /**
-     * @param ClassMethod|Closure|Function_ $node
+     * @param ClassMethod|Closure|Function_|ArrowFunction $node
      */
     public function refactor(Node $node): ?Node
     {