forbid calling impure callable in immutable context

vimeo · Jan 1, 2022 · 8049438 · 8049438
1 parent f9b3600
commit 8049438
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 1 deletion.
diff --git a/src/Psalm/Context.php b/src/Psalm/Context.php
@@ -353,11 +353,13 @@ class Context
 
     /**
      * @var bool
+     * Set by @psalm-immutable
      */
     public $mutation_free = false;
 
     /**
      * @var bool
+     * Set by @psalm-external-mutation-free
      */
     public $external_mutation_free = false;
 

diff --git a/src/Psalm/Internal/Analyzer/Statements/Expression/Call/FunctionCallAnalyzer.php b/src/Psalm/Internal/Analyzer/Statements/Expression/Call/FunctionCallAnalyzer.php
@@ -642,7 +642,7 @@ private static function getAnalyzeNamedExpression(
                 }
 
                 if ($var_type_part instanceof TClosure || $var_type_part instanceof TCallable) {
-                    if (!$var_type_part->is_pure && $context->pure) {
+                    if (!$var_type_part->is_pure && ($context->pure || $context->mutation_free)) {
                         IssueBuffer::maybeAdd(
                             new ImpureFunctionCall(
                                 'Cannot call an impure function from a mutation-free context',

diff --git a/tests/PureAnnotationTest.php b/tests/PureAnnotationTest.php
@@ -876,6 +876,38 @@ function testImpure(): void
                     ',
                 'error_message' => 'ImpureMethodCall',
             ],
+            'impureCallableInImmutableContext' => [
+                '<?php
+
+                    /**
+                     * @psalm-immutable
+                     */
+                    class Either
+                    {
+                        /**
+                         * @psalm-param callable $_
+                         */
+                        public function fold($_): void
+                        {
+                            $_();
+                        }
+                    }
+
+                    class Whatever
+                    {
+                        public function __construct()
+                        {
+                            $either = new Either();
+                            $either->fold(
+                                function (): void {}
+                            );
+                        }
+                    }
+
+                    new Whatever();
+                    ',
+                'error_message' => 'ImpureFunctionCall',
+            ],
         ];
     }
 }