handle return flag for a try/catch/finally (#4746)

* handle return flag for a try/catch/finally * add tests for psalter
vimeo · Dec 1, 2020 · f0c0ac0 · f0c0ac0
1 parent 98053ea
commit f0c0ac0
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 3 deletions.
diff --git a/src/Psalm/Internal/Analyzer/Statements/Block/TryAnalyzer.php b/src/Psalm/Internal/Analyzer/Statements/Block/TryAnalyzer.php
@@ -444,6 +444,7 @@ function ($fq_catch_class) use ($codebase): \Psalm\Type\Atomic\TNamedObject {
             $context->loop_scope->final_actions[] = ScopeAnalyzer::ACTION_NONE;
         }
 
+        $finally_has_returned = false;
         if ($stmt->finally) {
             if ($try_context->finally_scope) {
                 $finally_context = clone $context;
@@ -455,9 +456,7 @@ function ($fq_catch_class) use ($codebase): \Psalm\Type\Atomic\TNamedObject {
 
                 $statements_analyzer->analyze($stmt->finally->stmts, $finally_context);
 
-                if ($finally_context->has_returned) {
-                    $context->has_returned = true;
-                }
+                $finally_has_returned = $finally_context->has_returned;
 
                 /** @var string $var_id */
                 foreach ($finally_context->assigned_var_ids as $var_id => $_) {
@@ -500,6 +499,9 @@ function ($fq_catch_class) use ($codebase): \Psalm\Type\Atomic\TNamedObject {
             }
         }
 
+        $body_has_returned = !in_array(ScopeAnalyzer::ACTION_NONE, $stmt_control_actions, true);
+        $context->has_returned = ($body_has_returned && $all_catches_leave) || $finally_has_returned;
+
         return null;
     }
 }
diff --git a/tests/FileManipulation/ReturnTypeManipulationTest.php b/tests/FileManipulation/ReturnTypeManipulationTest.php
@@ -607,6 +607,38 @@ function foo(): string {
                 false,
                 true,
             ],
+            'tryCatchReturn' => [
+                '<?php
+                    function scope(){
+                        try{
+                            return func();
+                        }
+                        catch(Exception $e){
+                            return null;
+                        }
+                    }
+
+                    function func(): stdClass{
+                        return new stdClass();
+                    }',
+                '<?php
+                    function scope(): ?stdClass{
+                        try{
+                            return func();
+                        }
+                        catch(Exception $e){
+                            return null;
+                        }
+                    }
+
+                    function func(): stdClass{
+                        return new stdClass();
+                    }',
+                '7.3',
+                ['MissingReturnType'],
+                false,
+                true,
+            ],
         ];
     }
 }